feat:接入mcp,为模型提供图表信息

This commit is contained in:
2026-10-01 06:33:44 -07:00
parent 618290d852
commit f1b2a70ad1
44 changed files with 6531 additions and 207 deletions
+19 -6
View File
@@ -18,6 +18,11 @@ from app.core.logging import setup_logging
async def lifespan(application: FastAPI):
setup_logging()
from app.clients.decision import close_decision_engine
from app.clients.material_mcp import (
MaterialFieldServiceError,
start_material_server,
stop_material_server,
)
from app.clients.user_center import close_user_center
from app.observability.tracing import flush_tracing
from app.processor.graph import build_graph
@@ -28,9 +33,15 @@ async def lifespan(application: FastAPI):
# 按配置选择 MongoDB 或进程内存,会话记录与图检查点使用同一种存储
persistence = await open_persistence()
application.state.conversation_store = persistence.store
material_server = None
try:
# 物料目录启动时载入内存,agent 规划与前端接口共用,避免重复查询数据库
await warm_material_cache(persistence.db)
# 物料字段服务给模型查询字段;启动失败只记录,需要查字段的修改执行时会报错
try:
material_server = await start_material_server()
except MaterialFieldServiceError as error:
logger.error("{},修改组件属性和新增组件将无法执行", error)
application.state.agent_service = AgentService(
build_graph(persistence.checkpointer), persistence.store
)
@@ -38,19 +49,21 @@ async def lifespan(application: FastAPI):
finally:
# 共享客户端只在应用退出时关闭,追踪刷新放到线程避免阻塞事件循环
try:
await persistence.close()
await stop_material_server(material_server)
finally:
try:
await close_decision_engine()
await persistence.close()
finally:
try:
await close_user_center()
await close_decision_engine()
finally:
try:
await asyncio.to_thread(flush_tracing)
await close_user_center()
finally:
await logger.complete()
try:
await asyncio.to_thread(flush_tracing)
finally:
await logger.complete()
def create_app() -> FastAPI:
application = FastAPI(title="Screen Agent API", lifespan=lifespan)