refacotr(workflow): move WorkflowNodeExecutionStatus from models to core.workflow.entities
Signed-off-by: -LAN- <laipz8200@outlook.com>pull/18805/head
-LAN-
1 year ago
parent
7ccec5cd95
commit
a9dd15d260
@ -0,0 +1,3 @@
|
|||||||
|
from .node_execution_entities import WorkflowNodeExecutionStatus
|
||||||
|
|
||||||
|
__all__ = ["WorkflowNodeExecutionStatus"]
|
||||||
@ -0,0 +1,64 @@
|
|||||||
|
from datetime import datetime
|
||||||
|
from enum import StrEnum
|
||||||
|
from typing import Any, Optional
|
||||||
|
|
||||||
|
from pydantic import BaseModel, Field
|
||||||
|
|
||||||
|
|
||||||
|
class WorkflowNodeExecutionStatus(StrEnum):
|
||||||
|
"""
|
||||||
|
Workflow Node Execution Status Enum
|
||||||
|
"""
|
||||||
|
|
||||||
|
RUNNING = "running"
|
||||||
|
SUCCEEDED = "succeeded"
|
||||||
|
FAILED = "failed"
|
||||||
|
EXCEPTION = "exception"
|
||||||
|
RETRY = "retry"
|
||||||
|
|
||||||
|
|
||||||
|
class WorkflowNodeExecution(BaseModel):
|
||||||
|
"""
|
||||||
|
Core Workflow Node Execution Model
|
||||||
|
|
||||||
|
A minimal Pydantic model for workflow node execution that doesn't contain
|
||||||
|
fields like tenant_id, app_id, etc. that aren't needed by the pure workflow core.
|
||||||
|
|
||||||
|
This model contains only the essential fields needed for the workflow core functionality.
|
||||||
|
"""
|
||||||
|
|
||||||
|
id: str = Field(..., description="Execution ID")
|
||||||
|
node_execution_id: Optional[str] = Field(None, description="Node execution ID, used for tracking execution")
|
||||||
|
index: int = Field(..., description="Execution sequence number, used for displaying Tracing Node order")
|
||||||
|
predecessor_node_id: Optional[str] = Field(
|
||||||
|
None, description="Predecessor node ID, used for displaying execution path"
|
||||||
|
)
|
||||||
|
node_id: str = Field(..., description="Node ID")
|
||||||
|
node_type: str = Field(..., description="Node type, such as 'start'")
|
||||||
|
title: str = Field(..., description="Node title")
|
||||||
|
|
||||||
|
# Data fields
|
||||||
|
inputs: Optional[dict[str, Any]] = Field(None, description="All predecessor node variable content used in the node")
|
||||||
|
process_data: Optional[dict[str, Any]] = Field(None, description="Node process data")
|
||||||
|
outputs: Optional[dict[str, Any]] = Field(None, description="Node output variables")
|
||||||
|
|
||||||
|
# Status and error information
|
||||||
|
status: WorkflowNodeExecutionStatus = Field(WorkflowNodeExecutionStatus.RUNNING, description="Execution status")
|
||||||
|
error: Optional[str] = Field(None, description="Error reason if status is failed")
|
||||||
|
error_type: Optional[str] = Field(None, description="Error type if status is failed")
|
||||||
|
|
||||||
|
# Timing and performance
|
||||||
|
elapsed_time: float = Field(0.0, description="Time consumption (s)")
|
||||||
|
created_at: datetime = Field(default_factory=datetime.now, description="Run time")
|
||||||
|
finished_at: Optional[datetime] = Field(None, description="End time")
|
||||||
|
|
||||||
|
# Metadata
|
||||||
|
execution_metadata: Optional[dict[str, Any]] = Field(None, description="Execution metadata")
|
||||||
|
|
||||||
|
# For iteration/loop tracking
|
||||||
|
workflow_run_id: Optional[str] = Field(None, description="Workflow run ID")
|
||||||
|
|
||||||
|
class Config:
|
||||||
|
"""Pydantic model configuration"""
|
||||||
|
|
||||||
|
arbitrary_types_allowed = True
|
||||||
Loading…
Reference in New Issue