@musnows/scriverse 0.7.12 → 0.7.13

package/dist/ai.js

@@ -63,6 +63,9 @@ const interactiveStreamErrorCodes = new Set([
 function isInteractiveStreamError(error) {
     return error instanceof AppError && interactiveStreamErrorCodes.has(error.code);
 }
+function isAuthorNoteChapter(chapter) {
+    return String(chapter.chapterType ?? "") === "作者的话";
+}
 function interactiveStreamRequestCancelledError() {
     return new AppError(499, "AI_STREAM_REQUEST_CANCELLED", "AI 流式请求已取消");
 }
@@ -1256,10 +1259,15 @@ export class ContextBuilder {
         if (scope.type === "selection") {
             if (!scope.selection)
                 throw new AppError(400, "SELECTION_REQUIRED", "选中文本上下文不能为空");
+            const selectionChapter = scope.chapterId ? this.store.getChapter(scope.chapterId) : null;
+            if (selectionChapter && selectionChapter.workId !== workId)
+                throw new AppError(400, "CHAPTER_WORK_MISMATCH", "章节不属于当前作品");
             // 分析任务会在 selection 中放入服务端 CHAPTER 标记，不能转义
-            contentSections.push(wrapAiContextRegion("selection", `当前选中文本：\n${scope.selection}`, { escape: false }));
-            if (scope.chapterId)
-                this.appendChapter(contentSections, workId, scope.chapterId, false);
+            if (!selectionChapter || !isAuthorNoteChapter(selectionChapter)) {
+                contentSections.push(wrapAiContextRegion("selection", `当前选中文本：\n${scope.selection}`, { escape: false }));
+                if (scope.chapterId)
+                    this.appendChapter(contentSections, workId, scope.chapterId, false);
+            }
         }
         else if (scope.type === "chapter") {
             const chapterIds = [...new Set([
@@ -1272,8 +1280,9 @@ export class ContextBuilder {
                 this.appendPreviousChapterTail(contentSections, workId, scope.chapterId);
             for (const chapterId of chapterIds)
                 this.appendChapter(contentSections, workId, chapterId, true);
-            if (scope.selection)
+            if (scope.selection && (!scope.chapterId || !isAuthorNoteChapter(this.store.getChapter(scope.chapterId)))) {
                 contentSections.push(wrapAiContextRegion("selection", `当前选中文本（本次修改目标）：\n${scope.selection}`, { escape: false }));
+            }
         }
         else if (scope.type === "volume") {
             const volumeIds = [...new Set([
@@ -1288,7 +1297,7 @@ export class ContextBuilder {
                 const volume = volumes.find((item) => item.id === volumeId);
                 if (!volume)
                     throw notFound("卷");
-                const chapters = volume.chapters;
+                const chapters = volume.chapters.filter((chapter) => !isAuthorNoteChapter(chapter));
                 contentSections.push(wrapAiContextRegion("volume", `当前卷：${String(volume.title)}`));
                 for (const chapter of chapters) {
                     contentSections.push(wrapAiContextRegion("chapter", `[${String(volume.title)} / ${String(chapter.title)} | 版本 ${String(chapter.versionNo)}]\n${String(chapter.content)}`));
@@ -1300,7 +1309,7 @@ export class ContextBuilder {
             const volumes = tree.volumes;
             contentSections.push(wrapAiContextRegion("book", "全书正文（按问题相关度选取原文，完整结构见章节概要）："));
             for (const volume of volumes) {
-                for (const chapter of volume.chapters) {
+                for (const chapter of volume.chapters.filter((item) => !isAuthorNoteChapter(item))) {
                     contentSections.push(wrapAiContextRegion("chapter", `[# ${String(volume.title)} / ${String(chapter.title)} | 版本 ${String(chapter.versionNo)}]\n${String(chapter.content)}`));
                 }
             }
@@ -1384,8 +1393,9 @@ export class ContextBuilder {
                 if (chapter.workId !== workId)
                     throw new AppError(400, "CHAPTER_WORK_MISMATCH", "引用章节不属于当前作品");
             }
-            if (chapters.length) {
-                contentSections.push(wrapAiContextRegion("referenced_chapters", `作者主动引用的章节：\n${chapters
+            const eligibleChapters = chapters.filter((chapter) => !isAuthorNoteChapter(chapter));
+            if (eligibleChapters.length) {
+                contentSections.push(wrapAiContextRegion("referenced_chapters", `作者主动引用的章节：\n${eligibleChapters
                     .map((chapter) => `[${String(chapter.title)} | 版本 ${String(chapter.versionNo)}]\n${String(chapter.content)}`)
                     .join("\n\n")}`));
             }
@@ -1490,6 +1500,8 @@ export class ContextBuilder {
         const chapter = this.store.getChapter(chapterId);
         if (chapter.workId !== workId)
             throw new AppError(400, "CHAPTER_WORK_MISMATCH", "章节不属于当前作品");
+        if (isAuthorNoteChapter(chapter))
+            return;
         sections.push(wrapAiContextRegion("chapter", includeContent
             ? `当前章节：${String(chapter.title)} | 版本 ${String(chapter.versionNo)}\n${String(chapter.content)}`
             : `所在章节：${String(chapter.title)} | 版本 ${String(chapter.versionNo)}`));
@@ -1503,7 +1515,7 @@ export class ContextBuilder {
             return;
         const perVolumeBudget = Math.max(24, Math.floor(maximumTokens / volumes.length));
         for (const volume of volumes) {
-            const chapters = volume.chapters;
+            const chapters = volume.chapters.filter((chapter) => !isAuthorNoteChapter(chapter));
             const ranked = chapters.map((chapter, order) => {
                 const summary = summaryByChapterId.get(String(chapter.id)) ?? "";
                 const line = `- ${String(chapter.title)}：${summary || "尚无章节概要"}`;
@@ -1532,9 +1544,12 @@ export class ContextBuilder {
         }
     }
     appendPreviousChapterTail(sections, workId, chapterId) {
+        const current = this.store.getChapter(chapterId);
+        if (current.workId !== workId || isAuthorNoteChapter(current))
+            return;
         const tree = this.store.getWorkTree(workId);
         const chapters = tree.volumes
-            .flatMap((volume) => volume.chapters);
+            .flatMap((volume) => volume.chapters.filter((chapter) => !isAuthorNoteChapter(chapter)));
         const index = chapters.findIndex((chapter) => chapter.id === chapterId);
         if (index <= 0)
             return;
@@ -1545,6 +1560,9 @@ export class ContextBuilder {
         sections.push(wrapAiContextRegion("previous_chapter_tail", `上一章节结尾：${String(previous.title)} | 版本 ${String(previous.versionNo)}\n${content.slice(-5000)}`));
     }
     appendChapterKnowledge(sections, workId, chapterId) {
+        const chapter = this.store.getChapter(chapterId);
+        if (chapter.workId !== workId || isAuthorNoteChapter(chapter))
+            return;
         const outline = this.store.getChapterOutline(chapterId);
         if (outline) {
             sections.push(wrapAiContextRegion("chapter_outline", `当前章大纲（创作约束）：\n目标：${String(outline.goal) || "未填写"}\n冲突：${String(outline.conflict) || "未填写"}\n转折：${String(outline.turningPoint) || "未填写"}\n状态：${String(outline.status)}`));
@@ -3434,7 +3452,26 @@ export class AiManager {
             && titleModelId
             && (conversationBefore?.title === "新对话" || conversationBefore?.title === defaultTitle));
         const processStartedAt = process.hrtime.bigint();
-        const generated = await this.generate({ ...input, taskType: "chat" }, onDelta);
+        let persistedConversationMessage = null;
+        let streamedConversationContent = "";
+        const persistStreamDelta = (delta) => {
+            if (input.conversationId && input.assistantMessageRequestId && delta.length > 0) {
+                streamedConversationContent += delta;
+                persistedConversationMessage = this.store.upsertAiConversationAssistantMessage(input.conversationId, input.assistantMessageRequestId, streamedConversationContent);
+            }
+            onDelta(delta);
+        };
+        let generated;
+        try {
+            generated = await this.generate({ ...input, taskType: "chat" }, persistStreamDelta);
+        }
+        catch (error) {
+            if (persistedConversationMessage && input.conversationId && input.assistantMessageRequestId) {
+                const interruptionCode = error instanceof AppError ? error.code : "AI_STREAM_FAILED";
+                persistedConversationMessage = this.store.upsertAiConversationAssistantMessage(input.conversationId, input.assistantMessageRequestId, streamedConversationContent, { interrupted: true, interruptionCode }, true);
+            }
+            throw error;
+        }
         const processDurationMs = Math.min(86_400_000, Math.max(0, Math.round(Number(process.hrtime.bigint() - processStartedAt) / 1_000_000)));
         const chapter = input.scope.chapterId ? this.store.getChapter(input.scope.chapterId) : null;
         const suggestionId = id("suggestion");
@@ -3442,22 +3479,17 @@ export class AiManager {
        source_text, content, action, status, created_at, created_by_user_id) VALUES (?, ?, ?, ?, ?, 'chat', ?, ?, ?, 'note', 'pending', ?, ?)`, suggestionId, generated.callId, input.workId, chapter ? String(chapter.id) : null, chapter ? Number(chapter.versionNo) : null, input.instruction, input.scope.selection ?? "", generated.content, now(), currentRequestActor()?.userId ?? null);
         const modelDisplayName = typeof generated.model.displayName === "string" ? generated.model.displayName : undefined;
         const conversationMessage = input.conversationId && input.assistantMessageRequestId
-            ? this.store.addAiConversationMessage(input.conversationId, {
-                role: "assistant",
-                content: generated.content,
-                requestId: input.assistantMessageRequestId,
-                metadata: {
-                    ...(modelDisplayName ? { modelDisplayName } : {}),
-                    outputTokens: generated.outputTokens,
-                    processDurationMs,
-                    ...(generated.reasoningContent === undefined ? {} : { reasoningContent: generated.reasoningContent }),
-                    ...(generated.cacheHitPercent === undefined ? {} : { cacheHitPercent: generated.cacheHitPercent }),
-                    toolCalls: generated.toolCalls,
-                    processSteps: generated.processSteps,
-                    ...(generated.anthropicContent?.length ? { anthropicContent: generated.anthropicContent } : {})
-                }
-            })
-            : null;
+            ? this.store.upsertAiConversationAssistantMessage(input.conversationId, input.assistantMessageRequestId, generated.content, {
+                ...(modelDisplayName ? { modelDisplayName } : {}),
+                outputTokens: generated.outputTokens,
+                processDurationMs,
+                ...(generated.reasoningContent === undefined ? {} : { reasoningContent: generated.reasoningContent }),
+                ...(generated.cacheHitPercent === undefined ? {} : { cacheHitPercent: generated.cacheHitPercent }),
+                toolCalls: generated.toolCalls,
+                processSteps: generated.processSteps,
+                ...(generated.anthropicContent?.length ? { anthropicContent: generated.anthropicContent } : {})
+            }, true)
+            : persistedConversationMessage;
         if (shouldGenerateTitle && conversationMessage && input.conversationId) {
             void this.generateConversationTitle(input.workId, input.conversationId, titleModelId, firstUserContent, generated.content, defaultTitle).catch((error) => {
                 logger.warn("ai.conversation_title.failed", { workId: input.workId, conversationId: input.conversationId, error: aiErrorForLog(error) });
@@ -4606,7 +4638,7 @@ export class AiManager {
             };
         }
         const args = parsed.data;
-        const scopedChapterIds = scope && (scope.type === "chapter" || scope.type === "volume")
+        const scopedChapterIds = scope && (scope.type === "chapter" || scope.type === "volume" || scope.type === "book")
             ? new Set(this.getScopeChapters(workId, scope).map((chapter) => String(chapter.id)))
             : null;
         if (name === "recall_relationship") {
@@ -4820,7 +4852,7 @@ export class AiManager {
                     .map((item) => item.trim()).filter(Boolean).slice(0, 10);
                 const seenParagraphs = new Set();
                 for (const identityTerm of identityTerms) {
-                    for (const paragraph of this.store.searchChapterParagraphs(workId, identityTerm, 50)) {
+                    for (const paragraph of this.store.searchChapterParagraphs(workId, identityTerm, 50, { excludeAuthorNotes: true })) {
                         const key = `${String(paragraph.chapterId)}:${String(paragraph.paragraph)}`;
                         if (seenParagraphs.has(key))
                             continue;
@@ -4855,7 +4887,7 @@ export class AiManager {
         if (name === "story_index") {
             const { offset, limit, cursor } = args;
             const work = this.store.getWork(workId);
-            const chapterPage = this.store.getStoryIndexChapterPage(workId, offset, limit);
+            const chapterPage = this.store.getStoryIndexChapterPage(workId, offset, limit, { excludeAuthorNotes: true });
             const workRecords = structuralToolResultRecords([{
                     id: work.id,
                     title: work.title,
@@ -4914,6 +4946,8 @@ export class AiManager {
                     const chapter = this.store.getChapter(chapterId);
                     if (chapter.workId !== workId)
                         return { chapterId, error: { code: "CHAPTER_WORK_MISMATCH", message: "The requested chapter belongs to a different work." } };
+                    if (isAuthorNoteChapter(chapter))
+                        return { chapterId, error: { code: "CHAPTER_AUTHOR_NOTE_EXCLUDED", message: "Author notes are excluded from AI context." } };
                     const content = collapseAiBlankLines(String(chapter.content));
                     return { chapterId, title: chapter.title, versionNo: chapter.versionNo, ...(include !== "content" ? { summary: summaries.get(chapterId) ?? "" } : {}), ...(include !== "summary" ? { content } : {}) };
                 }
@@ -4931,7 +4965,7 @@ export class AiManager {
         }
         if (name === "grep") {
             const { keyword, limit, cursor } = args;
-            const matches = this.store.searchChapterParagraphs(workId, keyword, limit)
+            const matches = this.store.searchChapterParagraphs(workId, keyword, limit, { excludeAuthorNotes: true })
                 .filter((match) => !scopedChapterIds || scopedChapterIds.has(String(match.chapterId)));
             const records = structuralToolResultRecords(matches, maximumRecordChars);
             const result = paginateToolResultRecords(records, cursor, (page, pagination) => ({
@@ -4960,6 +4994,15 @@ export class AiManager {
                     : sourceType === "chapter-outline" ? "outline" : sourceType;
                 if (!requestedCategories.has(type))
                     return [];
+                if (sourceType === "chapter") {
+                    try {
+                        if (isAuthorNoteChapter(this.store.getChapter(String(item.id))))
+                            return [];
+                    }
+                    catch {
+                        return [];
+                    }
+                }
                 return [{
                         ...item,
                         ...this.hybridAiSearchDetails(workId, sourceType, String(item.id)),
@@ -7187,7 +7230,14 @@ export class AiManager {
                 if (String(chapter.workId) !== workId)
                     throw new AppError(400, "CHAPTER_WORK_MISMATCH", "章节不属于当前作品");
             }
-            return new Set(chapterIds);
+            return new Set(chapterIds.filter((chapterId) => {
+                try {
+                    return !isAuthorNoteChapter(this.store.getChapter(chapterId));
+                }
+                catch {
+                    return false;
+                }
+            }));
         }
         if (scope.type === "volume") {
             const volumeIds = [...new Set([
@@ -7215,6 +7265,8 @@ export class AiManager {
                 const chapter = this.store.getChapter(sourceId);
                 if (String(chapter.workId) !== workId)
                     return null;
+                if (isAuthorNoteChapter(chapter))
+                    return null;
                 return {
                     sourceType,
                     sourceId,
@@ -8121,8 +8173,7 @@ export class AiManager {
                 return null;
             if (scope.type === "volume" && scope.volumeId !== String(chapter.volume_id))
                 return null;
-            if ((scope.type === "book" || scope.type === "volume")
-                && (Boolean(chapter.excluded_from_analysis) || String(chapter.chapter_type) === "作者的话"))
+            if (Boolean(chapter.excluded_from_analysis) || String(chapter.chapter_type) === "作者的话")
                 return null;
             return String(chapter.version_no);
         }
@@ -9107,7 +9158,7 @@ export class AiManager {
         });
     }
     isAutomaticAnalysisChapter(chapter) {
-        return !chapter.excludedFromAnalysis && chapter.chapterType !== "作者的话";
+        return !chapter.excludedFromAnalysis && !isAuthorNoteChapter(chapter);
     }
     buildChapterChunks(chapters, maximumChars = 10_000) {
         const chunks = [];