fix: USE_OPENAI_ENDPOINT_FOR_ALL_MODELS

This commit is contained in:
Hk-Gosuto 2024-06-07 23:29:24 +08:00
parent 740ece39da
commit bc9679335d
4 changed files with 12 additions and 6 deletions

View File

@ -218,7 +218,7 @@ export function getHeaders(ignoreHeaders?: boolean) {
const modelConfig = useChatStore.getState().currentSession().mask.modelConfig;
const isGoogle =
modelConfig.model.startsWith("gemini") &&
!accessStore.useOpenAIEndpointForAllModels();
!accessStore.isUseOpenAIEndpointForAllModels;
if (!ignoreHeaders && !isGoogle) {
headers = {
"Content-Type": "application/json",

View File

@ -1,5 +1,11 @@
/* eslint-disable @next/next/no-img-element */
import { ChatMessage, ModelType, useAppConfig, useChatStore } from "../store";
import {
ChatMessage,
ModelType,
useAccessStore,
useAppConfig,
useChatStore,
} from "../store";
import Locale from "../locales";
import styles from "./exporter.module.scss";
import {

View File

@ -392,8 +392,7 @@ export const useChatStore = createPersistStore(
session.messages.push(botMessage);
});
const isEnableRAG = attachFiles && attachFiles?.length > 0;
var api: ClientApi;
api = new ClientApi(ModelProvider.GPT);
var api: ClientApi = getClientApi(modelConfig.model);
if (
config.pluginConfig.enable &&
session.mask.usePlugins &&
@ -660,7 +659,7 @@ export const useChatStore = createPersistStore(
const session = get().currentSession();
const modelConfig = session.mask.modelConfig;
var api: ClientApi = getClientApi(config.modelConfig.model);
var api: ClientApi = getClientApi(modelConfig.model);
// remove error messages if any
const messages = session.messages;

View File

@ -284,8 +284,9 @@ export function isSupportRAGModel(modelName: string) {
export function getClientApi(modelName: string): ClientApi {
const accessStore = useAccessStore.getState();
if (accessStore.useOpenAIEndpointForAllModels())
if (accessStore.isUseOpenAIEndpointForAllModels) {
return new ClientApi(ModelProvider.GPT);
}
var api: ClientApi;
if (modelName.startsWith("gemini")) {
api = new ClientApi(ModelProvider.GeminiPro);