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

feat: show req and resp ids in tui

Browse Source
This commit is contained in:
yunyaozhou
2026-07-01 10:03:20 +08:00
parent 58f3caeea5
commit 5c7bea431a
4 changed files with 41 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_payload.py
+16 -10
View File
@@ -94,8 +94,11 @@ class CodexRetryGatewayTUITests(unittest.TestCase):
"finished_at": "2026-06-30T00:00:01Z",
"usage_last_updated_at": "2026-06-30T00:00:01Z",
}
self.assertEqual(mod.primary_request_id(row), "resp_1")
self.assertEqual(mod.request_id_text(row), "r")
self.assertEqual(mod.response_id_text(row), "resp_1")
self.assertIn("thread thread_1", mod.render_request_detail(row))
self.assertIn("req r", mod.render_request_detail(row))
self.assertIn("resp resp_1", mod.render_request_detail(row))
self.assertIn("updated 1.0s", mod.render_request_detail(row))
self.assertEqual(mod.request_updated_elapsed(row), "1.0s")
self.assertEqual(mod.request_chunk_progress(row), "7 / 4.1KB")
@@ -136,15 +139,16 @@ class CodexRetryGatewayTUITests(unittest.TestCase):
"error": "Selected model is at capacity. Please try a different model.",
"finished_at": "2026-06-30T12:25:00Z",
}
compact = mod.request_row_cells(row, {"id": 18, "thread": 18, "path": 22, "model": 16, "usage": 24, "note": 22})
wide = mod.request_row_cells(row, {"id": 36, "thread": 36, "path": 40, "model": 24, "usage": 72, "note": 48})
self.assertEqual(compact[9], "0B")
self.assertEqual(wide[9], "0B")
compact = mod.request_row_cells(row, {"req_id": 18, "resp_id": 18, "thread": 18, "path": 22, "model": 16, "usage": 24, "note": 22})
wide = mod.request_row_cells(row, {"req_id": 36, "resp_id": 36, "thread": 36, "path": 40, "model": 24, "usage": 72, "note": 48})
self.assertEqual(compact[10], "0B")
self.assertEqual(wide[10], "0B")
self.assertLess(len(compact[1]), len(wide[1]))
self.assertLess(len(compact[2]), len(wide[2]))
self.assertLess(len(compact[5]), len(wide[5]))
self.assertLess(len(compact[8]), len(wide[8]))
self.assertLess(len(compact[15]), len(wide[15]))
self.assertLess(len(compact[3]), len(wide[3]))
self.assertLess(len(compact[6]), len(wide[6]))
self.assertLess(len(compact[9]), len(wide[9]))
self.assertLess(len(compact[16]), len(wide[16]))
def test_request_row_cells_show_request_bytes(self) -> None:
mod = load_module()
@@ -174,8 +178,10 @@ class CodexRetryGatewayTUITests(unittest.TestCase):
"finished_at": "2026-06-30T12:00:03Z",
}
cells = mod.request_row_cells(row)
self.assertEqual(cells[9], "1.5KB")
self.assertEqual(cells[10], "4.1KB")
self.assertEqual(cells[1], "req_1")
self.assertEqual(cells[2], "resp_1")
self.assertEqual(cells[10], "1.5KB")
self.assertEqual(cells[11], "4.1KB")
def test_request_rows_read_usage_from_nested_usage_object(self) -> None:
mod = load_module()