phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-25 02:34:29 +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({
model: 'gemini-pro'
model: 'gemini-1.5-pro'
}, requestOptions);
const prompt = 'Say "hello test" and nothing else';
@ -76,7 +76,7 @@ describe('Gemini AI Tests', () => {
};
const model = genAI.getGenerativeModel({
model: 'gemini-pro'
model: 'gemini-1.5-pro'
}, requestOptions);
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 modelParams = {
model: 'gemini-pro',
model: 'gemini-1.5-pro',
};
const requestOptions = {
baseUrl: 'http://127.0.0.1:4000/gemini',
customHeaders: {
"tags": "gemini-js-sdk,gemini-pro"
"tags": "gemini-js-sdk,gemini-1.5-pro"
}
};