diff --git a/tests/local_testing/test_function_calling.py b/tests/local_testing/test_function_calling.py index 71249898f79..1312c4dcc6b 100644 --- a/tests/local_testing/test_function_calling.py +++ b/tests/local_testing/test_function_calling.py @@ -48,7 +48,7 @@ def get_current_weather(location, unit="fahrenheit"): "gpt-3.5-turbo-1106", "mistral/mistral-large-latest", "claude-3-haiku-20240307", - "gemini/gemini-1.5-pro", + "gemini/gemini-2.5-flash-lite", "anthropic.claude-3-sonnet-20240229-v1:0", "cohere_chat/command-r", ],