diff --git a/emma/emma_backend_ollama.py b/emma/emma_backend_ollama.py index b422047..3e92844 100644 --- a/emma/emma_backend_ollama.py +++ b/emma/emma_backend_ollama.py @@ -19,7 +19,7 @@ class OllamaBackend: def embed(self, text: str) -> np.ndarray: r = requests.post(f"{self.base_url}/api/embeddings", - json={"model": self.embed_model, "prompt": text}, timeout=30) + json={"model": self.embed_model, "prompt": text}, timeout=120) r.raise_for_status() return np.array(r.json()["embedding"], dtype=np.float32) @@ -55,7 +55,7 @@ class OllamaBackend: ], "stream": False, }, - timeout=60, + timeout=120, ) r.raise_for_status() return r.json()["message"]["content"]