(test) model in SpendLogs table

2026-08-10 00:22:03 +00:00 · 2024-01-24 11:10:13 -08:00
parent a842e6520c
commit 3e53742a4f
1 changed files with 2 additions and 0 deletions
@@ -825,6 +825,7 @@ def test_call_with_key_over_budget_stream(prisma_client):
            )
            await track_cost_callback(
                kwargs={
+                    "model": "sagemaker-chatgpt-v-2",
                    "stream": True,
                    "complete_streaming_response": resp,
                    "litellm_params": {
@@ -850,6 +851,7 @@ def test_call_with_key_over_budget_stream(prisma_client):

            assert spend_log.request_id == request_id
            assert spend_log.spend == float("5e-05")
+            assert spend_log.model == "sagemaker-chatgpt-v-2"

            # use generated key to auth in
            result = await user_api_key_auth(request=request, api_key=bearer_token)