| status=event.status, | status=event.status, | ||||
| data=event.data, | data=event.data, | ||||
| metadata=event.metadata, | metadata=event.metadata, | ||||
| node_id=event.node_id, | |||||
| ) | ) | ||||
| ) | ) | ||||
| elif isinstance(event, ParallelBranchRunStartedEvent): | elif isinstance(event, ParallelBranchRunStartedEvent): |
| status: str | status: str | ||||
| data: Mapping[str, Any] | data: Mapping[str, Any] | ||||
| metadata: Optional[Mapping[str, Any]] = None | metadata: Optional[Mapping[str, Any]] = None | ||||
| node_id: str | |||||
| class QueueNodeRetryEvent(QueueNodeStartedEvent): | class QueueNodeRetryEvent(QueueNodeStartedEvent): |
| status: str | status: str | ||||
| data: Mapping[str, Any] | data: Mapping[str, Any] | ||||
| metadata: Optional[Mapping[str, Any]] = None | metadata: Optional[Mapping[str, Any]] = None | ||||
| node_id: str | |||||
| event: StreamEvent = StreamEvent.AGENT_LOG | event: StreamEvent = StreamEvent.AGENT_LOG | ||||
| data: Data | data: Data |
| status=event.status, | status=event.status, | ||||
| data=event.data, | data=event.data, | ||||
| metadata=event.metadata, | metadata=event.metadata, | ||||
| node_id=event.node_id, | |||||
| ), | ), | ||||
| ) | ) |
| status: str = Field(..., description="status") | status: str = Field(..., description="status") | ||||
| data: Mapping[str, Any] = Field(..., description="data") | data: Mapping[str, Any] = Field(..., description="data") | ||||
| metadata: Optional[Mapping[str, Any]] = Field(default=None, description="metadata") | metadata: Optional[Mapping[str, Any]] = Field(default=None, description="metadata") | ||||
| node_id: str = Field(..., description="agent node id") | |||||
| InNodeEvent = BaseNodeEvent | BaseParallelBranchEvent | BaseIterationEvent | BaseAgentEvent | BaseLoopEvent | InNodeEvent = BaseNodeEvent | BaseParallelBranchEvent | BaseIterationEvent | BaseAgentEvent | BaseLoopEvent |
| from core.workflow.entities.variable_pool import VariablePool, VariableValue | from core.workflow.entities.variable_pool import VariablePool, VariableValue | ||||
| from core.workflow.graph_engine.condition_handlers.condition_manager import ConditionManager | from core.workflow.graph_engine.condition_handlers.condition_manager import ConditionManager | ||||
| from core.workflow.graph_engine.entities.event import ( | from core.workflow.graph_engine.entities.event import ( | ||||
| BaseAgentEvent, | |||||
| BaseIterationEvent, | BaseIterationEvent, | ||||
| BaseLoopEvent, | BaseLoopEvent, | ||||
| GraphEngineEvent, | GraphEngineEvent, | ||||
| break | break | ||||
| yield event | yield event | ||||
| if event.parallel_id == parallel_id: | |||||
| if not isinstance(event, BaseAgentEvent) and event.parallel_id == parallel_id: | |||||
| if isinstance(event, ParallelBranchRunSucceededEvent): | if isinstance(event, ParallelBranchRunSucceededEvent): | ||||
| succeeded_count += 1 | succeeded_count += 1 | ||||
| if succeeded_count == len(futures): | if succeeded_count == len(futures): |
| data=message.message.data, | data=message.message.data, | ||||
| label=message.message.label, | label=message.message.label, | ||||
| metadata=message.message.metadata, | metadata=message.message.metadata, | ||||
| node_id=self.node_id, | |||||
| ) | ) | ||||
| # check if the agent log is already in the list | # check if the agent log is already in the list |