feat: add node id in agent log

2025-01-09 17:20:53 +08:00
6 changed files with 6 additions and 0 deletions
@@ -386,6 +386,7 @@ class WorkflowBasedAppRunner(AppRunner):
                    status=event.status,
                    data=event.data,
                    metadata=event.metadata,
+                    node_id=event.node_id,
                )
            )
        elif isinstance(event, ParallelBranchRunStartedEvent):
@@ -330,6 +330,7 @@ class QueueAgentLogEvent(AppQueueEvent):
    status: str
    data: Mapping[str, Any]
    metadata: Optional[Mapping[str, Any]] = None
+    node_id: str


 class QueueNodeRetryEvent(QueueNodeStartedEvent):
@@ -717,6 +717,7 @@ class AgentLogStreamResponse(StreamResponse):
        status: str
        data: Mapping[str, Any]
        metadata: Optional[Mapping[str, Any]] = None
+        node_id: str

    event: StreamEvent = StreamEvent.AGENT_LOG
    data: Data
@@ -863,5 +863,6 @@ class WorkflowCycleManage:
                status=event.status,
                data=event.data,
                metadata=event.metadata,
+                node_id=event.node_id,
            ),
        )
@@ -204,6 +204,7 @@ class AgentLogEvent(BaseAgentEvent):
    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="agent node id")


 InNodeEvent = BaseNodeEvent | BaseParallelBranchEvent | BaseIterationEvent | BaseAgentEvent
@@ -338,6 +338,7 @@ class ToolNode(BaseNode[ToolNodeData]):
                    data=message.message.data,
                    label=message.message.label,
                    metadata=message.message.metadata,
+                    node_id=self.node_id,
                )

                # check if the agent log is already in the list