Support model config in LLMs (#1495)

2024-07-18 21:51:40 +05:30
parent c411dc294e
commit 40c9abe484
15 changed files with 172 additions and 41 deletions
--- a/mem0/memory/main.py
+++ b/mem0/memory/main.py
@@ -82,7 +82,7 @@ class Memory(MemoryBase):
                f"Unsupported vector store type: {self.config.vector_store_type}"
            )

-        self.llm = LlmFactory.create(self.config.llm.provider)
+        self.llm = LlmFactory.create(self.config.llm.provider, self.config.llm.config)
        self.db = SQLiteManager(self.config.history_db_path)
        self.collection_name = self.config.collection_name
        self.vector_store.create_col(