From c07c0da01a9c3ca10011ca35a1ec3b2ae9a5a990 Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Sat, 10 Feb 2024 07:48:37 +0800 Subject: [PATCH] community[patch]: Fix Milvus add texts when ids=None (#17021) - **Description:** Fix Milvus add texts when ids=None (auto_id=True) Signed-off-by: Jael Gu Co-authored-by: Bagatur --- libs/community/langchain_community/vectorstores/milvus.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/libs/community/langchain_community/vectorstores/milvus.py b/libs/community/langchain_community/vectorstores/milvus.py index 365ada0ce6..0592e932af 100644 --- a/libs/community/langchain_community/vectorstores/milvus.py +++ b/libs/community/langchain_community/vectorstores/milvus.py @@ -583,7 +583,9 @@ class Milvus(VectorStore): # Grab end index end = min(i + batch_size, total_count) # Convert dict to list of lists batch for insertion - insert_list = [insert_dict[x][i:end] for x in self.fields] + insert_list = [ + insert_dict[x][i:end] for x in self.fields if x in insert_dict + ] # Insert into the collection. try: res: Collection