Kris
88588c3740
1. 新增多渠道配置项，支持多渠道网关管理 2. 添加用户来源字段，区分本地用户与渠道用户 3. 实现WebSocket聊天路由与Slack Webhook处理 4. 新增渠道消息记录与统计相关存储模型与仓库 5. 实现多渠道管理API与Webhook分发路由 6. 初始化渠道管理器并集成到应用生命周期 7. 添加第三方依赖包支持多渠道SDK 8. 优化鉴权逻辑，排除渠道用户的常规鉴权
118 lines
4.5 KiB
Python
118 lines
4.5 KiB
Python
import os
|
||
from contextlib import asynccontextmanager
|
||
|
||
from fastapi import FastAPI
|
||
from langgraph.checkpoint.postgres.aio import AsyncPostgresSaver
|
||
|
||
from yuxi.services.task_service import tasker
|
||
from yuxi.services.mcp_service import ensure_builtin_mcp_servers_in_db
|
||
from yuxi.services.model_provider_service import ensure_builtin_model_providers_in_db
|
||
from yuxi.services.subagent_service import init_builtin_subagents
|
||
from yuxi.services.run_queue_service import close_queue_clients, get_redis_client
|
||
from yuxi.storage.postgres.manager import pg_manager
|
||
from yuxi.knowledge import knowledge_base
|
||
from yuxi.utils import logger
|
||
from yuxi.agents.backends.sandbox import init_sandbox_provider, shutdown_sandbox_provider
|
||
from yuxi.channels.manager import channel_manager
|
||
from yuxi import get_version
|
||
|
||
|
||
@asynccontextmanager
|
||
async def lifespan(app: FastAPI):
|
||
"""FastAPI lifespan事件管理器"""
|
||
# 初始化数据库连接
|
||
try:
|
||
pg_manager.initialize()
|
||
await pg_manager.create_business_tables()
|
||
await pg_manager.ensure_business_schema()
|
||
await pg_manager.ensure_knowledge_schema()
|
||
except Exception as e:
|
||
logger.error(f"Failed to initialize database during startup: {e}")
|
||
|
||
# 确保内置 MCP 服务器定义存在于数据库
|
||
try:
|
||
await ensure_builtin_mcp_servers_in_db()
|
||
except Exception as e:
|
||
logger.error(f"Failed to ensure builtin MCP servers during startup: {e}")
|
||
|
||
# 初始化内置模型供应商配置
|
||
try:
|
||
async with pg_manager.get_async_session_context() as session:
|
||
await ensure_builtin_model_providers_in_db(session)
|
||
except Exception as e:
|
||
logger.error(f"Failed to ensure builtin model providers during startup: {e}")
|
||
|
||
# 初始化模型缓存（v2 模型选择使用）
|
||
try:
|
||
from yuxi.services.model_cache import model_cache
|
||
from yuxi.services.model_provider_service import get_all_model_providers
|
||
|
||
async with pg_manager.get_async_session_context() as session:
|
||
providers = await get_all_model_providers(session)
|
||
model_cache.rebuild(providers)
|
||
except Exception as e:
|
||
logger.error(f"Failed to initialize model cache during startup: {e}")
|
||
|
||
# 初始化内置 SubAgent
|
||
try:
|
||
await init_builtin_subagents()
|
||
except Exception as e:
|
||
logger.error(f"Failed to initialize builtin subagents during startup: {e}")
|
||
raise
|
||
|
||
# 初始化知识库管理器
|
||
if os.environ.get("LITE_MODE", "").lower() in ("true", "1"):
|
||
logger.info("LITE_MODE enabled, skipping knowledge base initialization")
|
||
else:
|
||
try:
|
||
await knowledge_base.initialize()
|
||
except Exception as e:
|
||
logger.error(f"Failed to initialize knowledge base manager: {e}")
|
||
|
||
# 预热 Redis（run 队列）
|
||
try:
|
||
redis = await get_redis_client()
|
||
await redis.ping()
|
||
except Exception as e:
|
||
logger.warning(f"Run queue redis unavailable on startup: {e}")
|
||
|
||
try:
|
||
init_sandbox_provider()
|
||
except Exception as e:
|
||
logger.error(f"Failed to initialize sandbox provider during startup: {e}")
|
||
|
||
# =========================================================
|
||
# 2. 核心修复：在这里执行一次 setup()，建完表就拉倒
|
||
# =========================================================
|
||
checkpointer = AsyncPostgresSaver(pg_manager.langgraph_pool)
|
||
await checkpointer.setup()
|
||
print("LangGraph Checkpoint tables verified/created!")
|
||
|
||
# 多渠道网关初始化
|
||
try:
|
||
await channel_manager.initialize()
|
||
app.state.channel_manager = channel_manager
|
||
except Exception as e:
|
||
logger.error(f"Failed to initialize channel manager during startup: {e}")
|
||
|
||
await tasker.start()
|
||
logger.info(f"""
|
||
|
||
░██ ░██ ░██
|
||
░██ ░██
|
||
░██ ░██ ░██ ░██ ░██ ░██ ░██
|
||
░████ ░██ ░██ ░██ ░██ ░██
|
||
░██ ░██ ░██ ░█████ ░██
|
||
░██ ░██ ░███ ░██ ░██ ░██
|
||
░██ ░█████░██ ░██ ░██ ░██ v{get_version()}
|
||
|
||
""")
|
||
logger.info("Yuxi backend startup complete")
|
||
yield
|
||
if hasattr(app.state, "channel_manager"):
|
||
await app.state.channel_manager.shutdown()
|
||
await tasker.shutdown()
|
||
shutdown_sandbox_provider()
|
||
await close_queue_clients()
|
||
await pg_manager.close()
|