test_call_with_key_over_model_budget

BerriAI · Dec 19, 2024 · 664aa53 · 664aa53
1 parent e646566
commit 664aa53
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/proxy_unit_tests/test_key_generate_prisma.py b/tests/proxy_unit_tests/test_key_generate_prisma.py
@@ -1697,7 +1697,10 @@ async def test():
             request = GenerateKeyRequest(
                 max_budget=1000,
                 model_max_budget={
-                    "chatgpt-v-2": 0.000001,
+                    "chatgpt-v-2": {
+                        "budget_limit": "0.000001",
+                        "time_period": "1d",
+                    },
                 },
                 metadata={"user_api_key": 0.0001},
             )