sessions: Dict[str, Dict[str, Union[CompiledGraph, str]]] = {
    # "user_id": {
    # "agent": CompiledGraph, # create_react_agent 返回的智能体实例
    # "session_id": str # UUID 字符串
    # }
}

# 客户端发起的智能体请求
class AgentRequest(BaseModel):
    user_id: str
    query: str
    system_message: Optional[str] = "你会使用工具来帮助用户。如果工具使用被拒绝，请提示用户。"

# 智能体给予的响应
class AgentResponse(BaseModel):

    # 一次会话的唯一标识符
    session_id: str

    # 三个状态：interrupted, completed, error
    status: str 

    #error时的提示消息
    message: Optional[str] = None

    #completed时的结果消息
    result: Optional[Dict[str, Any]] = None

    #interrupted时的中断消息
    interrupt_data: Optional[Dict[str, Any]] = None

# 客户端给予的反馈响应
class InterruptResponse(BaseModel):
    user_id: str
    session_id: str

    # 响应类型：accept, reject, edit
    response_type: str 

    # 如果是edit类型，可能需要额外的参数
    args: Optional[Dict[str, Any]] = None

@app.post("/agent/invoke", response_model=AgentResponse)
def invoke_agent(request: AgentRequest):
    """启动智能体处理用户请求 - 同步版本，等待执行完成或中断"""
    user_id = request.user_id
    if user_id notin sessions:

        # 只在创建新会话时生成新的会话ID
        session_id = str(uuid.uuid4())
        agent = create_tavily_search_agent(user_id)
        sessions[user_id] = {
            "agent": agent,
            "session_id": session_id
        }
    else:
        # 使用现有会话的ID和agent
        agent = sessions[user_id]["agent"]
        session_id = sessions[user_id]["session_id"]
        
    # 初始化智能体输入
    messages = ...
    config = {"configurable": {"thread_id": session_id}}

    try:
        result = agent.invoke(messages,config)
        return process_agent_result(session_id, result)
    ...

@app.post("/agent/resume", response_model=AgentResponse)
def resume_agent(response: InterruptResponse):

    """恢复被中断的智能体执行 - 同步版本，等待执行完成或再次中断"""
    user_id = response.user_id
    client_session_id = response.session_id
    
    # 检查用户会话是否存在
    if user_id notin sessions:
        raise HTTPException(status_code=404, detail=f"用户会话 {user_id} 不存在")
    
    # 检查会话ID是否匹配，由于未实现过期机制，也可以省略
    server_session_id = sessions[user_id]["session_id"]
    if server_session_id != client_session_id:
        raise HTTPException(status_code=400, detail="会话ID不匹配，可能是过期的请求")
    
    # 获取智能体和配置
    agent = sessions[user_id]["agent"]
    
    # 构造响应数据
    command_data = {
        "type": response.response_type
    }
    
    # 如果提供了参数，添加到响应数据中
    if response.args:
        command_data["args"] = response.args
    
    try:
        # 先恢复智能体执行
        result = agent.invoke(Command(resume=command_data), config={"configurable": {"thread_id": server_session_id}})

        # 再处理结果
        return process_agent_result(server_session_id, result)

......

@app.delete("/agent/session/{user_id}")
def delete_agent_session(user_id: str):
    ...
    del sessions[user_id]
    ...

def main():

...获取用户ID,模拟不同的客户端
   user_id = ...

    while True:
        try:
            # 获取用户查询问题
            query = ...
                
            # 调用智能体
            response = invoke_agent(user_id, query)
            
            # 处理智能体响应
            process_agent_response(response, user_id)

        ...

# 增强的会话存储结构
sessions = {
    "user_id": {
        "agent": agent_instance,
        "session_id": "uuid",
        "status": "idle|running|interrupted|completed|error", # 新增状态跟踪
        "last_response": AgentResponse, # 保存最后响应
        "last_query": str, # 保存最后查询
        "last_updated": timestamp # 时间戳
    }
}

...
sessions[user_id]["status"] = "running"
sessions[user_id]["last_query"] = request.query
sessions[user_id]["last_response"] = None
sessions[user_id]["last_updated"] = time.time()

......
sessions[user_id]["status"] = response.status
sessions[user_id]["last_response"] = response
sessions[user_id]["last_updated"] = time.time()