# ！/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


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()

test_router = APIRouter(prefix="/test", 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()