phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 18:54:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix pass through tests vertex

Browse source
This commit is contained in:
Ishaan Jaff 2025-02-25 20:21:16 -08:00
parent 89b5eeb345
commit fe3a13ecf9
2 changed files with 4 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

4
tests/pass_through_tests/test_gemini_with_spend.test.js
View file

@ -29,7 +29,7 @@ describe('Gemini AI Tests', () => {
}; };
const model = genAI.getGenerativeModel({ const model = genAI.getGenerativeModel({
model: 'gemini-pro' model: 'gemini-1.5-pro'
}, requestOptions); }, requestOptions);
const prompt = 'Say "hello test" and nothing else'; const prompt = 'Say "hello test" and nothing else';
@ -76,7 +76,7 @@ describe('Gemini AI Tests', () => {
}; };
const model = genAI.getGenerativeModel({ const model = genAI.getGenerativeModel({
model: 'gemini-pro' model: 'gemini-1.5-pro'
}, requestOptions); }, requestOptions);
const prompt = 'Say "hello test" and nothing else'; const prompt = 'Say "hello test" and nothing else';

4
tests/pass_through_tests/test_local_gemini.js
View file

@ -1,13 +1,13 @@
const { GoogleGenerativeAI, ModelParams, RequestOptions } = require("@google/generative-ai"); const { GoogleGenerativeAI, ModelParams, RequestOptions } = require("@google/generative-ai");
const modelParams = { const modelParams = {
model: 'gemini-pro', model: 'gemini-1.5-pro',
}; };
const requestOptions = { const requestOptions = {
baseUrl: 'http://127.0.0.1:4000/gemini', baseUrl: 'http://127.0.0.1:4000/gemini',
customHeaders: { customHeaders: {
"tags": "gemini-js-sdk,gemini-pro" "tags": "gemini-js-sdk,gemini-1.5-pro"
} }
}; };