@musnows/scriverse 0.9.5 → 0.9.7

package/dist/app.js

@@ -17,9 +17,10 @@ import { DEFAULT_AI_CHAT_TAB_LIMIT } from "./ai-chat-tab-limit.js";
 import { MAX_AI_STREAM_IDLE_TIMEOUT_SECONDS, MIN_AI_STREAM_IDLE_TIMEOUT_SECONDS, normalizeAiStreamIdleTimeoutSeconds } from "./ai-stream-timeout.js";
 import { AttachmentStorage } from "./attachment-storage.js";
 import { attachmentDownloadFileName, inlineContentDisposition } from "./attachment-download.js";
-import { AiManager } from "./ai.js";
+import { AI_MODEL_KINDS, AiManager } from "./ai.js";
 import { LiteLlmPriceCache } from "./ai-model-pricing.js";
 import { resolveMaxAgentToolCallLimit } from "./ai-tool-results.js";
+import { SEMANTIC_SOURCE_TYPES } from "./semantic-search.js";
 import { CHARACTER_EXTRACTION_MAX_ALIASES, CHARACTER_EXTRACTION_MAX_CANDIDATES, CHARACTER_EXTRACTION_MAX_IDENTITY_LENGTH, CHARACTER_EXTRACTION_MAX_NAME_LENGTH, CHARACTER_EXTRACTION_MAX_SPECIES_LENGTH } from "./character-extraction.js";
 import { AiWritePlanManager, AI_USER_QUESTION_STATUSES, AI_WRITE_PLAN_STATUSES, aiWriteToolDescriptions, aiWriteToolLabels, aiWriteToolsUpdateSchema, answerAiUserQuestionSchema, resolveAiWritePlanMaxOperations } from "./ai-write-plans.js";
 import { CredentialVault } from "./credential-vault.js";
@@ -32,6 +33,7 @@ import { isOfficialGoogleVertexBaseUrl, parseGoogleServiceAccount } from "./goog
 import { HYBRID_SEARCH_TYPES, MAXIMUM_WORK_SEARCH_QUERY_LENGTH, readableHybridSearchTypes } from "./hybrid-search.js";
 import { applyImportFileHints, parseNovelText } from "./parser.js";
 import { MAX_CHAPTER_LINE_IDS } from "./chapter-annotation-anchor.js";
+import { isChapterNumberTemplate } from "./chapter-title-numbering.js";
 import { aiConversationTaskTypes, attachmentPermissionModules, RECYCLE_BIN_RETENTION_DAYS, Store, versionedEntityTypes, WORK_AGENT_TOOL_IDS } from "./store.js";
 import { composeRoleplayStoredUserContent } from "./roleplay-turn.js";
 import { ROLEPLAY_MEMORY_CATEGORIES, ROLEPLAY_MEMORY_CERTAINTY, ROLEPLAY_MEMORY_IMPORTANCE, ROLEPLAY_MEMORY_STATUSES } from "./roleplay-memory.js";
@@ -485,6 +487,7 @@ const providerUpdateSchema = providerBaseSchema.partial().superRefine((value, ct
 const modelSchema = z.object({
     displayName: nonEmpty.max(200),
     modelId: nonEmpty.max(300),
+    modelKind: z.enum(AI_MODEL_KINDS).optional(),
     purposes: optionalStrings,
     contextNote: z.string().max(10_000).optional(),
     contextWindow: z.number().int().min(32_768, "模型上下文不能低于 32768 Token").max(2_000_000).optional(),
@@ -503,7 +506,7 @@ const aiPromptSchema = z.object({
     streamIdleTimeoutSeconds: z.number().int().min(MIN_AI_STREAM_IDLE_TIMEOUT_SECONDS).max(MAX_AI_STREAM_IDLE_TIMEOUT_SECONDS).optional()
 });
 const aiUsageQuerySchema = z.object({
-    timezoneOffset: z.coerce.number().int().min(-840).max(840).default(0)
+    timezoneOffset: z.coerce.number().int().min(-840).max(840).optional()
 }).strict();
 const adminAiConversationQuerySchema = z.object({
     page: z.string().optional(),
@@ -651,11 +654,38 @@ const workAiSettingsSchema = z.object({
     titleGenerationModelId: z.string().trim().max(200).optional(),
     imageToolModelId: identifier.nullable().optional()
 }).strict();
+const semanticSearchSettingsSchema = z.object({
+    enabled: z.boolean().optional(),
+    embeddingModelId: identifier.nullable().optional(),
+    rerankModelId: identifier.nullable().optional(),
+    vectorDimension: z.number().int().min(1).max(65_536).optional(),
+    recallLimit: z.number().int().min(1).max(200).optional(),
+    resultLimit: z.number().int().min(1).max(100).optional(),
+    budgetTokens: z.number().int().min(256).max(100_000).optional(),
+    channelWeight: z.number().min(0.1).max(5).optional()
+}).strict();
+const semanticSearchRequestSchema = z.object({
+    query: z.string().trim().min(1).max(2_000),
+    types: z.array(z.enum(SEMANTIC_SOURCE_TYPES)).max(SEMANTIC_SOURCE_TYPES.length).optional(),
+    limit: z.number().int().min(1).max(100).optional(),
+    includeKeyword: z.boolean().optional(),
+    currentChapterId: identifier.optional(),
+    selection: z.string().max(4_000).optional()
+}).strict();
+const semanticSnapshotSchema = z.object({
+    query: z.string().trim().min(1).max(2_000),
+    entryIds: z.array(identifier).min(1).max(30),
+    scope: jsonObject.optional(),
+    conversationId: identifier.optional()
+}).strict();
 const contextSchema = z.object({
     type: z.enum(["none", "selection", "chapter", "volume", "book", "settings-catalog", "entities"]),
     chapterId: identifier.optional(),
     volumeId: identifier.optional(),
     selection: z.string().max(200_000).optional(),
+    selectionStart: z.number().int().min(0).max(10_000_000).optional(),
+    selectionEnd: z.number().int().min(0).max(10_000_000).optional(),
+    writingChapterVersion: z.number().int().min(1).max(2_000_000_000).optional(),
     chapterIds: z.array(identifier).max(20).optional(),
     volumeIds: z.array(identifier).max(20).optional(),
     characterIds: optionalStrings,
@@ -663,6 +693,7 @@ const contextSchema = z.object({
     settingIds: optionalStrings,
     raceIds: optionalStrings,
     organizationIds: optionalStrings,
+    semanticSnapshotId: identifier.optional(),
     includeBookSummary: z.boolean().optional(),
     includeSettingInfo: z.boolean().optional()
 });
@@ -1021,7 +1052,7 @@ function redactAiCallContext(record, permissions) {
     const redactedScope = { ...scope };
     let restricted = false;
     if (permissions.prose === "none") {
-        for (const field of ["selection", "chapterId", "volumeId", "chapterIds", "includeBookSummary"]) {
+        for (const field of ["selection", "selectionStart", "selectionEnd", "writingChapterVersion", "chapterId", "volumeId", "chapterIds", "includeBookSummary"]) {
             if (field in redactedScope) {
                 delete redactedScope[field];
                 restricted = true;
@@ -1977,10 +2008,13 @@ export function createRuntime(options) {
     });
     app.get("/api/works/:workId/chapter-annotations", (request, response) => {
         const pagination = parsePagination(request.query);
+        const chapterId = request.query.chapterId === undefined ? undefined : parse(identifier, request.query.chapterId);
+        const query = request.query.q === undefined ? undefined : parse(z.string().trim().max(100), request.query.q);
+        const filters = { chapterId, query };
         const permissions = requestPermissions(request, String(request.params.workId));
         data(response, pagination
-            ? store.listWorkChapterAnnotationsPage(request.params.workId, pagination, readableChapterAnnotationKinds(permissions))
-            : store.listWorkChapterAnnotations(request.params.workId, readableChapterAnnotationKinds(permissions)));
+            ? store.listWorkChapterAnnotationsPage(request.params.workId, pagination, readableChapterAnnotationKinds(permissions), filters)
+            : store.listWorkChapterAnnotations(request.params.workId, readableChapterAnnotationKinds(permissions), filters));
     });
     app.post("/api/chapters/:chapterId/annotations", (request, response) => {
         const input = parse(z.object({
@@ -2017,6 +2051,12 @@ export function createRuntime(options) {
             z.object({ type: z.literal("move"), volumeId: identifier }).strict(),
             z.object({ type: z.literal("setType"), chapterType: chapterTypeSchema }).strict(),
             z.object({ type: z.literal("setAnalysisExclusion"), excludedFromAnalysis: z.boolean() }).strict(),
+            z.object({
+                type: z.literal("renumberTitles"),
+                template: z.string().min(1).max(50).refine(isChapterNumberTemplate, "标题格式必须且只能包含一个 {n} 占位符，且不能包含换行或控制字符"),
+                numberStyle: z.enum(["arabic", "chinese"]),
+                startAt: z.number().int().min(1).max(999_999)
+            }).strict(),
             z.object({ type: z.literal("delete") }).strict()
         ]);
         const input = parse(z.object({ chapters: selectedChapters, action }).strict(), request.body);
@@ -2779,8 +2819,8 @@ export function createRuntime(options) {
         data(response, settings);
     });
     app.get("/api/platform/ai/usage", (request, response) => {
-        const query = parse(aiUsageQuerySchema, request.query);
-        data(response, ai.getPlatformTokenUsage(query.timezoneOffset));
+        parse(aiUsageQuerySchema, request.query);
+        data(response, ai.getPlatformTokenUsage());
     });
     app.post("/api/platform/ai/usage/pricing/refresh", async (request, response) => {
         if (!request.authUser)
@@ -2846,13 +2886,33 @@ export function createRuntime(options) {
         data(response, { ...queued, queuedAt: new Date().toISOString() }, 202);
     });
     app.get("/api/works/:workId/ai-settings", (request, response) => data(response, store.getWorkAiSettings(request.params.workId)));
+    app.get("/api/works/:workId/ai-settings/mcp-servers", (request, response) => {
+        data(response, ai.getRemoteMcpSettings(request.params.workId));
+    });
+    app.put("/api/works/:workId/ai-settings/mcp-servers", async (request, response) => {
+        data(response, await ai.updateRemoteMcpSettings(request.params.workId, request.body));
+    });
     app.get("/api/works/:workId/ai-settings/usage", (request, response) => {
-        const query = parse(aiUsageQuerySchema, request.query);
-        data(response, ai.getWorkTokenUsage(request.params.workId, query.timezoneOffset));
+        parse(aiUsageQuerySchema, request.query);
+        data(response, ai.getWorkTokenUsage(request.params.workId));
     });
     app.get("/api/works/:workId/ai-settings/relationship-search-index", (request, response) => {
         data(response, ai.getRelationshipSearchIndexStatus(request.params.workId));
     });
+    app.get("/api/works/:workId/ai-settings/semantic-search-index", (request, response) => {
+        data(response, ai.getSemanticSearchIndexStatus(request.params.workId));
+    });
+    app.patch("/api/works/:workId/ai-settings/semantic-search", async (request, response) => {
+        data(response, await ai.updateSemanticSearchSettings(request.params.workId, parse(semanticSearchSettingsSchema, request.body)));
+    });
+    app.post("/api/works/:workId/ai-settings/semantic-search-index/sync", (request, response) => {
+        parse(z.object({}).strict(), request.body ?? {});
+        data(response, ai.syncSemanticSearchIndex(request.params.workId), 202);
+    });
+    app.post("/api/works/:workId/ai-settings/semantic-search-index/rebuild", (request, response) => {
+        parse(z.object({}).strict(), request.body ?? {});
+        data(response, ai.rebuildSemanticSearchIndex(request.params.workId), 202);
+    });
     app.post("/api/works/:workId/ai-settings/relationship-search-index/sync", (request, response) => {
         const workId = request.params.workId;
         const result = ai.syncRelationshipSearchIndex(workId);
@@ -3032,12 +3092,24 @@ export function createRuntime(options) {
             ignoreContextWarning: z.boolean().optional()
         }).strict(), request.body ?? {});
         const conversation = store.getAiConversation(request.params.conversationId);
+        const permissions = requestPermissions(request, String(conversation.workId));
+        const writingSkillRequest = conversation.roleplayCharacter
+            ? { skillName: null, instruction: input.instruction }
+            : ai.resolveWritingSkillInstruction(input.instruction);
+        if (writingSkillRequest.skillName && !canReadWorkModule(permissions, "prose")) {
+            throw new AppError(403, "WORK_MODULE_READ_DENIED", "你没有读取当前章节正文以使用写作 Skill 的权限");
+        }
+        const citedInstruction = instructionWithCitations(writingSkillRequest.instruction, input.citations ?? []);
+        const preparedScope = conversation.roleplayCharacter
+            ? input.scope
+            : ai.resolveWritingSkillScope(String(conversation.workId), "chat", input.instruction, input.scope);
         data(response, await ai.prepareConversationContext({
             conversationId: request.params.conversationId,
             workId: String(conversation.workId),
             modelId: input.modelId,
-            scope: input.scope,
-            instruction: instructionWithCitations(input.instruction, input.citations ?? [])
+            scope: preparedScope,
+            instruction: citedInstruction,
+            skillInstruction: input.instruction
         }, { ignoreWarning: input.ignoreContextWarning === true }));
     });
     app.post("/api/ai-conversations/:conversationId/compact", async (request, response) => {
@@ -3226,6 +3298,9 @@ export function createRuntime(options) {
         const pagination = parsePagination(request.query);
         data(response, pagination ? ai.listWorkModelsPage(request.params.workId, pagination) : ai.listWorkModels(request.params.workId));
     });
+    app.get("/api/works/:workId/semantic-models", (request, response) => {
+        data(response, ai.listWorkSemanticModels(request.params.workId));
+    });
     app.get("/api/works/:workId/task-defaults", (request, response) => {
         const pagination = parsePagination(request.query);
         data(response, pagination ? ai.listTaskDefaultsPage(request.params.workId, pagination) : ai.listTaskDefaults(request.params.workId));
@@ -3381,10 +3456,19 @@ export function createRuntime(options) {
         const storedUserContent = isRoleplay
             ? composeRoleplayStoredUserContent(sceneDirection, instructionText)
             : instructionText;
-        const resolvedInstruction = instructionWithCitations(instructionText, citations);
+        const permissions = requestPermissions(request, request.params.workId);
+        const writingSkillRequest = isRoleplay
+            ? { skillName: null, instruction: instructionText }
+            : ai.resolveWritingSkillInstruction(instructionText);
+        const resolvedInstruction = instructionWithCitations(writingSkillRequest.instruction, citations);
+        if (writingSkillRequest.skillName && !canReadWorkModule(permissions, "prose")) {
+            throw new AppError(403, "WORK_MODULE_READ_DENIED", "你没有读取当前章节正文以使用写作 Skill 的权限");
+        }
+        const requestScope = isRoleplay
+            ? input.scope
+            : ai.resolveWritingSkillScope(request.params.workId, "chat", instructionText, input.scope);
         const mentionSource = [sceneDirection, resolvedInstruction].filter(Boolean).join("\n");
         const modelId = resolveConversationModelId(request.params.workId, conversationId, input.modelId);
-        const permissions = requestPermissions(request, request.params.workId);
         const controller = new AbortController();
         let streamRequestId = null;
         let streamRequestFinished = false;
@@ -3439,8 +3523,9 @@ export function createRuntime(options) {
                     conversationId,
                     workId: request.params.workId,
                     modelId,
-                    scope: input.scope,
+                    scope: requestScope,
                     instruction: mentionSource,
+                    skillInstruction: instructionText,
                     excludeConversationMessageId: input.currentMessageId
                 }, { ignoreWarning: input.ignoreContextWarning === true });
                 preparedConversation = redactAiConversation(store.getAiConversationSummary(conversationId), permissions);
@@ -3461,12 +3546,12 @@ export function createRuntime(options) {
                 }
             }
             const resolvedScope = input.currentMessageId
-                ? input.scope
+                ? requestScope
                 : ai.resolveInstructionMentions({
                     workId: request.params.workId,
                     taskType: "chat",
                     instruction: mentionSource,
-                    scope: input.scope,
+                    scope: requestScope,
                     conversationId
                 });
             const mentionCharacterIds = [...new Set([
@@ -3485,11 +3570,12 @@ export function createRuntime(options) {
                     content: storedUserContent,
                     citations,
                     ...(input.currentMessageId ? { existingMessageId: input.currentMessageId } : {}),
-                    ...((modelId || mentionCharacterIds.length || mentionRaceIds.length || mentionOrganizationIds.length || input.imageAttachmentIds?.length) ? { metadata: {
+                    ...((modelId || mentionCharacterIds.length || mentionRaceIds.length || mentionOrganizationIds.length || input.imageAttachmentIds?.length || input.scope.semanticSnapshotId) ? { metadata: {
                             ...(modelId ? { modelId } : {}),
                             ...(mentionCharacterIds.length ? { mentionCharacterIds } : {}),
                             ...(mentionRaceIds.length ? { mentionRaceIds } : {}),
                             ...(mentionOrganizationIds.length ? { mentionOrganizationIds } : {}),
+                            ...(input.scope.semanticSnapshotId ? { semanticSnapshotId: input.scope.semanticSnapshotId } : {}),
                             ...(input.imageAttachmentIds?.length ? { chatImageAttachmentIds: [...new Set(input.imageAttachmentIds)] } : {})
                         } } : {})
                 }
@@ -3504,6 +3590,15 @@ export function createRuntime(options) {
                 }
                 if (begun.request.status === "completed" && begun.assistantMessage) {
                     const content = String(begun.assistantMessage.content ?? "");
+                    const assistantMetadata = begun.assistantMessage.metadata && typeof begun.assistantMessage.metadata === "object"
+                        ? begun.assistantMessage.metadata
+                        : {};
+                    const writingSuggestionId = typeof assistantMetadata.writingSuggestionId === "string"
+                        ? assistantMetadata.writingSuggestionId
+                        : "";
+                    const writingSuggestion = writingSuggestionId
+                        ? redactSuggestion(ai.getSuggestion(writingSuggestionId), permissions)
+                        : null;
                     if (content)
                         sendEvent("delta", { delta: content, replayed: true });
                     sendEvent("complete", {
@@ -3511,7 +3606,8 @@ export function createRuntime(options) {
                         conversationId,
                         conversationTitle: store.getAiConversationSummary(conversationId).title,
                         messageId: begun.assistantMessage.id,
-                        messageCreatedAt: begun.assistantMessage.createdAt
+                        messageCreatedAt: begun.assistantMessage.createdAt,
+                        ...(writingSuggestion ? { writingSuggestion } : {})
                     });
                 }
                 else {
@@ -3543,9 +3639,10 @@ export function createRuntime(options) {
             const suggestion = await ai.createStreamingChat({
                 workId: request.params.workId,
                 instruction: resolvedInstruction,
+                skillInstruction: instructionText,
                 ...(sceneDirection ? { sceneDirection } : {}),
                 // 仍由生成路径基于原始范围持久化累计注入，保证预解析不会吞掉本轮自动命中。
-                scope: input.scope,
+                scope: requestScope,
                 signal: controller.signal,
                 onToolCall: (toolCall, round) => sendEvent("tool_call", { ...toolCall, round }),
                 onProcessStep: (step) => sendEvent("process_step", step),
@@ -3578,6 +3675,7 @@ export function createRuntime(options) {
                 contextUsage: suggestion.contextUsage,
                 conversationId,
                 conversationTitle: suggestion.conversationTitle,
+                ...(suggestion.action !== "note" ? { writingSuggestion: redactSuggestion(suggestion, permissions) } : {}),
                 messageId: typeof suggestion.conversationMessage === "object" && suggestion.conversationMessage !== null
                     ? suggestion.conversationMessage.id
                     : undefined,
@@ -3640,6 +3738,11 @@ export function createRuntime(options) {
     });
     app.post("/api/suggestions/:suggestionId/accept", (request, response) => {
         const input = parse(z.object({ content: z.string().max(2_000_000).optional() }), request.body ?? {});
+        const suggestion = ai.getSuggestion(request.params.suggestionId);
+        const permissions = requestPermissions(request, String(suggestion.workId));
+        if (!canWriteWorkModule(permissions, "prose")) {
+            throw new AppError(403, "WORK_MODULE_WRITE_DENIED", "你没有将 AI 建议写入正文的权限");
+        }
         data(response, ai.acceptSuggestion(request.params.suggestionId, input.content));
     });
     app.post("/api/suggestions/:suggestionId/reject", (request, response) => {
@@ -3668,6 +3771,27 @@ export function createRuntime(options) {
             conversationOwnerUserId: request.authUser?.userId
         }));
     });
+    app.post("/api/works/:workId/semantic-search", async (request, response) => {
+        const input = parse(semanticSearchRequestSchema, request.body);
+        const permissions = requestPermissions(request, request.params.workId);
+        const allowedTypes = readableHybridSearchTypes(permissions)
+            .filter((type) => SEMANTIC_SOURCE_TYPES.includes(type));
+        data(response, await ai.semanticSearchStory(request.params.workId, input.query, {
+            allowedTypes,
+            types: input.types,
+            limit: input.limit,
+            includeKeyword: input.includeKeyword,
+            conversationOwnerUserId: request.authUser?.userId,
+            currentChapterId: input.currentChapterId,
+            selection: input.selection
+        }));
+    });
+    app.post("/api/works/:workId/semantic-search/snapshots", (request, response) => {
+        const input = parse(semanticSnapshotSchema, request.body);
+        if (input.conversationId)
+            assertRequestAiConversationOwner(request, input.conversationId);
+        data(response, ai.createSemanticContextSnapshot(request.params.workId, input), 201);
+    });
     app.head("/api/works/:workId/export", (request, response) => {
         parse(z.enum(["epub"]), request.query.format ?? "epub");
         store.getWork(request.params.workId);