diff --git a/packages/core/src/core/modelCheck.ts b/packages/core/src/core/modelCheck.ts index 807bd5c5..c9078aaf 100644 --- a/packages/core/src/core/modelCheck.ts +++ b/packages/core/src/core/modelCheck.ts @@ -28,7 +28,7 @@ export async function getEffectiveModel( const modelToTest = DEFAULT_GEMINI_MODEL; const fallbackModel = DEFAULT_GEMINI_FLASH_MODEL; - const endpoint = `https://generativelanguage.googleapis.com/v1beta/models/${modelToTest}:generateContent?key=${apiKey}`; + const endpoint = `https://generativelanguage.googleapis.com/v1beta/models/${modelToTest}:generateContent`; const body = JSON.stringify({ contents: [{ parts: [{ text: 'test' }] }], generationConfig: { @@ -45,7 +45,10 @@ export async function getEffectiveModel( try { const response = await fetch(endpoint, { method: 'POST', - headers: { 'Content-Type': 'application/json' }, + headers: { + 'Content-Type': 'application/json', + 'x-goog-api-key': apiKey, + }, body, signal: controller.signal, });