Add support for message history pagination, preserve scroll in rendering, and improve draft/message workflows.
This commit is contained in:
+2
-1
@@ -19,7 +19,7 @@ import {createChatsRouter} from "./routes/chats.js";
|
||||
import {createPagesRouter} from "./routes/pages.js";
|
||||
import {createPersonasRouter} from "./routes/personas.js";
|
||||
import {createSettingsRouter} from "./routes/settings.js";
|
||||
import {createSystemRouter} from "./routes/system.js";
|
||||
import {createHealthRouter, createSystemRouter} from "./routes/system.js";
|
||||
import {createAttachmentService} from "./services/attachmentService.js";
|
||||
import {createUploadsRouter} from "./routes/uploads.js";
|
||||
import {createVisionDebugService} from "./services/visionDebugService.js";
|
||||
@@ -271,6 +271,7 @@ export function createApp(options = {}) {
|
||||
);
|
||||
|
||||
app.use(observability.middleware);
|
||||
app.use(createHealthRouter({modelService}));
|
||||
app.use(express.json({limit: "1mb"}));
|
||||
app.use(session({
|
||||
store: options.sessionStore || (config.testMode ? undefined : new FileStore({path: config.sessionsDir, reapInterval: 3600})),
|
||||
|
||||
@@ -0,0 +1,80 @@
|
||||
export function createAccountStatements(db) {
|
||||
return {
|
||||
insertUser: db.prepare("INSERT INTO users (username, password) VALUES (?, ?)"),
|
||||
getUser: db.prepare("SELECT username, password, role, session_version FROM users WHERE username = ?"),
|
||||
updateUsername: db.prepare("UPDATE users SET username = ? WHERE username = ?"),
|
||||
updatePassword: db.prepare("UPDATE users SET password = ? WHERE username = ?"),
|
||||
incrementSessionVersion: db.prepare("UPDATE users SET session_version = COALESCE(session_version, 0) + 1 WHERE username = ?"),
|
||||
updateChatSessionsUsername: db.prepare("UPDATE chat_sessions SET username = ? WHERE username = ?"),
|
||||
updatePersonasUsername: db.prepare("UPDATE personas SET username = ? WHERE username = ?"),
|
||||
updatePersonasSourceCreator: db.prepare("UPDATE personas SET source_creator_username = ? WHERE source_creator_username = ?"),
|
||||
updateUserSettingsUsername: db.prepare("UPDATE user_settings SET username = ? WHERE username = ?"),
|
||||
updatePersonaMarketCreator: db.prepare("UPDATE persona_market SET creator_username = ? WHERE creator_username = ?"),
|
||||
updatePersonaVersionsUsername: db.prepare("UPDATE persona_versions SET username = ? WHERE username = ?"),
|
||||
updatePromptTemplatesUsername: db.prepare("UPDATE prompt_templates SET username = ? WHERE username = ?"),
|
||||
updateChatMemoriesUsername: db.prepare("UPDATE chat_memories SET username = ? WHERE username = ?"),
|
||||
updatePersonaChatStateUsername: db.prepare("UPDATE persona_chat_state SET username = ? WHERE username = ?"),
|
||||
updateMarketFavoritesUsername: db.prepare("UPDATE persona_market_favorites SET username = ? WHERE username = ?"),
|
||||
updateMarketRatingsUsername: db.prepare("UPDATE persona_market_ratings SET username = ? WHERE username = ?"),
|
||||
updateMarketReportsUsername: db.prepare("UPDATE persona_market_reports SET reporter_username = ? WHERE reporter_username = ?"),
|
||||
deleteChatMemoriesByUser: db.prepare("DELETE FROM chat_memories WHERE username = ?"),
|
||||
deleteChatSessionsByUser: db.prepare("DELETE FROM chat_sessions WHERE username = ?"),
|
||||
deletePersonasByUser: db.prepare("DELETE FROM personas WHERE username = ?"),
|
||||
deleteUserSettingsByUser: db.prepare("DELETE FROM user_settings WHERE username = ?"),
|
||||
deletePersonaVersionsByUser: db.prepare("DELETE FROM persona_versions WHERE username = ?"),
|
||||
deletePromptTemplatesByUser: db.prepare("DELETE FROM prompt_templates WHERE username = ?"),
|
||||
deleteMarketPersonasByCreator: db.prepare("DELETE FROM persona_market WHERE creator_username = ?"),
|
||||
deleteMarketFavoritesByUser: db.prepare("DELETE FROM persona_market_favorites WHERE username = ?"),
|
||||
deleteMarketRatingsByUser: db.prepare("DELETE FROM persona_market_ratings WHERE username = ?"),
|
||||
deleteMarketReportsByUser: db.prepare("DELETE FROM persona_market_reports WHERE reporter_username = ?"),
|
||||
deletePersonaChatStateByUser: db.prepare("DELETE FROM persona_chat_state WHERE username = ?"),
|
||||
deleteUser: db.prepare("DELETE FROM users WHERE username = ?")
|
||||
};
|
||||
}
|
||||
|
||||
export function createAccountRepository(db, statements) {
|
||||
const renameUserTransaction = db.transaction((currentUsername, nextUsername) => {
|
||||
// Existing username references do not all cascade on update. Check them at commit,
|
||||
// after every owned row has been moved to the new username.
|
||||
db.pragma("defer_foreign_keys = ON");
|
||||
statements.updateUsername.run(nextUsername, currentUsername);
|
||||
statements.updateChatSessionsUsername.run(nextUsername, currentUsername);
|
||||
statements.updatePersonasUsername.run(nextUsername, currentUsername);
|
||||
statements.updatePersonasSourceCreator.run(nextUsername, currentUsername);
|
||||
statements.updateUserSettingsUsername.run(nextUsername, currentUsername);
|
||||
statements.updatePersonaMarketCreator.run(nextUsername, currentUsername);
|
||||
statements.updatePersonaVersionsUsername.run(nextUsername, currentUsername);
|
||||
statements.updatePromptTemplatesUsername.run(nextUsername, currentUsername);
|
||||
statements.updateChatMemoriesUsername.run(nextUsername, currentUsername);
|
||||
statements.updatePersonaChatStateUsername.run(nextUsername, currentUsername);
|
||||
statements.updateMarketFavoritesUsername.run(nextUsername, currentUsername);
|
||||
statements.updateMarketRatingsUsername.run(nextUsername, currentUsername);
|
||||
statements.updateMarketReportsUsername.run(nextUsername, currentUsername);
|
||||
});
|
||||
|
||||
const deleteUserTransaction = db.transaction((username) => {
|
||||
statements.deleteMarketFavoritesByUser.run(username);
|
||||
statements.deleteMarketRatingsByUser.run(username);
|
||||
statements.deleteMarketReportsByUser.run(username);
|
||||
statements.deleteMarketPersonasByCreator.run(username);
|
||||
statements.deletePromptTemplatesByUser.run(username);
|
||||
statements.deletePersonaVersionsByUser.run(username);
|
||||
statements.deleteChatMemoriesByUser.run(username);
|
||||
statements.deletePersonaChatStateByUser.run(username);
|
||||
statements.deleteChatSessionsByUser.run(username);
|
||||
statements.deleteUserSettingsByUser.run(username);
|
||||
statements.deletePersonasByUser.run(username);
|
||||
return statements.deleteUser.run(username);
|
||||
});
|
||||
return {
|
||||
getUser: (username) => statements.getUser.get(username),
|
||||
insertUser: (username, passwordHash) => statements.insertUser.run(username, passwordHash),
|
||||
updatePassword: (passwordHash, username) => statements.updatePassword.run(passwordHash, username),
|
||||
incrementSessionVersion(username) {
|
||||
statements.incrementSessionVersion.run(username);
|
||||
return statements.getUser.get(username);
|
||||
},
|
||||
renameUser: (currentUsername, nextUsername) => renameUserTransaction(currentUsername, nextUsername),
|
||||
deleteUser: (username) => deleteUserTransaction(username)
|
||||
};
|
||||
}
|
||||
@@ -0,0 +1,559 @@
|
||||
export function createChatStatements(db) {
|
||||
return {
|
||||
insertChatSession: db.prepare(
|
||||
"INSERT INTO chat_sessions (username, title, assistant_persona_id, user_persona_id, folder_name, is_pinned, archived_at, scenario_prompt, scenario_summary) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)"
|
||||
),
|
||||
listChatSessions: db.prepare(`
|
||||
SELECT cs.id,
|
||||
CASE
|
||||
WHEN cs.assistant_persona_id IS NOT NULL AND cs.source_chat_id IS NULL THEN COALESCE(p.name, cs.title)
|
||||
ELSE cs.title
|
||||
END AS title,
|
||||
cs.title AS stored_title,
|
||||
cs.assistant_persona_id,
|
||||
p.name AS assistant_persona_name,
|
||||
cs.user_persona_id,
|
||||
up.name AS user_persona_name,
|
||||
cs.folder_name,
|
||||
cs.is_pinned,
|
||||
cs.archived_at,
|
||||
cs.scenario_prompt,
|
||||
cs.scenario_summary,
|
||||
cs.context_summary,
|
||||
cs.context_summary_message_id,
|
||||
cs.active_leaf_message_id,
|
||||
cs.source_chat_id,
|
||||
cs.branched_from_message_id,
|
||||
cs.preferred_model,
|
||||
cs.temperature,
|
||||
cs.context_length,
|
||||
cs.response_length,
|
||||
cs.system_instruction,
|
||||
cs.created_at,
|
||||
cs.updated_at
|
||||
FROM chat_sessions cs
|
||||
LEFT JOIN personas p ON p.id = cs.assistant_persona_id
|
||||
LEFT JOIN personas up ON up.id = cs.user_persona_id
|
||||
WHERE cs.username = ?
|
||||
ORDER BY cs.is_pinned DESC, datetime(cs.updated_at) DESC
|
||||
`),
|
||||
countChatSessions: db.prepare("SELECT COUNT(*) as count FROM chat_sessions WHERE username = ? AND archived_at IS NULL"),
|
||||
getChatSession: db.prepare(`
|
||||
SELECT cs.id,
|
||||
CASE
|
||||
WHEN cs.assistant_persona_id IS NOT NULL AND cs.source_chat_id IS NULL THEN COALESCE(p.name, cs.title)
|
||||
ELSE cs.title
|
||||
END AS title,
|
||||
cs.title AS stored_title,
|
||||
cs.assistant_persona_id,
|
||||
p.name AS assistant_persona_name,
|
||||
cs.user_persona_id,
|
||||
up.name AS user_persona_name,
|
||||
cs.folder_name,
|
||||
cs.is_pinned,
|
||||
cs.archived_at,
|
||||
cs.scenario_prompt,
|
||||
cs.scenario_summary,
|
||||
cs.context_summary,
|
||||
cs.context_summary_message_id,
|
||||
cs.active_leaf_message_id,
|
||||
cs.source_chat_id,
|
||||
cs.branched_from_message_id,
|
||||
cs.preferred_model,
|
||||
cs.temperature,
|
||||
cs.context_length,
|
||||
cs.response_length,
|
||||
cs.system_instruction,
|
||||
cs.created_at,
|
||||
cs.updated_at
|
||||
FROM chat_sessions cs
|
||||
LEFT JOIN personas p ON p.id = cs.assistant_persona_id
|
||||
LEFT JOIN personas up ON up.id = cs.user_persona_id
|
||||
WHERE cs.id = ?
|
||||
AND cs.username = ?
|
||||
`),
|
||||
getChatSessionByParticipants: db.prepare(`
|
||||
SELECT cs.id,
|
||||
CASE
|
||||
WHEN cs.assistant_persona_id IS NOT NULL AND cs.source_chat_id IS NULL THEN COALESCE(p.name, cs.title)
|
||||
ELSE cs.title
|
||||
END AS title,
|
||||
cs.title AS stored_title,
|
||||
cs.assistant_persona_id,
|
||||
p.name AS assistant_persona_name,
|
||||
cs.user_persona_id,
|
||||
up.name AS user_persona_name,
|
||||
cs.folder_name,
|
||||
cs.is_pinned,
|
||||
cs.archived_at,
|
||||
cs.scenario_prompt,
|
||||
cs.scenario_summary,
|
||||
cs.context_summary,
|
||||
cs.context_summary_message_id,
|
||||
cs.active_leaf_message_id,
|
||||
cs.source_chat_id,
|
||||
cs.branched_from_message_id,
|
||||
cs.preferred_model,
|
||||
cs.temperature,
|
||||
cs.context_length,
|
||||
cs.response_length,
|
||||
cs.system_instruction,
|
||||
cs.created_at,
|
||||
cs.updated_at
|
||||
FROM chat_sessions cs
|
||||
LEFT JOIN personas p ON p.id = cs.assistant_persona_id
|
||||
LEFT JOIN personas up ON up.id = cs.user_persona_id
|
||||
WHERE cs.username = ?
|
||||
AND cs.assistant_persona_id = ?
|
||||
AND ((cs.user_persona_id IS NULL AND ? IS NULL) OR cs.user_persona_id = ?)
|
||||
AND cs.source_chat_id IS NULL
|
||||
ORDER BY cs.is_pinned DESC, datetime(cs.updated_at) DESC
|
||||
LIMIT 1
|
||||
`),
|
||||
updateChatSessionTitle: db.prepare("UPDATE chat_sessions SET title = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ? AND username = ?"),
|
||||
updateChatSessionScene: db.prepare("UPDATE chat_sessions SET scenario_prompt = ?, scenario_summary = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ? AND username = ?"),
|
||||
updateChatContextSummary: db.prepare("UPDATE chat_sessions SET context_summary = ?, context_summary_message_id = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ? AND username = ?"),
|
||||
updateChatGenerationSettings: db.prepare(`
|
||||
UPDATE chat_sessions
|
||||
SET preferred_model = ?, temperature = ?, context_length = ?, response_length = ?, system_instruction = ?, updated_at = CURRENT_TIMESTAMP
|
||||
WHERE id = ? AND username = ?
|
||||
`),
|
||||
updateChatPreferredModel: db.prepare("UPDATE chat_sessions SET preferred_model = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ? AND username = ?"),
|
||||
updateChatActiveLeaf: db.prepare("UPDATE chat_sessions SET active_leaf_message_id = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ? AND username = ?"),
|
||||
updateChatBranchSource: db.prepare("UPDATE chat_sessions SET source_chat_id = ?, branched_from_message_id = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ? AND username = ?"),
|
||||
setChatActiveLeafById: db.prepare("UPDATE chat_sessions SET active_leaf_message_id = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ?"),
|
||||
updateChatSessionOrganization: db.prepare(`
|
||||
UPDATE chat_sessions
|
||||
SET folder_name = ?,
|
||||
is_pinned = ?,
|
||||
archived_at = ?,
|
||||
updated_at = CURRENT_TIMESTAMP
|
||||
WHERE id = ?
|
||||
AND username = ?
|
||||
`),
|
||||
touchChatSession: db.prepare("UPDATE chat_sessions SET updated_at = CURRENT_TIMESTAMP WHERE id = ? AND username = ?"),
|
||||
deleteChatSession: db.prepare("DELETE FROM chat_sessions WHERE id = ? AND username = ?"),
|
||||
insertChatMessage: db.prepare("INSERT INTO chat_messages (chat_id, role, content, model_name, retry_variants, retry_active_index, retry_retries_used, retry_prompt_message_id, delivery_status, error_message, parent_message_id) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)"),
|
||||
updateChatMessageDelivery: db.prepare("UPDATE chat_messages SET delivery_status = ?, error_message = ? WHERE id = ? AND chat_id = ?"),
|
||||
updateChatMessageSources: db.prepare("UPDATE chat_messages SET sources = ? WHERE id = ? AND chat_id = ?"),
|
||||
updateChatMessageThinking: db.prepare("UPDATE chat_messages SET thinking = ? WHERE id = ? AND chat_id = ?"),
|
||||
getChatMessages: db.prepare(`
|
||||
SELECT thinking, sources, id, role, content, model_name, retry_variants, retry_active_index, retry_retries_used, retry_prompt_message_id, delivery_status, error_message, parent_message_id, created_at
|
||||
FROM chat_messages
|
||||
WHERE chat_id = ?
|
||||
ORDER BY id
|
||||
`),
|
||||
getRecentChatMessages: db.prepare(`
|
||||
WITH RECURSIVE branch AS (
|
||||
SELECT cm.* FROM chat_messages cm
|
||||
JOIN chat_sessions cs ON cs.id = cm.chat_id
|
||||
WHERE cm.chat_id = ?
|
||||
AND cm.id = COALESCE(cs.active_leaf_message_id, (SELECT MAX(id) FROM chat_messages WHERE chat_id = cm.chat_id))
|
||||
UNION ALL
|
||||
SELECT parent.* FROM chat_messages parent
|
||||
JOIN branch child ON child.parent_message_id = parent.id
|
||||
)
|
||||
SELECT id, role, content FROM branch
|
||||
WHERE delivery_status != 'failed'
|
||||
ORDER BY id DESC
|
||||
LIMIT ?
|
||||
`),
|
||||
getRecentChatMessagesUpToId: db.prepare(`
|
||||
WITH RECURSIVE branch AS (
|
||||
SELECT * FROM chat_messages WHERE chat_id = ? AND id = ?
|
||||
UNION ALL
|
||||
SELECT parent.* FROM chat_messages parent
|
||||
JOIN branch child ON child.parent_message_id = parent.id
|
||||
)
|
||||
SELECT id, role, content FROM branch
|
||||
WHERE delivery_status != 'failed'
|
||||
ORDER BY id DESC
|
||||
LIMIT ?
|
||||
`),
|
||||
getChatBranchThroughMessage: db.prepare(`
|
||||
WITH RECURSIVE branch AS (
|
||||
SELECT * FROM chat_messages WHERE chat_id = ? AND id = ?
|
||||
UNION ALL
|
||||
SELECT parent.* FROM chat_messages parent
|
||||
JOIN branch child ON child.parent_message_id = parent.id
|
||||
)
|
||||
SELECT id, role, content, model_name, retry_variants, retry_active_index, retry_retries_used,
|
||||
retry_prompt_message_id, delivery_status, error_message, parent_message_id, created_at, sources, thinking
|
||||
FROM branch
|
||||
WHERE delivery_status = 'complete'
|
||||
ORDER BY id
|
||||
`),
|
||||
getActiveBranchMessages: db.prepare(`
|
||||
WITH RECURSIVE branch AS (
|
||||
SELECT cm.* FROM chat_messages cm
|
||||
JOIN chat_sessions cs ON cs.id = cm.chat_id
|
||||
WHERE cm.chat_id = ?
|
||||
AND cm.id = COALESCE(cs.active_leaf_message_id, (SELECT MAX(id) FROM chat_messages WHERE chat_id = cm.chat_id))
|
||||
UNION ALL
|
||||
SELECT parent.* FROM chat_messages parent
|
||||
JOIN branch child ON child.parent_message_id = parent.id
|
||||
)
|
||||
SELECT id, role, content, model_name, retry_variants, retry_active_index, retry_retries_used,
|
||||
retry_prompt_message_id, delivery_status, error_message, parent_message_id, created_at, sources, thinking
|
||||
FROM branch ORDER BY id
|
||||
`),
|
||||
getSiblingChatMessages: db.prepare(`
|
||||
SELECT id, role, content, parent_message_id
|
||||
FROM chat_messages
|
||||
WHERE chat_id = ? AND role = ? AND parent_message_id IS ?
|
||||
ORDER BY id
|
||||
`),
|
||||
getLatestDescendantMessage: db.prepare(`
|
||||
WITH RECURSIVE descendants AS (
|
||||
SELECT id FROM chat_messages WHERE chat_id = ? AND id = ?
|
||||
UNION ALL
|
||||
SELECT child.id FROM chat_messages child JOIN descendants parent ON child.parent_message_id = parent.id
|
||||
WHERE child.chat_id = ? AND child.delivery_status != 'failed'
|
||||
)
|
||||
SELECT cm.* FROM chat_messages cm JOIN descendants d ON d.id = cm.id ORDER BY cm.id DESC LIMIT 1
|
||||
`),
|
||||
getChatMessageById: db.prepare(`
|
||||
SELECT thinking, sources, id, role, content, model_name, retry_variants, retry_active_index, retry_retries_used, retry_prompt_message_id, delivery_status, error_message, parent_message_id, created_at
|
||||
FROM chat_messages
|
||||
WHERE id = ?
|
||||
AND chat_id = ?
|
||||
`),
|
||||
getChatMessageByIndex: db.prepare(`
|
||||
SELECT thinking, sources, id, role, content, model_name, retry_variants, retry_active_index, retry_retries_used, retry_prompt_message_id, delivery_status, error_message, parent_message_id, created_at
|
||||
FROM chat_messages
|
||||
WHERE chat_id = ?
|
||||
ORDER BY id
|
||||
LIMIT 1 OFFSET ?
|
||||
`),
|
||||
getLatestChatMessage: db.prepare(`
|
||||
SELECT id, role, content, model_name, retry_variants, retry_active_index, retry_retries_used, retry_prompt_message_id, delivery_status, error_message, parent_message_id, created_at
|
||||
FROM chat_messages
|
||||
WHERE chat_id = ?
|
||||
ORDER BY id DESC
|
||||
LIMIT 1
|
||||
`),
|
||||
getPreviousUserMessage: db.prepare(`
|
||||
SELECT id, role, content, model_name, retry_variants, retry_active_index, retry_retries_used, retry_prompt_message_id, delivery_status, error_message, parent_message_id, created_at
|
||||
FROM chat_messages
|
||||
WHERE chat_id = ?
|
||||
AND role = 'user'
|
||||
AND id < ?
|
||||
ORDER BY id DESC
|
||||
LIMIT 1
|
||||
`),
|
||||
listChatMemories: db.prepare(`
|
||||
SELECT id, username, chat_id, assistant_persona_id, user_persona_id, fact, created_at, updated_at
|
||||
FROM chat_memories
|
||||
WHERE chat_id = ?
|
||||
AND username = ?
|
||||
ORDER BY datetime(updated_at) DESC, id DESC
|
||||
`),
|
||||
getChatMemory: db.prepare(`
|
||||
SELECT id, username, chat_id, assistant_persona_id, user_persona_id, fact, created_at, updated_at
|
||||
FROM chat_memories
|
||||
WHERE id = ?
|
||||
AND chat_id = ?
|
||||
AND username = ?
|
||||
`),
|
||||
insertChatMemory: db.prepare(`
|
||||
INSERT INTO chat_memories (username, chat_id, assistant_persona_id, user_persona_id, fact)
|
||||
VALUES (?, ?, ?, ?, ?)
|
||||
`),
|
||||
updateChatMemory: db.prepare(`
|
||||
UPDATE chat_memories
|
||||
SET fact = ?,
|
||||
updated_at = CURRENT_TIMESTAMP
|
||||
WHERE id = ?
|
||||
AND chat_id = ?
|
||||
AND username = ?
|
||||
`),
|
||||
deleteChatMemory: db.prepare("DELETE FROM chat_memories WHERE id = ? AND chat_id = ? AND username = ?"),
|
||||
getPersonaChatState: db.prepare(`
|
||||
SELECT id,
|
||||
username,
|
||||
chat_id,
|
||||
assistant_persona_id,
|
||||
user_persona_id,
|
||||
relationship_notes,
|
||||
tone,
|
||||
current_location,
|
||||
goals,
|
||||
unresolved_threads,
|
||||
boundaries,
|
||||
created_at,
|
||||
updated_at
|
||||
FROM persona_chat_state
|
||||
WHERE chat_id = ?
|
||||
AND username = ?
|
||||
`),
|
||||
upsertPersonaChatState: db.prepare(`
|
||||
INSERT INTO persona_chat_state (
|
||||
username,
|
||||
chat_id,
|
||||
assistant_persona_id,
|
||||
user_persona_id,
|
||||
relationship_notes,
|
||||
tone,
|
||||
current_location,
|
||||
goals,
|
||||
unresolved_threads,
|
||||
boundaries
|
||||
)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||
ON CONFLICT(chat_id) DO UPDATE SET
|
||||
username = excluded.username,
|
||||
assistant_persona_id = excluded.assistant_persona_id,
|
||||
user_persona_id = excluded.user_persona_id,
|
||||
relationship_notes = excluded.relationship_notes,
|
||||
tone = excluded.tone,
|
||||
current_location = excluded.current_location,
|
||||
goals = excluded.goals,
|
||||
unresolved_threads = excluded.unresolved_threads,
|
||||
boundaries = excluded.boundaries,
|
||||
updated_at = CURRENT_TIMESTAMP
|
||||
`),
|
||||
updateChatMessage: db.prepare("UPDATE chat_messages SET content = ? WHERE id = ? AND chat_id = ?"),
|
||||
updateChatMessageWithRetryState: db.prepare("UPDATE chat_messages SET content = ?, retry_variants = ?, retry_active_index = ?, retry_retries_used = ?, retry_prompt_message_id = ? WHERE id = ? AND chat_id = ?"),
|
||||
deleteChatMessage: db.prepare("DELETE FROM chat_messages WHERE id = ? AND chat_id = ?"),
|
||||
deleteChatMessages: db.prepare("DELETE FROM chat_messages WHERE chat_id = ?"),
|
||||
searchChats: db.prepare(`
|
||||
SELECT cs.id,
|
||||
CASE
|
||||
WHEN cs.assistant_persona_id IS NOT NULL AND cs.source_chat_id IS NULL THEN COALESCE(p.name, cs.title)
|
||||
ELSE cs.title
|
||||
END AS title,
|
||||
cs.title AS stored_title,
|
||||
cs.assistant_persona_id,
|
||||
p.name AS assistant_persona_name,
|
||||
cs.user_persona_id,
|
||||
up.name AS user_persona_name,
|
||||
cs.folder_name,
|
||||
cs.is_pinned,
|
||||
cs.archived_at,
|
||||
cs.scenario_prompt,
|
||||
cs.scenario_summary,
|
||||
cs.context_summary,
|
||||
cs.context_summary_message_id,
|
||||
cs.active_leaf_message_id,
|
||||
cs.source_chat_id,
|
||||
cs.branched_from_message_id,
|
||||
cs.preferred_model,
|
||||
cs.temperature,
|
||||
cs.context_length,
|
||||
cs.response_length,
|
||||
cs.system_instruction,
|
||||
cs.created_at,
|
||||
cs.updated_at,
|
||||
cm.id AS matched_message_id,
|
||||
cm.content AS matched_content
|
||||
FROM chat_sessions cs
|
||||
LEFT JOIN personas p ON p.id = cs.assistant_persona_id
|
||||
LEFT JOIN personas up ON up.id = cs.user_persona_id
|
||||
LEFT JOIN chat_messages cm ON cm.id = (
|
||||
SELECT m.id FROM chat_messages m
|
||||
WHERE m.chat_id = cs.id AND LOWER(m.content) LIKE ?
|
||||
ORDER BY m.id DESC LIMIT 1
|
||||
)
|
||||
WHERE cs.username = ?
|
||||
AND (
|
||||
LOWER(CASE
|
||||
WHEN cs.assistant_persona_id IS NOT NULL AND cs.source_chat_id IS NULL THEN COALESCE(p.name, cs.title)
|
||||
ELSE cs.title
|
||||
END) LIKE ?
|
||||
OR cm.id IS NOT NULL
|
||||
)
|
||||
ORDER BY cs.is_pinned DESC, datetime(cs.updated_at) DESC
|
||||
`)
|
||||
};
|
||||
}
|
||||
|
||||
export function createChatRepository(db, config, statements) {
|
||||
const completeChatTurnTransaction = db.transaction((chatId, userMessageId, assistantContent, retryState, modelName, sources, thinking) => {
|
||||
const assistantResult = statements.insertChatMessage.run(
|
||||
chatId,
|
||||
"bot",
|
||||
String(assistantContent || ""),
|
||||
modelName || null,
|
||||
JSON.stringify(retryState.retryVariants),
|
||||
retryState.retryActiveIndex,
|
||||
retryState.retryRetriesUsed,
|
||||
retryState.retryPromptMessageId ?? null,
|
||||
"complete",
|
||||
null,
|
||||
userMessageId
|
||||
);
|
||||
statements.updateChatMessageDelivery.run("complete", null, userMessageId, chatId);
|
||||
if (sources) statements.updateChatMessageSources.run(JSON.stringify(sources), assistantResult.lastInsertRowid, chatId);
|
||||
if (thinking) statements.updateChatMessageThinking.run(thinking, assistantResult.lastInsertRowid, chatId);
|
||||
statements.setChatActiveLeafById.run(assistantResult.lastInsertRowid, chatId);
|
||||
return assistantResult;
|
||||
});
|
||||
|
||||
const insertBranchChatMessageTransaction = db.transaction((chatId, role, content, retryState, modelName, deliveryStatus, errorMessage, parentMessageId) => {
|
||||
const result = statements.insertChatMessage.run(
|
||||
chatId, role, String(content || ""), modelName || null,
|
||||
retryState ? JSON.stringify(retryState.retryVariants) : null,
|
||||
retryState?.retryActiveIndex ?? 0,
|
||||
retryState?.retryRetriesUsed ?? 0,
|
||||
retryState?.retryPromptMessageId ?? null,
|
||||
deliveryStatus,
|
||||
errorMessage,
|
||||
parentMessageId ?? null
|
||||
);
|
||||
statements.setChatActiveLeafById.run(result.lastInsertRowid, chatId);
|
||||
return result;
|
||||
});
|
||||
|
||||
const failChatTurnTransaction = db.transaction((chatId, userMessageId, errorMessage, parentMessageId) => {
|
||||
statements.updateChatMessageDelivery.run("failed", errorMessage, userMessageId, chatId);
|
||||
statements.setChatActiveLeafById.run(parentMessageId ?? null, chatId);
|
||||
});
|
||||
return {
|
||||
listChats: (username) => statements.listChatSessions.all(username),
|
||||
searchChats(username, query) {
|
||||
const like = `%${String(query || "").trim().toLowerCase()}%`;
|
||||
return statements.searchChats.all(like, username, like);
|
||||
},
|
||||
countChats: (username) => statements.countChatSessions.get(username).count,
|
||||
getChat: (chatId, username) => statements.getChatSession.get(chatId, username),
|
||||
getChatByParticipants: (username, assistantPersonaId, userPersonaId) => statements.getChatSessionByParticipants.get(username, assistantPersonaId, userPersonaId, userPersonaId),
|
||||
createChat(username, title, assistantPersonaId = null, userPersonaId = null, scenarioPrompt = null, scenarioSummary = null, organization = {}) {
|
||||
const chatId = statements.insertChatSession.run(
|
||||
username,
|
||||
title,
|
||||
assistantPersonaId,
|
||||
userPersonaId,
|
||||
organization.folderName ?? null,
|
||||
organization.isPinned ? 1 : 0,
|
||||
organization.archivedAt ?? null,
|
||||
scenarioPrompt,
|
||||
scenarioSummary
|
||||
).lastInsertRowid;
|
||||
return statements.getChatSession.get(chatId, username);
|
||||
},
|
||||
updateChatTitle: (chatId, username, title) => statements.updateChatSessionTitle.run(title, chatId, username),
|
||||
updateChatScene: (chatId, username, scenarioPrompt, scenarioSummary) => statements.updateChatSessionScene.run(scenarioPrompt, scenarioSummary, chatId, username),
|
||||
updateChatContextSummary: (chatId, username, summary, messageId) => statements.updateChatContextSummary.run(summary, Number(messageId || 0), chatId, username),
|
||||
getPersonaChatState: (chatId, username) => statements.getPersonaChatState.get(chatId, username) || null,
|
||||
updatePersonaChatState(username, chat, state) {
|
||||
statements.upsertPersonaChatState.run(
|
||||
username,
|
||||
chat.id,
|
||||
chat.assistant_persona_id ?? null,
|
||||
chat.user_persona_id ?? null,
|
||||
state.relationship_notes ?? null,
|
||||
state.tone ?? null,
|
||||
state.current_location ?? null,
|
||||
state.goals ?? null,
|
||||
state.unresolved_threads ?? null,
|
||||
state.boundaries ?? null
|
||||
);
|
||||
return statements.getPersonaChatState.get(chat.id, username);
|
||||
},
|
||||
updateChatOrganization(chatId, username, {folderName = null, isPinned = false, archivedAt = null}) {
|
||||
return statements.updateChatSessionOrganization.run(folderName, isPinned ? 1 : 0, archivedAt, chatId, username);
|
||||
},
|
||||
touchChat: (chatId, username) => statements.touchChatSession.run(chatId, username),
|
||||
deleteChat: (chatId, username) => statements.deleteChatSession.run(chatId, username),
|
||||
listChatMessages: (chatId) => statements.getChatMessages.all(chatId),
|
||||
listActiveBranchMessages: (chatId) => statements.getActiveBranchMessages.all(chatId),
|
||||
listChatBranchThroughMessage: (chatId, messageId) => statements.getChatBranchThroughMessage.all(chatId, messageId),
|
||||
listSiblingChatMessages: (chatId, role, parentMessageId) => statements.getSiblingChatMessages.all(chatId, role, parentMessageId ?? null),
|
||||
getLatestDescendantMessage: (chatId, messageId) => statements.getLatestDescendantMessage.get(chatId, messageId, chatId),
|
||||
getRecentChatMessages: (chatId) => statements.getRecentChatMessages.all(chatId, config.chatHistoryLimit),
|
||||
getRecentChatMessagesUpToId: (chatId, messageId) => statements.getRecentChatMessagesUpToId.all(chatId, messageId, config.chatHistoryLimit),
|
||||
getChatMessage: (chatId, messageId) => statements.getChatMessageById.get(messageId, chatId),
|
||||
getChatMessageByIndex: (chatId, index) => statements.getChatMessageByIndex.get(chatId, index),
|
||||
getLatestChatMessage: (chatId) => statements.getLatestChatMessage.get(chatId),
|
||||
getPreviousUserMessage: (chatId, messageId) => statements.getPreviousUserMessage.get(chatId, messageId),
|
||||
listChatMemories: (chatId, username) => statements.listChatMemories.all(chatId, username),
|
||||
getChatMemory: (chatId, memoryId, username) => statements.getChatMemory.get(memoryId, chatId, username),
|
||||
createChatMemory(username, chat, fact) {
|
||||
const result = statements.insertChatMemory.run(
|
||||
username,
|
||||
chat.id,
|
||||
chat.assistant_persona_id ?? null,
|
||||
chat.user_persona_id ?? null,
|
||||
String(fact || "").trim()
|
||||
);
|
||||
return statements.getChatMemory.get(result.lastInsertRowid, chat.id, username);
|
||||
},
|
||||
updateChatMemory: (chatId, memoryId, username, fact) => statements.updateChatMemory.run(String(fact || "").trim(), memoryId, chatId, username),
|
||||
deleteChatMemory: (chatId, memoryId, username) => statements.deleteChatMemory.run(memoryId, chatId, username),
|
||||
insertChatMessage(chatId, role, content, retryState = null, modelName = null, deliveryStatus = "complete", errorMessage = null, parentMessageId = null) {
|
||||
return statements.insertChatMessage.run(
|
||||
chatId,
|
||||
role,
|
||||
String(content || ""),
|
||||
modelName || null,
|
||||
retryState ? JSON.stringify(retryState.retryVariants) : null,
|
||||
retryState?.retryActiveIndex ?? 0,
|
||||
retryState?.retryRetriesUsed ?? 0,
|
||||
retryState?.retryPromptMessageId ?? null,
|
||||
deliveryStatus,
|
||||
errorMessage,
|
||||
parentMessageId
|
||||
);
|
||||
},
|
||||
insertBranchChatMessage: (chatId, role, content, retryState = null, modelName = null, deliveryStatus = "complete", errorMessage = null, parentMessageId = null) =>
|
||||
insertBranchChatMessageTransaction(chatId, role, content, retryState, modelName, deliveryStatus, errorMessage, parentMessageId),
|
||||
updateChatMessageDelivery: (chatId, messageId, deliveryStatus, errorMessage = null) =>
|
||||
statements.updateChatMessageDelivery.run(deliveryStatus, errorMessage, messageId, chatId),
|
||||
completeChatTurn: (chatId, userMessageId, assistantContent, retryState, modelName = null, sources = null, thinking = null) =>
|
||||
completeChatTurnTransaction(chatId, userMessageId, assistantContent, retryState, modelName, sources, thinking),
|
||||
updateChatMessageThinking: (chatId, messageId, thinking) =>
|
||||
statements.updateChatMessageThinking.run(thinking, messageId, chatId),
|
||||
updateChatMessageSources: (chatId, messageId, sources) =>
|
||||
statements.updateChatMessageSources.run(JSON.stringify(sources), messageId, chatId),
|
||||
failChatTurn: (chatId, userMessageId, errorMessage, parentMessageId = null) =>
|
||||
failChatTurnTransaction(chatId, userMessageId, errorMessage, parentMessageId),
|
||||
updateChatActiveLeaf: (chatId, username, messageId) => statements.updateChatActiveLeaf.run(messageId, chatId, username),
|
||||
updateChatBranchSource: (chatId, username, sourceChatId, messageId) =>
|
||||
statements.updateChatBranchSource.run(sourceChatId, messageId ?? null, chatId, username),
|
||||
updateChatGenerationSettings: (chatId, username, settings) => statements.updateChatGenerationSettings.run(
|
||||
settings.preferredModel ?? null,
|
||||
settings.temperature ?? null,
|
||||
settings.contextLength ?? null,
|
||||
settings.responseLength ?? null,
|
||||
settings.systemInstruction ?? null,
|
||||
chatId,
|
||||
username
|
||||
),
|
||||
updateChatPreferredModel: (chatId, username, model) => statements.updateChatPreferredModel.run(model || null, chatId, username),
|
||||
listMessageVersions: (messageId) => db.prepare("SELECT * FROM chat_message_versions WHERE message_id = ? ORDER BY id").all(messageId),
|
||||
saveMessageVersion: (chatId, messageId, version = null, selectedIndex = null) => db.transaction(() => {
|
||||
const row = statements.getChatMessageById.get(messageId, chatId);
|
||||
if (!row) throw new Error("Message not found");
|
||||
let versions = db.prepare("SELECT * FROM chat_message_versions WHERE message_id = ? ORDER BY id").all(messageId);
|
||||
const insert = db.prepare("INSERT INTO chat_message_versions (message_id, content, thinking, sources, model_name) VALUES (?, ?, ?, ?, ?)");
|
||||
if (!versions.length) {
|
||||
db.prepare("INSERT INTO chat_message_versions (message_id, content, thinking, sources, model_name, created_at) VALUES (?, ?, ?, ?, ?, ?)")
|
||||
.run(messageId, row.content, row.thinking, row.sources, row.model_name, row.created_at);
|
||||
} else {
|
||||
// Edits keep their existing semantics: update the selected answer.
|
||||
db.prepare("UPDATE chat_message_versions SET content = ?, thinking = ?, sources = ?, model_name = ? WHERE id = ?")
|
||||
.run(row.content, row.thinking, row.sources, row.model_name, versions[row.retry_active_index]?.id ?? versions[0].id);
|
||||
}
|
||||
if (version) insert.run(messageId, version.content, version.thinking || null, JSON.stringify(version.sources ?? null), version.modelName);
|
||||
versions = db.prepare("SELECT * FROM chat_message_versions WHERE message_id = ? ORDER BY id").all(messageId);
|
||||
const index = version ? versions.length - 1 : selectedIndex;
|
||||
if (!Number.isInteger(index) || !versions[index]) throw new Error("Version is out of range");
|
||||
const active = versions[index];
|
||||
db.prepare(`UPDATE chat_messages SET content = ?, thinking = ?, sources = ?, model_name = ?,
|
||||
retry_variants = ?, retry_active_index = ?, retry_retries_used = ? WHERE id = ? AND chat_id = ?`)
|
||||
.run(active.content, active.thinking, active.sources, active.model_name, JSON.stringify(versions.map(item => item.content)), index, versions.length - 1, messageId, chatId);
|
||||
})(),
|
||||
updateChatMessage: (chatId, messageId, content) => statements.updateChatMessage.run(content, messageId, chatId),
|
||||
updateChatMessageRetryState: (chatId, messageId, payload) => statements.updateChatMessageWithRetryState.run(
|
||||
payload.content,
|
||||
JSON.stringify(payload.retryVariants),
|
||||
payload.retryActiveIndex,
|
||||
payload.retryRetriesUsed,
|
||||
payload.retryPromptMessageId ?? null,
|
||||
messageId,
|
||||
chatId
|
||||
),
|
||||
deleteChatMessage: (chatId, messageId) => statements.deleteChatMessage.run(messageId, chatId),
|
||||
clearChatMessages: (chatId) => statements.deleteChatMessages.run(chatId)
|
||||
};
|
||||
}
|
||||
@@ -0,0 +1,65 @@
|
||||
import {attachmentMetadata} from "./attachments.js";
|
||||
|
||||
/** Bounded history reads and batched metadata for a single page. */
|
||||
export function createHistoryRepository(db) {
|
||||
const activeLeaf = db.prepare(`SELECT COALESCE(active_leaf_message_id,
|
||||
(SELECT MAX(id) FROM chat_messages WHERE chat_id = chat_sessions.id)) AS id
|
||||
FROM chat_sessions WHERE id = ?`);
|
||||
const ancestors = `WITH RECURSIVE branch(id, parent_message_id) AS (
|
||||
SELECT id, parent_message_id FROM chat_messages WHERE chat_id = ? AND id = ?
|
||||
UNION SELECT p.id, p.parent_message_id FROM chat_messages p JOIN branch c ON p.id = c.parent_message_id WHERE p.chat_id = ?
|
||||
)`;
|
||||
const onBranch = db.prepare(`${ancestors} SELECT * FROM branch WHERE id = ?`);
|
||||
const page = db.prepare(`WITH RECURSIVE branch AS (
|
||||
SELECT *, 0 AS depth FROM chat_messages WHERE chat_id = ? AND id = ?
|
||||
UNION ALL SELECT p.*, c.depth + 1 FROM chat_messages p JOIN branch c ON p.id = c.parent_message_id
|
||||
WHERE p.chat_id = ? AND c.depth < ?
|
||||
) SELECT * FROM branch ORDER BY depth DESC`);
|
||||
const versions = db.prepare("SELECT * FROM chat_message_versions WHERE message_id IN (SELECT value FROM json_each(?)) ORDER BY id");
|
||||
const attachments = db.prepare(`SELECT a.*, m.message_id FROM attachments a JOIN message_attachments m ON a.id = m.attachment_id
|
||||
WHERE m.message_id IN (SELECT value FROM json_each(?)) ORDER BY m.position`);
|
||||
const siblings = db.prepare(`WITH families AS (SELECT DISTINCT role, parent_message_id FROM chat_messages
|
||||
WHERE id IN (SELECT value FROM json_each(?)) AND chat_id = ?)
|
||||
SELECT cm.id, ROW_NUMBER() OVER family - 1 AS siblingIndex, COUNT(*) OVER family AS siblingCount,
|
||||
LAG(cm.id) OVER family AS previousSiblingId, LEAD(cm.id) OVER family AS nextSiblingId
|
||||
FROM chat_messages cm JOIN families f ON cm.role = f.role AND cm.parent_message_id IS f.parent_message_id
|
||||
WHERE cm.chat_id = ? WINDOW family AS (PARTITION BY cm.role, cm.parent_message_id ORDER BY cm.id
|
||||
ROWS BETWEEN UNBOUNDED PRECEDING AND UNBOUNDED FOLLOWING)`);
|
||||
const group = (rows, format = row => row) => {
|
||||
const result = new Map();
|
||||
for (const row of rows) result.set(row.message_id, [...(result.get(row.message_id) || []), format(row)]);
|
||||
return result;
|
||||
};
|
||||
return {
|
||||
getActiveBranchPage(chatId, {limit = 50, before = null, leaf = null, around = null} = {}) {
|
||||
const leafId = activeLeaf.get(chatId)?.id ?? null;
|
||||
if (leaf != null && leaf !== leafId) return {error: "The conversation branch changed. Reload the latest messages.", status: 409};
|
||||
let startId = leafId;
|
||||
let matchOnBranch = null;
|
||||
if (around != null) {
|
||||
matchOnBranch = Boolean(onBranch.get(chatId, leafId, chatId, around));
|
||||
if (matchOnBranch) startId = around;
|
||||
}
|
||||
if (before != null) {
|
||||
const cursor = onBranch.get(chatId, leafId, chatId, before);
|
||||
if (!cursor) return {error: "History cursor is not on this branch.", status: 400};
|
||||
startId = cursor.parent_message_id;
|
||||
}
|
||||
const rows = startId == null ? [] : page.all(chatId, startId, chatId, limit);
|
||||
const hasMore = rows.length > limit;
|
||||
if (hasMore) rows.shift();
|
||||
return {rows, matchOnBranch, pagination: {leafId, hasMore, before: hasMore ? rows[0].id : null,
|
||||
hasNewer: around != null && matchOnBranch && around !== leafId}};
|
||||
},
|
||||
getHistoryMetadata(chatId, ids) {
|
||||
const wanted = new Set(ids);
|
||||
const encoded = JSON.stringify(ids);
|
||||
return {versions: group(versions.all(encoded)), attachments: group(attachments.all(encoded), attachmentMetadata),
|
||||
siblings: new Map(siblings.all(encoded, chatId, chatId).filter(row => wanted.has(row.id)).map(row => [row.id, row]))};
|
||||
},
|
||||
listBranchNodes(chatId, {after = 0, limit = 200} = {}) {
|
||||
return db.prepare(`SELECT id, role, parent_message_id AS parentMessageId, substr(content, 1, 180) AS preview,
|
||||
retry_retries_used + 1 AS versionCount FROM chat_messages WHERE chat_id = ? AND id > ? ORDER BY id LIMIT ?`).all(chatId, after, limit + 1);
|
||||
}
|
||||
};
|
||||
}
|
||||
@@ -0,0 +1,195 @@
|
||||
import {buildPersonaSnapshot} from "./personaSnapshot.js";
|
||||
|
||||
export function createPersonaStatements(db) {
|
||||
return {
|
||||
clearAssistantPersonaFromChats: db.prepare(`
|
||||
UPDATE chat_sessions
|
||||
SET assistant_persona_id = NULL,
|
||||
updated_at = CURRENT_TIMESTAMP
|
||||
WHERE username = ?
|
||||
AND assistant_persona_id = ?
|
||||
`),
|
||||
clearUserPersonaFromChats: db.prepare(`
|
||||
UPDATE chat_sessions
|
||||
SET user_persona_id = NULL,
|
||||
updated_at = CURRENT_TIMESTAMP
|
||||
WHERE username = ?
|
||||
AND user_persona_id = ?
|
||||
`),
|
||||
listPersonasByType: db.prepare(`
|
||||
SELECT id, name, pronouns, appearance, background, details, example_dialogues, persona_type,
|
||||
source_market_id, source_creator_username, created_at, updated_at
|
||||
FROM personas
|
||||
WHERE username = ?
|
||||
AND persona_type = ?
|
||||
ORDER BY datetime(updated_at) DESC
|
||||
`),
|
||||
listAllPersonas: db.prepare(`
|
||||
SELECT id, username, name, pronouns, appearance, background, details, example_dialogues, persona_type,
|
||||
source_market_id, source_creator_username, created_at, updated_at
|
||||
FROM personas
|
||||
WHERE username = ?
|
||||
ORDER BY datetime(updated_at) DESC
|
||||
`),
|
||||
getPersona: db.prepare(`
|
||||
SELECT id, name, pronouns, appearance, background, details, example_dialogues, persona_type,
|
||||
source_market_id, source_creator_username, created_at, updated_at
|
||||
FROM personas
|
||||
WHERE id = ?
|
||||
AND username = ?
|
||||
`),
|
||||
insertPersona: db.prepare("INSERT INTO personas (username, name, pronouns, appearance, background, details, example_dialogues, persona_type) VALUES (?, ?, ?, ?, ?, ?, ?, ?)"),
|
||||
updatePersona: db.prepare(`
|
||||
UPDATE personas
|
||||
SET name = ?,
|
||||
pronouns = ?,
|
||||
appearance = ?,
|
||||
background = ?,
|
||||
details = ?,
|
||||
example_dialogues = ?,
|
||||
updated_at = CURRENT_TIMESTAMP
|
||||
WHERE id = ?
|
||||
AND username = ?
|
||||
`),
|
||||
deletePersona: db.prepare("DELETE FROM personas WHERE id = ? AND username = ?"),
|
||||
getNextPersonaVersionNumber: db.prepare("SELECT COALESCE(MAX(version_number), 0) + 1 AS nextVersion FROM persona_versions WHERE persona_id = ?"),
|
||||
insertPersonaVersion: db.prepare("INSERT INTO persona_versions (persona_id, username, version_number, snapshot_json) VALUES (?, ?, ?, ?)"),
|
||||
listPersonaVersions: db.prepare(`
|
||||
SELECT id, persona_id, username, version_number, snapshot_json, created_at
|
||||
FROM persona_versions
|
||||
WHERE persona_id = ?
|
||||
AND username = ?
|
||||
ORDER BY version_number DESC
|
||||
`),
|
||||
getPersonaVersion: db.prepare(`
|
||||
SELECT id, persona_id, username, version_number, snapshot_json, created_at
|
||||
FROM persona_versions
|
||||
WHERE id = ?
|
||||
AND persona_id = ?
|
||||
AND username = ?
|
||||
`),
|
||||
getActivePersonaId: db.prepare("SELECT active_persona_id, active_user_persona_id FROM user_settings WHERE username = ?"),
|
||||
setActiveUserPersona: db.prepare(`
|
||||
INSERT INTO user_settings (username, active_user_persona_id)
|
||||
VALUES (?, ?)
|
||||
ON CONFLICT(username) DO UPDATE SET active_user_persona_id = excluded.active_user_persona_id
|
||||
`),
|
||||
getActiveUserPersona: db.prepare(`
|
||||
SELECT p.id, p.name, p.pronouns, p.appearance, p.background, p.details, p.example_dialogues
|
||||
FROM user_settings us
|
||||
JOIN personas p ON p.id = us.active_user_persona_id
|
||||
WHERE us.username = ?
|
||||
`),
|
||||
getAssistantPersonaForChat: db.prepare(`
|
||||
SELECT p.id, p.name, p.pronouns, p.appearance, p.background, p.details, p.example_dialogues
|
||||
FROM chat_sessions cs
|
||||
JOIN personas p ON p.id = cs.assistant_persona_id
|
||||
WHERE cs.id = ?
|
||||
AND cs.username = ?
|
||||
`),
|
||||
getUserPersonaForChat: db.prepare(`
|
||||
SELECT p.id, p.name, p.pronouns, p.appearance, p.background, p.details, p.example_dialogues
|
||||
FROM chat_sessions cs
|
||||
JOIN personas p ON p.id = cs.user_persona_id
|
||||
WHERE cs.id = ?
|
||||
AND cs.username = ?
|
||||
`),
|
||||
deleteMarketPersonaByPersonaId: db.prepare("DELETE FROM persona_market WHERE persona_id = ? AND creator_username = ?")
|
||||
};
|
||||
}
|
||||
|
||||
export function createPersonaRepository(db, statements) {
|
||||
const savePersonaVersion = db.transaction((personaId, username) => {
|
||||
const persona = statements.getPersona.get(personaId, username);
|
||||
if (!persona) return null;
|
||||
const versionNumber = statements.getNextPersonaVersionNumber.get(personaId).nextVersion;
|
||||
statements.insertPersonaVersion.run(personaId, username, versionNumber, buildPersonaSnapshot(persona));
|
||||
return versionNumber;
|
||||
});
|
||||
|
||||
const clonePersonaTransaction = db.transaction((personaId, username) => {
|
||||
const source = statements.getPersona.get(personaId, username);
|
||||
if (!source) return null;
|
||||
const cloneId = statements.insertPersona.run(
|
||||
username,
|
||||
`${source.name} Copy`,
|
||||
source.pronouns,
|
||||
source.appearance,
|
||||
source.background,
|
||||
source.details,
|
||||
source.example_dialogues,
|
||||
source.persona_type
|
||||
).lastInsertRowid;
|
||||
const clone = statements.getPersona.get(cloneId, username);
|
||||
statements.insertPersonaVersion.run(cloneId, username, 1, buildPersonaSnapshot(clone));
|
||||
return clone;
|
||||
});
|
||||
return {
|
||||
listPersonasByType: (username, personaType) => statements.listPersonasByType.all(username, personaType),
|
||||
listAllPersonas: (username) => statements.listAllPersonas.all(username),
|
||||
getPersona: (personaId, username) => statements.getPersona.get(personaId, username),
|
||||
createPersona(username, payload, personaType) {
|
||||
const personaId = statements.insertPersona.run(
|
||||
username,
|
||||
payload.name,
|
||||
payload.pronouns,
|
||||
payload.appearance,
|
||||
payload.background,
|
||||
payload.details,
|
||||
payload.exampleDialogues,
|
||||
personaType
|
||||
).lastInsertRowid;
|
||||
const persona = statements.getPersona.get(personaId, username);
|
||||
statements.insertPersonaVersion.run(personaId, username, 1, buildPersonaSnapshot(persona));
|
||||
return persona;
|
||||
},
|
||||
updatePersona(personaId, username, payload) {
|
||||
savePersonaVersion(personaId, username);
|
||||
return statements.updatePersona.run(
|
||||
payload.name,
|
||||
payload.pronouns,
|
||||
payload.appearance,
|
||||
payload.background,
|
||||
payload.details,
|
||||
payload.exampleDialogues,
|
||||
personaId,
|
||||
username
|
||||
);
|
||||
},
|
||||
restorePersonaVersion(personaId, versionId, username) {
|
||||
const version = statements.getPersonaVersion.get(versionId, personaId, username);
|
||||
if (!version) return null;
|
||||
savePersonaVersion(personaId, username);
|
||||
const snapshot = JSON.parse(version.snapshot_json);
|
||||
statements.updatePersona.run(
|
||||
snapshot.name,
|
||||
snapshot.pronouns ?? null,
|
||||
snapshot.appearance ?? null,
|
||||
snapshot.background ?? null,
|
||||
snapshot.details ?? null,
|
||||
snapshot.example_dialogues ?? null,
|
||||
personaId,
|
||||
username
|
||||
);
|
||||
return statements.getPersona.get(personaId, username);
|
||||
},
|
||||
listPersonaVersions(personaId, username) {
|
||||
return statements.listPersonaVersions.all(personaId, username).map((version) => ({
|
||||
...version,
|
||||
snapshot: JSON.parse(version.snapshot_json)
|
||||
}));
|
||||
},
|
||||
clonePersona: (personaId, username) => clonePersonaTransaction(personaId, username),
|
||||
deletePersona: (personaId, username) => statements.deletePersona.run(personaId, username),
|
||||
clearPersonaReferences(username, personaId) {
|
||||
statements.clearAssistantPersonaFromChats.run(username, personaId);
|
||||
statements.clearUserPersonaFromChats.run(username, personaId);
|
||||
statements.deleteMarketPersonaByPersonaId.run(personaId, username);
|
||||
},
|
||||
getActivePersonaIds: (username) => statements.getActivePersonaId.get(username) || {},
|
||||
setActiveUserPersona: (username, personaId) => statements.setActiveUserPersona.run(username, personaId),
|
||||
getActiveUserPersona: (username) => statements.getActiveUserPersona.get(username),
|
||||
getAssistantPersonaForChat: (chatId, username) => statements.getAssistantPersonaForChat.get(chatId, username),
|
||||
getUserPersonaForChat: (chatId, username) => statements.getUserPersonaForChat.get(chatId, username)
|
||||
};
|
||||
}
|
||||
@@ -0,0 +1,11 @@
|
||||
export const buildPersonaSnapshot = (persona) => JSON.stringify({
|
||||
name: persona.name,
|
||||
pronouns: persona.pronouns,
|
||||
appearance: persona.appearance,
|
||||
background: persona.background,
|
||||
details: persona.details,
|
||||
example_dialogues: persona.example_dialogues,
|
||||
persona_type: persona.persona_type,
|
||||
source_market_id: persona.source_market_id ?? null,
|
||||
source_creator_username: persona.source_creator_username ?? null
|
||||
});
|
||||
+12
-820
@@ -1,3 +1,8 @@
|
||||
import {buildPersonaSnapshot} from "./personaSnapshot.js";
|
||||
import {createHistoryRepository} from "./historyRepository.js";
|
||||
import {createAccountStatements, createAccountRepository} from "./accountRepository.js";
|
||||
import {createChatStatements, createChatRepository} from "./chatRepository.js";
|
||||
import {createPersonaStatements, createPersonaRepository} from "./personaRepository.js";
|
||||
/**
|
||||
* @file Bündelt Datenbankzugriffe auf Konten, Chats, Nachrichten, Personas und den Marktplatz.
|
||||
*/
|
||||
@@ -13,18 +18,6 @@ const parseJsonArray = (value) => {
|
||||
}
|
||||
};
|
||||
|
||||
const buildPersonaSnapshot = (persona) => JSON.stringify({
|
||||
name: persona.name,
|
||||
pronouns: persona.pronouns,
|
||||
appearance: persona.appearance,
|
||||
background: persona.background,
|
||||
details: persona.details,
|
||||
example_dialogues: persona.example_dialogues,
|
||||
persona_type: persona.persona_type,
|
||||
source_market_id: persona.source_market_id ?? null,
|
||||
source_creator_username: persona.source_creator_username ?? null
|
||||
});
|
||||
|
||||
const MARKET_SORT_ORDERS = {
|
||||
best: "pm.favorite_count DESC, rating_average DESC, pm.usage_count DESC, datetime(pm.created_at) DESC, pm.id DESC",
|
||||
newest: "datetime(pm.created_at) DESC, pm.id DESC",
|
||||
@@ -45,413 +38,10 @@ function normalizeMarketPersonaType(personaType) {
|
||||
|
||||
export function createRepositories(db, config) {
|
||||
const statements = {
|
||||
insertUser: db.prepare("INSERT INTO users (username, password) VALUES (?, ?)"),
|
||||
getUser: db.prepare("SELECT username, password, role, session_version FROM users WHERE username = ?"),
|
||||
updateUsername: db.prepare("UPDATE users SET username = ? WHERE username = ?"),
|
||||
updatePassword: db.prepare("UPDATE users SET password = ? WHERE username = ?"),
|
||||
incrementSessionVersion: db.prepare("UPDATE users SET session_version = COALESCE(session_version, 0) + 1 WHERE username = ?"),
|
||||
insertChatSession: db.prepare(
|
||||
"INSERT INTO chat_sessions (username, title, assistant_persona_id, user_persona_id, folder_name, is_pinned, archived_at, scenario_prompt, scenario_summary) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)"
|
||||
),
|
||||
listChatSessions: db.prepare(`
|
||||
SELECT cs.id,
|
||||
CASE
|
||||
WHEN cs.assistant_persona_id IS NOT NULL AND cs.source_chat_id IS NULL THEN COALESCE(p.name, cs.title)
|
||||
ELSE cs.title
|
||||
END AS title,
|
||||
cs.title AS stored_title,
|
||||
cs.assistant_persona_id,
|
||||
p.name AS assistant_persona_name,
|
||||
cs.user_persona_id,
|
||||
up.name AS user_persona_name,
|
||||
cs.folder_name,
|
||||
cs.is_pinned,
|
||||
cs.archived_at,
|
||||
cs.scenario_prompt,
|
||||
cs.scenario_summary,
|
||||
cs.context_summary,
|
||||
cs.context_summary_message_id,
|
||||
cs.active_leaf_message_id,
|
||||
cs.source_chat_id,
|
||||
cs.branched_from_message_id,
|
||||
cs.preferred_model,
|
||||
cs.temperature,
|
||||
cs.context_length,
|
||||
cs.response_length,
|
||||
cs.system_instruction,
|
||||
cs.created_at,
|
||||
cs.updated_at
|
||||
FROM chat_sessions cs
|
||||
LEFT JOIN personas p ON p.id = cs.assistant_persona_id
|
||||
LEFT JOIN personas up ON up.id = cs.user_persona_id
|
||||
WHERE cs.username = ?
|
||||
ORDER BY cs.is_pinned DESC, datetime(cs.updated_at) DESC
|
||||
`),
|
||||
countChatSessions: db.prepare("SELECT COUNT(*) as count FROM chat_sessions WHERE username = ? AND archived_at IS NULL"),
|
||||
getChatSession: db.prepare(`
|
||||
SELECT cs.id,
|
||||
CASE
|
||||
WHEN cs.assistant_persona_id IS NOT NULL AND cs.source_chat_id IS NULL THEN COALESCE(p.name, cs.title)
|
||||
ELSE cs.title
|
||||
END AS title,
|
||||
cs.title AS stored_title,
|
||||
cs.assistant_persona_id,
|
||||
p.name AS assistant_persona_name,
|
||||
cs.user_persona_id,
|
||||
up.name AS user_persona_name,
|
||||
cs.folder_name,
|
||||
cs.is_pinned,
|
||||
cs.archived_at,
|
||||
cs.scenario_prompt,
|
||||
cs.scenario_summary,
|
||||
cs.context_summary,
|
||||
cs.context_summary_message_id,
|
||||
cs.active_leaf_message_id,
|
||||
cs.source_chat_id,
|
||||
cs.branched_from_message_id,
|
||||
cs.preferred_model,
|
||||
cs.temperature,
|
||||
cs.context_length,
|
||||
cs.response_length,
|
||||
cs.system_instruction,
|
||||
cs.created_at,
|
||||
cs.updated_at
|
||||
FROM chat_sessions cs
|
||||
LEFT JOIN personas p ON p.id = cs.assistant_persona_id
|
||||
LEFT JOIN personas up ON up.id = cs.user_persona_id
|
||||
WHERE cs.id = ?
|
||||
AND cs.username = ?
|
||||
`),
|
||||
getChatSessionByParticipants: db.prepare(`
|
||||
SELECT cs.id,
|
||||
CASE
|
||||
WHEN cs.assistant_persona_id IS NOT NULL AND cs.source_chat_id IS NULL THEN COALESCE(p.name, cs.title)
|
||||
ELSE cs.title
|
||||
END AS title,
|
||||
cs.title AS stored_title,
|
||||
cs.assistant_persona_id,
|
||||
p.name AS assistant_persona_name,
|
||||
cs.user_persona_id,
|
||||
up.name AS user_persona_name,
|
||||
cs.folder_name,
|
||||
cs.is_pinned,
|
||||
cs.archived_at,
|
||||
cs.scenario_prompt,
|
||||
cs.scenario_summary,
|
||||
cs.context_summary,
|
||||
cs.context_summary_message_id,
|
||||
cs.active_leaf_message_id,
|
||||
cs.source_chat_id,
|
||||
cs.branched_from_message_id,
|
||||
cs.preferred_model,
|
||||
cs.temperature,
|
||||
cs.context_length,
|
||||
cs.response_length,
|
||||
cs.system_instruction,
|
||||
cs.created_at,
|
||||
cs.updated_at
|
||||
FROM chat_sessions cs
|
||||
LEFT JOIN personas p ON p.id = cs.assistant_persona_id
|
||||
LEFT JOIN personas up ON up.id = cs.user_persona_id
|
||||
WHERE cs.username = ?
|
||||
AND cs.assistant_persona_id = ?
|
||||
AND ((cs.user_persona_id IS NULL AND ? IS NULL) OR cs.user_persona_id = ?)
|
||||
AND cs.source_chat_id IS NULL
|
||||
ORDER BY cs.is_pinned DESC, datetime(cs.updated_at) DESC
|
||||
LIMIT 1
|
||||
`),
|
||||
updateChatSessionTitle: db.prepare("UPDATE chat_sessions SET title = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ? AND username = ?"),
|
||||
updateChatSessionScene: db.prepare("UPDATE chat_sessions SET scenario_prompt = ?, scenario_summary = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ? AND username = ?"),
|
||||
updateChatContextSummary: db.prepare("UPDATE chat_sessions SET context_summary = ?, context_summary_message_id = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ? AND username = ?"),
|
||||
updateChatGenerationSettings: db.prepare(`
|
||||
UPDATE chat_sessions
|
||||
SET preferred_model = ?, temperature = ?, context_length = ?, response_length = ?, system_instruction = ?, updated_at = CURRENT_TIMESTAMP
|
||||
WHERE id = ? AND username = ?
|
||||
`),
|
||||
updateChatPreferredModel: db.prepare("UPDATE chat_sessions SET preferred_model = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ? AND username = ?"),
|
||||
updateChatActiveLeaf: db.prepare("UPDATE chat_sessions SET active_leaf_message_id = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ? AND username = ?"),
|
||||
updateChatBranchSource: db.prepare("UPDATE chat_sessions SET source_chat_id = ?, branched_from_message_id = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ? AND username = ?"),
|
||||
setChatActiveLeafById: db.prepare("UPDATE chat_sessions SET active_leaf_message_id = ?, updated_at = CURRENT_TIMESTAMP WHERE id = ?"),
|
||||
updateChatSessionOrganization: db.prepare(`
|
||||
UPDATE chat_sessions
|
||||
SET folder_name = ?,
|
||||
is_pinned = ?,
|
||||
archived_at = ?,
|
||||
updated_at = CURRENT_TIMESTAMP
|
||||
WHERE id = ?
|
||||
AND username = ?
|
||||
`),
|
||||
touchChatSession: db.prepare("UPDATE chat_sessions SET updated_at = CURRENT_TIMESTAMP WHERE id = ? AND username = ?"),
|
||||
deleteChatSession: db.prepare("DELETE FROM chat_sessions WHERE id = ? AND username = ?"),
|
||||
clearAssistantPersonaFromChats: db.prepare(`
|
||||
UPDATE chat_sessions
|
||||
SET assistant_persona_id = NULL,
|
||||
updated_at = CURRENT_TIMESTAMP
|
||||
WHERE username = ?
|
||||
AND assistant_persona_id = ?
|
||||
`),
|
||||
clearUserPersonaFromChats: db.prepare(`
|
||||
UPDATE chat_sessions
|
||||
SET user_persona_id = NULL,
|
||||
updated_at = CURRENT_TIMESTAMP
|
||||
WHERE username = ?
|
||||
AND user_persona_id = ?
|
||||
`),
|
||||
insertChatMessage: db.prepare("INSERT INTO chat_messages (chat_id, role, content, model_name, retry_variants, retry_active_index, retry_retries_used, retry_prompt_message_id, delivery_status, error_message, parent_message_id) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)"),
|
||||
updateChatMessageDelivery: db.prepare("UPDATE chat_messages SET delivery_status = ?, error_message = ? WHERE id = ? AND chat_id = ?"),
|
||||
updateChatMessageSources: db.prepare("UPDATE chat_messages SET sources = ? WHERE id = ? AND chat_id = ?"),
|
||||
updateChatMessageThinking: db.prepare("UPDATE chat_messages SET thinking = ? WHERE id = ? AND chat_id = ?"),
|
||||
...createAccountStatements(db),
|
||||
...createChatStatements(db),
|
||||
...createPersonaStatements(db),
|
||||
updateChatMessageLinks: db.prepare("UPDATE chat_messages SET parent_message_id = ?, retry_prompt_message_id = ? WHERE id = ? AND chat_id = ?"),
|
||||
getChatMessages: db.prepare(`
|
||||
SELECT thinking, sources, id, role, content, model_name, retry_variants, retry_active_index, retry_retries_used, retry_prompt_message_id, delivery_status, error_message, parent_message_id, created_at
|
||||
FROM chat_messages
|
||||
WHERE chat_id = ?
|
||||
ORDER BY id
|
||||
`),
|
||||
getRecentChatMessages: db.prepare(`
|
||||
WITH RECURSIVE branch AS (
|
||||
SELECT cm.* FROM chat_messages cm
|
||||
JOIN chat_sessions cs ON cs.id = cm.chat_id
|
||||
WHERE cm.chat_id = ?
|
||||
AND cm.id = COALESCE(cs.active_leaf_message_id, (SELECT MAX(id) FROM chat_messages WHERE chat_id = cm.chat_id))
|
||||
UNION ALL
|
||||
SELECT parent.* FROM chat_messages parent
|
||||
JOIN branch child ON child.parent_message_id = parent.id
|
||||
)
|
||||
SELECT id, role, content FROM branch
|
||||
WHERE delivery_status != 'failed'
|
||||
ORDER BY id DESC
|
||||
LIMIT ?
|
||||
`),
|
||||
getRecentChatMessagesUpToId: db.prepare(`
|
||||
WITH RECURSIVE branch AS (
|
||||
SELECT * FROM chat_messages WHERE chat_id = ? AND id = ?
|
||||
UNION ALL
|
||||
SELECT parent.* FROM chat_messages parent
|
||||
JOIN branch child ON child.parent_message_id = parent.id
|
||||
)
|
||||
SELECT id, role, content FROM branch
|
||||
WHERE delivery_status != 'failed'
|
||||
ORDER BY id DESC
|
||||
LIMIT ?
|
||||
`),
|
||||
getChatBranchThroughMessage: db.prepare(`
|
||||
WITH RECURSIVE branch AS (
|
||||
SELECT * FROM chat_messages WHERE chat_id = ? AND id = ?
|
||||
UNION ALL
|
||||
SELECT parent.* FROM chat_messages parent
|
||||
JOIN branch child ON child.parent_message_id = parent.id
|
||||
)
|
||||
SELECT id, role, content, model_name, retry_variants, retry_active_index, retry_retries_used,
|
||||
retry_prompt_message_id, delivery_status, error_message, parent_message_id, created_at, sources, thinking
|
||||
FROM branch
|
||||
WHERE delivery_status = 'complete'
|
||||
ORDER BY id
|
||||
`),
|
||||
getActiveBranchMessages: db.prepare(`
|
||||
WITH RECURSIVE branch AS (
|
||||
SELECT cm.* FROM chat_messages cm
|
||||
JOIN chat_sessions cs ON cs.id = cm.chat_id
|
||||
WHERE cm.chat_id = ?
|
||||
AND cm.id = COALESCE(cs.active_leaf_message_id, (SELECT MAX(id) FROM chat_messages WHERE chat_id = cm.chat_id))
|
||||
UNION ALL
|
||||
SELECT parent.* FROM chat_messages parent
|
||||
JOIN branch child ON child.parent_message_id = parent.id
|
||||
)
|
||||
SELECT id, role, content, model_name, retry_variants, retry_active_index, retry_retries_used,
|
||||
retry_prompt_message_id, delivery_status, error_message, parent_message_id, created_at, sources, thinking
|
||||
FROM branch ORDER BY id
|
||||
`),
|
||||
getSiblingChatMessages: db.prepare(`
|
||||
SELECT id, role, content, parent_message_id
|
||||
FROM chat_messages
|
||||
WHERE chat_id = ? AND role = ? AND parent_message_id IS ?
|
||||
ORDER BY id
|
||||
`),
|
||||
getLatestDescendantMessage: db.prepare(`
|
||||
WITH RECURSIVE descendants AS (
|
||||
SELECT id FROM chat_messages WHERE chat_id = ? AND id = ?
|
||||
UNION ALL
|
||||
SELECT child.id FROM chat_messages child JOIN descendants parent ON child.parent_message_id = parent.id
|
||||
WHERE child.chat_id = ? AND child.delivery_status != 'failed'
|
||||
)
|
||||
SELECT cm.* FROM chat_messages cm JOIN descendants d ON d.id = cm.id ORDER BY cm.id DESC LIMIT 1
|
||||
`),
|
||||
getChatMessageById: db.prepare(`
|
||||
SELECT thinking, sources, id, role, content, model_name, retry_variants, retry_active_index, retry_retries_used, retry_prompt_message_id, delivery_status, error_message, parent_message_id, created_at
|
||||
FROM chat_messages
|
||||
WHERE id = ?
|
||||
AND chat_id = ?
|
||||
`),
|
||||
getChatMessageByIndex: db.prepare(`
|
||||
SELECT thinking, sources, id, role, content, model_name, retry_variants, retry_active_index, retry_retries_used, retry_prompt_message_id, delivery_status, error_message, parent_message_id, created_at
|
||||
FROM chat_messages
|
||||
WHERE chat_id = ?
|
||||
ORDER BY id
|
||||
LIMIT 1 OFFSET ?
|
||||
`),
|
||||
getLatestChatMessage: db.prepare(`
|
||||
SELECT id, role, content, model_name, retry_variants, retry_active_index, retry_retries_used, retry_prompt_message_id, delivery_status, error_message, parent_message_id, created_at
|
||||
FROM chat_messages
|
||||
WHERE chat_id = ?
|
||||
ORDER BY id DESC
|
||||
LIMIT 1
|
||||
`),
|
||||
getPreviousUserMessage: db.prepare(`
|
||||
SELECT id, role, content, model_name, retry_variants, retry_active_index, retry_retries_used, retry_prompt_message_id, delivery_status, error_message, parent_message_id, created_at
|
||||
FROM chat_messages
|
||||
WHERE chat_id = ?
|
||||
AND role = 'user'
|
||||
AND id < ?
|
||||
ORDER BY id DESC
|
||||
LIMIT 1
|
||||
`),
|
||||
listChatMemories: db.prepare(`
|
||||
SELECT id, username, chat_id, assistant_persona_id, user_persona_id, fact, created_at, updated_at
|
||||
FROM chat_memories
|
||||
WHERE chat_id = ?
|
||||
AND username = ?
|
||||
ORDER BY datetime(updated_at) DESC, id DESC
|
||||
`),
|
||||
getChatMemory: db.prepare(`
|
||||
SELECT id, username, chat_id, assistant_persona_id, user_persona_id, fact, created_at, updated_at
|
||||
FROM chat_memories
|
||||
WHERE id = ?
|
||||
AND chat_id = ?
|
||||
AND username = ?
|
||||
`),
|
||||
insertChatMemory: db.prepare(`
|
||||
INSERT INTO chat_memories (username, chat_id, assistant_persona_id, user_persona_id, fact)
|
||||
VALUES (?, ?, ?, ?, ?)
|
||||
`),
|
||||
updateChatMemory: db.prepare(`
|
||||
UPDATE chat_memories
|
||||
SET fact = ?,
|
||||
updated_at = CURRENT_TIMESTAMP
|
||||
WHERE id = ?
|
||||
AND chat_id = ?
|
||||
AND username = ?
|
||||
`),
|
||||
deleteChatMemory: db.prepare("DELETE FROM chat_memories WHERE id = ? AND chat_id = ? AND username = ?"),
|
||||
getPersonaChatState: db.prepare(`
|
||||
SELECT id,
|
||||
username,
|
||||
chat_id,
|
||||
assistant_persona_id,
|
||||
user_persona_id,
|
||||
relationship_notes,
|
||||
tone,
|
||||
current_location,
|
||||
goals,
|
||||
unresolved_threads,
|
||||
boundaries,
|
||||
created_at,
|
||||
updated_at
|
||||
FROM persona_chat_state
|
||||
WHERE chat_id = ?
|
||||
AND username = ?
|
||||
`),
|
||||
upsertPersonaChatState: db.prepare(`
|
||||
INSERT INTO persona_chat_state (
|
||||
username,
|
||||
chat_id,
|
||||
assistant_persona_id,
|
||||
user_persona_id,
|
||||
relationship_notes,
|
||||
tone,
|
||||
current_location,
|
||||
goals,
|
||||
unresolved_threads,
|
||||
boundaries
|
||||
)
|
||||
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||
ON CONFLICT(chat_id) DO UPDATE SET
|
||||
username = excluded.username,
|
||||
assistant_persona_id = excluded.assistant_persona_id,
|
||||
user_persona_id = excluded.user_persona_id,
|
||||
relationship_notes = excluded.relationship_notes,
|
||||
tone = excluded.tone,
|
||||
current_location = excluded.current_location,
|
||||
goals = excluded.goals,
|
||||
unresolved_threads = excluded.unresolved_threads,
|
||||
boundaries = excluded.boundaries,
|
||||
updated_at = CURRENT_TIMESTAMP
|
||||
`),
|
||||
updateChatMessage: db.prepare("UPDATE chat_messages SET content = ? WHERE id = ? AND chat_id = ?"),
|
||||
updateChatMessageWithRetryState: db.prepare("UPDATE chat_messages SET content = ?, retry_variants = ?, retry_active_index = ?, retry_retries_used = ?, retry_prompt_message_id = ? WHERE id = ? AND chat_id = ?"),
|
||||
deleteChatMessage: db.prepare("DELETE FROM chat_messages WHERE id = ? AND chat_id = ?"),
|
||||
deleteChatMessages: db.prepare("DELETE FROM chat_messages WHERE chat_id = ?"),
|
||||
searchChats: db.prepare(`
|
||||
SELECT cs.id,
|
||||
CASE
|
||||
WHEN cs.assistant_persona_id IS NOT NULL AND cs.source_chat_id IS NULL THEN COALESCE(p.name, cs.title)
|
||||
ELSE cs.title
|
||||
END AS title,
|
||||
cs.title AS stored_title,
|
||||
cs.assistant_persona_id,
|
||||
p.name AS assistant_persona_name,
|
||||
cs.user_persona_id,
|
||||
up.name AS user_persona_name,
|
||||
cs.folder_name,
|
||||
cs.is_pinned,
|
||||
cs.archived_at,
|
||||
cs.scenario_prompt,
|
||||
cs.scenario_summary,
|
||||
cs.context_summary,
|
||||
cs.context_summary_message_id,
|
||||
cs.active_leaf_message_id,
|
||||
cs.source_chat_id,
|
||||
cs.branched_from_message_id,
|
||||
cs.preferred_model,
|
||||
cs.temperature,
|
||||
cs.context_length,
|
||||
cs.response_length,
|
||||
cs.system_instruction,
|
||||
cs.created_at,
|
||||
cs.updated_at,
|
||||
cm.id AS matched_message_id,
|
||||
cm.content AS matched_content
|
||||
FROM chat_sessions cs
|
||||
LEFT JOIN personas p ON p.id = cs.assistant_persona_id
|
||||
LEFT JOIN personas up ON up.id = cs.user_persona_id
|
||||
LEFT JOIN chat_messages cm ON cm.id = (
|
||||
SELECT m.id FROM chat_messages m
|
||||
WHERE m.chat_id = cs.id AND LOWER(m.content) LIKE ?
|
||||
ORDER BY m.id DESC LIMIT 1
|
||||
)
|
||||
WHERE cs.username = ?
|
||||
AND (
|
||||
LOWER(CASE
|
||||
WHEN cs.assistant_persona_id IS NOT NULL AND cs.source_chat_id IS NULL THEN COALESCE(p.name, cs.title)
|
||||
ELSE cs.title
|
||||
END) LIKE ?
|
||||
OR cm.id IS NOT NULL
|
||||
)
|
||||
ORDER BY cs.is_pinned DESC, datetime(cs.updated_at) DESC
|
||||
`),
|
||||
listPersonasByType: db.prepare(`
|
||||
SELECT id, name, pronouns, appearance, background, details, example_dialogues, persona_type,
|
||||
source_market_id, source_creator_username, created_at, updated_at
|
||||
FROM personas
|
||||
WHERE username = ?
|
||||
AND persona_type = ?
|
||||
ORDER BY datetime(updated_at) DESC
|
||||
`),
|
||||
listAllPersonas: db.prepare(`
|
||||
SELECT id, username, name, pronouns, appearance, background, details, example_dialogues, persona_type,
|
||||
source_market_id, source_creator_username, created_at, updated_at
|
||||
FROM personas
|
||||
WHERE username = ?
|
||||
ORDER BY datetime(updated_at) DESC
|
||||
`),
|
||||
getPersona: db.prepare(`
|
||||
SELECT id, name, pronouns, appearance, background, details, example_dialogues, persona_type,
|
||||
source_market_id, source_creator_username, created_at, updated_at
|
||||
FROM personas
|
||||
WHERE id = ?
|
||||
AND username = ?
|
||||
`),
|
||||
insertPersona: db.prepare("INSERT INTO personas (username, name, pronouns, appearance, background, details, example_dialogues, persona_type) VALUES (?, ?, ?, ?, ?, ?, ?, ?)"),
|
||||
insertMarketPersona: db.prepare("INSERT INTO personas (username, name, pronouns, appearance, background, details, example_dialogues, persona_type, source_market_id, source_creator_username) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)"),
|
||||
getPersonaBySourceMarket: db.prepare(`
|
||||
SELECT id
|
||||
@@ -459,61 +49,6 @@ export function createRepositories(db, config) {
|
||||
WHERE username = ?
|
||||
AND source_market_id = ?
|
||||
`),
|
||||
updatePersona: db.prepare(`
|
||||
UPDATE personas
|
||||
SET name = ?,
|
||||
pronouns = ?,
|
||||
appearance = ?,
|
||||
background = ?,
|
||||
details = ?,
|
||||
example_dialogues = ?,
|
||||
updated_at = CURRENT_TIMESTAMP
|
||||
WHERE id = ?
|
||||
AND username = ?
|
||||
`),
|
||||
deletePersona: db.prepare("DELETE FROM personas WHERE id = ? AND username = ?"),
|
||||
getNextPersonaVersionNumber: db.prepare("SELECT COALESCE(MAX(version_number), 0) + 1 AS nextVersion FROM persona_versions WHERE persona_id = ?"),
|
||||
insertPersonaVersion: db.prepare("INSERT INTO persona_versions (persona_id, username, version_number, snapshot_json) VALUES (?, ?, ?, ?)"),
|
||||
listPersonaVersions: db.prepare(`
|
||||
SELECT id, persona_id, username, version_number, snapshot_json, created_at
|
||||
FROM persona_versions
|
||||
WHERE persona_id = ?
|
||||
AND username = ?
|
||||
ORDER BY version_number DESC
|
||||
`),
|
||||
getPersonaVersion: db.prepare(`
|
||||
SELECT id, persona_id, username, version_number, snapshot_json, created_at
|
||||
FROM persona_versions
|
||||
WHERE id = ?
|
||||
AND persona_id = ?
|
||||
AND username = ?
|
||||
`),
|
||||
getActivePersonaId: db.prepare("SELECT active_persona_id, active_user_persona_id FROM user_settings WHERE username = ?"),
|
||||
setActiveUserPersona: db.prepare(`
|
||||
INSERT INTO user_settings (username, active_user_persona_id)
|
||||
VALUES (?, ?)
|
||||
ON CONFLICT(username) DO UPDATE SET active_user_persona_id = excluded.active_user_persona_id
|
||||
`),
|
||||
getActiveUserPersona: db.prepare(`
|
||||
SELECT p.id, p.name, p.pronouns, p.appearance, p.background, p.details, p.example_dialogues
|
||||
FROM user_settings us
|
||||
JOIN personas p ON p.id = us.active_user_persona_id
|
||||
WHERE us.username = ?
|
||||
`),
|
||||
getAssistantPersonaForChat: db.prepare(`
|
||||
SELECT p.id, p.name, p.pronouns, p.appearance, p.background, p.details, p.example_dialogues
|
||||
FROM chat_sessions cs
|
||||
JOIN personas p ON p.id = cs.assistant_persona_id
|
||||
WHERE cs.id = ?
|
||||
AND cs.username = ?
|
||||
`),
|
||||
getUserPersonaForChat: db.prepare(`
|
||||
SELECT p.id, p.name, p.pronouns, p.appearance, p.background, p.details, p.example_dialogues
|
||||
FROM chat_sessions cs
|
||||
JOIN personas p ON p.id = cs.user_persona_id
|
||||
WHERE cs.id = ?
|
||||
AND cs.username = ?
|
||||
`),
|
||||
listMarketPersonas: db.prepare(`
|
||||
SELECT pm.id, pm.persona_id, pm.creator_username, pm.name, pm.pronouns, pm.appearance, pm.background, pm.details,
|
||||
pm.example_dialogues, pm.persona_type, pm.tags, pm.usage_count, pm.favorite_count, pm.rating_total, pm.rating_count,
|
||||
@@ -691,135 +226,9 @@ export function createRepositories(db, config) {
|
||||
WHERE creator_username = ?
|
||||
AND soft_deleted_at IS NULL
|
||||
`),
|
||||
updateChatSessionsUsername: db.prepare("UPDATE chat_sessions SET username = ? WHERE username = ?"),
|
||||
updatePersonasUsername: db.prepare("UPDATE personas SET username = ? WHERE username = ?"),
|
||||
updatePersonasSourceCreator: db.prepare("UPDATE personas SET source_creator_username = ? WHERE source_creator_username = ?"),
|
||||
updateUserSettingsUsername: db.prepare("UPDATE user_settings SET username = ? WHERE username = ?"),
|
||||
updatePersonaMarketCreator: db.prepare("UPDATE persona_market SET creator_username = ? WHERE creator_username = ?"),
|
||||
updatePersonaVersionsUsername: db.prepare("UPDATE persona_versions SET username = ? WHERE username = ?"),
|
||||
updatePromptTemplatesUsername: db.prepare("UPDATE prompt_templates SET username = ? WHERE username = ?"),
|
||||
updateChatMemoriesUsername: db.prepare("UPDATE chat_memories SET username = ? WHERE username = ?"),
|
||||
updatePersonaChatStateUsername: db.prepare("UPDATE persona_chat_state SET username = ? WHERE username = ?"),
|
||||
updateMarketFavoritesUsername: db.prepare("UPDATE persona_market_favorites SET username = ? WHERE username = ?"),
|
||||
updateMarketRatingsUsername: db.prepare("UPDATE persona_market_ratings SET username = ? WHERE username = ?"),
|
||||
updateMarketReportsUsername: db.prepare("UPDATE persona_market_reports SET reporter_username = ? WHERE reporter_username = ?"),
|
||||
deleteChatMemoriesByUser: db.prepare("DELETE FROM chat_memories WHERE username = ?"),
|
||||
deleteChatSessionsByUser: db.prepare("DELETE FROM chat_sessions WHERE username = ?"),
|
||||
deletePersonasByUser: db.prepare("DELETE FROM personas WHERE username = ?"),
|
||||
deleteUserSettingsByUser: db.prepare("DELETE FROM user_settings WHERE username = ?"),
|
||||
deletePersonaVersionsByUser: db.prepare("DELETE FROM persona_versions WHERE username = ?"),
|
||||
deletePromptTemplatesByUser: db.prepare("DELETE FROM prompt_templates WHERE username = ?"),
|
||||
deleteMarketPersonasByCreator: db.prepare("DELETE FROM persona_market WHERE creator_username = ?"),
|
||||
deleteMarketFavoritesByUser: db.prepare("DELETE FROM persona_market_favorites WHERE username = ?"),
|
||||
deleteMarketRatingsByUser: db.prepare("DELETE FROM persona_market_ratings WHERE username = ?"),
|
||||
deleteMarketReportsByUser: db.prepare("DELETE FROM persona_market_reports WHERE reporter_username = ?"),
|
||||
deletePersonaChatStateByUser: db.prepare("DELETE FROM persona_chat_state WHERE username = ?"),
|
||||
deleteUser: db.prepare("DELETE FROM users WHERE username = ?"),
|
||||
deleteMarketPersonaByPersonaId: db.prepare("DELETE FROM persona_market WHERE persona_id = ? AND creator_username = ?"),
|
||||
incrementMarketUsageCount: db.prepare("UPDATE persona_market SET usage_count = usage_count + 1 WHERE id = ?")
|
||||
};
|
||||
|
||||
const renameUserTransaction = db.transaction((currentUsername, nextUsername) => {
|
||||
// Existing username references do not all cascade on update. Check them at commit,
|
||||
// after every owned row has been moved to the new username.
|
||||
db.pragma("defer_foreign_keys = ON");
|
||||
statements.updateUsername.run(nextUsername, currentUsername);
|
||||
statements.updateChatSessionsUsername.run(nextUsername, currentUsername);
|
||||
statements.updatePersonasUsername.run(nextUsername, currentUsername);
|
||||
statements.updatePersonasSourceCreator.run(nextUsername, currentUsername);
|
||||
statements.updateUserSettingsUsername.run(nextUsername, currentUsername);
|
||||
statements.updatePersonaMarketCreator.run(nextUsername, currentUsername);
|
||||
statements.updatePersonaVersionsUsername.run(nextUsername, currentUsername);
|
||||
statements.updatePromptTemplatesUsername.run(nextUsername, currentUsername);
|
||||
statements.updateChatMemoriesUsername.run(nextUsername, currentUsername);
|
||||
statements.updatePersonaChatStateUsername.run(nextUsername, currentUsername);
|
||||
statements.updateMarketFavoritesUsername.run(nextUsername, currentUsername);
|
||||
statements.updateMarketRatingsUsername.run(nextUsername, currentUsername);
|
||||
statements.updateMarketReportsUsername.run(nextUsername, currentUsername);
|
||||
});
|
||||
|
||||
const deleteUserTransaction = db.transaction((username) => {
|
||||
statements.deleteMarketFavoritesByUser.run(username);
|
||||
statements.deleteMarketRatingsByUser.run(username);
|
||||
statements.deleteMarketReportsByUser.run(username);
|
||||
statements.deleteMarketPersonasByCreator.run(username);
|
||||
statements.deletePromptTemplatesByUser.run(username);
|
||||
statements.deletePersonaVersionsByUser.run(username);
|
||||
statements.deleteChatMemoriesByUser.run(username);
|
||||
statements.deletePersonaChatStateByUser.run(username);
|
||||
statements.deleteChatSessionsByUser.run(username);
|
||||
statements.deleteUserSettingsByUser.run(username);
|
||||
statements.deletePersonasByUser.run(username);
|
||||
return statements.deleteUser.run(username);
|
||||
});
|
||||
|
||||
const completeChatTurnTransaction = db.transaction((chatId, userMessageId, assistantContent, retryState, modelName, sources, thinking) => {
|
||||
const assistantResult = statements.insertChatMessage.run(
|
||||
chatId,
|
||||
"bot",
|
||||
String(assistantContent || ""),
|
||||
modelName || null,
|
||||
JSON.stringify(retryState.retryVariants),
|
||||
retryState.retryActiveIndex,
|
||||
retryState.retryRetriesUsed,
|
||||
retryState.retryPromptMessageId ?? null,
|
||||
"complete",
|
||||
null,
|
||||
userMessageId
|
||||
);
|
||||
statements.updateChatMessageDelivery.run("complete", null, userMessageId, chatId);
|
||||
if (sources) statements.updateChatMessageSources.run(JSON.stringify(sources), assistantResult.lastInsertRowid, chatId);
|
||||
if (thinking) statements.updateChatMessageThinking.run(thinking, assistantResult.lastInsertRowid, chatId);
|
||||
statements.setChatActiveLeafById.run(assistantResult.lastInsertRowid, chatId);
|
||||
return assistantResult;
|
||||
});
|
||||
|
||||
const insertBranchChatMessageTransaction = db.transaction((chatId, role, content, retryState, modelName, deliveryStatus, errorMessage, parentMessageId) => {
|
||||
const result = statements.insertChatMessage.run(
|
||||
chatId, role, String(content || ""), modelName || null,
|
||||
retryState ? JSON.stringify(retryState.retryVariants) : null,
|
||||
retryState?.retryActiveIndex ?? 0,
|
||||
retryState?.retryRetriesUsed ?? 0,
|
||||
retryState?.retryPromptMessageId ?? null,
|
||||
deliveryStatus,
|
||||
errorMessage,
|
||||
parentMessageId ?? null
|
||||
);
|
||||
statements.setChatActiveLeafById.run(result.lastInsertRowid, chatId);
|
||||
return result;
|
||||
});
|
||||
|
||||
const failChatTurnTransaction = db.transaction((chatId, userMessageId, errorMessage, parentMessageId) => {
|
||||
statements.updateChatMessageDelivery.run("failed", errorMessage, userMessageId, chatId);
|
||||
statements.setChatActiveLeafById.run(parentMessageId ?? null, chatId);
|
||||
});
|
||||
|
||||
const savePersonaVersion = db.transaction((personaId, username) => {
|
||||
const persona = statements.getPersona.get(personaId, username);
|
||||
if (!persona) return null;
|
||||
const versionNumber = statements.getNextPersonaVersionNumber.get(personaId).nextVersion;
|
||||
statements.insertPersonaVersion.run(personaId, username, versionNumber, buildPersonaSnapshot(persona));
|
||||
return versionNumber;
|
||||
});
|
||||
|
||||
const clonePersonaTransaction = db.transaction((personaId, username) => {
|
||||
const source = statements.getPersona.get(personaId, username);
|
||||
if (!source) return null;
|
||||
const cloneId = statements.insertPersona.run(
|
||||
username,
|
||||
`${source.name} Copy`,
|
||||
source.pronouns,
|
||||
source.appearance,
|
||||
source.background,
|
||||
source.details,
|
||||
source.example_dialogues,
|
||||
source.persona_type
|
||||
).lastInsertRowid;
|
||||
const clone = statements.getPersona.get(cloneId, username);
|
||||
statements.insertPersonaVersion.run(cloneId, username, 1, buildPersonaSnapshot(clone));
|
||||
return clone;
|
||||
});
|
||||
|
||||
const importWorkspaceTransaction = db.transaction((username, workspace) => {
|
||||
const imported = {personas: 0, chats: 0, templates: 0};
|
||||
const personaIds = new Map();
|
||||
@@ -965,227 +374,10 @@ export function createRepositories(db, config) {
|
||||
});
|
||||
|
||||
return {
|
||||
getUser: (username) => statements.getUser.get(username),
|
||||
insertUser: (username, passwordHash) => statements.insertUser.run(username, passwordHash),
|
||||
updatePassword: (passwordHash, username) => statements.updatePassword.run(passwordHash, username),
|
||||
incrementSessionVersion(username) {
|
||||
statements.incrementSessionVersion.run(username);
|
||||
return statements.getUser.get(username);
|
||||
},
|
||||
renameUser: (currentUsername, nextUsername) => renameUserTransaction(currentUsername, nextUsername),
|
||||
deleteUser: (username) => deleteUserTransaction(username),
|
||||
listChats: (username) => statements.listChatSessions.all(username),
|
||||
searchChats(username, query) {
|
||||
const like = `%${String(query || "").trim().toLowerCase()}%`;
|
||||
return statements.searchChats.all(like, username, like);
|
||||
},
|
||||
countChats: (username) => statements.countChatSessions.get(username).count,
|
||||
getChat: (chatId, username) => statements.getChatSession.get(chatId, username),
|
||||
getChatByParticipants: (username, assistantPersonaId, userPersonaId) => statements.getChatSessionByParticipants.get(username, assistantPersonaId, userPersonaId, userPersonaId),
|
||||
createChat(username, title, assistantPersonaId = null, userPersonaId = null, scenarioPrompt = null, scenarioSummary = null, organization = {}) {
|
||||
const chatId = statements.insertChatSession.run(
|
||||
username,
|
||||
title,
|
||||
assistantPersonaId,
|
||||
userPersonaId,
|
||||
organization.folderName ?? null,
|
||||
organization.isPinned ? 1 : 0,
|
||||
organization.archivedAt ?? null,
|
||||
scenarioPrompt,
|
||||
scenarioSummary
|
||||
).lastInsertRowid;
|
||||
return statements.getChatSession.get(chatId, username);
|
||||
},
|
||||
updateChatTitle: (chatId, username, title) => statements.updateChatSessionTitle.run(title, chatId, username),
|
||||
updateChatScene: (chatId, username, scenarioPrompt, scenarioSummary) => statements.updateChatSessionScene.run(scenarioPrompt, scenarioSummary, chatId, username),
|
||||
updateChatContextSummary: (chatId, username, summary, messageId) => statements.updateChatContextSummary.run(summary, Number(messageId || 0), chatId, username),
|
||||
getPersonaChatState: (chatId, username) => statements.getPersonaChatState.get(chatId, username) || null,
|
||||
updatePersonaChatState(username, chat, state) {
|
||||
statements.upsertPersonaChatState.run(
|
||||
username,
|
||||
chat.id,
|
||||
chat.assistant_persona_id ?? null,
|
||||
chat.user_persona_id ?? null,
|
||||
state.relationship_notes ?? null,
|
||||
state.tone ?? null,
|
||||
state.current_location ?? null,
|
||||
state.goals ?? null,
|
||||
state.unresolved_threads ?? null,
|
||||
state.boundaries ?? null
|
||||
);
|
||||
return statements.getPersonaChatState.get(chat.id, username);
|
||||
},
|
||||
updateChatOrganization(chatId, username, {folderName = null, isPinned = false, archivedAt = null}) {
|
||||
return statements.updateChatSessionOrganization.run(folderName, isPinned ? 1 : 0, archivedAt, chatId, username);
|
||||
},
|
||||
touchChat: (chatId, username) => statements.touchChatSession.run(chatId, username),
|
||||
deleteChat: (chatId, username) => statements.deleteChatSession.run(chatId, username),
|
||||
listChatMessages: (chatId) => statements.getChatMessages.all(chatId),
|
||||
listActiveBranchMessages: (chatId) => statements.getActiveBranchMessages.all(chatId),
|
||||
listChatBranchThroughMessage: (chatId, messageId) => statements.getChatBranchThroughMessage.all(chatId, messageId),
|
||||
listSiblingChatMessages: (chatId, role, parentMessageId) => statements.getSiblingChatMessages.all(chatId, role, parentMessageId ?? null),
|
||||
getLatestDescendantMessage: (chatId, messageId) => statements.getLatestDescendantMessage.get(chatId, messageId, chatId),
|
||||
getRecentChatMessages: (chatId) => statements.getRecentChatMessages.all(chatId, config.chatHistoryLimit),
|
||||
getRecentChatMessagesUpToId: (chatId, messageId) => statements.getRecentChatMessagesUpToId.all(chatId, messageId, config.chatHistoryLimit),
|
||||
getChatMessage: (chatId, messageId) => statements.getChatMessageById.get(messageId, chatId),
|
||||
getChatMessageByIndex: (chatId, index) => statements.getChatMessageByIndex.get(chatId, index),
|
||||
getLatestChatMessage: (chatId) => statements.getLatestChatMessage.get(chatId),
|
||||
getPreviousUserMessage: (chatId, messageId) => statements.getPreviousUserMessage.get(chatId, messageId),
|
||||
listChatMemories: (chatId, username) => statements.listChatMemories.all(chatId, username),
|
||||
getChatMemory: (chatId, memoryId, username) => statements.getChatMemory.get(memoryId, chatId, username),
|
||||
createChatMemory(username, chat, fact) {
|
||||
const result = statements.insertChatMemory.run(
|
||||
username,
|
||||
chat.id,
|
||||
chat.assistant_persona_id ?? null,
|
||||
chat.user_persona_id ?? null,
|
||||
String(fact || "").trim()
|
||||
);
|
||||
return statements.getChatMemory.get(result.lastInsertRowid, chat.id, username);
|
||||
},
|
||||
updateChatMemory: (chatId, memoryId, username, fact) => statements.updateChatMemory.run(String(fact || "").trim(), memoryId, chatId, username),
|
||||
deleteChatMemory: (chatId, memoryId, username) => statements.deleteChatMemory.run(memoryId, chatId, username),
|
||||
insertChatMessage(chatId, role, content, retryState = null, modelName = null, deliveryStatus = "complete", errorMessage = null, parentMessageId = null) {
|
||||
return statements.insertChatMessage.run(
|
||||
chatId,
|
||||
role,
|
||||
String(content || ""),
|
||||
modelName || null,
|
||||
retryState ? JSON.stringify(retryState.retryVariants) : null,
|
||||
retryState?.retryActiveIndex ?? 0,
|
||||
retryState?.retryRetriesUsed ?? 0,
|
||||
retryState?.retryPromptMessageId ?? null,
|
||||
deliveryStatus,
|
||||
errorMessage,
|
||||
parentMessageId
|
||||
);
|
||||
},
|
||||
insertBranchChatMessage: (chatId, role, content, retryState = null, modelName = null, deliveryStatus = "complete", errorMessage = null, parentMessageId = null) =>
|
||||
insertBranchChatMessageTransaction(chatId, role, content, retryState, modelName, deliveryStatus, errorMessage, parentMessageId),
|
||||
updateChatMessageDelivery: (chatId, messageId, deliveryStatus, errorMessage = null) =>
|
||||
statements.updateChatMessageDelivery.run(deliveryStatus, errorMessage, messageId, chatId),
|
||||
completeChatTurn: (chatId, userMessageId, assistantContent, retryState, modelName = null, sources = null, thinking = null) =>
|
||||
completeChatTurnTransaction(chatId, userMessageId, assistantContent, retryState, modelName, sources, thinking),
|
||||
updateChatMessageThinking: (chatId, messageId, thinking) =>
|
||||
statements.updateChatMessageThinking.run(thinking, messageId, chatId),
|
||||
updateChatMessageSources: (chatId, messageId, sources) =>
|
||||
statements.updateChatMessageSources.run(JSON.stringify(sources), messageId, chatId),
|
||||
failChatTurn: (chatId, userMessageId, errorMessage, parentMessageId = null) =>
|
||||
failChatTurnTransaction(chatId, userMessageId, errorMessage, parentMessageId),
|
||||
updateChatActiveLeaf: (chatId, username, messageId) => statements.updateChatActiveLeaf.run(messageId, chatId, username),
|
||||
updateChatBranchSource: (chatId, username, sourceChatId, messageId) =>
|
||||
statements.updateChatBranchSource.run(sourceChatId, messageId ?? null, chatId, username),
|
||||
updateChatGenerationSettings: (chatId, username, settings) => statements.updateChatGenerationSettings.run(
|
||||
settings.preferredModel ?? null,
|
||||
settings.temperature ?? null,
|
||||
settings.contextLength ?? null,
|
||||
settings.responseLength ?? null,
|
||||
settings.systemInstruction ?? null,
|
||||
chatId,
|
||||
username
|
||||
),
|
||||
updateChatPreferredModel: (chatId, username, model) => statements.updateChatPreferredModel.run(model || null, chatId, username),
|
||||
listMessageVersions: (messageId) => db.prepare("SELECT * FROM chat_message_versions WHERE message_id = ? ORDER BY id").all(messageId),
|
||||
saveMessageVersion: (chatId, messageId, version = null, selectedIndex = null) => db.transaction(() => {
|
||||
const row = statements.getChatMessageById.get(messageId, chatId);
|
||||
if (!row) throw new Error("Message not found");
|
||||
let versions = db.prepare("SELECT * FROM chat_message_versions WHERE message_id = ? ORDER BY id").all(messageId);
|
||||
const insert = db.prepare("INSERT INTO chat_message_versions (message_id, content, thinking, sources, model_name) VALUES (?, ?, ?, ?, ?)");
|
||||
if (!versions.length) {
|
||||
db.prepare("INSERT INTO chat_message_versions (message_id, content, thinking, sources, model_name, created_at) VALUES (?, ?, ?, ?, ?, ?)")
|
||||
.run(messageId, row.content, row.thinking, row.sources, row.model_name, row.created_at);
|
||||
} else {
|
||||
// Edits keep their existing semantics: update the selected answer.
|
||||
db.prepare("UPDATE chat_message_versions SET content = ?, thinking = ?, sources = ?, model_name = ? WHERE id = ?")
|
||||
.run(row.content, row.thinking, row.sources, row.model_name, versions[row.retry_active_index]?.id ?? versions[0].id);
|
||||
}
|
||||
if (version) insert.run(messageId, version.content, version.thinking || null, JSON.stringify(version.sources ?? null), version.modelName);
|
||||
versions = db.prepare("SELECT * FROM chat_message_versions WHERE message_id = ? ORDER BY id").all(messageId);
|
||||
const index = version ? versions.length - 1 : selectedIndex;
|
||||
if (!Number.isInteger(index) || !versions[index]) throw new Error("Version is out of range");
|
||||
const active = versions[index];
|
||||
db.prepare(`UPDATE chat_messages SET content = ?, thinking = ?, sources = ?, model_name = ?,
|
||||
retry_variants = ?, retry_active_index = ?, retry_retries_used = ? WHERE id = ? AND chat_id = ?`)
|
||||
.run(active.content, active.thinking, active.sources, active.model_name, JSON.stringify(versions.map(item => item.content)), index, versions.length - 1, messageId, chatId);
|
||||
})(),
|
||||
updateChatMessage: (chatId, messageId, content) => statements.updateChatMessage.run(content, messageId, chatId),
|
||||
updateChatMessageRetryState: (chatId, messageId, payload) => statements.updateChatMessageWithRetryState.run(
|
||||
payload.content,
|
||||
JSON.stringify(payload.retryVariants),
|
||||
payload.retryActiveIndex,
|
||||
payload.retryRetriesUsed,
|
||||
payload.retryPromptMessageId ?? null,
|
||||
messageId,
|
||||
chatId
|
||||
),
|
||||
deleteChatMessage: (chatId, messageId) => statements.deleteChatMessage.run(messageId, chatId),
|
||||
clearChatMessages: (chatId) => statements.deleteChatMessages.run(chatId),
|
||||
listPersonasByType: (username, personaType) => statements.listPersonasByType.all(username, personaType),
|
||||
listAllPersonas: (username) => statements.listAllPersonas.all(username),
|
||||
getPersona: (personaId, username) => statements.getPersona.get(personaId, username),
|
||||
createPersona(username, payload, personaType) {
|
||||
const personaId = statements.insertPersona.run(
|
||||
username,
|
||||
payload.name,
|
||||
payload.pronouns,
|
||||
payload.appearance,
|
||||
payload.background,
|
||||
payload.details,
|
||||
payload.exampleDialogues,
|
||||
personaType
|
||||
).lastInsertRowid;
|
||||
const persona = statements.getPersona.get(personaId, username);
|
||||
statements.insertPersonaVersion.run(personaId, username, 1, buildPersonaSnapshot(persona));
|
||||
return persona;
|
||||
},
|
||||
updatePersona(personaId, username, payload) {
|
||||
savePersonaVersion(personaId, username);
|
||||
return statements.updatePersona.run(
|
||||
payload.name,
|
||||
payload.pronouns,
|
||||
payload.appearance,
|
||||
payload.background,
|
||||
payload.details,
|
||||
payload.exampleDialogues,
|
||||
personaId,
|
||||
username
|
||||
);
|
||||
},
|
||||
restorePersonaVersion(personaId, versionId, username) {
|
||||
const version = statements.getPersonaVersion.get(versionId, personaId, username);
|
||||
if (!version) return null;
|
||||
savePersonaVersion(personaId, username);
|
||||
const snapshot = JSON.parse(version.snapshot_json);
|
||||
statements.updatePersona.run(
|
||||
snapshot.name,
|
||||
snapshot.pronouns ?? null,
|
||||
snapshot.appearance ?? null,
|
||||
snapshot.background ?? null,
|
||||
snapshot.details ?? null,
|
||||
snapshot.example_dialogues ?? null,
|
||||
personaId,
|
||||
username
|
||||
);
|
||||
return statements.getPersona.get(personaId, username);
|
||||
},
|
||||
listPersonaVersions(personaId, username) {
|
||||
return statements.listPersonaVersions.all(personaId, username).map((version) => ({
|
||||
...version,
|
||||
snapshot: JSON.parse(version.snapshot_json)
|
||||
}));
|
||||
},
|
||||
clonePersona: (personaId, username) => clonePersonaTransaction(personaId, username),
|
||||
deletePersona: (personaId, username) => statements.deletePersona.run(personaId, username),
|
||||
clearPersonaReferences(username, personaId) {
|
||||
statements.clearAssistantPersonaFromChats.run(username, personaId);
|
||||
statements.clearUserPersonaFromChats.run(username, personaId);
|
||||
statements.deleteMarketPersonaByPersonaId.run(personaId, username);
|
||||
},
|
||||
getActivePersonaIds: (username) => statements.getActivePersonaId.get(username) || {},
|
||||
setActiveUserPersona: (username, personaId) => statements.setActiveUserPersona.run(username, personaId),
|
||||
getActiveUserPersona: (username) => statements.getActiveUserPersona.get(username),
|
||||
getAssistantPersonaForChat: (chatId, username) => statements.getAssistantPersonaForChat.get(chatId, username),
|
||||
getUserPersonaForChat: (chatId, username) => statements.getUserPersonaForChat.get(chatId, username),
|
||||
...createHistoryRepository(db),
|
||||
...createAccountRepository(db, statements),
|
||||
...createChatRepository(db, config, statements),
|
||||
...createPersonaRepository(db, statements),
|
||||
listMarketPersonas({admin = false, username = null, personaType = null, sort = "best"} = {}) {
|
||||
const normalizedPersonaType = normalizeMarketPersonaType(personaType);
|
||||
const normalizedSort = normalizeMarketSort(sort);
|
||||
|
||||
@@ -299,6 +299,8 @@ export function runMigrations(db, {dropLegacyChats = true} = {}) {
|
||||
ensureColumn("chat_messages", chatMessageColumns, "delivery_status", "ALTER TABLE chat_messages ADD COLUMN delivery_status TEXT DEFAULT 'complete'");
|
||||
ensureColumn("chat_messages", chatMessageColumns, "error_message", "ALTER TABLE chat_messages ADD COLUMN error_message TEXT");
|
||||
ensureColumn("chat_messages", chatMessageColumns, "parent_message_id", "ALTER TABLE chat_messages ADD COLUMN parent_message_id INTEGER");
|
||||
db.exec(`CREATE INDEX IF NOT EXISTS chat_messages_family ON chat_messages(chat_id, parent_message_id, role, id);
|
||||
CREATE INDEX IF NOT EXISTS chat_messages_chat_order ON chat_messages(chat_id, id);`);
|
||||
db.prepare("UPDATE chat_messages SET retry_active_index = 0 WHERE retry_active_index IS NULL").run();
|
||||
db.prepare("UPDATE chat_messages SET retry_retries_used = 0 WHERE retry_retries_used IS NULL").run();
|
||||
db.prepare("UPDATE chat_messages SET delivery_status = 'complete' WHERE delivery_status IS NULL").run();
|
||||
|
||||
@@ -5,6 +5,7 @@ import express from "express";
|
||||
import {asyncHandler} from "../middleware/asyncHandler.js";
|
||||
import bcrypt from "bcrypt";
|
||||
import {validateBody} from "../middleware/validate.js";
|
||||
import {newPasswordError} from "../services/passwordPolicy.js";
|
||||
|
||||
const credentialsValidator = (body) => {
|
||||
const username = String(body?.username || "").trim();
|
||||
@@ -22,6 +23,8 @@ export function createAuthRouter({repositories, authRateLimiters, config}) {
|
||||
|
||||
router.post("/register", ...authRateLimiters, validateBody(credentialsValidator), asyncHandler(async (req, res) => {
|
||||
const {username, password, inviteCode} = req.validatedBody;
|
||||
const passwordError = newPasswordError(password);
|
||||
if (passwordError) return res.status(400).json({error: passwordError, code: "PASSWORD_TOO_LONG"});
|
||||
if (config.registrationInviteCode && inviteCode !== config.registrationInviteCode) {
|
||||
return res.status(403).json({error: "Valid invite code is required"});
|
||||
}
|
||||
|
||||
+53
-5
@@ -217,10 +217,39 @@ export function createChatsRouter({repositories, chatService, modelService, conf
|
||||
const chatId = Number(req.params.id);
|
||||
const chat = repositories.getChat(chatId, req.session.user);
|
||||
if (!chat) return res.status(404).json({error: "Chat not found"});
|
||||
return res.json({
|
||||
messages: chatService.listActiveBranchMessages(chatId),
|
||||
chat
|
||||
});
|
||||
const options = {};
|
||||
for (const key of ["limit", "before", "leaf", "around"]) {
|
||||
if (req.query[key] === undefined) continue;
|
||||
const value = Number(req.query[key]);
|
||||
if (!Number.isSafeInteger(value) || value < 1 || (key === "limit" && value > 100)) {
|
||||
return res.status(400).json({error: `Invalid history ${key}`});
|
||||
}
|
||||
options[key] = value;
|
||||
}
|
||||
const result = chatService.getHistoryPage(chatId, options);
|
||||
if (result.error) return res.status(result.status).json(result);
|
||||
return res.json({...result, chat});
|
||||
});
|
||||
|
||||
router.get("/chats/:id/branches", (req, res) => {
|
||||
const chatId = Number(req.params.id);
|
||||
const chat = repositories.getChat(chatId, req.session.user);
|
||||
if (!chat) return res.status(404).json({error: "Chat not found"});
|
||||
const after = Number(req.query.after || 0);
|
||||
if (!Number.isSafeInteger(after) || after < 0) return res.status(400).json({error: "Invalid branch cursor"});
|
||||
const nodes = repositories.listBranchNodes(chatId, {after});
|
||||
const hasMore = nodes.length > 200;
|
||||
if (hasMore) nodes.pop();
|
||||
return res.json({nodes, activeLeafId: chat.active_leaf_message_id || nodes.at(-1)?.id || null,
|
||||
next: hasMore ? nodes.at(-1).id : null});
|
||||
});
|
||||
|
||||
router.get("/chats/:id/messages/:messageId/versions", (req, res) => {
|
||||
const chatId = Number(req.params.id);
|
||||
if (!repositories.getChat(chatId, req.session.user)) return res.status(404).json({error: "Chat not found"});
|
||||
const message = repositories.getChatMessage(chatId, Number(req.params.messageId));
|
||||
if (!message) return res.status(404).json({error: "Message not found"});
|
||||
return res.json({message: chatService.formatChatMessage(message)});
|
||||
});
|
||||
|
||||
router.put("/chats/:id/generation-settings", validateBody(generationSettingsValidator), (req, res) => {
|
||||
@@ -406,7 +435,26 @@ export function createChatsRouter({repositories, chatService, modelService, conf
|
||||
const chatId = Number(req.params.id);
|
||||
const chat = repositories.getChat(chatId, req.session.user);
|
||||
if (!chat) return res.status(404).json({error: "Chat not found"});
|
||||
return res.json({memories: repositories.listChatMemories(chatId, req.session.user)});
|
||||
const payload = chatService.buildMessagesPayload({user: req.session.user, chatId, session: chat});
|
||||
const characters = payload.messagesPayload.reduce((total, message) => total + String(message.content || "").length, 0);
|
||||
return res.json({memories: repositories.listChatMemories(chatId, req.session.user), summary: chat.context_summary || "",
|
||||
summaryMessageId: chat.context_summary_message_id || 0, summaryMaxChars: config.chatSummaryMaxChars,
|
||||
context: {estimatedTokens: Math.ceil(characters / 4), contextLength: chat.context_length || null,
|
||||
recentMessageLimit: config.chatHistoryLimit, memoryFactLimit: 30}});
|
||||
});
|
||||
|
||||
router.put("/chats/:id/summary", (req, res) => {
|
||||
const chatId = Number(req.params.id);
|
||||
const chat = repositories.getChat(chatId, req.session.user);
|
||||
if (!chat) return res.status(404).json({error: "Chat not found"});
|
||||
if (chatService.isGenerating(chatId)) return res.status(409).json({error: "Wait for the response to finish before editing its summary."});
|
||||
const summary = req.body?.summary;
|
||||
if (typeof summary !== "string" || summary.length > config.chatSummaryMaxChars) {
|
||||
return res.status(400).json({error: `Summary must be text of at most ${config.chatSummaryMaxChars} characters.`});
|
||||
}
|
||||
if (req.body.previousSummary !== (chat.context_summary || "")) return res.status(409).json({error: "The summary changed. Reopen memory to review it before saving."});
|
||||
repositories.updateChatContextSummary(chatId, req.session.user, summary.trim() || null, chat.context_summary_message_id);
|
||||
return res.json({summary: summary.trim()});
|
||||
});
|
||||
|
||||
router.post("/chats/:id/memories", validateBody(memoryValidator), (req, res) => {
|
||||
|
||||
@@ -6,6 +6,7 @@ import {asyncHandler} from "../middleware/asyncHandler.js";
|
||||
import bcrypt from "bcrypt";
|
||||
import {requireLogin} from "../middleware/auth.js";
|
||||
import {validateBody} from "../middleware/validate.js";
|
||||
import {newPasswordError} from "../services/passwordPolicy.js";
|
||||
|
||||
const usernameValidator = (body) => {
|
||||
const username = String(body?.username || "").trim();
|
||||
@@ -72,6 +73,8 @@ export function createSettingsRouter({repositories}) {
|
||||
router.put("/settings/password", validateBody(passwordValidator), asyncHandler(async (req, res) => {
|
||||
const username = req.session.user;
|
||||
const {currentPassword, newPassword} = req.validatedBody;
|
||||
const passwordError = newPasswordError(newPassword);
|
||||
if (passwordError) return res.status(400).json({error: passwordError, code: "PASSWORD_TOO_LONG"});
|
||||
const user = repositories.getUser(username);
|
||||
if (!user) return res.status(404).json({error: "User not found"});
|
||||
const valid = await bcrypt.compare(currentPassword, user.password);
|
||||
|
||||
@@ -25,14 +25,20 @@ function buildConfigWarnings(config) {
|
||||
return warnings;
|
||||
}
|
||||
|
||||
export function createSystemRouter({modelService, config}) {
|
||||
export function createHealthRouter({modelService}) {
|
||||
const router = express.Router();
|
||||
|
||||
router.get("/health", asyncHandler(async (req, res) => {
|
||||
const model = await modelService.checkHealth();
|
||||
return res.status(model.ok ? 200 : 503).json({ok: model.ok, model});
|
||||
return res.status(model.ok ? 200 : 503).json({ok: Boolean(model.ok)});
|
||||
}));
|
||||
|
||||
return router;
|
||||
}
|
||||
|
||||
export function createSystemRouter({modelService, config}) {
|
||||
const router = express.Router();
|
||||
|
||||
router.get("/admin/diagnostics", requireLogin, requireAdmin, asyncHandler(async (req, res) => {
|
||||
const diagnostics = await modelService.getDiagnostics();
|
||||
return res.status(diagnostics.model.ok ? 200 : 503).json({
|
||||
|
||||
+19
-16
@@ -27,10 +27,10 @@ import { supportsReasoning } from "../../public/js/modelCapabilities.js";
|
||||
import {VISION_EVIDENCE_GUIDANCE} from "./visionAnalysis.js";
|
||||
|
||||
export function createChatService(repositories, modelService, config, webSearchProvider = new SearXNGSearchProvider(config), attachmentService = null, visionDebugService = null) {
|
||||
const formatChatMessage = row => {
|
||||
const formatChatMessage = (row, related = null) => {
|
||||
const message = formatBaseChatMessage(row);
|
||||
if (message?.role === "bot") {
|
||||
const versions = repositories.listMessageVersions(row.id);
|
||||
const versions = related?.versions ? related.versions.get(row.id) || [] : repositories.listMessageVersions(row.id);
|
||||
message.versionCount = versions.length || 1;
|
||||
message.versionIndex = versions.length ? Number(row.retry_active_index || 0) : 0;
|
||||
if (versions.length) message.responseVersions = versions.map(version => ({
|
||||
@@ -38,7 +38,7 @@ export function createChatService(repositories, modelService, config, webSearchP
|
||||
sources: parseSources(version.sources), modelName: version.model_name, createdAt: version.created_at
|
||||
}));
|
||||
}
|
||||
const attachments = row && attachmentService?.list(row.id);
|
||||
const attachments = row && (related?.attachments ? related.attachments.get(row.id) : attachmentService?.list(row.id));
|
||||
return message && attachments?.length ? { ...message, attachments } : message;
|
||||
};
|
||||
const chatGenerationQueues = new Map();
|
||||
@@ -198,6 +198,9 @@ export function createChatService(repositories, modelService, config, webSearchP
|
||||
|
||||
const latestSummarized = summarizable[summarizable.length - 1];
|
||||
if (!latestSummarized || !summary) return;
|
||||
const current = repositories.getChat(chatId, user);
|
||||
// A manual correction or branch switch during generation wins over the stale summary.
|
||||
if (!current || current.context_summary !== session.context_summary || current.context_summary_message_id !== session.context_summary_message_id) return;
|
||||
repositories.updateChatContextSummary(chatId, user, clampSummary(summary), latestSummarized.id);
|
||||
};
|
||||
|
||||
@@ -573,18 +576,17 @@ export function createChatService(repositories, modelService, config, webSearchP
|
||||
}
|
||||
});
|
||||
|
||||
const listActiveBranchMessages = (chatId) => repositories.listActiveBranchMessages(chatId).map((row) => {
|
||||
const message = formatChatMessage(row);
|
||||
const siblings = repositories.listSiblingChatMessages(chatId, row.role, row.parent_message_id);
|
||||
const siblingIndex = siblings.findIndex((sibling) => sibling.id === row.id);
|
||||
return {
|
||||
...message,
|
||||
siblingIndex,
|
||||
siblingCount: siblings.length,
|
||||
previousSiblingId: siblingIndex > 0 ? siblings[siblingIndex - 1].id : null,
|
||||
nextSiblingId: siblingIndex >= 0 && siblingIndex < siblings.length - 1 ? siblings[siblingIndex + 1].id : null
|
||||
};
|
||||
});
|
||||
const formatHistory = (chatId, rows) => {
|
||||
const metadata = repositories.getHistoryMetadata(chatId, rows.map(row => row.id));
|
||||
return rows.map(row => ({...formatChatMessage(row, metadata), ...metadata.siblings.get(row.id)}));
|
||||
};
|
||||
const listActiveBranchMessages = chatId => formatHistory(chatId, repositories.listActiveBranchMessages(chatId));
|
||||
const getHistoryPage = (chatId, options) => {
|
||||
const result = repositories.getActiveBranchPage(chatId, options);
|
||||
if (result.error) return result;
|
||||
const {rows, ...details} = result;
|
||||
return {...details, messages: formatHistory(chatId, rows)};
|
||||
};
|
||||
|
||||
const cloneChatThroughMessage = ({ user, sourceChat, messageId = null }) => {
|
||||
if (repositories.countChats(user) >= config.chatLimit) {
|
||||
@@ -779,9 +781,10 @@ export function createChatService(repositories, modelService, config, webSearchP
|
||||
const leaf = includeDescendants ? repositories.getLatestDescendantMessage(chatId, messageId) : message;
|
||||
repositories.updateChatActiveLeaf(chatId, user, leaf?.id || messageId);
|
||||
repositories.updateChatContextSummary(chatId, user, null, 0);
|
||||
return { messages: listActiveBranchMessages(chatId) };
|
||||
return getHistoryPage(chatId);
|
||||
},
|
||||
listActiveBranchMessages,
|
||||
getHistoryPage,
|
||||
stopGeneration({ user, chatId }) {
|
||||
if (!repositories.getChat(chatId, user)) return { error: "Chat not found", status: 404 };
|
||||
const controller = activeGenerations.get(chatId);
|
||||
|
||||
@@ -0,0 +1,7 @@
|
||||
// bcrypt uses the first 72 UTF-8 bytes. Apply this only when creating a new hash:
|
||||
// existing credentials must remain usable for login and current-password checks.
|
||||
export function newPasswordError(password) {
|
||||
return Buffer.byteLength(password, "utf8") > 72
|
||||
? "Password must be at most 72 UTF-8 bytes (some characters use multiple bytes)"
|
||||
: null;
|
||||
}
|
||||
Reference in New Issue
Block a user