fix: type errors

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

api/core/workflow/graph_events/agent.py

@@ -1,5 +1,5 @@
 from collections.abc import Mapping
-from typing import Any, Optional
+from typing import Any
 
 from pydantic import Field
 
@@ -14,4 +14,4 @@ class NodeRunAgentLogEvent(GraphAgentNodeEventBase):
     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")
+    metadata: Mapping[str, Any] | None = Field(default=None, description="metadata")

api/core/workflow/graph_events/graph.py

@@ -1,4 +1,4 @@
-from typing import Any, Optional
+from typing import Any
 
 from pydantic import Field
 
@@ -10,7 +10,7 @@ class GraphRunStartedEvent(BaseGraphEvent):
 
 
 class GraphRunSucceededEvent(BaseGraphEvent):
-    outputs: Optional[dict[str, Any]] = None
+    outputs: dict[str, Any] | None = None
 
 
 class GraphRunFailedEvent(BaseGraphEvent):
@@ -20,11 +20,11 @@ class GraphRunFailedEvent(BaseGraphEvent):
 
 class GraphRunPartialSucceededEvent(BaseGraphEvent):
     exceptions_count: int = Field(..., description="exception count")
-    outputs: Optional[dict[str, Any]] = None
+    outputs: dict[str, Any] | None = None
 
 
 class GraphRunAbortedEvent(BaseGraphEvent):
     """Event emitted when a graph run is aborted by user command."""
 
-    reason: Optional[str] = Field(default=None, description="reason for abort")
-    outputs: Optional[dict[str, Any]] = Field(default=None, description="partial outputs if any")
+    reason: str | None = Field(default=None, description="reason for abort")
+    outputs: dict[str, Any] | None = Field(default=None, description="partial outputs if any")

api/core/workflow/graph_events/iteration.py

@@ -1,6 +1,6 @@
 from collections.abc import Mapping
 from datetime import datetime
-from typing import Any, Optional
+from typing import Any
 
 from pydantic import Field
 
@@ -10,31 +10,31 @@ from .base import GraphNodeEventBase
 class NodeRunIterationStartedEvent(GraphNodeEventBase):
     node_title: str
     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
+    inputs: Mapping[str, Any] | None = None
+    metadata: Mapping[str, Any] | None = None
+    predecessor_node_id: str | None = None
 
 
 class NodeRunIterationNextEvent(GraphNodeEventBase):
     node_title: str
     index: int = Field(..., description="index")
-    pre_iteration_output: Optional[Any] = None
+    pre_iteration_output: Any = None
 
 
 class NodeRunIterationSucceededEvent(GraphNodeEventBase):
     node_title: str
     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
+    inputs: Mapping[str, Any] | None = None
+    outputs: Mapping[str, Any] | None = None
+    metadata: Mapping[str, Any] | None = None
     steps: int = 0
 
 
 class NodeRunIterationFailedEvent(GraphNodeEventBase):
     node_title: str
     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
+    inputs: Mapping[str, Any] | None = None
+    outputs: Mapping[str, Any] | None = None
+    metadata: Mapping[str, Any] | None = None
     steps: int = 0
     error: str = Field(..., description="failed reason")

api/core/workflow/graph_events/loop.py

@@ -1,6 +1,6 @@
 from collections.abc import Mapping
 from datetime import datetime
-from typing import Any, Optional
+from typing import Any
 
 from pydantic import Field
 
@@ -10,31 +10,31 @@ from .base import GraphNodeEventBase
 class NodeRunLoopStartedEvent(GraphNodeEventBase):
     node_title: str
     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
+    inputs: Mapping[str, Any] | None = None
+    metadata: Mapping[str, Any] | None = None
+    predecessor_node_id: str | None = None
 
 
 class NodeRunLoopNextEvent(GraphNodeEventBase):
     node_title: str
     index: int = Field(..., description="index")
-    pre_loop_output: Optional[Any] = None
+    pre_loop_output: Any = None
 
 
 class NodeRunLoopSucceededEvent(GraphNodeEventBase):
     node_title: str
     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
+    inputs: Mapping[str, Any] | None = None
+    outputs: Mapping[str, Any] | None = None
+    metadata: Mapping[str, Any] | None = None
     steps: int = 0
 
 
 class NodeRunLoopFailedEvent(GraphNodeEventBase):
     node_title: str
     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
+    inputs: Mapping[str, Any] | None = None
+    outputs: Mapping[str, Any] | None = None
+    metadata: Mapping[str, Any] | None = None
     steps: int = 0
     error: str = Field(..., description="failed reason")

api/core/workflow/graph_events/node.py

@@ -1,6 +1,5 @@
 from collections.abc import Sequence
 from datetime import datetime
-from typing import Optional
 
 from pydantic import Field
 
@@ -12,9 +11,9 @@ from .base import GraphNodeEventBase
 
 class NodeRunStartedEvent(GraphNodeEventBase):
     node_title: str
-    predecessor_node_id: Optional[str] = None
-    parallel_mode_run_id: Optional[str] = None
-    agent_strategy: Optional[AgentNodeStrategyInit] = None
+    predecessor_node_id: str | None = None
+    parallel_mode_run_id: str | None = None
+    agent_strategy: AgentNodeStrategyInit | None = None
     start_at: datetime = Field(..., description="node start time")
 
     # FIXME(-LAN-): only for ToolNode