38 lines
1.4 KiB
TypeScript
38 lines
1.4 KiB
TypeScript
import type {StoredChatMessage} from "./conversation-types";
|
||
|
||
export const untitledConversationLabel = "未命名会话";
|
||
|
||
function messageText(message: StoredChatMessage) {
|
||
return message.parts
|
||
.filter((part) => part.type === "text" && typeof part.text === "string")
|
||
.map((part) => String(part.text))
|
||
.join("\n")
|
||
.trim();
|
||
}
|
||
|
||
export function conversationTitlePrompt(messages: StoredChatMessage[]) {
|
||
const transcript = messages
|
||
.filter((message) => message.role === "user" || message.role === "assistant")
|
||
.slice(0, 4)
|
||
.map((message) => `${message.role === "user" ? "用户" : "助手"}：${messageText(message).slice(0, 2000)}`)
|
||
.filter((line) => !line.endsWith("："))
|
||
.join("\n\n");
|
||
return `请为下面的对话生成一个简洁、具体的中文标题。只输出标题，不要解释，不要加引号、Markdown 或“标题：”前缀。标题不超过 30 个字符，不要使用斜杠。\n\n${transcript}`;
|
||
}
|
||
|
||
export function normalizeGeneratedConversationTitle(value: string) {
|
||
let title = value
|
||
.replace(/<think>[\s\S]*?<\/think>/gi, "")
|
||
.trim()
|
||
.split(/\r?\n/)[0]
|
||
.replace(/^#{1,6}\s*/, "")
|
||
.replace(/^(?:标题|title)\s*[：:]\s*/i, "")
|
||
.replace(/^[`'“”‘’\"]+|[`'“”‘’\"]+$/g, "")
|
||
.replace(/[\u0000-\u001f]/g, " ")
|
||
.replace(/\s+/g, " ")
|
||
.replaceAll("/", "／")
|
||
.trim();
|
||
title = [...title].slice(0, 60).join("").trim();
|
||
return title;
|
||
}
|