feat: Add schemes

This commit is contained in:
2026-02-24 23:36:44 -05:00
parent 42dc2099d2
commit 5453d4eed8
3 changed files with 35 additions and 9 deletions

View File

@@ -2,15 +2,24 @@ from fastapi import APIRouter, Depends, Request
from langchain.messages import HumanMessage from langchain.messages import HumanMessage
from ...dependencies import get_agent from ...dependencies import get_agent
from ....logger import logger from ....logger import logger
from ....schemes.chat_schemes import SendMessageScheme
import httpx import httpx
import json import json
router = APIRouter() router = APIRouter()
async def send_whatsapp_message(instance_url: str, instance_name: str, apikey: str, jid: str, text: str): async def send_whatsapp_message(request_data: SendMessageScheme):
""" """
Envía un mensaje de texto a través de la API de WhatsApp. Envía un mensaje de texto a través de la API de WhatsApp.
""" """
instance_url = request_data["instance_url"]
instance_name = request_data["instance_name"]
apikey = request_data["apikey"]
jid = request_data["jid"]
text = request_data["text"]
delay = request_data.get("delay", 1200)
endpoint = f"{instance_url}/message/sendText/{instance_name}" endpoint = f"{instance_url}/message/sendText/{instance_name}"
headers = { headers = {
"apikey": apikey, "apikey": apikey,
@@ -20,7 +29,7 @@ async def send_whatsapp_message(instance_url: str, instance_name: str, apikey: s
payload = { payload = {
"number": jid, "number": jid,
"text": text, "text": text,
"delay": 1200, "delay": delay,
"linkPreview": False "linkPreview": False
} }
@@ -38,10 +47,15 @@ async def send_whatsapp_message(instance_url: str, instance_name: str, apikey: s
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") logger.info("Received webhook request")
body = await request.json() body = await request.json()
logger.info(f"Webhook payload: {json.dumps(body)}")
try: try:
data = body.get("data", {}) data = body.get("data", {})
key = data.get("key", {}) key = data.get("key", {})
remote_jid = data.get("key", {}).get("remoteJid")
instance_name = body.get("instance")
server_url = body.get("server_url")
api_key = body.get("apikey")
if key.get("fromMe") is True: if key.get("fromMe") is True:
logger.info("Mensaje enviado por mí (bot), Ignorando.") logger.info("Mensaje enviado por mí (bot), Ignorando.")
@@ -70,12 +84,14 @@ async def webhook_chat(request: Request, agent = Depends(get_agent)):
clean_jid = thread_id.split('@')[0] clean_jid = thread_id.split('@')[0]
await send_whatsapp_message( await send_whatsapp_message(
instance_url="http://localhost:8080", SendMessageScheme(
instance_name="ALEJANDRO AYALA", instance_url=server_url,
apikey="98B6B820AFC3-4544-9A4C-011B0719C2D1", instance_name=instance_name,
jid=clean_jid, apikey=api_key,
text=agent_response_content jid=clean_jid,
) text=agent_response_content,
delay=1200
))
except Exception as e: except Exception as e:
logger.error(f"Error processing webhook: {e}") logger.error(f"Error processing webhook: {e}")

View File

View File

@@ -0,0 +1,10 @@
from typing import TypedDict
class SendMessageScheme(TypedDict):
instance_url: str
instance_name: str
apikey: str
jid: str
delay: int = 1200
text: str