- Introduce a new workflow editor component for visualizing and managing workflows, allowing users to add nodes and define connections. - Implement backend support for node types, including validation and constraints for workflow graphs. - Add new API endpoints for retrieving node types and their specifications. - Enhance the AssistantPage to integrate the workflow editor, enabling users to create and edit workflows directly. - Update frontend components to support new workflow functionalities, including condition edges and generic nodes. - Refactor existing code to accommodate the new workflow features and improve overall structure.
60 lines
1.5 KiB
Python
60 lines
1.5 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 输出(裸音频流)
|
|
"""
|
|
|
|
from contextlib import asynccontextmanager
|
|
|
|
import config
|
|
import uvicorn
|
|
from db.session import init_db
|
|
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 init_db() # MVP:启动建表;表稳定后切 alembic
|
|
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)
|