test(test_key_generate_prisma.py): fix test

This commit is contained in:
Krrish Dholakia 2024-01-22 21:12:03 -08:00 committed by ishaan-jaff
parent 4ab99adaa2
commit 7f27026f87

View file

@ -738,6 +738,7 @@ def test_call_with_key_over_budget(prisma_client):
"user_api_key_user_id": user_id,
}
},
"response_cost": 0.00002,
},
completion_response=resp,
start_time=datetime.now(),
@ -813,6 +814,7 @@ def test_call_with_key_over_budget_stream(prisma_client):
"user_api_key_user_id": user_id,
}
},
"response_cost": 0.00002,
},
completion_response=ModelResponse(),
start_time=datetime.now(),