feat: session-level model config

This commit is contained in:
Yidadaa
2023-04-22 01:13:23 +08:00
parent a3ca8ea5c4
commit 4cdb2f0fa3
8 changed files with 187 additions and 167 deletions

View File

@@ -334,14 +334,14 @@ export const useChatStore = create<ChatStore>()(
// get short term and unmemoried long term memory
const shortTermMemoryMessageIndex = Math.max(
0,
n - config.historyMessageCount,
n - config.modelConfig.historyMessageCount,
);
const longTermMemoryMessageIndex = session.lastSummarizeIndex;
const oldestIndex = Math.max(
shortTermMemoryMessageIndex,
longTermMemoryMessageIndex,
);
const threshold = config.compressMessageLengthThreshold;
const threshold = config.modelConfig.compressMessageLengthThreshold;
// get recent messages as many as possible
const reversedRecentMessages = [];
@@ -410,7 +410,7 @@ export const useChatStore = create<ChatStore>()(
if (historyMsgLength > config?.modelConfig?.max_tokens ?? 4000) {
const n = toBeSummarizedMsgs.length;
toBeSummarizedMsgs = toBeSummarizedMsgs.slice(
Math.max(0, n - config.historyMessageCount),
Math.max(0, n - config.modelConfig.historyMessageCount),
);
}
@@ -423,11 +423,12 @@ export const useChatStore = create<ChatStore>()(
"[Chat History] ",
toBeSummarizedMsgs,
historyMsgLength,
config.compressMessageLengthThreshold,
config.modelConfig.compressMessageLengthThreshold,
);
if (
historyMsgLength > config.compressMessageLengthThreshold &&
historyMsgLength >
config.modelConfig.compressMessageLengthThreshold &&
session.sendMemory
) {
requestChatStream(

View File

@@ -16,8 +16,6 @@ export enum Theme {
}
const DEFAULT_CONFIG = {
historyMessageCount: 4,
compressMessageLengthThreshold: 1000,
sendBotMessages: true as boolean,
submitKey: SubmitKey.CtrlEnter as SubmitKey,
avatar: "1f603",
@@ -34,6 +32,8 @@ const DEFAULT_CONFIG = {
temperature: 1,
max_tokens: 2000,
presence_penalty: 0,
historyMessageCount: 4,
compressMessageLengthThreshold: 1000,
},
};