diff --git a/litellm/model_prices_and_context_window_backup.json b/litellm/model_prices_and_context_window_backup.json index aa3074a76dd..dad1f38e946 100644 --- a/litellm/model_prices_and_context_window_backup.json +++ b/litellm/model_prices_and_context_window_backup.json @@ -63871,7 +63871,7 @@ "supports_function_calling": true, "source": "https://api.openinfer.ai/v1/models" }, - "openinfer/@oi/Gemma4-31B": { + "openinfer/@oi/Gemma4-31B-It": { "max_tokens": 8192, "max_input_tokens": 262144, "max_output_tokens": 8192, diff --git a/model_prices_and_context_window.json b/model_prices_and_context_window.json index aa3074a76dd..dad1f38e946 100644 --- a/model_prices_and_context_window.json +++ b/model_prices_and_context_window.json @@ -63871,7 +63871,7 @@ "supports_function_calling": true, "source": "https://api.openinfer.ai/v1/models" }, - "openinfer/@oi/Gemma4-31B": { + "openinfer/@oi/Gemma4-31B-It": { "max_tokens": 8192, "max_input_tokens": 262144, "max_output_tokens": 8192, diff --git a/tests/test_litellm/llms/openai_like/test_openinfer_provider.py b/tests/test_litellm/llms/openai_like/test_openinfer_provider.py index 367b82836f6..b94cce7fff4 100644 --- a/tests/test_litellm/llms/openai_like/test_openinfer_provider.py +++ b/tests/test_litellm/llms/openai_like/test_openinfer_provider.py @@ -107,7 +107,7 @@ class TestOpenInferProviderConfig: ("openinfer/@oi/Llama-3.2-1B-Instruct", 2e-08, 2e-08), ("openinfer/@oi/Qwen3.5-9B", 1.5e-07, 1.8e-07), ("openinfer/@oi/Qwen3.5-27B", 7.2e-07, 7.2e-07), - ("openinfer/@oi/Gemma4-31B", 5.2e-07, 7.5e-07), + ("openinfer/@oi/Gemma4-31B-It", 5.2e-07, 7.5e-07), ), ) def test_catalog_token_rates_match_vendor_per_million_prices(