update worklow events logs. (#19871)
Signed-off-by: zhanluxianshen <zhanluxianshen@163.com>
This commit is contained in:
@@ -232,14 +232,14 @@ class WorkflowLoggingCallback(WorkflowCallback):
|
|||||||
Publish loop started
|
Publish loop started
|
||||||
"""
|
"""
|
||||||
self.print_text("\n[LoopRunStartedEvent]", color="blue")
|
self.print_text("\n[LoopRunStartedEvent]", color="blue")
|
||||||
self.print_text(f"Loop Node ID: {event.loop_id}", color="blue")
|
self.print_text(f"Loop Node ID: {event.loop_node_id}", color="blue")
|
||||||
|
|
||||||
def on_workflow_loop_next(self, event: LoopRunNextEvent) -> None:
|
def on_workflow_loop_next(self, event: LoopRunNextEvent) -> None:
|
||||||
"""
|
"""
|
||||||
Publish loop next
|
Publish loop next
|
||||||
"""
|
"""
|
||||||
self.print_text("\n[LoopRunNextEvent]", color="blue")
|
self.print_text("\n[LoopRunNextEvent]", color="blue")
|
||||||
self.print_text(f"Loop Node ID: {event.loop_id}", color="blue")
|
self.print_text(f"Loop Node ID: {event.loop_node_id}", color="blue")
|
||||||
self.print_text(f"Loop Index: {event.index}", color="blue")
|
self.print_text(f"Loop Index: {event.index}", color="blue")
|
||||||
|
|
||||||
def on_workflow_loop_completed(self, event: LoopRunSucceededEvent | LoopRunFailedEvent) -> None:
|
def on_workflow_loop_completed(self, event: LoopRunSucceededEvent | LoopRunFailedEvent) -> None:
|
||||||
@@ -250,7 +250,7 @@ class WorkflowLoggingCallback(WorkflowCallback):
|
|||||||
"\n[LoopRunSucceededEvent]" if isinstance(event, LoopRunSucceededEvent) else "\n[LoopRunFailedEvent]",
|
"\n[LoopRunSucceededEvent]" if isinstance(event, LoopRunSucceededEvent) else "\n[LoopRunFailedEvent]",
|
||||||
color="blue",
|
color="blue",
|
||||||
)
|
)
|
||||||
self.print_text(f"Node ID: {event.loop_id}", color="blue")
|
self.print_text(f"Loop Node ID: {event.loop_node_id}", color="blue")
|
||||||
|
|
||||||
def print_text(self, text: str, color: Optional[str] = None, end: str = "\n") -> None:
|
def print_text(self, text: str, color: Optional[str] = None, end: str = "\n") -> None:
|
||||||
"""Print text with highlighting and no end characters."""
|
"""Print text with highlighting and no end characters."""
|
||||||
|
@@ -334,7 +334,7 @@ class Graph(BaseModel):
|
|||||||
|
|
||||||
parallel = GraphParallel(
|
parallel = GraphParallel(
|
||||||
start_from_node_id=start_node_id,
|
start_from_node_id=start_node_id,
|
||||||
parent_parallel_id=parent_parallel.id if parent_parallel else None,
|
parent_parallel_id=parent_parallel_id,
|
||||||
parent_parallel_start_node_id=parent_parallel.start_from_node_id if parent_parallel else None,
|
parent_parallel_start_node_id=parent_parallel.start_from_node_id if parent_parallel else None,
|
||||||
)
|
)
|
||||||
parallel_mapping[parallel.id] = parallel
|
parallel_mapping[parallel.id] = parallel
|
||||||
|
Reference in New Issue
Block a user