diff --git a/src/airunner_nexus/client.py b/src/airunner_nexus/client.py index 3074309..2300594 100644 --- a/src/airunner_nexus/client.py +++ b/src/airunner_nexus/client.py @@ -4,8 +4,8 @@ import time from datetime import datetime -from ainexus.llm.agent import Agent -from ainexus.settings import DEFAULT_HOST, DEFAULT_PORT, PACKET_SIZE, USER_NAME, BOT_NAME, LLM_INSTRUCTIONS +from airunner_nexus.llm.agent import Agent +from airunner_nexus.settings import DEFAULT_HOST, DEFAULT_PORT, PACKET_SIZE, USER_NAME, BOT_NAME, LLM_INSTRUCTIONS class Client: diff --git a/src/airunner_nexus/llm/llm_handler.py b/src/airunner_nexus/llm/llm_handler.py index 915f31f..d6651a9 100644 --- a/src/airunner_nexus/llm/llm_handler.py +++ b/src/airunner_nexus/llm/llm_handler.py @@ -5,9 +5,9 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig from transformers.generation.streamers import TextIteratorStreamer -from ainexus import settings -from ainexus.llm.external_condition_stopping_criteria import ExternalConditionStoppingCriteria -from ainexus.settings import MODEL_BASE_PATH, MODELS +from airunner_nexus import settings +from airunner_nexus.llm.external_condition_stopping_criteria import ExternalConditionStoppingCriteria +from airunner_nexus.settings import MODEL_BASE_PATH, MODELS class LLMHandler():#RagMixin): diff --git a/src/airunner_nexus/llm/rag_mixin.py b/src/airunner_nexus/llm/rag_mixin.py index 9a83891..951bcfc 100644 --- a/src/airunner_nexus/llm/rag_mixin.py +++ b/src/airunner_nexus/llm/rag_mixin.py @@ -9,8 +9,8 @@ from llama_index.core import Settings from llama_index.core.readers.json import JSONReader -from ainexus.llm.enums import AgentState -from ainexus.llm.external_condition_stopping_criteria import ExternalConditionStoppingCriteria +from airunner_nexus.llm.enums import AgentState +from airunner_nexus.llm.external_condition_stopping_criteria import ExternalConditionStoppingCriteria class RagMixin: diff --git a/src/airunner_nexus/server.py b/src/airunner_nexus/server.py index 3b7fb55..cb1a524 100644 --- a/src/airunner_nexus/server.py +++ b/src/airunner_nexus/server.py @@ -6,11 +6,11 @@ import queue from typing import Optional -from ainexus import settings -from ainexus.llm.llm_handler import LLMHandler -from ainexus.logger import logger -from ainexus.exceptions import FailedToSendError, NoConnectionToClientError -import ainexus.messagecodes as codes +from airunner_nexus import settings +from airunner_nexus.llm.llm_handler import LLMHandler +from airunner_nexus.logger import logger +from airunner_nexus.exceptions import FailedToSendError, NoConnectionToClientError +import airunner_nexus.messagecodes as codes class Server: