This commit is contained in:
H0llyW00dzZ
2024-10-20 19:57:32 +08:00
committed by GitHub
3 changed files with 33 additions and 15 deletions

View File

@@ -652,10 +652,19 @@ export const useChatStore = createPersistStore(
onFinish(message) {
if (!isValidMessage(message)) return;
get().updateCurrentSession(
(session) =>
(session.topic =
message.length > 0 ? trimTopic(message) : DEFAULT_TOPIC),
);
(session) => {
session.topic =
message.length > 0 ? trimTopic(message) : DEFAULT_TOPIC;
// Add system message after summarizing the topic
// which is powerful based of fine-tuning
const systemMessage: ChatMessage = {
date: new Date().toLocaleString(),
id: nanoid(),
role: "system",
content: `${Locale.FineTuned.Sysmessage} ${session.topic}`,
};
session.messages = [systemMessage, ...session.messages];
});
},
});
}