Skip to content

Commit

Permalink
fix imports
Browse files Browse the repository at this point in the history
  • Loading branch information
w4ffl35 committed Aug 15, 2024
1 parent 4962b54 commit d49a74c
Show file tree
Hide file tree
Showing 4 changed files with 12 additions and 12 deletions.
4 changes: 2 additions & 2 deletions src/airunner_nexus/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down
6 changes: 3 additions & 3 deletions src/airunner_nexus/llm/llm_handler.py
Original file line number Diff line number Diff line change
Expand Up @@ -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):
Expand Down
4 changes: 2 additions & 2 deletions src/airunner_nexus/llm/rag_mixin.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down
10 changes: 5 additions & 5 deletions src/airunner_nexus/server.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down

0 comments on commit d49a74c

Please sign in to comment.