state.py

# Tencent is pleased to support the open source community by making tRPC-Agent-Python available.
#
# Copyright (C) 2026 Tencent. All rights reserved.
#
# tRPC-Agent-Python is licensed under Apache-2.0.
"""State schema for graph interrupt example."""
from typing import Any
from typing_extensions import Annotated

from trpc_agent_sdk.dsl.graph import State
from trpc_agent_sdk.dsl.graph import append_list


class InterruptState(State):
    """Custom state for interrupt approval flow."""

    request_text: str
    suggested_action: str
    approval_status: str
    approval_note: str
    summary_request: str
    approval_summary: str
    context_note: str

    node_execution_history: Annotated[list[dict[str, Any]], append_list]