feat: 任务详情 LLM 请求数(cost_records 存 completions)

This commit is contained in:
host committed 2026-10-30 12:58:41 +08:00
1 parent 2ff8975f43
commit ccb5a8638d
9 files changed
+215 -77

No files matched your search

+13 -1
View File
@@ -140,7 +140,8 @@ class TestCostStore(unittest.TestCase):
def test_record_computes_total_and_day(self):
row_id = self.store.record(
"t1", "p1", "gpt-6.1-sol", input_tokens=100, output_tokens=50,
usd_micros=2000, source="actual", fallback_count=1)
usd_micros=2000, source="actual", fallback_count=1,
completions=7)
self.assertIsInstance(row_id, int)
report = self.store.report_by_task("t1")
self.assertEqual(report["total_tokens"], 150)
@@ -149,6 +150,16 @@ class TestCostStore(unittest.TestCase):
self.assertEqual(report["fallback_count"], 1)
self.assertEqual(report["actual_tokens"], 150)
self.assertEqual(report["estimated_tokens"], 0)
self.assertEqual(report["llm_requests"], 7)
def test_llm_requests_sums_completions_not_turns(self):
self.store.record("t1", "p1", "m", input_tokens=10,
output_tokens=0, source="actual", completions=3)
self.store.record("t1", "p1", "m", input_tokens=20,
output_tokens=0, source="actual", completions=5)
report = self.store.report_by_task("t1")
self.assertEqual(report["turns"], 2)
self.assertEqual(report["llm_requests"], 8)
def test_invalid_source_rejected(self):
with self.assertRaises(ValueError):
@@ -196,6 +207,7 @@ class TestCostStore(unittest.TestCase):
"estimated_tokens": 0,
"cached_input_tokens": 0,
"unpriced_actual_tokens": 0,
"llm_requests": 0,
})