test: fix test

krrishdholakia · krrishdholakia · commit 0e747aaaf1f9 · 2025-09-16T19:20:12.000-07:00
diff --git a/.gitignore b/.gitignore
@@ -95,4 +95,5 @@ test.py
 litellm_config.yaml
 .cursor
 .vscode/launch.json
-litellm/proxy/to_delete_loadtest_work/*
+litellm/proxy/to_delete_loadtest_work/*
+update_model_cost_map.py
diff --git a/tests/local_testing/test_completion_cost.py b/tests/local_testing/test_completion_cost.py
@@ -1068,6 +1068,7 @@ def test_completion_cost_prompt_caching(model, custom_llm_provider):
         (
             response_1.usage.prompt_tokens
             - response_1.usage.prompt_tokens_details.cached_tokens
+            - response_1.usage.prompt_tokens_details.cache_creation_tokens
         )
         * _model_info["input_cost_per_token"]
         + (response_1.usage.prompt_tokens_details.cached_tokens or 0)

Original file line number	Diff line number	Diff line change
`@@ -1068,6 +1068,7 @@ def test_completion_cost_prompt_caching(model, custom_llm_provider):`
`1068`	`1068`	`(`
`1069`	`1069`	`response_1.usage.prompt_tokens`
`1070`	`1070`	`- response_1.usage.prompt_tokens_details.cached_tokens`
	`1071`	`+ - response_1.usage.prompt_tokens_details.cache_creation_tokens`
`1071`	`1072`	`)`
`1072`	`1073`	`* _model_info["input_cost_per_token"]`
`1073`	`1074`	`+ (response_1.usage.prompt_tokens_details.cached_tokens or 0)`