fix: show request timing and nested usage
This commit is contained in:
@@ -109,6 +109,34 @@ class CodexRetryGatewayTUITests(unittest.TestCase):
|
||||
}
|
||||
self.assertEqual(mod.request_usage_summary(row), "in 75 | out 40 | cached 25 (25%)")
|
||||
|
||||
def test_request_rows_read_usage_from_nested_usage_object(self) -> None:
|
||||
mod = load_module()
|
||||
payload = {
|
||||
"entries": [
|
||||
{
|
||||
"seq": 1,
|
||||
"request_id": "req_nested",
|
||||
"path": "/responses",
|
||||
"usage": {
|
||||
"input_tokens": 120,
|
||||
"output_tokens": 33,
|
||||
"total_tokens": 153,
|
||||
"cached_tokens": 20,
|
||||
},
|
||||
}
|
||||
]
|
||||
}
|
||||
rows = mod.normalize_request_rows(payload)
|
||||
self.assertEqual(rows[0]["input_tokens"], 120)
|
||||
self.assertEqual(rows[0]["output_tokens"], 33)
|
||||
self.assertEqual(rows[0]["total_tokens"], 153)
|
||||
self.assertEqual(rows[0]["cached_tokens"], 20)
|
||||
|
||||
def test_format_duration_ms_as_seconds(self) -> None:
|
||||
mod = load_module()
|
||||
self.assertEqual(mod.format_duration_ms_as_seconds(1532), "1.5s")
|
||||
self.assertEqual(mod.format_duration_ms_as_seconds(120000), "120s")
|
||||
|
||||
def test_profile_editor_payload_round_trip(self) -> None:
|
||||
mod = load_module()
|
||||
row = {
|
||||
|
||||
Reference in New Issue
Block a user