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)