@musnows/scriverse 0.5.12 → 0.6.0

package/dist/store.js

@@ -123,6 +123,10 @@ export const versionedEntityTypes = [
     "chapter-outline",
     "foreshadow"
 ];
+export function defaultAiConversationTitle(prompt) {
+    const normalized = prompt.replace(/\s+/gu, " ").trim();
+    return Array.from(normalized).slice(0, 15).join("") || "新对话";
+}
 function isRecord(value) {
     return Boolean(value) && typeof value === "object" && !Array.isArray(value);
 }
@@ -689,6 +693,9 @@ export class Store {
             agentTools: json(String(row?.agent_tools_json ?? '["story_index","read_chapters","search_story_entities","grep","read_character_sections","search_drafts"]'), ["story_index", "read_chapters", "search_story_entities", "grep", "read_character_sections", "search_drafts"])
                 .map((tool) => tool === "query_story_knowledge" ? "search_story_entities" : tool)
                 .filter((tool, index, tools) => tools.indexOf(tool) === index),
+            titleGenerationModelId: row?.title_generation_model_id === null || row?.title_generation_model_id === undefined
+                ? null
+                : String(row.title_generation_model_id),
             updatedAt: String(row?.updated_at ?? "")
         };
     }
@@ -705,12 +712,15 @@ export class Store {
         const nextBookSummaryContextPercent = input.bookSummaryContextPercent ?? Number(current.bookSummaryContextPercent);
         const nextContextCompactThreshold = input.contextCompactThreshold ?? Number(current.contextCompactThreshold);
         const nextAgentTools = input.agentTools ?? current.agentTools;
+        const nextTitleGenerationModelId = input.titleGenerationModelId === undefined
+            ? (current.titleGenerationModelId ? String(current.titleGenerationModelId) : null)
+            : input.titleGenerationModelId?.trim() || null;
         this.db.run(`INSERT INTO work_ai_settings (
          work_id, system_prompt, auto_run_enabled, auto_run_concurrency, auto_run_batch_limit,
          auto_run_daily_task_limit, auto_run_failure_threshold, auto_run_paused, auto_run_pause_reason,
          auto_run_resume_at, auto_run_consecutive_failures, book_summary_context_percent,
-         context_compact_threshold, agent_tools_json, updated_at
-       ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
+         context_compact_threshold, agent_tools_json, title_generation_model_id, updated_at
+       ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
        ON CONFLICT(work_id) DO UPDATE SET
          system_prompt = excluded.system_prompt,
          auto_run_enabled = excluded.auto_run_enabled,
@@ -725,7 +735,8 @@ export class Store {
          book_summary_context_percent = excluded.book_summary_context_percent,
          context_compact_threshold = excluded.context_compact_threshold,
          agent_tools_json = excluded.agent_tools_json,
-         updated_at = excluded.updated_at`, workId, nextPrompt, nextEnabled ? 1 : 0, Math.min(8, Math.max(1, nextConcurrency)), Math.min(200, Math.max(1, nextBatchLimit)), Math.min(10_000, Math.max(0, nextDailyTaskLimit)), Math.min(10, Math.max(1, nextFailureThreshold)), current.autoRunPaused ? 1 : 0, String(current.autoRunPauseReason ?? ""), current.autoRunResumeAt === null ? null : String(current.autoRunResumeAt), Math.max(0, Number(current.autoRunConsecutiveFailures) || 0), Math.min(90, Math.max(1, nextBookSummaryContextPercent)), Math.min(90, Math.max(50, nextContextCompactThreshold)), JSON.stringify(nextAgentTools), timestamp);
+         title_generation_model_id = excluded.title_generation_model_id,
+         updated_at = excluded.updated_at`, workId, nextPrompt, nextEnabled ? 1 : 0, Math.min(8, Math.max(1, nextConcurrency)), Math.min(200, Math.max(1, nextBatchLimit)), Math.min(10_000, Math.max(0, nextDailyTaskLimit)), Math.min(10, Math.max(1, nextFailureThreshold)), current.autoRunPaused ? 1 : 0, String(current.autoRunPauseReason ?? ""), current.autoRunResumeAt === null ? null : String(current.autoRunResumeAt), Math.max(0, Number(current.autoRunConsecutiveFailures) || 0), Math.min(90, Math.max(1, nextBookSummaryContextPercent)), Math.min(90, Math.max(50, nextContextCompactThreshold)), JSON.stringify(nextAgentTools), nextTitleGenerationModelId, timestamp);
         this.audit(workId, "work.ai-settings.updated", "work-ai-settings", workId, {
             systemPromptChanged: input.systemPrompt !== undefined,
             autoRunEnabled: nextEnabled,
@@ -735,7 +746,8 @@ export class Store {
             autoRunFailureThreshold: Math.min(10, Math.max(1, nextFailureThreshold)),
             bookSummaryContextPercent: Math.min(90, Math.max(1, nextBookSummaryContextPercent)),
             contextCompactThreshold: Math.min(90, Math.max(50, nextContextCompactThreshold)),
-            agentTools: nextAgentTools
+            agentTools: nextAgentTools,
+            titleGenerationModelId: nextTitleGenerationModelId
         });
         return this.getWorkAiSettings(workId);
     }
@@ -4127,7 +4139,7 @@ export class Store {
             throw notFound("AI 对话");
         if (requiredString(conversation, "work_id") !== workId)
             throw new AppError(400, "CONVERSATION_WORK_MISMATCH", "AI 对话不属于当前作品");
-        const rows = this.db.all("SELECT id, role, content FROM ai_conversation_messages WHERE conversation_id = ? ORDER BY created_at, rowid", conversationId);
+        const rows = this.db.all("SELECT id, role, content, metadata_json FROM ai_conversation_messages WHERE conversation_id = ? ORDER BY created_at, rowid", conversationId);
         const compactedMessageCount = Math.min(rows.length, Math.max(0, numberValue(conversation, "compacted_message_count")));
         return {
             workId,
@@ -4140,10 +4152,23 @@ export class Store {
                 .map((message) => ({
                 id: requiredString(message, "id"),
                 role: requiredString(message, "role") === "assistant" ? "assistant" : "user",
-                content: requiredString(message, "content")
+                content: requiredString(message, "content"),
+                metadata: json(requiredString(message, "metadata_json"), {})
             }))
         };
     }
+    getAiConversationTitleContext(conversationId, workId) {
+        const conversation = this.db.get("SELECT title, work_id FROM ai_conversations WHERE id = ?", conversationId);
+        if (!conversation)
+            throw notFound("AI 对话");
+        if (requiredString(conversation, "work_id") !== workId)
+            throw new AppError(400, "CONVERSATION_WORK_MISMATCH", "AI 对话不属于当前作品");
+        const messages = this.db.all("SELECT role, content FROM ai_conversation_messages WHERE conversation_id = ? ORDER BY created_at, rowid", conversationId).map((message) => ({
+            role: requiredString(message, "role") === "assistant" ? "assistant" : "user",
+            content: requiredString(message, "content")
+        }));
+        return { title: requiredString(conversation, "title"), messages };
+    }
     setAiConversationContextWarning(conversationId, pending) {
         const conversation = this.db.get("SELECT id FROM ai_conversations WHERE id = ?", conversationId);
         if (!conversation)
@@ -4157,6 +4182,14 @@ export class Store {
         this.db.run("UPDATE ai_conversations SET compacted_summary = ?, compacted_message_count = ?, context_warning_at = NULL, updated_at = ? WHERE id = ?", summary, Math.max(0, compactedMessageCount), now(), conversationId);
         return this.getAiConversation(conversationId);
     }
+    setAiConversationTitle(conversationId, title) {
+        const conversation = this.db.get("SELECT id FROM ai_conversations WHERE id = ?", conversationId);
+        if (!conversation)
+            throw notFound("AI 对话");
+        const normalizedTitle = title.replace(/\s+/gu, " ").trim().slice(0, 200) || "新对话";
+        this.db.run("UPDATE ai_conversations SET title = ?, updated_at = ? WHERE id = ?", normalizedTitle, now(), conversationId);
+        return this.getAiConversation(conversationId);
+    }
     addAiConversationMessage(conversationId, input) {
         const conversation = this.db.get("SELECT * FROM ai_conversations WHERE id = ?", conversationId);
         if (!conversation)
@@ -4170,7 +4203,7 @@ export class Store {
         const messageId = id("message");
         const timestamp = now();
         const title = requiredString(conversation, "title") === "新对话" && input.role === "user"
-            ? input.content.replace(/\s+/gu, " ").trim().slice(0, 36) || "新对话"
+            ? defaultAiConversationTitle(input.content)
             : requiredString(conversation, "title");
         this.db.transaction(() => {
             this.db.run("INSERT INTO ai_conversation_messages (id, conversation_id, role, content, citations_json, metadata_json, request_id, created_at, created_by_user_id) VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?) ON CONFLICT(conversation_id, request_id) WHERE request_id IS NOT NULL DO NOTHING", messageId, conversationId, input.role, input.content, JSON.stringify(input.citations ?? []), JSON.stringify(input.metadata ?? {}), requestId, timestamp, currentRequestActor()?.userId ?? null);