From d0a76d753234e84d48478b2df7a48b00732d1563 Mon Sep 17 00:00:00 2001 From: Kurt Heiden Date: Tue, 29 Oct 2024 08:29:26 -0600 Subject: [PATCH] Update ollama.py --- simplemind/providers/ollama.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/simplemind/providers/ollama.py b/simplemind/providers/ollama.py index de1bfdc..2373e06 100644 --- a/simplemind/providers/ollama.py +++ b/simplemind/providers/ollama.py @@ -10,7 +10,7 @@ NOT_IMPLEMENTED_REASON = """ # Alternate python dependency may be required """ class Ollama: - __name__ = PROVIDER_NAME + NAME = PROVIDER_NAME DEFAULT_MODEL = DEFAULT_MODEL def __init__(self, host_url: str = None): @@ -19,6 +19,8 @@ class Ollama: @property def client(self): """The raw Ollama client.""" + if not self.host_url: + raise ValueError("No ollama host url provided") return ol.Client( timeout=TIMEOUT, host=self.host_url)