test(test_key_generate_dynamodb.py): fix test

This commit is contained in:
Krrish Dholakia 2024-01-22 21:44:13 -08:00
parent 5e0d99b2ef
commit 4babd67c79

View file

@ -452,6 +452,7 @@ def test_call_with_key_over_budget_stream(custom_db_client):
"user_api_key_user_id": user_id,
}
},
"response_cost": 0.00002,
},
completion_response=ModelResponse(),
)