OneTeam/NaliiaBot
16
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

chore: Move dependencies to build Agent

Browse Source
This commit is contained in:
aserrador
2026-03-04 19:01:23 -05:00
parent e440c97743
commit 21d0f439c7
2 changed files with 11 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/naliiabotapi/api/dependencies.py
View File

@@ -1,7 +1,4 @@
from naliiabot.bot.agent.agent import Agent, AgentConfig from naliiabot.bot.agent.agent import Agent
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 psycopg_pool import AsyncConnectionPool from psycopg_pool import AsyncConnectionPool
from psycopg.rows import dict_row from psycopg.rows import dict_row

15
src/naliiabotapi/main.py
View File

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