|
@@ -1,3 +1,4 @@
|
|
|
+from collections.abc import Mapping
|
|
|
from datetime import datetime
|
|
|
from typing import Any, Optional
|
|
|
|
|
@@ -140,8 +141,8 @@ class BaseIterationEvent(GraphEngineEvent):
|
|
|
|
|
|
class IterationRunStartedEvent(BaseIterationEvent):
|
|
|
start_at: datetime = Field(..., description="start at")
|
|
|
- inputs: Optional[dict[str, Any]] = None
|
|
|
- metadata: Optional[dict[str, Any]] = None
|
|
|
+ inputs: Optional[Mapping[str, Any]] = None
|
|
|
+ metadata: Optional[Mapping[str, Any]] = None
|
|
|
predecessor_node_id: Optional[str] = None
|
|
|
|
|
|
|
|
@@ -153,18 +154,18 @@ class IterationRunNextEvent(BaseIterationEvent):
|
|
|
|
|
|
class IterationRunSucceededEvent(BaseIterationEvent):
|
|
|
start_at: datetime = Field(..., description="start at")
|
|
|
- inputs: Optional[dict[str, Any]] = None
|
|
|
- outputs: Optional[dict[str, Any]] = None
|
|
|
- metadata: Optional[dict[str, Any]] = None
|
|
|
+ inputs: Optional[Mapping[str, Any]] = None
|
|
|
+ outputs: Optional[Mapping[str, Any]] = None
|
|
|
+ metadata: Optional[Mapping[str, Any]] = None
|
|
|
steps: int = 0
|
|
|
iteration_duration_map: Optional[dict[str, float]] = None
|
|
|
|
|
|
|
|
|
class IterationRunFailedEvent(BaseIterationEvent):
|
|
|
start_at: datetime = Field(..., description="start at")
|
|
|
- inputs: Optional[dict[str, Any]] = None
|
|
|
- outputs: Optional[dict[str, Any]] = None
|
|
|
- metadata: Optional[dict[str, Any]] = None
|
|
|
+ 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")
|
|
|
|