harshith-7/discover-agent

1import uvicorn
2from fastapi import FastAPI, Request
3from fastapi.responses import JSONResponse
4from helper import helper
5import logging
6
7app = FastAPI()
8
9@app.post("/get_answer")
10async def get_answer(request: Request):
11    body = await request.json()
12    query = body.get('query')
13    print("Sending to generate the answer")
14    answers = helper.generate(query)
15    for answer in answers:
16        answer.pretty_print()
17    print("Sending response")
18    return JSONResponse(content={'answer': answers[-1].content}, status_code=200)
19
20if __name__=="__main__":
21    uvicorn.run(app, host="localhost", port=2911)