Fix bug in building messages with async models

Co-authored-by: Sukhbinder Singh <sukhbindersingh@gmail.com>
taketwo · Dec 20, 2024 · 39f73a3 · 39f73a3
1 parent e5d981b
commit 39f73a3
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/llm_ollama.py b/llm_ollama.py
@@ -166,7 +166,9 @@ def build_messages(self, prompt, conversation):
                     for attachment in prev_response.attachments
                 ]
 
-            messages.append({"role": "assistant", "content": prev_response.text()})
+            messages.append(
+                {"role": "assistant", "content": prev_response.text_or_raise()}
+            )
         if prompt.system and prompt.system != current_system:
             messages.append({"role": "system", "content": prompt.system})
         messages.append({"role": "user", "content": prompt.prompt})

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "llm-ollama"
-version = "0.8.0"
+version = "0.8.1"
 description = "LLM plugin providing access to local Ollama models"
 readme = "README.md"
 authors = [{ name = "Sergey Alexandrov" }]