| | |
| | | from contextlib import asynccontextmanager |
| | | |
| | | from fastapi import FastAPI |
| | | from app.api.auth import router as auth_router |
| | | from app.api.chat import router as chat_router |
| | |
| | | from app.api.files import router as files_router |
| | | from app.api.report import router as report_router |
| | | from app.models.base_model import init_db |
| | | from app.task.fetch_agent import sync_agents, initialize_agents |
| | | |
| | | |
| | | # 使用 Lifespan 事件处理程序 |
| | | @asynccontextmanager |
| | | async def lifespan(app: FastAPI): |
| | | # 初始化代理 |
| | | initialize_agents() |
| | | # 在应用启动时同步代理 |
| | | sync_agents() |
| | | yield |
| | | # 在应用关闭时执行清理操作(如果需要) |
| | | pass |
| | | |
| | | init_db() |
| | | app = FastAPI( |
| | | title="basic_rag_gateway", |
| | | version="0.1", |
| | | description="", |
| | | title="basic_rag_gateway", |
| | | version="0.1", |
| | | description="", |
| | | lifespan=lifespan |
| | | ) |
| | | |
| | | app.include_router(auth_router, prefix='/api/auth', tags=["auth"]) |