Spaces:
Sleeping
Sleeping
File size: 1,830 Bytes
1f725d8 5551822 1f725d8 5551822 1f725d8 5551822 1f725d8 5551822 1f725d8 5551822 1f725d8 5551822 1f725d8 5551822 1f725d8 5551822 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 | from fastapi import APIRouter, Request, Query
import logging
import logging
from src.MultiRag.pipeline.run_pipeline import RunPipeline
from src.MultiRag.graph.builder import graph
from src.MultiRag.models.rag_model import Content
from api.MultiRag.controllers.loadUserContent_component import load_user_content
from exception import MyException
router = APIRouter()
run_pipeline = RunPipeline()
async def run_agent(user_id, thread_id, userQuery: str):
logging.info(f"Starting AIAgents application for thread: {thread_id}")
try:
temp_user_content = await load_user_content(thread_id)
res = await run_pipeline.initiate(
thread_id=thread_id,
query=userQuery,
userContent=temp_user_content
)
return res
except Exception as e:
logging.error(f"Application failed: {e}")
raise MyException("AIAgents application failed") from e
finally:
logging.info("AIAgents application finished.")
@router.post("/chat")
async def chat(req: Request, message: str = Query(...)):
try:
user_id = req.headers.get("user_id")
thread_id = req.headers.get("thread_id") or user_id
if not user_id:
return {"data": "User ID missing in headers"}
res = await run_agent(user_id, thread_id, message)
# Extract the last message content to send to frontend
messages = res.get("messages", [])
if messages:
last_msg = messages[-1]
content = last_msg.content if hasattr(last_msg, 'content') else str(last_msg)
return {"data": content}
return {"data": "No response from agent."}
except Exception as e:
logging.error(f"Chat endpoint error: {e}")
return {"data": "Chat failed"}
|