Rag_Production / src /app /back-end.py
mhungtr211
Deploy bản RAG sạch lên Hugging Face
ac7ebac
Raw
History Blame Contribute Delete
372 Bytes
from fastapi import FastAPI
from src.generation.rag import Rag
from contextlib import asynccontextmanager
from src.api.routers import router
@asynccontextmanager
async def lifespan(app: FastAPI):
# VectorDB chỉ dùng để build index (chạy offline 1 lần)
app.state.rag_service = Rag()
yield
app = FastAPI(lifespan=lifespan)
app.include_router(router)