03b532dd09
- Integrate Alembic for managing database schema migrations, replacing the previous SQL schema management. - Update the FastAPI application to synchronize interface definitions at startup. - Modify the Docker Compose command to run Alembic migrations before starting the API. - Enhance the Makefile with new commands for database migration and revision management. - Remove outdated SQL schema and seed files, transitioning to a more dynamic migration approach. - Add initial migration scripts and configuration for Alembic, ensuring a structured database evolution.
61 lines
1.6 KiB
Python
61 lines
1.6 KiB
Python
"""FastAPI 入口。挂载路由,放行前端跨域,启动时同步接口定义。
|
|
|
|
启动: uv run --with-requirements requirements.txt uvicorn app:app --reload --port 8000
|
|
|
|
路由分组(对齐 dograh 的 routes/ 结构):
|
|
/health 健康检查
|
|
/api/assistants 助手 CRUD
|
|
/api/interface-definitions 接口定义
|
|
/api/model-resources 模型资源 CRUD
|
|
/ws/voice WebRTC 输出(浏览器)
|
|
/ws/stream WS 输出(裸音频流)
|
|
/api/webrtc/ice-servers WebRTC STUN/TURN 配置
|
|
"""
|
|
|
|
from contextlib import asynccontextmanager
|
|
|
|
import config
|
|
import uvicorn
|
|
from db.session import sync_interface_definitions
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
|
|
from routes import (
|
|
assistants,
|
|
health,
|
|
knowledge_bases,
|
|
model_registry,
|
|
node_types,
|
|
voice_webrtc,
|
|
voice_ws,
|
|
)
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(_app: FastAPI):
|
|
await sync_interface_definitions()
|
|
yield
|
|
|
|
|
|
app = FastAPI(title="AI Video Assistant 平台 - 后端", lifespan=lifespan)
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=config.CORS_ORIGINS,
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
app.include_router(health.router)
|
|
app.include_router(assistants.router)
|
|
app.include_router(knowledge_bases.router)
|
|
app.include_router(model_registry.router)
|
|
app.include_router(node_types.router)
|
|
app.include_router(voice_webrtc.router)
|
|
app.include_router(voice_ws.router)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
uvicorn.run("app:app", host=config.HOST, port=config.PORT, reload=True)
|