feat: project current account rate multipliers
This commit is contained in:
@@ -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)
|
||||
|
||||
Reference in New Issue
Block a user