diff --git a/libs/community/langchain_community/vectorstores/milvus.py b/libs/community/langchain_community/vectorstores/milvus.py index 365ada0ce6a..0592e932af6 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