Clean up AI code

Author: dandavison

tests/helpers/__init__.py

@@ -301,42 +301,28 @@ async def print_history(handle: WorkflowHandle):
 
 async def print_interleaved_histories(*handles: WorkflowHandle) -> None:
     """
-    Print the interleaved history events from multiple workflow handles in columns,
-    with each workflow's events in its own column, sorted chronologically.
-    Each workflow's events are numbered independently.
+    Print the interleaved history events from multiple workflow handles in columns.
     """
-    # Collect all events from all handles with their sequence numbers
     all_events: list[tuple[WorkflowHandle, HistoryEvent, int]] = []
-
     for handle in handles:
         event_num = 1
         async for event in handle.fetch_history_events():
             all_events.append((handle, event, event_num))
             event_num += 1
-
-    # Sort by event time
     all_events.sort(key=lambda item: item[1].event_time.ToDatetime())
-
-    # Prepare column headers
     col_width = 40
-    headers = [handle.id for handle in handles]
 
-    # Print headers
-    print("\nInterleaved histories:")
-    print("-" * (col_width * len(handles) + len(handles) - 1))
-    print(" | ".join(f"{h[: col_width - 3]:<{col_width - 3}}" for h in headers))
-    print("-" * (col_width * len(handles) + len(handles) - 1))
+    def _format_row(items: list[str], truncate: bool = False) -> str:
+        if truncate:
+            items = [item[: col_width - 3] for item in items]
+        return " | ".join(f"{item:<{col_width - 3}}" for item in items)
 
-    # Print events
+    headers = [handle.id for handle in handles]
+    print(_format_row(headers, truncate=True))
+    print("-" * (col_width * len(handles) + len(handles) - 1))
     for handle, event, event_num in all_events:
         event_type = EventType.Name(event.event_type).removeprefix("EVENT_TYPE_")
-
-        # Create row with empty cells
         row = [""] * len(handles)
-
-        # Find which column this event belongs to
         col_idx = handles.index(handle)
         row[col_idx] = f"{event_num:2}: {event_type[: col_width - 5]}"
-
-        # Print the row
-        print(" | ".join(f"{cell:<{col_width - 3}}" for cell in row))
+        print(_format_row(row))