diff --git a/app/client/api.ts b/app/client/api.ts index d12e013d8..19e77645d 100644 --- a/app/client/api.ts +++ b/app/client/api.ts @@ -170,15 +170,15 @@ export class ClientApi { export function getHeaders(ignoreHeaders?: boolean) { const accessStore = useAccessStore.getState(); let headers: Record = {}; - if (!ignoreHeaders) { + const modelConfig = useChatStore.getState().currentSession().mask.modelConfig; + const isGoogle = modelConfig.model === "gemini-pro"; + if (!ignoreHeaders && !isGoogle) { headers = { "Content-Type": "application/json", "x-requested-with": "XMLHttpRequest", Accept: "application/json", }; } - const modelConfig = useChatStore.getState().currentSession().mask.modelConfig; - const isGoogle = modelConfig.model === "gemini-pro"; const isAzure = accessStore.provider === ServiceProvider.Azure; const authHeader = isGoogle ? "x-goog-api-key" diff --git a/app/client/platforms/google.ts b/app/client/platforms/google.ts index 3d28c89be..05de1f1e9 100644 --- a/app/client/platforms/google.ts +++ b/app/client/platforms/google.ts @@ -187,7 +187,9 @@ export class GeminiProApi implements LLMApi { } path(path: string): string { const accessStore = useAccessStore.getState(); - const isGoogle = accessStore.provider === ServiceProvider.Google; + const isGoogle = + accessStore.useCustomConfig && + accessStore.provider === ServiceProvider.Google; if (isGoogle && !accessStore.isValidGoogle()) { throw Error(