diff --git a/Docker/src/server.py b/Docker/src/server.py index 861fd16..128a582 100644 --- a/Docker/src/server.py +++ b/Docker/src/server.py @@ -1,7 +1,6 @@ import logging import os from concurrent import futures -from typing import Any import brunix_pb2 import brunix_pb2_grpc @@ -13,20 +12,10 @@ from utils.llm_factory import create_chat_model from utils.emb_factory import create_embedding_model from graph import build_graph -# PROJECT_ROOT = Path(__file__).resolve().parents[2] -# if str(PROJECT_ROOT) not in sys.path: -# sys.path.insert(0, str(PROJECT_ROOT)) logging.basicConfig(level=logging.INFO) logger = logging.getLogger("brunix-engine") - -def _provider_kwargs(provider: str, base_url: str) -> dict[str, Any]: - if provider == "ollama": - return {"base_url": base_url} - return {} - - class BrunixEngine(brunix_pb2_grpc.AssistanceEngineServicer): def __init__(self): self.llm = create_chat_model( diff --git a/src/utils/__init__.py b/src/utils/__init__.py new file mode 100644 index 0000000..e69de29 diff --git a/src/emb_factory.py b/src/utils/emb_factory.py similarity index 100% rename from src/emb_factory.py rename to src/utils/emb_factory.py diff --git a/src/llm_factory.py b/src/utils/llm_factory.py similarity index 100% rename from src/llm_factory.py rename to src/utils/llm_factory.py