LQAgentPlatform/views/__init__.py

# ！/usr/bin/ python
# -*- coding: utf-8 -*-
'''
@Project    : lq-agent-api
@File       :__init__.py.py
@IDE        :PyCharm
@Author     :
@Date       :2025/7/10 17:04
'''

import uuid
from contextlib import asynccontextmanager
from contextvars import ContextVar

from fastapi import FastAPI, APIRouter

from function.load_mcp_server import LoadMcpServer

#mcp_server = LoadMcpServer()
mcp_server = None

@asynccontextmanager
async def lifespan(app: FastAPI):
    # 启动时加载工具
    #await mcp_server.get_mcp_tools()

    yield
    # 关闭时清理
    if mcp_server and mcp_server.cleanup:
        await mcp_server.close()

cattle_router = APIRouter(prefix="/queryex", tags=["agent"])
current_operation_id: ContextVar[str] = ContextVar("operation_id", default=str(uuid.uuid4()))



def get_operation_id() -> str:
    """依赖项：获取当前操作ID"""
    return current_operation_id.get()