app.py

from fastapi import FastAPI, Request
import uvicorn
from dotenv import load_dotenv
from pydantic import BaseModel
from supervisor_workflow import SupervisorWorkflow
from crag_agent import process_request_crag

load_dotenv()

app = FastAPI()

supervised_workflow = SupervisorWorkflow()
supervised_chain = supervised_workflow.gen_chain()

# TODO manage request body
class SupervisorData(BaseModel):
    message: str
    user_id: str
    thread_id: str

# TODO provide the thread_id and the user_id
@app.post("/supervisor")
async def supervisor(supervisor_request: SupervisorData):
    request_body = supervisor_request.model_dump()
    message = request_body["message"]
    reply = supervised_chain.invoke(
        message,
        {
            "recursion_limit": 150,
            "user_id": "restebance@gmail.com",
            "thread_id": "16"
        },
    )
    print(reply['response'])
    return {
        "data": {
            "message": reply['response']
        }
    }