vent_agent_2.0/api/interrupts.py

"""中断展示与恢复共用一个载荷和标识，批准范围等于卡片展示范围。"""
import hashlib
import json

from fastapi import HTTPException


def pack(item):
    value = getattr(item, "value", item)
    payload = dict(value) if isinstance(value, dict) else {"message": str(value)}
    if getattr(item, "id", None):
        payload["_interrupt_id"] = item.id
    return payload


def token_for(payload):
    return hashlib.sha256(json.dumps(payload, sort_keys=True, ensure_ascii=False, default=str).encode()).hexdigest()


def interrupt_frame(writer, payload):
    payload = payload if isinstance(payload, dict) else {"message": str(payload)}
    token = token_for(payload)
    if payload.get("kind") == "ask_user":
        return writer.frame("interrupt", interrupt_token=token,
                            ask_user={"questions": payload["questions"]})
    actions = payload.get("action_requests") or []
    if actions:
        first = actions[0]
        return writer.interrupt_tool_approval(
            tool=first.get("name", ""), args=first.get("args"),
            reason=first.get("description", ""), side_effect="批准后执行卡片中的全部操作",
            actions=actions, pending_count=len(actions), interrupt_token=token,
        )
    return writer.frame("interrupt", message=payload.get("message", "智能体暂停，等待输入"),
                        interrupt_token=token)


def resume_value(request, state):
    from api.chat_routes import _to_official_resume
    items = [item for task in getattr(state, "tasks", ()) for item in getattr(task, "interrupts", ())]
    if not items:
        raise HTTPException(status_code=409, detail={"code": "no_pending_interrupt", "message": "没有待确认请求"})
    selected = next((item for item in items if token_for(pack(item)) == request.interrupt_token), None)
    if selected is None:
        raise HTTPException(status_code=409, detail={"code": "stale_interrupt", "message": "确认请求已变化，请刷新后重试"})
    payload = pack(selected)
    if payload.get("kind") == "ask_user":
        if request.status == "cancelled":
            value = {"cancelled": True}
        else:
            if not request.answers or len(request.answers) != len(payload["questions"]) or any(not a.strip() for a in request.answers):
                raise HTTPException(status_code=400, detail={"code": "invalid_answers", "message": "请回答全部问题"})
            value = {"answers": request.answers}
    else:
        if request.approved is None and request.status not in {"cancelled", "error"}:
            raise HTTPException(status_code=400, detail={"code": "invalid_decision", "message": "请选择批准或拒绝"})
        value = _to_official_resume(request, len(payload.get("action_requests") or []))
    interrupt_id = getattr(selected, "id", None)
    return {interrupt_id: value} if interrupt_id else value