server.py

from dotenv import load_dotenv 
load_dotenv()
from rag_queue.queue.worker import process_query
from .client.rq_client import queue
from fastapi import FastAPI, Query

app = FastAPI()

@app.get("/")
def read_root():
    return {"status": "Server is up and running!"}

@app.post("/chat")
def chat(
        query: str = Query(..., description="The question to ask the AI assistant.")
):
   job = queue.enqueue(process_query, query)
   
   return {"status": "queued", "job_id": job.id}

@app.get("/job-status")
def get_job_result(
        job_id: str = Query(..., description="Job ID")
):
    job = queue.fetch_job(job_id = job_id)
    result = job.return_value()

    return {"result": result}