diff --git a/tests/pass_through_tests/test_local_gemini.js b/tests/pass_through_tests/test_local_gemini.js index 038af626d..7043a5ab4 100644 --- a/tests/pass_through_tests/test_local_gemini.js +++ b/tests/pass_through_tests/test_local_gemini.js @@ -5,7 +5,10 @@ const modelParams = { }; const requestOptions = { - baseUrl: 'http://127.0.0.1:4000/gemini' + baseUrl: 'http://127.0.0.1:4000/gemini', + customHeaders: { + "tags": "gemini-js-sdk,gemini-pro" + } }; const genAI = new GoogleGenerativeAI("sk-1234"); // litellm proxy API key