diff --git a/litellm/tests/test_completion_cost.py b/litellm/tests/test_completion_cost.py index 4e22f7ffd..354342021 100644 --- a/litellm/tests/test_completion_cost.py +++ b/litellm/tests/test_completion_cost.py @@ -103,7 +103,7 @@ def test_cost_azure_gpt_35(): ), ) ], - model="azure/gpt-35-turbo", # azure always has model written like this + model="gpt-35-turbo", # azure always has model written like this usage=Usage(prompt_tokens=21, completion_tokens=17, total_tokens=38), )