From a5fe02bdf28d252e259d505477cddbd22d68eb42 Mon Sep 17 00:00:00 2001 From: Hemang Date: Thu, 6 Nov 2025 11:06:31 +0100 Subject: [PATCH] Change gemini-2.0-flash to gemini-2.5-flash --- 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 f50bf8a..a51fd85 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.0-flash", "anthropic/claude-3-5-haiku-20241022"] +EXAMPLE_MODELS = ["openai/gpt-4o", "gemini/gemini-2.5-flash", "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 ee831a9..04b2880 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.0-flash": "GEMINI_API_KEY", + "gemini/gemini-2.5-flash": "GEMINI_API_KEY", "anthropic/claude-3-5-haiku-20241022": "ANTHROPIC_API_KEY", }