Parse index name from chunk metadata source attribute

Author: max-svistunov

src/utils/responses.py

@@ -846,6 +846,11 @@ def _resolve_source_for_result(
         if attr_store_id:
             return rag_id_mapping.get(attr_store_id, attr_store_id)
 
+        # Fallback: read index name embedded directly by rag-content
+        attr_source: Optional[str] = attributes.get("source")
+        if attr_source:
+            return rag_id_mapping.get(attr_source, attr_source)
+
     return None
 
 

tests/unit/utils/test_responses.py

@@ -2044,6 +2044,69 @@ def test_multiple_stores_attribute_not_in_mapping(
         )
         assert source == "vs-unknown"
 
+    def test_multiple_stores_source_attribute_fallback(
+        self, mocker: MockerFixture
+    ) -> None:
+        """Test resolution falls back to source attribute when no vector_store_id."""
+        mock_result = mocker.Mock()
+        mock_result.filename = "file-abc123"
+        mock_result.attributes = {"source": "ocp-documentation"}
+
+        source = _resolve_source_for_result(
+            mock_result,
+            ["vs-001", "vs-002"],
+            {"ocp-documentation": "ocp-4.18-docs"},
+        )
+        assert source == "ocp-4.18-docs"
+
+    def test_multiple_stores_source_attribute_unmapped(
+        self, mocker: MockerFixture
+    ) -> None:
+        """Test resolution returns raw source attribute when not in mapping."""
+        mock_result = mocker.Mock()
+        mock_result.filename = "file-abc123"
+        mock_result.attributes = {"source": "custom-index"}
+
+        source = _resolve_source_for_result(
+            mock_result,
+            ["vs-001", "vs-002"],
+            {},
+        )
+        assert source == "custom-index"
+
+    def test_multiple_stores_vector_store_id_preferred_over_source(
+        self, mocker: MockerFixture
+    ) -> None:
+        """Test vector_store_id takes precedence over source attribute."""
+        mock_result = mocker.Mock()
+        mock_result.filename = "file-abc123"
+        mock_result.attributes = {
+            "vector_store_id": "vs-002",
+            "source": "should-not-be-used",
+        }
+
+        source = _resolve_source_for_result(
+            mock_result,
+            ["vs-001", "vs-002"],
+            {"vs-002": "rhel-9-docs"},
+        )
+        assert source == "rhel-9-docs"
+
+    def test_multiple_stores_no_vector_store_id_no_source(
+        self, mocker: MockerFixture
+    ) -> None:
+        """Test resolution returns None when neither vector_store_id nor source present."""
+        mock_result = mocker.Mock()
+        mock_result.filename = "file-abc123"
+        mock_result.attributes = {"title": "some doc"}
+
+        source = _resolve_source_for_result(
+            mock_result,
+            ["vs-001", "vs-002"],
+            {"vs-001": "ocp-docs"},
+        )
+        assert source is None
+
 
 class TestBuildChunkAttributes:
     """Tests for _build_chunk_attributes function."""