21 lines
442 B
Python
21 lines
442 B
Python
from fastapi import FastAPI
|
|
from fastapi.responses import RedirectResponse
|
|
from langserve import add_routes
|
|
from rag_chain import build_rag_chain
|
|
|
|
app = FastAPI()
|
|
|
|
|
|
@app.get("/")
|
|
async def redirect_root_to_docs():
|
|
return RedirectResponse("/docs")
|
|
|
|
|
|
# Edit this to add the chain you want to add
|
|
add_routes(app, build_rag_chain(), path="/rag")
|
|
|
|
if __name__ == "__main__":
|
|
import uvicorn
|
|
|
|
uvicorn.run(app, host="0.0.0.0", port=8000)
|