From 21d0f439c75d797072e4ae7bef85cfeb5f4582ed Mon Sep 17 00:00:00 2001 From: aserrador Date: Wed, 4 Mar 2026 19:01:23 -0500 Subject: [PATCH] chore: Move dependencies to build Agent --- src/naliiabotapi/api/dependencies.py | 5 +---- src/naliiabotapi/main.py | 15 ++++++++++----- 2 files changed, 11 insertions(+), 9 deletions(-) diff --git a/src/naliiabotapi/api/dependencies.py b/src/naliiabotapi/api/dependencies.py index 89da0f5..5e5868d 100644 --- a/src/naliiabotapi/api/dependencies.py +++ b/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 diff --git a/src/naliiabotapi/main.py b/src/naliiabotapi/main.py index 6159154..91f88cd 100644 --- a/src/naliiabotapi/main.py +++ b/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,