chore: Move dependencies to build Agent

src/naliiabotapi/api/dependencies.py

@@ -1,7 +1,4 @@
-from naliiabot.bot.agent.agent import Agent, AgentConfig
-from naliiabot.bot.factories.llm_factory import LLMFactory
-from naliiabot.bot.prompts.load_prompt import get_prompt_template
-from naliiabot.bot.tools.naliia_tools import NaliiaTools
+from naliiabot.bot.agent.agent import Agent
 from psycopg_pool import AsyncConnectionPool
 from psycopg.rows import dict_row
 

src/naliiabotapi/main.py

@@ -1,4 +1,9 @@
+import os
 from fastapi import FastAPI
+from naliiabot.bot.agent.agent import AgentConfig
+from naliiabot.bot.factories.llm_factory import LLMFactory
+from naliiabot.bot.prompts.load_prompt import get_prompt_template
+from naliiabot.bot.tools.naliia_tools import NaliiaTools
 from naliiabotapi.api.v1.endpoints.chat import router as chat_router
 from naliiabotapi.api.v1.webhooks.chat_hook import router as chat_hook_router
 import naliiabotapi.api.dependencies as deps
@@ -15,11 +20,11 @@ async def lifespan(app: FastAPI):
         checkpointer = AsyncPostgresSaver(conn)
         await checkpointer.setup()
 
-        model_name = "anthropic"
-        model = deps.LLMFactory(model_name).get_model()
-        naliia_prompt = deps.get_prompt_template("NALIIA_PROMPT")
-        naliia_tools = deps.NaliiaTools().get_tools()
-        config = deps.AgentConfig(system_prompt=naliia_prompt)
+        model_name = os.getenv("LLM_MODEL", "anthropic")
+        model = LLMFactory(model_name).get_model()
+        naliia_prompt = get_prompt_template("NALIIA_PROMPT")
+        naliia_tools = NaliiaTools().get_tools()
+        config = AgentConfig(system_prompt=naliia_prompt)
 
         deps.agent_instance = deps.Agent(
             model=model,