File size: 942 Bytes
c7256ee
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
from fastapi import FastAPI
from fastapi.middleware.cors import CORSMiddleware
from backend.routes.health import router as health_router
from backend.routes.predict import router as predict_router
from backend.routes.predict_stream import router as predict_stream_router
from backend.routes.title import router as title_router
from backend.services.startup import initialize_runtime_state
from backend.state import state

# fastapi configs defined here
# all the router objects are imported here
#--@Qamar

app = FastAPI(title="RAG-AS3 API", version="0.1.0")

app.add_middleware(
    CORSMiddleware,
    allow_origins=["*"],
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)

app.include_router(health_router)
app.include_router(title_router)
app.include_router(predict_router)
app.include_router(predict_stream_router)


@app.on_event("startup")
def startup_event() -> None:
    initialize_runtime_state(state)