Update to milvus.py to load the configuration also in the init_collec…

…tion function
Significant-Gravitas · Apr 24, 2023 · b7defd2 · b7defd2
1 parent 04f6ffd
commit b7defd2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/autogpt/memory/milvus.py b/autogpt/memory/milvus.py
@@ -31,7 +31,7 @@ def __init__(self, cfg: Config) -> None:
             secure=self.secure,
         )
 
-        self.init_collection()
+        self.init_collection(cfg)
 
     def configure(self, cfg: Config) -> None:
         # init with configuration.
@@ -69,7 +69,7 @@ def configure(self, cfg: Config) -> None:
                 "params": {},
             }
 
-    def init_collection(self) -> None:
+    def init_collection(self, cfg: Config) -> None:
         """Initialize collection in vector database."""
         fields = [
             FieldSchema(name="pk", dtype=DataType.INT64, is_primary=True, auto_id=True),