From bc87753f2de56ea7a3e1ae159aec05d401230890 Mon Sep 17 00:00:00 2001
From: acano <acturnes@mrhouston.net>
Date: Tue, 3 Mar 2026 14:15:54 +0100
Subject: [PATCH] Implement embedding and chat model factories for multiple
 providers

---
 Docker/src/server.py           | 11 -----------
 src/utils/__init__.py          |  0
 src/{ => utils}/emb_factory.py |  0
 src/{ => utils}/llm_factory.py |  0
 4 files changed, 11 deletions(-)
 create mode 100644 src/utils/__init__.py
 rename src/{ => utils}/emb_factory.py (100%)
 rename src/{ => utils}/llm_factory.py (100%)

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