40 lines
997 B
Python
40 lines
997 B
Python
import uvicorn
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from app.api.routes import router as api_router
|
|
from app.core.config import settings
|
|
from app.core.database import engine, Base
|
|
from app.services.lightrag_wrapper import LightRAGWrapper
|
|
|
|
# Create tables
|
|
Base.metadata.create_all(bind=engine)
|
|
|
|
app = FastAPI(
|
|
title=settings.PROJECT_NAME,
|
|
openapi_url=f"{settings.API_V1_STR}/openapi.json"
|
|
)
|
|
|
|
# CORS
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=["*"],
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
# Include Router
|
|
app.include_router(api_router, prefix=settings.API_V1_STR)
|
|
|
|
@app.on_event("startup")
|
|
async def startup_event():
|
|
# Initialize LightRAG
|
|
wrapper = LightRAGWrapper()
|
|
await wrapper.initialize()
|
|
|
|
@app.get("/")
|
|
def root():
|
|
return {"message": "Welcome to LightRAG Service Wrapper"}
|
|
|
|
if __name__ == "__main__":
|
|
uvicorn.run("main:app", host="0.0.0.0", port=8000, reload=True)
|