@musnows/scriverse 1.0.2 → 1.0.4

package/dist/app.js

@@ -1056,7 +1056,7 @@ function redactAiCallContext(record, permissions) {
     const redactedScope = { ...scope };
     let restricted = false;
     if (permissions.prose === "none") {
-        for (const field of ["selection", "selectionStart", "selectionEnd", "writingChapterVersion", "chapterId", "volumeId", "chapterIds", "includeBookSummary"]) {
+        for (const field of ["selection", "selectionStart", "selectionEnd", "writingChapterVersion", "chapterId", "volumeId", "chapterIds", "volumeIds", "includeBookSummary"]) {
             if (field in redactedScope) {
                 delete redactedScope[field];
                 restricted = true;
@@ -1126,6 +1126,10 @@ function redactAiConversationMessage(item, permissions) {
             delete readableMetadata.mentionRaceIds;
         if (permissions.organizations === "none")
             delete readableMetadata.mentionOrganizationIds;
+        if (permissions.settings === "none")
+            delete readableMetadata.mentionSettingIds;
+        if (permissions.settings === "none")
+            delete readableMetadata.mentionContextSettingIds;
         if (Object.keys(readableMetadata).length === Object.keys(metadata).length)
             return item;
         return { ...message, metadata: readableMetadata };
@@ -1207,11 +1211,14 @@ export function publicAiStreamError(error) {
                 "monthStartedAt"
             ].filter((key) => details[key] !== undefined).map((key) => [key, details[key]]))
             : undefined;
+        const publicPendingQuestionDetails = error.code === "AI_QUESTION_PENDING" && typeof details?.questionId === "string"
+            ? { questionId: details.questionId }
+            : undefined;
         return {
             code: error.code,
             message: error.message,
             status: error.status,
-            ...(publicQuotaDetails ? { details: publicQuotaDetails } : {}),
+            ...(publicQuotaDetails || publicPendingQuestionDetails ? { details: publicQuotaDetails ?? publicPendingQuestionDetails } : {}),
             ...((error.status < 500 || error.code === "AI_CALL_FAILED") && typeof details?.failure === "string" ? { failure: details.failure } : {}),
             ...(typeof details?.callId === "string" ? { callId: details.callId } : {}),
             ...(typeof details?.providerName === "string" ? { providerName: details.providerName } : {}),
@@ -3434,6 +3441,9 @@ export function createRuntime(options) {
                 interrupted: z.boolean().optional(),
                 interruptionCode: z.string().max(100).optional(),
                 interruptionMessage: z.string().max(500).optional(),
+                errorCode: z.string().max(100).optional(),
+                errorStatus: z.number().int().min(100).max(599).optional(),
+                pendingQuestionId: identifier.optional(),
                 toolCalls: z.array(aiToolCallResultSchema).max(12).optional(),
                 processSteps: z.array(aiProcessStepSchema).max(50).optional()
             }).optional()
@@ -3568,7 +3578,7 @@ export function createRuntime(options) {
     app.get("/api/works/:workId/ai/questions/:questionId", (request, response) => {
         data(response, aiWritePlanManager.getQuestion(request.params.questionId, request.params.workId, planViewer()));
     });
-    const resumeQuestionWorkflow = async (questionId, workId, viewer) => {
+    const resumeQuestionWorkflow = async (questionId, workId, viewer, stream = {}) => {
         const continuation = aiWritePlanManager.claimQuestionContinuation(questionId, workId, viewer);
         if (!continuation)
             return;
@@ -3599,26 +3609,88 @@ export function createRuntime(options) {
                     : {}),
                 ...(typeof continuation.round === "number" ? { round: continuation.round } : {}),
                 ...(Array.isArray(continuation.toolMessages) ? { toolMessages: continuation.toolMessages } : {})
-            });
+            }, stream);
             aiWritePlanManager.finishQuestionContinuation(questionId, { callId: resumed.callId ?? null, completed: true });
+            return resumed;
         }
         catch (error) {
             aiWritePlanManager.finishQuestionContinuation(questionId, { message: error instanceof Error ? error.message : "恢复失败" }, true);
             throw error;
         }
     };
+    const respondWithQuestionContinuation = async (request, response, viewer) => {
+        const { questionId, workId } = request.params;
+        if (!request.get("Accept")?.includes("text/event-stream")) {
+            await resumeQuestionWorkflow(questionId, workId, viewer);
+            data(response, aiWritePlanManager.getQuestion(questionId, workId, viewer));
+            return;
+        }
+        const controller = new AbortController();
+        response.status(200);
+        response.setHeader("Content-Type", "text/event-stream; charset=utf-8");
+        response.setHeader("Cache-Control", "no-cache, no-transform");
+        response.setHeader("Connection", "keep-alive");
+        response.setHeader("X-Accel-Buffering", "no");
+        response.flushHeaders();
+        const sendEvent = (event, payload) => {
+            if (!response.writableEnded && !response.destroyed)
+                response.write(`event: ${event}\ndata: ${JSON.stringify(payload)}\n\n`);
+        };
+        const heartbeat = setInterval(() => {
+            if (!response.writableEnded && !response.destroyed)
+                response.write(": heartbeat\n\n");
+        }, aiStreamHeartbeatIntervalMs);
+        heartbeat.unref();
+        response.on("close", () => {
+            clearInterval(heartbeat);
+            if (!response.writableEnded)
+                controller.abort(new Error("浏览器已中断流式请求"));
+        });
+        try {
+            const resumed = await resumeQuestionWorkflow(questionId, workId, viewer, {
+                signal: controller.signal,
+                onStart: (message) => sendEvent("continuation", message),
+                onDelta: (delta) => sendEvent("delta", { delta }),
+                onToolCall: (toolCall, round) => sendEvent("tool_call", { ...toolCall, round }),
+                onProcessStep: (step) => sendEvent("process_step", step),
+                onContextCompacted: (event) => sendEvent("context_compacted", event)
+            });
+            const message = resumed?.conversationMessage;
+            const metadata = message?.metadata;
+            sendEvent("complete", {
+                warningOnly: !resumed,
+                model: resumed?.model,
+                outputTokens: metadata?.outputTokens ?? resumed?.outputTokens,
+                cacheHitPercent: metadata?.cacheHitPercent ?? resumed?.cacheHitPercent,
+                processDurationMs: metadata?.processDurationMs ?? resumed?.processDurationMs,
+                toolCalls: metadata?.toolCalls ?? resumed?.toolCalls,
+                processSteps: metadata?.processSteps ?? resumed?.processSteps,
+                contextUsage: resumed?.contextUsage,
+                messageId: message?.id,
+                messageCreatedAt: message?.createdAt,
+                question: aiWritePlanManager.getQuestion(questionId, workId, viewer)
+            });
+        }
+        catch (error) {
+            if (!controller.signal.aborted)
+                sendEvent("error", publicAiStreamError(error));
+        }
+        finally {
+            clearInterval(heartbeat);
+            if (!response.writableEnded && !response.destroyed)
+                response.end();
+        }
+    };
     app.post("/api/works/:workId/ai/questions/:questionId/answer", async (request, response) => {
         const input = parse(answerAiUserQuestionSchema, request.body ?? {});
         const viewer = planViewer();
         aiWritePlanManager.answerQuestion(request.params.questionId, request.params.workId, viewer, input);
-        await resumeQuestionWorkflow(request.params.questionId, request.params.workId, viewer);
-        data(response, aiWritePlanManager.getQuestion(request.params.questionId, request.params.workId, viewer));
+        await respondWithQuestionContinuation(request, response, viewer);
     });
     app.post("/api/works/:workId/ai/questions/:questionId/reject", async (request, response) => {
         const viewer = planViewer();
         aiWritePlanManager.rejectQuestion(request.params.questionId, request.params.workId, viewer);
-        await resumeQuestionWorkflow(request.params.questionId, request.params.workId, viewer);
-        data(response, aiWritePlanManager.getQuestion(request.params.questionId, request.params.workId, viewer));
+        await respondWithQuestionContinuation(request, response, viewer);
     });
     app.get("/api/works/:workId/providers", (request, response) => {
         store.getWork(request.params.workId);
@@ -3923,6 +3995,9 @@ export function createRuntime(options) {
                 ])];
             const mentionRaceIds = [...new Set(resolvedScope.raceIds ?? [])];
             const mentionOrganizationIds = [...new Set(resolvedScope.organizationIds ?? [])];
+            const mentionSettingIds = [...new Set(resolvedScope.settingIds ?? [])];
+            const mentionChapterIds = [...new Set(resolvedScope.chapterIds ?? [])];
+            const mentionContextSettingIds = resolvedScope.includeSettingInfo === true ? ["include-setting-info"] : [];
             const begun = store.beginAiConversationStreamRequest({
                 workId: request.params.workId,
                 conversationId,
@@ -3933,11 +4008,14 @@ export function createRuntime(options) {
                     content: storedUserContent,
                     citations,
                     ...(input.currentMessageId ? { existingMessageId: input.currentMessageId } : {}),
-                    ...((modelId || mentionCharacterIds.length || mentionRaceIds.length || mentionOrganizationIds.length || input.imageAttachmentIds?.length || input.scope.semanticSnapshotId) ? { metadata: {
+                    ...((modelId || mentionCharacterIds.length || mentionRaceIds.length || mentionOrganizationIds.length || mentionSettingIds.length || mentionChapterIds.length || mentionContextSettingIds.length || input.imageAttachmentIds?.length || input.scope.semanticSnapshotId) ? { metadata: {
                             ...(modelId ? { modelId } : {}),
                             ...(mentionCharacterIds.length ? { mentionCharacterIds } : {}),
                             ...(mentionRaceIds.length ? { mentionRaceIds } : {}),
                             ...(mentionOrganizationIds.length ? { mentionOrganizationIds } : {}),
+                            ...(mentionSettingIds.length ? { mentionSettingIds } : {}),
+                            ...(mentionChapterIds.length ? { mentionChapterIds } : {}),
+                            ...(mentionContextSettingIds.length ? { mentionContextSettingIds } : {}),
                             ...(input.scope.semanticSnapshotId ? { semanticSnapshotId: input.scope.semanticSnapshotId } : {}),
                             ...(input.imageAttachmentIds?.length ? { chatImageAttachmentIds: [...new Set(input.imageAttachmentIds)] } : {})
                         } } : {})
@@ -3996,8 +4074,11 @@ export function createRuntime(options) {
             const currentMessageId = String(begun.userMessage?.id ?? input.currentMessageId ?? "");
             if (begun.userMessage)
                 sendEvent("user_message", { message: redactAiConversationMessage(begun.userMessage, permissions) });
-            if (aiWritePlanManager.latestPendingQuestion(conversationId)) {
-                throw new AppError(409, "AI_QUESTION_PENDING", "当前对话仍有待回答问题，请先回答或拒绝后再继续");
+            const pendingQuestion = aiWritePlanManager.latestPendingQuestion(conversationId);
+            if (pendingQuestion) {
+                throw new AppError(409, "AI_QUESTION_PENDING", "当前对话仍有待回答问题，请先回答或拒绝后再继续", {
+                    questionId: pendingQuestion.id
+                });
             }
             const suggestion = await ai.createStreamingChat({
                 workId: request.params.workId,