xinyin025/ChatGPT-Next-Web
1
0
Fork 0
mirror of https://github.com/ChatGPTNextWeb/ChatGPT-Next-Web.git synced 2025-10-01 23:56:39 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

修改消息压缩使用的模型url

Browse Source
This commit is contained in:
sijinhui 2024-07-19 18:03:38 +08:00
parent 69c434c047
commit 60813d448f
2 changed files with 21 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
app/client/platforms/openai.ts
View File

@ -119,6 +119,7 @@ export class ChatGPTApi implements LLMApi {
providerName: options.config.providerName,
},
};
console.log('-------', modelConfig, options)
const requestPayload: RequestPayload = {
messages,
stream: options.config.stream,

26
app/store/chat.ts
View File

@ -97,7 +97,10 @@ function createEmptySession(): ChatSession {
const ChatFetchTaskPool: Record<string, any> = {};
function getSummarizeModel(currentModel: string) {
function getSummarizeModel(currentModel: string): {
name: string,
providerName: string | undefined,
} {
// if it is using gpt-* models, force to use 3.5 to summarize
if (currentModel.startsWith("gpt")) {
const configStore = useAppConfig.getState();
@ -110,12 +113,21 @@ function getSummarizeModel(currentModel: string) {
const summarizeModel = allModel.find(
(m) => m.name === SUMMARIZE_MODEL && m.available,
);
return summarizeModel?.name ?? currentModel;
return {
name: summarizeModel?.name ?? currentModel,
providerName: summarizeModel?.provider?.providerName,
}
}
if (currentModel.startsWith("gemini")) {
return GEMINI_SUMMARIZE_MODEL;
return {
name: GEMINI_SUMMARIZE_MODEL,
providerName: ServiceProvider.Google,
}
}
return {
name: currentModel,
providerName: undefined,
}
return currentModel;
}
function countMessages(msgs: ChatMessage[]) {
@ -905,7 +917,8 @@ export const useChatStore = createPersistStore(
api.llm.chat({
messages: topicMessages,
config: {
model: getSummarizeModel(session.mask.modelConfig.model),
model: getSummarizeModel(session.mask.modelConfig.model).name,
providerName: getSummarizeModel(session.mask.modelConfig.model).providerName,
stream: false,
},
onFinish(message) {
@ -967,7 +980,8 @@ export const useChatStore = createPersistStore(
config: {
...modelcfg,
stream: true,
model: getSummarizeModel(session.mask.modelConfig.model),
model: getSummarizeModel(session.mask.modelConfig.model).name,
providerName: getSummarizeModel(session.mask.modelConfig.model).providerName,
},
onUpdate(message) {
session.memoryPrompt = message;