Support async client (#1980)

2024-10-22 12:42:55 +05:30
parent c5d298eec8
commit fbf1d8c372
11 changed files with 213 additions and 58 deletions
--- a/mem0/vector_stores/milvus.py
+++ b/mem0/vector_stores/milvus.py
@@ -76,11 +76,8 @@ class MilvusDB(VectorStoreBase):
            schema = CollectionSchema(fields, enable_dynamic_field=True)

            index = self.client.prepare_index_params(
-                field_name="vectors",
-                metric_type=metric_type,
-                index_type="AUTOINDEX",
-                index_name="vector_index"  
-                )
+                field_name="vectors", metric_type=metric_type, index_type="AUTOINDEX", index_name="vector_index"
+            )
            self.client.create_collection(collection_name=collection_name, schema=schema, index_params=index)

    def insert(self, ids, vectors, payloads, **kwargs: Optional[dict[str, any]]):