propagate metadata

libs/community/tests/unit_tests/vectorstores/test_databricks_vector_search.py

@@ -605,6 +605,7 @@ def test_similarity_score_threshold(index_details: dict, threshold: float) -> No
     for idx, result in enumerate(result_with_scores):
         assert result.score >= threshold
         assert result.page_content == search_result[idx].page_content
+        assert result.metadata == search_result[idx].metadata
 
 
 @pytest.mark.requires("databricks", "databricks.vector_search")

libs/core/langchain_core/vectorstores.py

@@ -712,7 +712,11 @@ class VectorStoreRetriever(BaseRetriever):
             )
             if include_score:
                 return [
-                    DocumentSearchHit(page_content=doc.page_content, score=score)
+                    DocumentSearchHit(
+                        page_content=doc.page_content,
+                        metadata=doc.metadata,
+                        score=score,
+                    )
                     for doc, score in docs_and_similarities
                 ]
             docs = [doc for doc, _ in docs_and_similarities]
@@ -748,7 +752,11 @@ class VectorStoreRetriever(BaseRetriever):
             )
             if include_score:
                 return [
-                    DocumentSearchHit(page_content=doc.page_content, score=score)
+                    DocumentSearchHit(
+                        page_content=doc.page_content,
+                        metadata=doc.metadata,
+                        score=score,
+                    )
                     for doc, score in docs_and_similarities
                 ]
             docs = [doc for doc, _ in docs_and_similarities]

libs/langchain/langchain/retrievers/self_query/base.py

@@ -199,7 +199,9 @@ class SelfQueryRetriever(BaseRetriever):
                 query, **search_kwargs
             )
             return [
-                DocumentSearchHit(page_content=doc.page_content, score=score)
+                DocumentSearchHit(
+                    page_content=doc.page_content, metadata=doc.metadata, score=score
+                )
                 for doc, score in docs_and_scores
             ]
         else:
@@ -214,7 +216,9 @@ class SelfQueryRetriever(BaseRetriever):
                 query, **search_kwargs
             )
             return [
-                DocumentSearchHit(page_content=doc.page_content, score=score)
+                DocumentSearchHit(
+                    page_content=doc.page_content, metadata=doc.metadata, score=score
+                )
                 for doc, score in docs_and_scores
             ]
         else: