Fix Assistant

artitw · Oct 7, 2024 · d0768e0 · d0768e0
1 parent e8bb2e2
commit d0768e0
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
   name="text2text",
-  version="1.6.0",
+  version="1.6.1",
   author="artitw",
   author_email="artitw@gmail.com",
   description="Text2Text: Crosslingual NLP/G toolkit",

diff --git a/text2text/assistant.py b/text2text/assistant.py
@@ -9,9 +9,12 @@
 from llama_index.core.llms import ChatMessage
 
 def ollama_version():
-  result = subprocess.check_output(["ollama", "-v"], stderr=subprocess.STDOUT).decode("utf-8")
-  if result.startswith("ollama version "):
-    return result.replace("ollama version ", "")
+  try:
+    result = subprocess.check_output(["ollama", "-v"], stderr=subprocess.STDOUT).decode("utf-8")
+    if result.startswith("ollama version "):
+      return result.replace("ollama version ", "")
+  except Exception as e:
+    warnings.warn(str(e))
   return ""
 
 class Assistant(object):