agent_jrxml/agent/state.py

"""LangGraph JRXML 生成代理工作流的状态定义。"""

from typing import TypedDict, List


class AgentState(TypedDict, total=False):
    # 核心工作流字段
    conversation_history: List[dict]
    current_jrxml: str
    user_input: str
    status: str
    error_msg: str
    natural_explanation: str
    retry_count: int
    user_modification_request: str
    final_jrxml: str
    stage: str
    retrieved_context: str

    # 需求1：智能上下文压缩
    full_conversation_history: List[dict]
    compressed_history: str
    current_token_count: int

    # 需求2：多会话持久化
    session_id: str
    session_name: str
    created_at: str
    updated_at: str

    # 需求3：意图识别
    intent: str
    history_states: List[dict]

    # 需求4：JRXML 版本历史（用于下载历史版本）
    jrxml_versions: List[dict]

    # 需求5：错误自增长（记录修正前的状态，供 validate 节点判断是否入知识库）
    last_error_case: dict