feat: color requests by thread id

codex_retry_gateway_tui.py

@@ -5,6 +5,7 @@ from __future__ import annotations
 import argparse
 import datetime as dt
 from concurrent.futures import ThreadPoolExecutor
+import hashlib
 import importlib.metadata
 import json
 import os
@@ -81,6 +82,19 @@ REQUEST_TABLE_COLUMN_KEY_SET = set(REQUEST_TABLE_COLUMN_KEYS)
 REQUEST_TABLE_CELL_INDEX = {key: index for index, key in enumerate(REQUEST_TABLE_COLUMN_KEYS)}
 DEFAULT_REQUEST_TABLE_VISIBLE_COLUMNS = REQUEST_TABLE_COLUMN_KEYS
 DEFAULT_REQUEST_TABLE_SORT_COLUMN = "seq"
+THREAD_COLOR_PALETTE = (
+    "#5fd7ff",
+    "#87d75f",
+    "#ffaf5f",
+    "#d787ff",
+    "#5fafff",
+    "#ff87d7",
+    "#d7af5f",
+    "#5fd7af",
+    "#af87ff",
+    "#87afd7",
+)
+THREAD_COLOR_COLUMN_KEYS = frozenset({"req_id", "resp_id", "thread"})
 REQUEST_TABLE_WIDTH_PROFILES = {
     "compact": {
         "seq": 7,
@@ -990,6 +1004,37 @@ def request_row_cells(row: dict[str, Any], text_limits: dict[str, int] | None =
     )
 
 
+def request_thread_color(thread_id: Any) -> str | None:
+    text = str(thread_id or "").strip()
+    if not text:
+        return None
+    digest = hashlib.blake2b(text.encode("utf-8"), digest_size=4).digest()
+    color_index = int.from_bytes(digest[:2], "big") % len(THREAD_COLOR_PALETTE)
+    return THREAD_COLOR_PALETTE[color_index]
+
+
+def request_visible_cells(
+    row: dict[str, Any],
+    cells: tuple[Any, ...],
+    visible_column_keys: list[str],
+) -> list[Any]:
+    visible_cells = [cells[REQUEST_TABLE_CELL_INDEX[column_key]] for column_key in visible_column_keys]
+    thread_color = request_thread_color(row.get("thread_id"))
+    if not thread_color:
+        return visible_cells
+    try:
+        from rich.text import Text
+    except ImportError:
+        return visible_cells
+    styled_cells: list[Any] = []
+    for column_key, cell in zip(visible_column_keys, visible_cells):
+        if column_key in THREAD_COLOR_COLUMN_KEYS:
+            styled_cells.append(Text(str(cell), style=f"bold {thread_color}"))
+        else:
+            styled_cells.append(cell)
+    return styled_cells
+
+
 def request_table_column_label(column_key: str) -> str:
     column = REQUEST_TABLE_COLUMN_BY_KEY.get(column_key)
     return str(column["label"]) if column else column_key
@@ -2113,13 +2158,14 @@ def run_textual(
             selected_row_index = table.cursor_row if table.cursor_row is not None and table.cursor_row >= 0 else 0
             selected_row_key = self._selected_request_key()
             text_limits = self._request_text_limits(table)
+            visible_column_keys = self._visible_request_column_keys()
             table.clear()
             self.request_by_key = {}
             for row in self.requests:
                 key = self._request_row_key(row)
                 self.request_by_key[key] = row
                 cells = request_row_cells(row, text_limits)
-                visible_cells = [cells[REQUEST_TABLE_CELL_INDEX[column_key]] for column_key in self._visible_request_column_keys()]
+                visible_cells = request_visible_cells(row, cells, visible_column_keys)
                 table.add_row(*visible_cells, key=key)
             if self.requests:
                 if not self._select_request_row(selected_row_key):