From edbd166ec646aa37fa0b883ca2dd33a3316c9d75 Mon Sep 17 00:00:00 2001 From: Kurt Heiden Date: Mon, 28 Oct 2024 12:46:28 -0600 Subject: [PATCH] Update ollama.py --- simplemind/integrations/ollama.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/simplemind/integrations/ollama.py b/simplemind/integrations/ollama.py index 6bb1d7e..7581c61 100644 --- a/simplemind/integrations/ollama.py +++ b/simplemind/integrations/ollama.py @@ -32,7 +32,7 @@ class Ollama(BaseClientProvider): @property def available_models(self): - """Returns the available models from the OpenAI client.""" + """Returns the available models from the Ollama client.""" def gen(): for model in self.client.list().get('models'): @@ -41,7 +41,7 @@ class Ollama(BaseClientProvider): return [g for g in gen()] def test_connection(self): - """Test the connection to ollama. Returns True if successful.""" + """Test the connection to Ollama. Returns True if successful.""" return bool(len(self.available_models)) @@ -72,7 +72,7 @@ class Ollama(BaseClientProvider): ) def message(self, message=None, message_history=None, response_model=False, **kwargs): - """Generates a response from the OpenAI client.""" + """Generates a response from the Ollama client.""" use_instructor = bool(response_model) client = self.instructor_client if use_instructor else self.client