Spaces:
Sleeping
Sleeping
File size: 1,743 Bytes
61d9463 |
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 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
import os
from fastapi import FastAPI, Depends
from fastapi.middleware.cors import CORSMiddleware
from backend.app.routes import users_route, events_route, travels_route
from backend.app.helper.security import get_api_key
from backend.app.config import get_settings
from backend.app.routes import chatbot_route
settings = get_settings()
app = FastAPI(
title="AI API",
description="API for AI micro services",
version="1.0.0"
)
# Add CORS middleware
app.add_middleware(
CORSMiddleware,
allow_origins=["*"],
allow_credentials=True,
allow_methods=["*"],
allow_headers=["*"],
)
# Add health check endpoint
@app.get("/")
async def root():
return {"status": "alive", "message": "API is running"}
# avoiding the conflictions
os.environ['KMP_DUPLICATE_LIB_OK'] = 'TRUE'
@app.on_event("startup")
async def startup_event():
print("Application startup...")
# asyncio.create_task(run_embedding_worker_loop(interval_seconds=30))
@app.on_event("shutdown")
async def shutdown_event():
print("Application shutdown...")
users_route.user_db.db.close()
events_route.event_db.db.close()
travels_route.travel_db.db.close()
# Include routers with API key dependency
app.include_router(
events_route.router,
prefix="/events",
tags=["events"],
dependencies=[Depends(get_api_key)]
)
app.include_router(
travels_route.router,
prefix="/travels",
tags=["travels"],
dependencies=[Depends(get_api_key)]
)
app.include_router(
users_route.router,
prefix="/users",
tags=["users"],
dependencies=[Depends(get_api_key)]
)
app.include_router(
chatbot_route.router,
prefix="/chatbot",
tags=["chatbot"],
dependencies=[Depends(get_api_key)]
)
|