community[patch]: Return PK in similarity search Document (#17561)

Issue: #17390 Co-authored-by: hasan <[email protected]>
langchain-ai · Feb 22, 2024 · 7248e98 · 7248e98
1 parent 1ec8199
commit 7248e98
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/libs/community/langchain_community/vectorstores/milvus.py b/libs/community/langchain_community/vectorstores/milvus.py
@@ -739,8 +739,8 @@ def similarity_search_with_score_by_vector(
         if param is None:
             param = self.search_params
 
-        # Determine result metadata fields.
-        output_fields = [x for x in self.fields if x != self._primary_field]
+        # Determine result metadata fields with PK.
+        output_fields = self.fields[:]
         output_fields.remove(self._vector_field)
 
         # Perform the search.