From b0b322c7e5a3ba987a406ff4ce17d55ab80a22b4 Mon Sep 17 00:00:00 2001 From: AJ-ing Date: Wed, 9 Sep 2026 16:12:37 +1000 Subject: [PATCH] fix(openinfer): use vendor Gemma4-31B-It catalog id Co-authored-by: Cursor --- litellm/model_prices_and_context_window_backup.json | 2 +- model_prices_and_context_window.json | 2 +- tests/test_litellm/llms/openai_like/test_openinfer_provider.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) 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(