From 20ff3dc8c9db425f7362faf6499e2c93bbd5d32b Mon Sep 17 00:00:00 2001 From: aserrador Date: Mon, 16 Feb 2026 10:50:40 -0500 Subject: [PATCH] feat: Add tools strategy --- src/naliiabot/bot/tools/naliia_tools.py | 13 +++++++++++++ src/naliiabotapi/api/dependencies.py | 9 ++++++++- 2 files changed, 21 insertions(+), 1 deletion(-) create mode 100644 src/naliiabot/bot/tools/naliia_tools.py diff --git a/src/naliiabot/bot/tools/naliia_tools.py b/src/naliiabot/bot/tools/naliia_tools.py new file mode 100644 index 0000000..ae9d748 --- /dev/null +++ b/src/naliiabot/bot/tools/naliia_tools.py @@ -0,0 +1,13 @@ +from langchain.tools import tool + + +class NaliiaTools: + + @tool + def verificar_usuario_registrado(self): + """ + Verifica si el usuario que esta contactandose es un + usuario ya conocido y registrado en la base de datos. + """ + + return True diff --git a/src/naliiabotapi/api/dependencies.py b/src/naliiabotapi/api/dependencies.py index 26f48d9..52fc540 100644 --- a/src/naliiabotapi/api/dependencies.py +++ b/src/naliiabotapi/api/dependencies.py @@ -1,6 +1,7 @@ import os from src.naliiabot.bot.agent.agent import Agent, AgentConfig +from src.naliiabot.bot.tools.naliia_tools import NaliiaTools from src.naliiabot.bot.factories.llm_factory import LLMFactory from src.naliiabot.bot.prompts.load_prompt import get_prompt_template @@ -16,6 +17,12 @@ def get_agent(): NALIIA_PROMPT = get_prompt_template( "NALIIA_PROMPT") + naliia_tools = [ + t for t in dir(NaliiaTools()) if not t.startswith('__') and not t.endswith('__') + ] + config = AgentConfig(system_prompt=NALIIA_PROMPT) model = LLMFactory(MODEL_NAME).get_model() - return Agent(model=model) \ No newline at end of file + + raise Exception(naliia_tools) + return Agent(model=model, config=config, tools=naliia_tools)