feat: queue-based graph engine

Signed-off-by: -LAN- <laipz8200@outlook.com>

api/core/workflow/node_events/__init__.py

@@ -0,0 +1,40 @@
+from .agent import AgentLogEvent
+from .base import NodeEventBase, NodeRunResult
+from .iteration import (
+    IterationFailedEvent,
+    IterationNextEvent,
+    IterationStartedEvent,
+    IterationSucceededEvent,
+)
+from .loop import (
+    LoopFailedEvent,
+    LoopNextEvent,
+    LoopStartedEvent,
+    LoopSucceededEvent,
+)
+from .node import (
+    ModelInvokeCompletedEvent,
+    RunRetrieverResourceEvent,
+    RunRetryEvent,
+    StreamChunkEvent,
+    StreamCompletedEvent,
+)
+
+__all__ = [
+    "AgentLogEvent",
+    "IterationFailedEvent",
+    "IterationNextEvent",
+    "IterationStartedEvent",
+    "IterationSucceededEvent",
+    "LoopFailedEvent",
+    "LoopNextEvent",
+    "LoopStartedEvent",
+    "LoopSucceededEvent",
+    "ModelInvokeCompletedEvent",
+    "NodeEventBase",
+    "NodeRunResult",
+    "RunRetrieverResourceEvent",
+    "RunRetryEvent",
+    "StreamChunkEvent",
+    "StreamCompletedEvent",
+]

api/core/workflow/node_events/agent.py

@@ -0,0 +1,18 @@
+from collections.abc import Mapping
+from typing import Any, Optional
+
+from pydantic import Field
+
+from .base import NodeEventBase
+
+
+class AgentLogEvent(NodeEventBase):
+    message_id: str = Field(..., description="id")
+    label: str = Field(..., description="label")
+    node_execution_id: str = Field(..., description="node execution id")
+    parent_id: str | None = Field(..., description="parent id")
+    error: str | None = Field(..., description="error")
+    status: str = Field(..., description="status")
+    data: Mapping[str, Any] = Field(..., description="data")
+    metadata: Optional[Mapping[str, Any]] = Field(default=None, description="metadata")
+    node_id: str = Field(..., description="node id")

api/core/workflow/node_events/base.py

@@ -0,0 +1,35 @@
+from collections.abc import Mapping
+from typing import Any
+
+from pydantic import BaseModel, Field
+
+from core.model_runtime.entities.llm_entities import LLMUsage
+from core.workflow.enums import WorkflowNodeExecutionMetadataKey, WorkflowNodeExecutionStatus
+
+
+class NodeEventBase(BaseModel):
+    """Base class for all node events"""
+
+    pass
+
+
+class NodeRunResult(BaseModel):
+    """
+    Node Run Result.
+    """
+
+    status: WorkflowNodeExecutionStatus = WorkflowNodeExecutionStatus.PENDING
+
+    inputs: Mapping[str, Any] = Field(default_factory=dict)
+    process_data: Mapping[str, Any] = Field(default_factory=dict)
+    outputs: Mapping[str, Any] = Field(default_factory=dict)
+    metadata: Mapping[WorkflowNodeExecutionMetadataKey, Any] = Field(default_factory=dict)
+    llm_usage: LLMUsage = Field(default_factory=LLMUsage.empty_usage)
+
+    edge_source_handle: str = "source"  # source handle id of node with multiple branches
+
+    error: str = ""
+    error_type: str = ""
+
+    # single step node run retry
+    retry_index: int = 0

api/core/workflow/node_events/iteration.py

@@ -0,0 +1,36 @@
+from collections.abc import Mapping
+from datetime import datetime
+from typing import Any, Optional
+
+from pydantic import Field
+
+from .base import NodeEventBase
+
+
+class IterationStartedEvent(NodeEventBase):
+    start_at: datetime = Field(..., description="start at")
+    inputs: Optional[Mapping[str, Any]] = None
+    metadata: Optional[Mapping[str, Any]] = None
+    predecessor_node_id: Optional[str] = None
+
+
+class IterationNextEvent(NodeEventBase):
+    index: int = Field(..., description="index")
+    pre_iteration_output: Optional[Any] = None
+
+
+class IterationSucceededEvent(NodeEventBase):
+    start_at: datetime = Field(..., description="start at")
+    inputs: Optional[Mapping[str, Any]] = None
+    outputs: Optional[Mapping[str, Any]] = None
+    metadata: Optional[Mapping[str, Any]] = None
+    steps: int = 0
+
+
+class IterationFailedEvent(NodeEventBase):
+    start_at: datetime = Field(..., description="start at")
+    inputs: Optional[Mapping[str, Any]] = None
+    outputs: Optional[Mapping[str, Any]] = None
+    metadata: Optional[Mapping[str, Any]] = None
+    steps: int = 0
+    error: str = Field(..., description="failed reason")

api/core/workflow/node_events/loop.py

@@ -0,0 +1,36 @@
+from collections.abc import Mapping
+from datetime import datetime
+from typing import Any, Optional
+
+from pydantic import Field
+
+from .base import NodeEventBase
+
+
+class LoopStartedEvent(NodeEventBase):
+    start_at: datetime = Field(..., description="start at")
+    inputs: Optional[Mapping[str, Any]] = None
+    metadata: Optional[Mapping[str, Any]] = None
+    predecessor_node_id: Optional[str] = None
+
+
+class LoopNextEvent(NodeEventBase):
+    index: int = Field(..., description="index")
+    pre_loop_output: Optional[Any] = None
+
+
+class LoopSucceededEvent(NodeEventBase):
+    start_at: datetime = Field(..., description="start at")
+    inputs: Optional[Mapping[str, Any]] = None
+    outputs: Optional[Mapping[str, Any]] = None
+    metadata: Optional[Mapping[str, Any]] = None
+    steps: int = 0
+
+
+class LoopFailedEvent(NodeEventBase):
+    start_at: datetime = Field(..., description="start at")
+    inputs: Optional[Mapping[str, Any]] = None
+    outputs: Optional[Mapping[str, Any]] = None
+    metadata: Optional[Mapping[str, Any]] = None
+    steps: int = 0
+    error: str = Field(..., description="failed reason")

api/core/workflow/node_events/node.py

@@ -0,0 +1,40 @@
+from collections.abc import Sequence
+from datetime import datetime
+
+from pydantic import Field
+
+from core.model_runtime.entities.llm_entities import LLMUsage
+from core.rag.entities.citation_metadata import RetrievalSourceMetadata
+from core.workflow.node_events import NodeRunResult
+
+from .base import NodeEventBase
+
+
+class RunRetrieverResourceEvent(NodeEventBase):
+    retriever_resources: Sequence[RetrievalSourceMetadata] = Field(..., description="retriever resources")
+    context: str = Field(..., description="context")
+
+
+class ModelInvokeCompletedEvent(NodeEventBase):
+    text: str
+    usage: LLMUsage
+    finish_reason: str | None = None
+
+
+class RunRetryEvent(NodeEventBase):
+    error: str = Field(..., description="error")
+    retry_index: int = Field(..., description="Retry attempt number")
+    start_at: datetime = Field(..., description="Retry start time")
+
+
+class StreamChunkEvent(NodeEventBase):
+    # Spec-compliant fields
+    selector: Sequence[str] = Field(
+        ..., description="selector identifying the output location (e.g., ['nodeA', 'text'])"
+    )
+    chunk: str = Field(..., description="the actual chunk content")
+    is_final: bool = Field(default=False, description="indicates if this is the last chunk")
+
+
+class StreamCompletedEvent(NodeEventBase):
+    node_run_result: NodeRunResult = Field(..., description="run result")