From a6bed68aa57490af7dd0fa2c22aba67f33d0f516 Mon Sep 17 00:00:00 2001 From: Hemang Date: Thu, 6 Nov 2025 11:45:00 +0100 Subject: [PATCH] Update gemini model name for litellm --- README.md | 2 +- tests/integration/litellm/test_chat_without_tool_call.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index a51fd85..e788029 100644 --- a/README.md +++ b/README.md @@ -160,7 +160,7 @@ import random import os base_url = "/api/v1/gateway/litellm/{add-your-dataset-name-here}" -EXAMPLE_MODELS = ["openai/gpt-4o", "gemini/gemini-2.5-flash", "anthropic/claude-3-5-haiku-20241022"] +EXAMPLE_MODELS = ["openai/gpt-4o", "gemini/gemini-2.5-flash-preview-09-2025", "anthropic/claude-3-5-haiku-20241022"] model = random.choice(SAMPLE_MODELS) base_url += "/" + model.split("/")[0] # append /gemini /openai or /anthropic. diff --git a/tests/integration/litellm/test_chat_without_tool_call.py b/tests/integration/litellm/test_chat_without_tool_call.py index 04b2880..f535ebd 100644 --- a/tests/integration/litellm/test_chat_without_tool_call.py +++ b/tests/integration/litellm/test_chat_without_tool_call.py @@ -10,7 +10,7 @@ from litellm import completion MODEL_API_KEYS = { "openai/gpt-4o": "OPENAI_API_KEY", - "gemini/gemini-2.5-flash": "GEMINI_API_KEY", + "gemini/gemini-2.5-flash-preview-09-2025": "GEMINI_API_KEY", "anthropic/claude-3-5-haiku-20241022": "ANTHROPIC_API_KEY", }