diff --git a/setup.py b/setup.py index 443dfd0..b18df30 100755 --- 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 index d1ca066..e1877ac 100644 --- 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):