gahow/aituner
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Use time-based trace window ids

Browse Source
This commit is contained in:
Gahow Wang
2026-04-04 22:09:43 +08:00
parent 4e1401f50c
commit 7b7eaafd78
4 changed files with 32 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_core_flow.py
View File

@@ -344,10 +344,10 @@ class CoreFlowTests(unittest.TestCase):
windows_payload = json.loads((output_root / "windows.json").read_text(encoding="utf-8"))
windows = {item["window_id"]: item for item in windows_payload["windows"]}
self.assertIn("chat_w20260311_peak_1000", windows)
self.assertEqual(windows["chat_w20260311_peak_1000"]["num_requests"], 1)
self.assertIn("chat_w20260311_1000", windows)
self.assertEqual(windows["chat_w20260311_1000"]["num_requests"], 1)
trace_path = output_root / windows["chat_w20260311_peak_1000"]["trace_file"]
trace_path = output_root / windows["chat_w20260311_1000"]["trace_file"]
rows = [json.loads(line) for line in trace_path.read_text(encoding="utf-8").splitlines()]
self.assertEqual(len(rows), 1)
self.assertEqual(rows[0]["prompt"], "real prompt")