Underground-Digital/Workflow-Engine

1from typing import Any2 3from pydantic import BaseModel, Field4 5from core.model_runtime.entities.llm_entities import LLMUsage6from core.workflow.entities.variable_pool import VariablePool7from core.workflow.graph_engine.entities.runtime_route_state import RuntimeRouteState8 9 10class GraphRuntimeState(BaseModel):11    variable_pool: VariablePool = Field(..., description="variable pool")12    """variable pool"""13 14    start_at: float = Field(..., description="start time")15    """start time"""16    total_tokens: int = 017    """total tokens"""18    llm_usage: LLMUsage = LLMUsage.empty_usage()19    """llm usage info"""20    outputs: dict[str, Any] = {}21    """outputs"""22 23    node_run_steps: int = 024    """node run steps"""25 26    node_run_state: RuntimeRouteState = RuntimeRouteState()27    """node run state"""28