diff --git a/tests/pass_through_tests/test_gemini_with_spend.test.js b/tests/pass_through_tests/test_gemini_with_spend.test.js index 84010ecee1c..7a6584d1618 100644 --- a/tests/pass_through_tests/test_gemini_with_spend.test.js +++ b/tests/pass_through_tests/test_gemini_with_spend.test.js @@ -29,7 +29,7 @@ describe('Gemini AI Tests', () => { }; const model = genAI.getGenerativeModel({ - model: 'gemini-1.5-pro' + model: 'gemini-2.5-flash-lite' }, requestOptions); const prompt = 'Say "hello test" and nothing else'; @@ -77,7 +77,7 @@ describe('Gemini AI Tests', () => { }; const model = genAI.getGenerativeModel({ - model: 'gemini-1.5-pro' + model: 'gemini-2.5-flash-lite' }, requestOptions); const prompt = 'Say "hello test" and nothing else'; diff --git a/tests/pass_through_tests/test_local_gemini.js b/tests/pass_through_tests/test_local_gemini.js index 1f3f7f8a0d5..0a72ca5cd7b 100644 --- a/tests/pass_through_tests/test_local_gemini.js +++ b/tests/pass_through_tests/test_local_gemini.js @@ -1,13 +1,13 @@ const { GoogleGenerativeAI, ModelParams, RequestOptions } = require("@google/generative-ai"); const modelParams = { - model: 'gemini-1.5-pro', + model: 'gemini-2.5-flash-lite', }; const requestOptions = { baseUrl: 'http://127.0.0.1:4000/gemini', customHeaders: { - "tags": "gemini-js-sdk,gemini-1.5-pro" + "tags": "gemini-js-sdk,gemini-2.5-flash-lite" } }; diff --git a/tests/pass_through_tests/test_local_vertex.js b/tests/pass_through_tests/test_local_vertex.js index a94c6746e9a..231858cdf9a 100644 --- a/tests/pass_through_tests/test_local_vertex.js +++ b/tests/pass_through_tests/test_local_vertex.js @@ -20,7 +20,7 @@ const requestOptions = { }; const generativeModel = vertexAI.getGenerativeModel( - { model: 'gemini-1.5-pro' }, + { model: 'gemini-2.5-flash-lite' }, requestOptions ); diff --git a/tests/pass_through_tests/test_vertex.test.js b/tests/pass_through_tests/test_vertex.test.js index dccd6494025..c10889e3a60 100644 --- a/tests/pass_through_tests/test_vertex.test.js +++ b/tests/pass_through_tests/test_vertex.test.js @@ -75,7 +75,7 @@ describe('Vertex AI Tests', () => { }; const generativeModel = vertexAI.getGenerativeModel( - { model: 'gemini-1.5-pro' }, + { model: 'gemini-2.5-flash-lite' }, requestOptions ); @@ -103,7 +103,7 @@ describe('Vertex AI Tests', () => { const vertexAI = new VertexAI({project: 'pathrise-convert-1606954137718', location: 'us-central1', apiEndpoint: "localhost:4000/vertex-ai"}); const customHeaders = new Headers({"x-litellm-api-key": "sk-1234"}); const requestOptions = {customHeaders: customHeaders}; - const generativeModel = vertexAI.getGenerativeModel({model: 'gemini-1.5-pro'}, requestOptions); + const generativeModel = vertexAI.getGenerativeModel({model: 'gemini-2.5-flash-lite'}, requestOptions); const request = {contents: [{role: 'user', parts: [{text: 'What is 2+2?'}]}]}; const result = await generativeModel.generateContent(request); diff --git a/tests/pass_through_tests/test_vertex_ai.py b/tests/pass_through_tests/test_vertex_ai.py index 4c321122968..dbcf93ee55d 100644 --- a/tests/pass_through_tests/test_vertex_ai.py +++ b/tests/pass_through_tests/test_vertex_ai.py @@ -104,7 +104,7 @@ async def test_basic_vertex_ai_pass_through_with_spendlog(): api_transport="rest", ) - model = GenerativeModel(model_name="gemini-1.5-pro") + model = GenerativeModel(model_name="gemini-2.5-flash-lite") response = model.generate_content("hi") print("response", response) @@ -136,7 +136,7 @@ async def test_basic_vertex_ai_pass_through_streaming_with_spendlog(): api_transport="rest", ) - model = GenerativeModel(model_name="gemini-1.5-pro") + model = GenerativeModel(model_name="gemini-2.5-flash-lite") response = model.generate_content("hi", stream=True) for chunk in response: @@ -192,7 +192,7 @@ async def test_vertex_ai_pass_through_endpoint_context_caching(): ] cached_content = caching.CachedContent.create( - model_name="gemini-1.5-pro-001", + model_name="gemini-2.5-flash-lite-001", system_instruction=system_instruction, contents=contents, ttl=datetime.timedelta(minutes=60), diff --git a/tests/pass_through_tests/test_vertex_with_spend.test.js b/tests/pass_through_tests/test_vertex_with_spend.test.js index 9d10ab4a25d..acfeffd6b9f 100644 --- a/tests/pass_through_tests/test_vertex_with_spend.test.js +++ b/tests/pass_through_tests/test_vertex_with_spend.test.js @@ -85,7 +85,7 @@ describe('Vertex AI Tests', () => { }; const generativeModel = vertexAI.getGenerativeModel( - { model: 'gemini-1.5-pro' }, + { model: 'gemini-2.5-flash-lite' }, requestOptions ); @@ -142,7 +142,7 @@ describe('Vertex AI Tests', () => { }; const generativeModel = vertexAI.getGenerativeModel( - { model: 'gemini-1.5-pro' }, + { model: 'gemini-2.5-flash-lite' }, requestOptions );