From ff399e552806f58bbeada71ca6b75b47d7b53f2c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?V=C6=B0=C6=A1ng=20H=E1=BB=AFu=20H=C6=B0ng=20=28Hans=29?=
 <92346765+vuonghuuhung@users.noreply.github.com>
Date: Tue, 19 Aug 2025 17:59:26 +0700
Subject: [PATCH] Fix: Ollama checking model exists (#2682)

---
 mem0/embeddings/ollama.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mem0/embeddings/ollama.py b/mem0/embeddings/ollama.py
index 6f6d11810..49b7c2e94 100644
--- a/mem0/embeddings/ollama.py
+++ b/mem0/embeddings/ollama.py
@@ -36,7 +36,7 @@ class OllamaEmbedding(EmbeddingBase):
         Ensure the specified model exists locally. If not, pull it from Ollama.
         """
         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)
 
     def embed(self, text, memory_action: Optional[Literal["add", "search", "update"]] = None):