From 0a62f08bff738893813764c3059cfe0eb96f6c8d Mon Sep 17 00:00:00 2001 From: artitw Date: Mon, 7 Oct 2024 04:26:27 +0000 Subject: [PATCH] Fix Assistant --- setup.py | 2 +- text2text/__init__.py | 3 ++- text2text/assistant.py | 4 ++-- 3 files changed, 5 insertions(+), 4 deletions(-) diff --git a/setup.py b/setup.py index 8b47ec4..0393f2b 100755 --- a/setup.py +++ b/setup.py @@ -5,7 +5,7 @@ setuptools.setup( name="text2text", - version="1.6.3", + version="1.6.4", author="artitw", author_email="artitw@gmail.com", description="Text2Text: Crosslingual NLP/G toolkit", diff --git a/text2text/__init__.py b/text2text/__init__.py index 3abf6b7..8c6e6f8 100755 --- a/text2text/__init__.py +++ b/text2text/__init__.py @@ -13,4 +13,5 @@ from .variator import Variator from .identifier import Identifier from .server import Server -from .handler import Handler \ No newline at end of file +from .handler import Handler +from . import utils \ No newline at end of file diff --git a/text2text/assistant.py b/text2text/assistant.py index 55fec9e..a3707ad 100644 --- a/text2text/assistant.py +++ b/text2text/assistant.py @@ -6,7 +6,7 @@ import warnings import importlib.resources -from . import utils +import text2text as t2t from llama_index.llms.ollama import Ollama from llama_index.core.llms import ChatMessage @@ -64,7 +64,7 @@ def load_model(self): if return_code != 0: raise Exception("Cannot install lshw.") - inp_file = importlib.resources.files(utils) / 'ollama_install.sh' + inp_file = importlib.resources.files(t2t.utils) / 'ollama_install.sh' with inp_file.open("rt") as f: install_script = f.read() result = run_sh(install_script)