feat: Implement WebSocket support for chat functionality and refactor chat service
- Added WebSocket endpoint for real-time chat interactions. - Refactored ChatPage component to utilize WebSocket for sending and receiving messages. - Updated chat service to handle streaming responses from the LLM agent. - Introduced error handling for WebSocket connections and message processing. - Modified Editor_Test to include AppShellWithMenu for better layout. - Adjusted file path in generar_tree.py for correct directory structure. - Created llm_chat_endpoint_v1.py and llm_chat_srvc.py for handling chat requests and responses. - Established logging for WebSocket interactions and errors.
This commit is contained in:
@@ -0,0 +1,35 @@
|
||||
from fastapi import WebSocket, APIRouter, WebSocketDisconnect
|
||||
from backend.domains.llms.llm_chat_srvc import construir_agente_llm
|
||||
from src.Logger.logger_db import LoggerDB, logger
|
||||
from entrypoint.init_db import db_credencial
|
||||
import json
|
||||
|
||||
LoggerDB(db_credencial, "logger_llm_ws", created_by="sistema")
|
||||
|
||||
router = APIRouter()
|
||||
agente = construir_agente_llm()
|
||||
|
||||
@router.websocket("/ws/chat")
|
||||
async def chat_ws(websocket: WebSocket):
|
||||
await websocket.accept()
|
||||
try:
|
||||
data = await websocket.receive_text()
|
||||
parsed = json.loads(data)
|
||||
prompt = parsed.get("prompt")
|
||||
if not prompt:
|
||||
await websocket.send_text("⚠️ Prompt vacío.")
|
||||
await websocket.close()
|
||||
return
|
||||
|
||||
# ✅ Solución: hacer await antes de iterar
|
||||
respuesta_gen = await agente.interactuar_en_bucle(prompt=prompt, stream=True)
|
||||
async for token in respuesta_gen:
|
||||
await websocket.send_text(token)
|
||||
|
||||
await websocket.close()
|
||||
|
||||
except WebSocketDisconnect:
|
||||
logger.info("🔌 WebSocket desconectado por el cliente.")
|
||||
except Exception as e:
|
||||
logger.exception("❌ Error en WebSocket:")
|
||||
await websocket.close()
|
||||
Reference in New Issue
Block a user