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

fix: PEP8 Style

Browse Source
This commit is contained in:
aserrador
2026-03-04 16:20:56 -05:00
parent aab046b2ef
commit 6fc6338413
17 changed files with 1086 additions and 217 deletions
Download Patch File Download Diff File Expand all files Collapse all files

833
poetry.lock generated
View File

File diff suppressed because it is too large Load Diff

4
pyproject.toml
View File

@@ -15,7 +15,9 @@ dependencies = [
"dotenv (>=0.9.9,<0.10.0)",
"fastapi[standard] (>=0.128.5,<0.129.0)",
"langgraph-checkpoint-postgres (>=3.0.4,<4.0.0)",
"psycopg[binary] (>=3.3.3,<4.0.0)"
"psycopg[binary] (>=3.3.3,<4.0.0)",
"fastmcp (>=3.1.0,<4.0.0)",
"flake8 (>=7.3.0,<8.0.0)"
]
[tool.poetry]

18
src/naliiabot/bot/agent/agent.py
View File

@@ -6,24 +6,22 @@ from langchain_core.runnables import RunnableConfig
from typing import Literal, Callable, Any, Union
from .schemas import MessagesState
from dataclasses import dataclass
import logging
@dataclass
class AgentConfig:
"""Configuración inmutable del agente para fácil testing."""
system_prompt: str = (
"helpful assistant that can call tools when needed. Always respond with a message. "
"helpful assistant that can call tools when needed."
" Always respond with a message."
)
max_iterations: int = 10
max_iterations: int = 200
timeout_seconds: float = 30.0
class Agent:
"""
Agente conversacional basado en LangGraph.
Soporta tools pasadas como lista de instancias.
"""
@@ -73,7 +71,9 @@ class Agent:
"""Expone el grafo para inspección en tests."""
return self._build_agent()
async def ainvoke(self, state: dict, config: RunnableConfig | None = None) -> dict:
async def ainvoke(
self, state: dict, config: RunnableConfig | None = None
) -> dict:
"""
Versión asíncrona de invoke, necesaria para checkpointers de DB.
"""
@@ -86,7 +86,9 @@ class Agent:
# Llamamos al método ainvoke del grafo compilado
return await self._compiled_agent.ainvoke(state, config=config)
def invoke(self, state: dict, config: RunnableConfig | None = None) -> dict:
def invoke(
self, state: dict, config: RunnableConfig | None = None
) -> dict:
"""
Ejecuta el agente con el estado inicial.
@@ -98,7 +100,7 @@ class Agent:
"""
if not config:
config : RunnableConfig = {"configurable": {"thread_id": "1"}}
config: RunnableConfig = {"configurable": {"thread_id": "1"}}
if "messages" not in state:
raise ValueError("State must contain 'messages' key")

22
src/naliiabot/bot/prompts/prompts/prompts_template.yaml
View File

@@ -1,7 +1,7 @@
prompts:
NALIIA_PROMPT: |
"""
Eres Naliia, asistente virtual de un centro de belleza.
Eres Naliia, asistente virtual.
Responde usando formato Markdown:
- Usa **negrita** para énfasis.
@@ -9,19 +9,11 @@ prompts:
- Usa saltos de línea entre secciones.
- Destaca opciones importantes con `código inline`.
Ejemplo de formato:
¡Hola! Bienvenido/a, soy **Naliia**, tu asistente virtual.
*Lo que NUNCA debes hacer:*
Puedo ayudarte con:
**Agendar una cita**
Ver horarios disponibles y reservar tu servicio
**Información sobre servicios y productos**
Conocer nuestros tratamientos y productos
**Cancelar una cita**
Si necesitas cancelar alguna reserva previa
¿En qué puedo ayudarte hoy?
❌ Dar diagnósticos médicos
❌ Recomendar tratamientos clínicos sin evaluación
❌ Minimizar riesgos ("es super seguro, no pasa nada")
❌ Prometer resultados específicos
❌ Agendar procedimientos invasivos sin mencionar evaluación previa
"""

99
src/naliiabot/bot/tools/naliia_tools.py
View File

@@ -1,67 +1,70 @@
import asyncio
from datetime import datetime
from fastmcp import Client
from langchain.tools import tool
from ..logger import logger
from .schemes import ScheduleSchema
MCP_SERVER_URL = "http://localhost:8000/mcp"
_MCP_CLIENT = Client(MCP_SERVER_URL)
class NaliiaTools:
def __init__(self):
self._session_active = False
def get_tools(self):
return [
self.verificar_usuario_registrado,
self.registrar_usuario_nuevo,
self.consultar_sedes,
self.consultar_agenda_disponible,
self.agendar_cita
self.get_current_datetime,
self.schedule_appointment
]
@tool
def verificar_usuario_registrado(phone: str) -> bool:
def get_current_datetime() -> datetime:
"""
Verifica si el usuario esta registrado.
Args:
phone: Numero de contacto del cliente ejemplo 30123334
Consulta la fecha actual, no se aceptan fechas
en el pasado con respecto a esta fecha.
"""
logger.info("Llamando a Verificar usuario.")
return datetime.now()
return False
@tool
def registrar_usuario_nuevo(full_name: str, phone: str) -> bool:
@tool(args_schema=ScheduleSchema)
def schedule_appointment(
schedule_date, schedule_time,
service_center, customer,
professional, description
) -> bool:
"""
En caso de que no se haya podido verificar el usuario, sera necesario
registrarlo como un usuario nuevo.
Args:
full_name: Nombre completo del cliente.
phone: Numero de contacto del cliente.
"""
logger.info("Llamando a Registrar Nuevo Usuario.")
return True
@tool
def consultar_sedes() -> list:
"""
Brinda informacion al usuario acerca de las sedes disponibles y sus horarios.
"""
logger.info("Llamando a consultar sedes.")
return []
@tool
def consultar_agenda_disponible() -> list:
"""
Ayuda a consultar al cliente los horarios de atencion disponibles para agendar su cita.
"""
logger.info("Llamando a consultar agenda disponible.")
return []
@tool
def agendar_cita() -> bool:
"""
Ayuda al cliente a confirmar su cita.
Permite al Cliente al Agendar una Cita
"""
logger.info("Llamando a agendar cita.")
logger.info([
schedule_date,
schedule_time,
service_center,
customer,
professional,
description
])
example_schedule = {
'professional': 6,
'description': "Bien Bonito Todo!",
'customer': 4,
'date': '2026-03-25 14:00',
'service_center': 11
}
async def call_tool():
async with _MCP_CLIENT:
result = await _MCP_CLIENT.call_tool(
"create_schedule", example_schedule)
logger.info(result)
asyncio.run(call_tool())
return True

37
src/naliiabot/bot/tools/schemes.py Normal file
View File

@@ -0,0 +1,37 @@
from datetime import date, time
from typing import Optional
from pydantic import BaseModel, Field, PositiveInt
class ScheduleSchema(BaseModel):
"""
Scheme for Schedule Appointment
"""
schedule_date: date = Field(
description="Date in format DD-MM-YYYY")
schedule_time: time = Field(
description="Time in format 12h (HH:MM)")
service_center: Optional[PositiveInt] = Field(
None, description="ID Service center Example: 1")
customer: Optional[PositiveInt] = Field(
None, description="ID Customer Example: 1")
professional: Optional[PositiveInt] = Field(
None, description="ID Professional Example: 1")
description: str = Field(
default="General Schedule",
max_length=255,
description="Max 255 characters."
)
class Config:
json_schema_extra = {
"example": {
"schedule_date": "15-03-2026",
"schedule_time": "02:00",
"service_center": 1,
"professional": 12,
"customer": 999,
"description": "Review"
}
}

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

@@ -1,17 +1,12 @@
import os
from naliiabot.bot.agent.agent import Agent, AgentConfig
from naliiabot.bot.tools.naliia_tools import NaliiaTools
from naliiabot.bot.factories.llm_factory import LLMFactory
from naliiabot.bot.prompts.load_prompt import get_prompt_template
from langgraph.checkpoint.postgres.aio import AsyncPostgresSaver
from naliiabot.bot.agent.agent import Agent
from psycopg_pool import AsyncConnectionPool
from psycopg.rows import dict_row
from ..logger import logger
from .settings import _settings
from .settings import _settings as st
_connection_string = f"postgresql://{_settings.DB_USER}:{_settings.DB_PASSWORD}@localhost:5432/{_settings.DB_NAME}"
_protocol = "postgresql://"
_connection_string =\
f"{_protocol}{st.DB_USER}:{st.DB_PASSWORD}@localhost:5432/{st.DB_NAME}"
agent_instance: Agent | None = None
db_pool = None

12
src/naliiabotapi/api/v1/endpoints/chat.py
View File

@@ -4,8 +4,9 @@ from langchain_core.messages import HumanMessage
router = APIRouter()
@router.post("/chat")
def chat(messages: dict, agent = Depends(get_agent)):
def chat(messages: dict, agent=Depends(get_agent)):
"""
Simulate a chat response based on the input message.
@@ -16,7 +17,12 @@ def chat(messages: dict, agent = Depends(get_agent)):
dict: A dictionary containing the response message.
"""
messages = [HumanMessage(content=messages["messages"])]
response_message = agent.invoke({"messages": messages})
messages = [
HumanMessage(
content=messages["messages"]
)]
response_message = agent.invoke(
{"messages": messages}
)
return {"response": response_message}

16
src/naliiabotapi/api/v1/webhooks/chat_hook.py
View File

@@ -9,6 +9,7 @@ import json
router = APIRouter()
async def send_whatsapp_message(request_data: SendMessageScheme):
"""
Envía un mensaje de texto a través de la API de WhatsApp.
@@ -35,8 +36,13 @@ async def send_whatsapp_message(request_data: SendMessageScheme):
async with httpx.AsyncClient() as client:
try:
response = await client.post(endpoint, json=payload, headers=headers)
response = await client.post(
endpoint,
json=payload,
headers=headers
)
response.raise_for_status()
return response.json()
except Exception as e:
logger.error(f"Error enviando mensaje a WhatsApp: {e}")
@@ -44,7 +50,7 @@ async def send_whatsapp_message(request_data: SendMessageScheme):
@router.post("/webhook")
async def webhook_chat(request: Request, agent = Depends(get_agent)):
async def webhook_chat(request: Request, agent=Depends(get_agent)):
logger.info("Received webhook request")
body = await request.json()
logger.info(f"Webhook payload: {json.dumps(body)}")
@@ -74,7 +80,9 @@ async def webhook_chat(request: Request, agent = Depends(get_agent)):
messages = [HumanMessage(content=user_message)]
agent_response = await agent.ainvoke({"messages": messages}, config=config)
agent_response = await agent.ainvoke(
{"messages": messages}, config=config
)
agent_response_content = agent_response["messages"][-1].content
clean_jid = thread_id.split('@')[0]
@@ -87,7 +95,7 @@ async def webhook_chat(request: Request, agent = Depends(get_agent)):
jid=clean_jid,
text=agent_response_content,
delay=1200
))
))
except Exception as e:
logger.error(f"Error processing webhook: {e}")

8
src/naliiabotapi/main.py
View File

@@ -16,10 +16,10 @@ async def lifespan(app: FastAPI):
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 = deps.LLMFactory(model_name).get_model()
deps.agent_instance = deps.Agent(
model=model,
@@ -35,7 +35,9 @@ async def lifespan(app: FastAPI):
app = FastAPI(
title="NaliiaBot API",
description="API for NaliiaBot, a chatbot that provides customer service and related topics.",
description=(
"API for NaliiaBot, a chatbot that provides "
"customer service and related topics."),
version="1.0.0",
lifespan=lifespan
)
@@ -43,10 +45,12 @@ app = FastAPI(
app.include_router(chat_router)
app.include_router(chat_hook_router)
@app.get("/")
def read_root():
return {"Hello": "World"}
@app.get("/health")
def health_check():
return {"status": "ok"}

17
src/ui/src/ui/app.py
View File

@@ -1,10 +1,8 @@
import streamlit as st
from agent_client import AgentClient
from logger import logger
import asyncio
import json
import httpx
import random
import time
async def main():
@@ -18,7 +16,6 @@ async def main():
with st.chat_message(message["role"]):
st.markdown(message["content"])
if prompt := st.chat_input("What is up?"):
st.chat_message("user").markdown(prompt)
st.session_state.messages.append({"role": "user", "content": prompt})
@@ -26,10 +23,8 @@ async def main():
agent_client = AgentClient()
response = await agent_client.send_message(prompt)
try:
response.raise_for_status()
data = response.json()
except json.JSONDecodeError:
raise RuntimeError(
f"Respuesta inválida (no JSON): {response.text}"
@@ -41,8 +36,10 @@ async def main():
raw_response = response.json()['response']['messages'][-1]['content']
with st.chat_message("assistant"):
formated_response = st.markdown(raw_response)
st.session_state.messages.append({"role": "assistant", "content": raw_response})
st.session_state.messages.append({
"role": "assistant",
"content": raw_response
})
if __name__ == "__main__": import asyncio; asyncio.run(main())
if __name__ == "__main__":
asyncio.run(main())

186
tests/conftest.py
View File

@@ -6,13 +6,15 @@ from src.naliiabot.bot.factories.llm_factory import LLMFactory
from httpx import ASGITransport, AsyncClient
from naliiabotapi.main import app
@pytest.fixture(scope="function")
async def async_client():
"""Fixture para un cliente HTTP asíncrono."""
async with AsyncClient(
transport=ASGITransport(app=app),
base_url="http://localhost:8010") as client:
base_url="http://localhost:8010"
) as client:
yield client
@@ -80,94 +82,94 @@ def send_message_payload():
"remoteJidAlt": "5512992544490@s.whatsapp.net",
"fromMe": False, "id": "AC4367597BDC50537133B9C8848B2E62",
"participant": "", "addressingMode": "lid"
},
"pushName": "Alejandro",
"status": "DELIVERY_ACK",
"message": {
"conversation": "Hola",
"messageContextInfo": {
"threadId": [],
"deviceListMetadata": {
"senderKeyIndexes": [],
"recipientKeyIndexes": [],
"senderKeyHash": {
"0": 154,
"1": 106,
"2": 69,
"3": 189,
"4": 90,
"5": 105,
"6": 172,
"7": 243,
"8": 27,
"9": 98
},
"senderTimestamp": {
"low": 1772156772,
"high": 0,
"unsigned": "true"
},
"recipientKeyHash": {
"0": 16,
"1": 98,
"2": 75,
"3": 100,
"4": 131,
"5": 43,
"6": 24,
"7": 225,
"8": 244,
"9": 126
},
"recipientTimestamp": {
"low": 1772155597,
"high": 0,
"unsigned": True
}
},
"deviceListMetadataVersion": 2,
"messageSecret": {
"0": 100,
"1": 47,
"2": 112,
"3": 133,
"4": 255,
"5": 156,
"6": 129,
"7": 57,
"8": 239,
"9": 219,
"10": 74,
"11": 171,
"12": 212,
"13": 150,
"14": 112,
"15": 83,
"16": 123,
"17": 78,
"18": 189,
"19": 13,
"20": 51,
"21": 133,
"22": 129,
"23": 6,
"24": 197,
"25": 119,
"26": 197,
"27": 56,
"28": 87,
"29": 176,
"30": 135,
"31": 62
}}},
"messageType": "conversation",
"messageTimestamp": 1772168493,
"instanceId": "75a877f1-d772-422b-a6f1-6ac470af5663",
"source": "android"
},
"destination": "http://192.168.58.110:8010/webhook",
"date_time": "2026-02-27T02:01:34.178Z",
"sender": "573016859278@s.whatsapp.net",
"server_url": "http://localhost:8080",
"apikey": "98B6B820AFC3-4544-9A4C-011B0719C2D1"
}
},
"pushName": "Alejandro",
"status": "DELIVERY_ACK",
"message": {
"conversation": "Hola",
"messageContextInfo": {
"threadId": [],
"deviceListMetadata": {
"senderKeyIndexes": [],
"recipientKeyIndexes": [],
"senderKeyHash": {
"0": 154,
"1": 106,
"2": 69,
"3": 189,
"4": 90,
"5": 105,
"6": 172,
"7": 243,
"8": 27,
"9": 98
},
"senderTimestamp": {
"low": 1772156772,
"high": 0,
"unsigned": "true"
},
"recipientKeyHash": {
"0": 16,
"1": 98,
"2": 75,
"3": 100,
"4": 131,
"5": 43,
"6": 24,
"7": 225,
"8": 244,
"9": 126
},
"recipientTimestamp": {
"low": 1772155597,
"high": 0,
"unsigned": True
}
},
"deviceListMetadataVersion": 2,
"messageSecret": {
"0": 100,
"1": 47,
"2": 112,
"3": 133,
"4": 255,
"5": 156,
"6": 129,
"7": 57,
"8": 239,
"9": 219,
"10": 74,
"11": 171,
"12": 212,
"13": 150,
"14": 112,
"15": 83,
"16": 123,
"17": 78,
"18": 189,
"19": 13,
"20": 51,
"21": 133,
"22": 129,
"23": 6,
"24": 197,
"25": 119,
"26": 197,
"27": 56,
"28": 87,
"29": 176,
"30": 135,
"31": 62
}}},
"messageType": "conversation",
"messageTimestamp": 1772168493,
"instanceId": "75a877f1-d772-422b-a6f1-6ac470af5663",
"source": "android"
},
"destination": "http://192.168.58.110:8010/webhook",
"date_time": "2026-02-27T02:01:34.178Z",
"sender": "573016859278@s.whatsapp.net",
"server_url": "http://localhost:8080",
"apikey": "98B6B820AFC3-4544-9A4C-011B0719C2D1"
}

5
tests/test_agent.py
View File

@@ -375,8 +375,9 @@ class TestIntegrationFlows:
assert "messages" in result
assert "llm_calls" in result
assert result["llm_calls"] == 1
assert len(result["messages"]) == 2 # Input + Response
assert result["messages"][-1].content == "¡Hola! ¿En qué puedo ayudarte?"
assert len(result["messages"]) == 2
assert result["messages"][-1].content == (
"¡Hola! ¿En qué puedo ayudarte?")
def test_tool_use_flow(self, agent_with_tool):
"""

4
tests/test_api.py
View File

@@ -1,7 +1,6 @@
import pytest
from fastapi.testclient import TestClient
from src.naliiabotapi.main import app
from langchain_core.messages import HumanMessage
@pytest.fixture
@@ -10,16 +9,19 @@ def client():
return client
def test_root(client):
response = client.get("/")
assert response.status_code == 200
def test_health(client):
response = client.get("/health")
assert response.status_code == 200
def test_chat(client):
response = client.post("/chat", json={"messages": "Hello, how are you?"})

5
tests/test_webhook.py
View File

@@ -26,7 +26,10 @@ class TestWebhookChatBot:
# Sobrescribir la dependencia del agente en la app
app.dependency_overrides[get_agent] = lambda: mock_agent
async with AsyncClient(transport=ASGITransport(app=app), base_url="http://localhost:8010") as client:
async with AsyncClient(
transport=ASGITransport(app=app),
base_url="http://localhost:8010"
) as client:
response = await client.post("/webhook", json=send_message_payload)
# Limpiar override para no afectar a otros tests