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

15
src/naliiabotapi/main.py
View File

@@ -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,