shujakuin/codex-retry-gateway-tui
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: improve retry first column controls

Browse Source
This commit is contained in:
yunyaozhou
2026-07-07 20:28:13 +08:00
parent c5fa238598
commit 184b6b65b0
2 changed files with 132 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_payload.py
+11 -5
View File
@@ -230,6 +230,7 @@ class CodexRetryGatewayTUITests(unittest.TestCase):
"reasoning_retry_current_width": 2,
"reasoning_retry_current_firsts": [
{
"round": 3,
"slot": 1,
"first_response_delay_ms": 1200,
"outcome": "ok",
@@ -238,6 +239,7 @@ class CodexRetryGatewayTUITests(unittest.TestCase):
"reasoning_tokens": 516,
},
{
"round": 3,
"slot": 2,
"first_delay_ms": 1800,
"outcome": "retry",
@@ -254,10 +256,11 @@ class CodexRetryGatewayTUITests(unittest.TestCase):
self.assertEqual(row["reasoning_retry_current_width"], 2)
self.assertEqual(len(row["reasoning_retry_current_firsts"]), 2)
self.assertEqual(mod.request_retry_round_text(row), "3(2)")
self.assertIn("1:1.2s", mod.request_first_text(row))
self.assertIn("slot 1", mod.request_retry_wave_summary(row))
self.assertEqual(mod.request_first_text(row), "1.2s 1.8s")
self.assertIn("round 3 slot 1", mod.request_retry_wave_summary(row))
self.assertIn("retry round 3(2)", mod.render_request_detail(row))
self.assertIn("current wave slot 1 first 1.2s", mod.render_request_detail(row))
self.assertIn("first 1.2s 1.8s", mod.render_request_detail(row))
self.assertIn("current wave round 3 slot 1 first 1.2s", mod.render_request_detail(row))
self.assertEqual(mod.request_row_cells(row)[20], "3(2)")
def test_request_table_preferences_round_trip(self) -> None:
@@ -273,12 +276,14 @@ class CodexRetryGatewayTUITests(unittest.TestCase):
{
"columns": ["round", "seq", "status", "unknown"],
"hidden_columns": ["status", "unknown"],
"widths": {"round": 14, "status": 9, "unknown": 99},
}
)
self.assertEqual(saved_path, prefs_path)
loaded = mod.load_request_table_preferences()
self.assertEqual(loaded["columns"][:3], ["round", "seq", "status"])
self.assertEqual(loaded["hidden_columns"], ["status"])
self.assertEqual(loaded["widths"], {"round": 14, "status": 9})
def test_request_table_preferences_from_text_hides_and_reorders_columns(self) -> None:
mod = load_module()
@@ -286,9 +291,9 @@ class CodexRetryGatewayTUITests(unittest.TestCase):
"\n".join(
[
"# comment",
"thread",
"thread 24",
"seq",
"# cache",
"# cache 9",
"in",
"out",
"",
@@ -297,6 +302,7 @@ class CodexRetryGatewayTUITests(unittest.TestCase):
)
self.assertEqual(prefs["columns"][:6], ["thread", "seq", "cache", "in", "out", "req_id"])
self.assertIn("cache", prefs["hidden_columns"])
self.assertEqual(prefs["widths"], {"thread": 24, "cache": 9})
def test_request_table_preferences_expand_legacy_usage_column(self) -> None:
mod = load_module()