feat: project current account rate multipliers

This commit is contained in:
2026-08-05 03:07:21 +08:00
parent 51802d07cf
commit 492e6c882f
5 changed files with 46 additions and 12 deletions
+6
View File
@@ -67,6 +67,8 @@ def workspace_payload_fixture() -> dict[str, object]:
"provider": "openai",
"kind": "quota_limited",
"usable": True,
"account_rate_multiplier": 0.06,
"account_rate_multiplier_cny": 0.003,
}
],
},
@@ -105,6 +107,8 @@ def workspace_payload_fixture() -> dict[str, object]:
"api_key_name": "wmy",
"account_id": "9",
"account_name": "oai-quota-code-plan",
"account_rate_multiplier": 0.06,
"account_rate_multiplier_cny": 0.003,
"model": "gpt-5.5",
"input_tokens": 100,
"output_tokens": 50,
@@ -803,7 +807,9 @@ class WorkspaceTests(unittest.TestCase):
normalized_accounts = mod.normalize_account_rows(accounts, pricing_payload=pricing)
normalized_logs = mod.normalize_log_rows(logs)
self.assertEqual(normalized_accounts[0]["id"], 9007199254740993)
self.assertEqual(normalized_accounts[0]["account_multiplier"], 0.003)
self.assertEqual(normalized_logs[0]["id"], 9007199254740997)
self.assertEqual(normalized_logs[0]["account_multiplier"], 0.003)
self.assertEqual(normalized_logs[0]["cost"], 0.25)
self.assertEqual(mod.as_int("9007199254740999"), 9007199254740999)
self.assertEqual(len(keys["items"]), 1)