xinyin025/ChatGPT-Next-Web
1
0
Fork 0
mirror of https://github.com/ChatGPTNextWeb/ChatGPT-Next-Web.git synced 2025-11-27 19:19:22 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Added summary model for claude LLMs.

Browse Source
This commit is contained in:
Chenglong Wang
2024-08-14 16:35:37 +08:00
committed by GitHub
parent cf1c8e8f2a
commit 2a20091f78
2 changed files with 20 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
app/store/chat.ts
View File

@@ -12,6 +12,7 @@ import {
StoreKey,
SUMMARIZE_MODEL,
GEMINI_SUMMARIZE_MODEL,
CLAUDE_SUMMARIZE_MODEL,
} from "../constant";
import { getClientApi } from "../client/api";
import type {
@@ -92,21 +93,31 @@ function createEmptySession(): ChatSession {
function getSummarizeModel(currentModel: string) {
// if it is using gpt-* models, force to use 4o-mini to summarize
const configStore = useAppConfig.getState();
const accessStore = useAccessStore.getState();
const allModel = collectModelsWithDefaultModel(
configStore.models,
[configStore.customModels, accessStore.customModels].join(","),
accessStore.defaultModel,
);
if (currentModel.startsWith("gpt")) {
const configStore = useAppConfig.getState();
const accessStore = useAccessStore.getState();
const allModel = collectModelsWithDefaultModel(
configStore.models,
[configStore.customModels, accessStore.customModels].join(","),
accessStore.defaultModel,
);
const summarizeModel = allModel.find(
(m) => m.name === SUMMARIZE_MODEL && m.available,
);
return summarizeModel?.name ?? currentModel;
}
if (currentModel.startsWith("gemini")) {
return GEMINI_SUMMARIZE_MODEL;
const summarizeModel = allModel.find(
(m) => m.name === GEMINI_SUMMARIZE_MODEL && m.available,
);
return summarizeModel?.name ?? currentModel;
}
if (currentModel.startsWith("claude")) {
const summarizeModel = allModel.find(
(m) => m.name === CLAUDE_SUMMARIZE_MODEL && m.available,
);
return summarizeModel?.name ?? currentModel;
}
return currentModel;
}