JDSS/mem0
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix: Ollama checking model exists (#2682)

Browse Source
This commit is contained in:
Vương Hữu Hưng (Hans)
2025-08-19 16:29:26 +05:30
committed by GitHub
parent e7013764f7
commit ff399e5528
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mem0/embeddings/ollama.py
+1 -1
View File
@@ -36,7 +36,7 @@ class OllamaEmbedding(EmbeddingBase):
Ensure the specified model exists locally. If not, pull it from Ollama. Ensure the specified model exists locally. If not, pull it from Ollama.
""" """
local_models = self.client.list()["models"] local_models = self.client.list()["models"]
if not any(model.get("name") == self.config.model for model in local_models): if not any(model.get("name") == self.config.model or model.get("model") == self.config.model for model in local_models):
self.client.pull(self.config.model) self.client.pull(self.config.model)
def embed(self, text, memory_action: Optional[Literal["add", "search", "update"]] = None): def embed(self, text, memory_action: Optional[Literal["add", "search", "update"]] = None):