File size: 465 Bytes
a5d2270 9548472 a5d2270 905d2fe 9548472 3b255c5 905d2fe 9548472 a5d2270 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
from fastapi import FastAPI, Request
from rag_system import RAGSystem
app = FastAPI()
rag = RAGSystem(vector_db_path="vector_db")
rag.load_vectorstore()
rag.load_llm()
rag.get_prompt_template()
@app.post("/ask")
async def ask_question(request: Request):
data = await request.json()
question = data.get("question")
if not question:
return {"error": "Question is required"}
answer = rag.ask_question(question)
return {"answer": answer}
|