01ea590f76
新增部门级智能体配置管理，支持创建、切换、编辑和删除配置 添加运行时配置中间件，支持动态加载模型、工具和知识库 重构智能体上下文处理，支持部门ID和配置切换 实现前端配置管理界面，支持配置的增删改查和设为默认 添加后端API和数据库模型支持多配置管理
39 lines
1.1 KiB
Python
39 lines
1.1 KiB
Python
from contextlib import asynccontextmanager
|
|
|
|
from fastapi import FastAPI
|
|
|
|
from src.services.task_service import tasker
|
|
from src.services.mcp_service import init_mcp_servers
|
|
from src.storage.postgres.manager import pg_manager
|
|
from src.knowledge import knowledge_base
|
|
from src.utils import logger
|
|
|
|
|
|
@asynccontextmanager
|
|
async def lifespan(app: FastAPI):
|
|
"""FastAPI lifespan事件管理器"""
|
|
# 初始化数据库连接
|
|
try:
|
|
pg_manager.initialize()
|
|
await pg_manager.create_business_tables()
|
|
await pg_manager.ensure_knowledge_schema()
|
|
except Exception as e:
|
|
logger.error(f"Failed to initialize database during startup: {e}")
|
|
|
|
# 初始化 MCP 服务器配置
|
|
try:
|
|
await init_mcp_servers()
|
|
except Exception as e:
|
|
logger.error(f"Failed to initialize MCP servers during startup: {e}")
|
|
|
|
# 初始化知识库管理器
|
|
try:
|
|
await knowledge_base.initialize()
|
|
except Exception as e:
|
|
logger.error(f"Failed to initialize knowledge base manager: {e}")
|
|
|
|
await tasker.start()
|
|
yield
|
|
await tasker.shutdown()
|
|
await pg_manager.close()
|