community[patch]: Return PK in similarity search Document (langchain-…

…ai#17561) Issue: langchain-ai#17390 Co-authored-by: hasan <[email protected]>
al1p · Feb 27, 2024 · 557b1bc · 557b1bc
1 parent 4677149
commit 557b1bc
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.