From 9379cb1038bd5d1d077712a324e3d6f0edf29982 Mon Sep 17 00:00:00 2001 From: yuneng-jiang Date: Wed, 11 Mar 2026 16:46:56 -0700 Subject: [PATCH] [Fix] Replace deprecated models in function calling tests Replace deprecated model references in test_proxy_function_calling_support_consistency: - claude-3-5-sonnet-20240620 -> claude-sonnet-4-6 - gemini-pro -> gemini-2.5-pro - gemini/gemini-1.5-pro -> gemini/gemini-2.5-pro - gemini/gemini-1.5-flash -> gemini/gemini-2.5-flash Co-Authored-By: Claude Opus 4.6 --- tests/test_litellm/test_utils.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/tests/test_litellm/test_utils.py b/tests/test_litellm/test_utils.py index 44b7ffb30d..fe112b8488 100644 --- a/tests/test_litellm/test_utils.py +++ b/tests/test_litellm/test_utils.py @@ -1225,14 +1225,14 @@ class TestProxyFunctionCalling: ), # Anthropic models (Claude supports function calling) ( - "claude-3-5-sonnet-20240620", - "litellm_proxy/claude-3-5-sonnet-20240620", + "claude-sonnet-4-6", + "litellm_proxy/claude-sonnet-4-6", True, ), # Google models - ("gemini-pro", "litellm_proxy/gemini-pro", True), - ("gemini/gemini-1.5-pro", "litellm_proxy/gemini/gemini-1.5-pro", True), - ("gemini/gemini-1.5-flash", "litellm_proxy/gemini/gemini-1.5-flash", True), + ("gemini-2.5-pro", "litellm_proxy/gemini-2.5-pro", True), + ("gemini/gemini-2.5-pro", "litellm_proxy/gemini/gemini-2.5-pro", True), + ("gemini/gemini-2.5-flash", "litellm_proxy/gemini/gemini-2.5-flash", True), # Groq models (mixed support) ("groq/gemma-7b-it", "litellm_proxy/groq/gemma-7b-it", True), (