update ollama code to get model.model property instead of model.name

embedchain/embedchain/embedder/ollama.py

@@ -21,7 +21,7 @@ def __init__(self, config: Optional[OllamaEmbedderConfig] = None):
 
         client = Client(host=config.base_url)
         local_models = client.list()["models"]
-        if not any(model.get("name") == self.config.model for model in local_models):
+        if not any(model.get("model") == self.config.model for model in local_models):
             logger.info(f"Pulling {self.config.model} from Ollama!")
             client.pull(self.config.model)
         embeddings = OllamaEmbeddings(model=self.config.model, base_url=config.base_url)

embedchain/embedchain/llm/ollama.py

@@ -28,7 +28,7 @@ def __init__(self, config: Optional[BaseLlmConfig] = None):
 
         client = Client(host=config.base_url)
         local_models = client.list()["models"]
-        if not any(model.get("name") == self.config.model for model in local_models):
+        if not any(model.get("model") == self.config.model for model in local_models):
             logger.info(f"Pulling {self.config.model} from Ollama!")
             client.pull(self.config.model)
 

mem0/embeddings/ollama.py

@@ -36,7 +36,7 @@ def _ensure_model_exists(self):
         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("model") == self.config.model for model in local_models):
             self.client.pull(self.config.model)
 
     def embed(self, text):

mem0/llms/ollama.py

@@ -23,7 +23,7 @@ def _ensure_model_exists(self):
         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("model") == self.config.model for model in local_models):
             self.client.pull(self.config.model)
 
     def _parse_response(self, response, tools):

tests/embeddings/test_ollama_embeddings.py

@@ -8,7 +8,7 @@
 def mock_ollama_client():
     with patch("mem0.embeddings.ollama.Client") as mock_ollama:
         mock_client = Mock()
-        mock_client.list.return_value = {"models": [{"name": "nomic-embed-text"}]}
+        mock_client.list.return_value = {"models": [{"model": "nomic-embed-text"}]}
         mock_ollama.return_value = mock_client
         yield mock_client