mirror of
https://github.com/BerriAI/litellm.git
synced 2026-09-28 01:32:17 +00:00
e48f813d4d
gemini-2.5-flash-lite retires on 2026-10-20 per Google's Vertex model lifecycle table, which would turn the google_generate_content_endpoint_testing CircleCI job red on every PR from that day. gemini-3.5-flash-lite is the newest flash-lite on both AI Studio and Vertex, with a retirement of 2027-07-21 or later. Vertex serves gemini-3.5-flash-lite only on the global, us, and eu endpoints, never on a single region, so the direct SDK Vertex tests now pass vertex_location global like the suite's proxy config already does. The 429 backoff test derives its mocked generateContent URL from litellm's own Gemini URL builder, since litellm routes Gemini 3 and newer models to v1alpha instead of v1beta. Co-authored-by: mateo-berri <277851410+mateo-berri@users.noreply.github.com>
17 lines
501 B
Python
17 lines
501 B
Python
from base_google_genai_proxy_sdk_test import BaseGoogleGenAIProxySDKTest
|
|
from base_google_test import BaseGoogleGenAITest
|
|
|
|
|
|
class TestVertexAIGenerateContent(BaseGoogleGenAITest, BaseGoogleGenAIProxySDKTest):
|
|
"""Test Vertex AI"""
|
|
|
|
@property
|
|
def model_config(self):
|
|
return {
|
|
"model": "vertex_ai/gemini-3.5-flash-lite",
|
|
"vertex_location": "global",
|
|
}
|
|
|
|
@property
|
|
def proxy_model_name(self) -> str:
|
|
return "vertex-gemini-3.5-flash-lite"
|