From 6927e81a187786ae0c1221982aba163b378a5c61 Mon Sep 17 00:00:00 2001 From: HoaNQ9 Date: Fri, 2 Feb 2024 12:26:10 +0700 Subject: [PATCH] Change text pk by using the parameter primary_field --- libs/community/langchain_community/vectorstores/milvus.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/community/langchain_community/vectorstores/milvus.py b/libs/community/langchain_community/vectorstores/milvus.py index cbae18a5102573..b1af2d9c4cc026 100644 --- a/libs/community/langchain_community/vectorstores/milvus.py +++ b/libs/community/langchain_community/vectorstores/milvus.py @@ -1012,14 +1012,14 @@ def get_ids(self, try: query_result = self.col.query( expr=expr, - output_fields=["pk"] + output_fields=[self._primary_field] ) except MilvusException as exc: logger.error( "Failed to get ids: %s error: %s", self.collection_name, exc ) raise exc - ids = [item.get("pk") for item in query_result] + ids = [item.get(self._primary_field) for item in query_result] return ids def upsert(self,