fix: gemini

This commit is contained in:
Hk-Gosuto 2023-12-29 17:59:49 +08:00
parent 20d0717d0c
commit 6c71c83af7
2 changed files with 6 additions and 4 deletions

View File

@ -170,15 +170,15 @@ export class ClientApi {
export function getHeaders(ignoreHeaders?: boolean) {
const accessStore = useAccessStore.getState();
let headers: Record<string, string> = {};
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"

View File

@ -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(