@musnows/scriverse 0.9.9 → 1.0.0

package/dist/ai-write-plans.js

@@ -50,7 +50,7 @@ export const aiWriteToolDescriptions = {
     outlines: "允许侧边栏 AI 编辑章节大纲以及创建或编辑伏笔（不能删除）。",
     annotations: "允许侧边栏 AI 复用现有批注能力，在正文指定位置创建评论或待办。",
     analysis_tasks: "允许侧边栏 AI 触发已有类型的分析任务进入现有队列。",
-    ask_user_questions: "允许侧边栏 AI 通过 AskUserQuestions 向用户提出单选问题。"
+    ask_user_questions: "允许侧边栏 AI 通过 AskUserQuestions 向用户批量提出单选问题。"
 };
 /** 全部关闭时的默认开关状态。 */
 export function defaultAiWriteToolToggles() {
@@ -388,22 +388,43 @@ export const createAiWritePlanInputSchema = z.object({
     aiSummary: z.string().trim().min(1).max(2000),
     operations: z.unknown()
 }).strict();
-/** 提问选项数量限制：一次一个问题和 2-6 个预设选项。 */
+/** 单次工具调用可批量提出 1-5 个问题，每题包含 2-6 个预设选项。 */
+export const MIN_AI_QUESTIONS_PER_CALL = 1;
+export const MAX_AI_QUESTIONS_PER_CALL = 5;
 export const MIN_AI_QUESTION_OPTIONS = 2;
 export const MAX_AI_QUESTION_OPTIONS = 6;
 export const MAX_AI_QUESTION_ANSWER_CHARS = 3000;
-export const askAiUserQuestionInputSchema = z.object({
+const aiUserQuestionItemSchema = z.object({
     question: z.string().trim().min(1).max(2000),
     options: z.array(z.string().trim().min(1).max(200))
         .min(MIN_AI_QUESTION_OPTIONS)
         .max(MAX_AI_QUESTION_OPTIONS)
 }).strict();
-export const answerAiUserQuestionSchema = z.object({
+export const askAiUserQuestionInputSchema = z.union([
+    z.object({
+        questions: z.array(aiUserQuestionItemSchema)
+            .min(MIN_AI_QUESTIONS_PER_CALL)
+            .max(MAX_AI_QUESTIONS_PER_CALL)
+    }).strict(),
+    aiUserQuestionItemSchema
+]).transform((input) => "questions" in input ? input : { questions: [input] });
+const aiUserQuestionAnswerSchema = z.object({
     selectedOption: z.number().int().min(0).optional(),
-    customAnswer: z.string().trim().min(1).max(MAX_AI_QUESTION_ANSWER_CHARS).optional()
+    customAnswer: z.string().trim()
+        .min(1, "自定义回答不能为空")
+        .max(MAX_AI_QUESTION_ANSWER_CHARS, `自定义回答不能超过 ${MAX_AI_QUESTION_ANSWER_CHARS} 个字符`)
+        .optional()
 }).strict().refine(
 // 至少提供一种回答；选择预设项后仍可附带自定义补充信息。
 (input) => input.selectedOption !== undefined || input.customAnswer !== undefined, { message: "必须选择预设选项或填写自定义回答" });
+export const answerAiUserQuestionSchema = z.union([
+    z.object({
+        answers: z.array(aiUserQuestionAnswerSchema)
+            .min(MIN_AI_QUESTIONS_PER_CALL)
+            .max(MAX_AI_QUESTIONS_PER_CALL)
+    }).strict(),
+    aiUserQuestionAnswerSchema
+]).transform((input) => "answers" in input ? input : { answers: [input] });
 // ---------------------------------------------------------------------------
 // 标签与展示辅助
 // ---------------------------------------------------------------------------
@@ -897,17 +918,76 @@ function planInputError(index, error) {
     const firstIssue = error.issues[0];
     return new AppError(400, "AI_PLAN_OPERATION_INVALID", `第 ${index + 1} 个操作的 ${firstIssue?.path.join(".") || "input"} 字段无效：${firstIssue?.message ?? "输入不符合要求"}`);
 }
-function resolvedQuestionAnswer(row) {
-    const options = json(row.options_json, []);
-    const selectedOption = row.selected_option === null ? null : Number(row.selected_option);
+function storedQuestionDefinitions(row) {
+    const stored = json(row.questions_json, []);
+    const questions = stored.flatMap((value) => {
+        if (!value || typeof value !== "object" || Array.isArray(value))
+            return [];
+        const record = value;
+        const question = typeof record.question === "string" ? record.question.trim() : "";
+        const options = Array.isArray(record.options)
+            ? record.options.filter((option) => typeof option === "string" && option.trim().length > 0)
+            : [];
+        return question && options.length >= MIN_AI_QUESTION_OPTIONS ? [{ question, options }] : [];
+    });
+    if (questions.length > 0)
+        return questions;
+    return [{ question: row.question, options: json(row.options_json, []) }];
+}
+function resolveQuestionAnswer(options, selectedOption, customAnswer) {
     const selectedOptionLabel = selectedOption !== null && selectedOption >= 0 && selectedOption < options.length
         ? options[selectedOption] ?? null
         : null;
-    const customAnswer = row.is_custom_answer === 1 ? row.answer_text : "";
     const answerText = selectedOptionLabel
         ? (customAnswer ? `${selectedOptionLabel}\n补充信息：${customAnswer}` : selectedOptionLabel)
-        : (customAnswer || row.answer_text);
-    return { selectedOption, selectedOptionLabel, customAnswer, answerText };
+        : customAnswer;
+    return {
+        selectedOption,
+        selectedOptionLabel,
+        customAnswer,
+        answerText,
+        isCustomAnswer: customAnswer.length > 0
+    };
+}
+function resolvedQuestionAnswers(row) {
+    const questions = storedQuestionDefinitions(row);
+    const storedAnswers = json(row.answers_json, []);
+    return questions.map((definition, index) => {
+        const value = storedAnswers[index];
+        const record = value && typeof value === "object" && !Array.isArray(value)
+            ? value
+            : null;
+        const storedSelectedOption = record?.selectedOption;
+        const selectedOption = Number.isInteger(storedSelectedOption)
+            ? Number(storedSelectedOption)
+            : index === 0 && row.selected_option !== null ? Number(row.selected_option) : null;
+        const customAnswer = typeof record?.customAnswer === "string"
+            ? record.customAnswer
+            : index === 0 && row.is_custom_answer === 1 ? row.answer_text : "";
+        return {
+            index,
+            question: definition.question,
+            options: definition.options.map((label, optionIndex) => ({ index: optionIndex, label, recommended: optionIndex === 0 })),
+            ...resolveQuestionAnswer(definition.options, selectedOption, customAnswer)
+        };
+    });
+}
+function combinedQuestionAnswerText(questions) {
+    const answered = questions.filter((question) => question.answerText);
+    if (answered.length === 0)
+        return "";
+    if (questions.length === 1)
+        return questions[0]?.answerText ?? "";
+    return answered.map((question) => `问题 ${question.index + 1}：${question.question}\n回答：${question.answerText}`).join("\n\n");
+}
+function firstQuestionAnswer(row) {
+    const answer = resolvedQuestionAnswers(row)[0];
+    return {
+        selectedOption: answer?.selectedOption ?? null,
+        selectedOptionLabel: answer?.selectedOptionLabel ?? null,
+        customAnswer: answer?.customAnswer ?? "",
+        answerText: answer?.answerText ?? ""
+    };
 }
 export class AiWritePlanManager {
     database;
@@ -1972,7 +2052,10 @@ export class AiWritePlanManager {
     // --------------------------------------------------------------- 用户提问
     createQuestion(input) {
         this.assertToolEnabled(input.workId, "ask_user_questions");
-        const parsed = askAiUserQuestionInputSchema.parse({ question: input.question, options: input.options });
+        const parsed = askAiUserQuestionInputSchema.parse(input.questions === undefined
+            ? { question: input.question, options: input.options }
+            : { questions: input.questions });
+        const firstQuestion = parsed.questions[0];
         const questionId = randomId("aiQ");
         const timestamp = now();
         const expiresAt = isoFromNow(timestamp, this.questionTtlMs);
@@ -1984,47 +2067,43 @@ export class AiWritePlanManager {
                 throw new AppError(409, "AI_QUESTION_PENDING", "当前对话已有一个待回答问题");
             this.database.run(`INSERT INTO ai_user_questions (
           id, work_id, conversation_id, initiator_user_id, recipient_user_id, question,
-          options_json, status, tool_call_id, created_at, expires_at
-        ) VALUES (?, ?, ?, ?, ?, ?, ?, 'pending', ?, ?, ?)`, questionId, input.workId, input.conversationId, input.initiator?.userId ?? null, input.recipientUserId, parsed.question, JSON.stringify(parsed.options), input.toolCallId ?? null, timestamp, expiresAt);
+          options_json, questions_json, status, tool_call_id, created_at, expires_at
+        ) VALUES (?, ?, ?, ?, ?, ?, ?, ?, 'pending', ?, ?, ?)`, questionId, input.workId, input.conversationId, input.initiator?.userId ?? null, input.recipientUserId, firstQuestion.question, JSON.stringify(firstQuestion.options), JSON.stringify(parsed.questions), input.toolCallId ?? null, timestamp, expiresAt);
         });
         this.store.audit(input.workId, "ai.question.asked", "ai_user_question", questionId, {
             conversationId: input.conversationId,
-            askedBy: input.initiator?.userId ?? null
+            askedBy: input.initiator?.userId ?? null,
+            questionCount: parsed.questions.length
         });
         return this.getQuestion(questionId, input.workId, input.initiator);
     }
     answerQuestion(questionId, workId, respondent, payload) {
         const row = this.assertAnswerable(questionId, workId, respondent);
-        const options = json(row.options_json, []);
-        const customAnswer = payload.customAnswer?.trim() ?? "";
-        if (payload.customAnswer !== undefined && !customAnswer) {
-            throw new AppError(400, "AI_QUESTION_CUSTOM_ANSWER_INVALID", "自定义回答不能为空");
-        }
-        if (customAnswer.length > MAX_AI_QUESTION_ANSWER_CHARS) {
-            throw new AppError(400, "AI_QUESTION_CUSTOM_ANSWER_TOO_LONG", `自定义回答不能超过 ${MAX_AI_QUESTION_ANSWER_CHARS} 个字符`);
+        const questions = storedQuestionDefinitions(row);
+        const parsed = answerAiUserQuestionSchema.parse(payload);
+        if (parsed.answers.length !== questions.length) {
+            throw new AppError(400, "AI_QUESTION_ANSWERS_INCOMPLETE", `必须一次提交全部 ${questions.length} 个问题的回答`);
         }
-        let selectedOption = null;
-        let selectedOptionLabel = "";
-        if (payload.selectedOption !== undefined) {
-            if (payload.selectedOption < 0 || payload.selectedOption >= options.length) {
-                throw new AppError(400, "AI_QUESTION_OPTION_INVALID", "选择的选项编号无效");
+        const answers = parsed.answers.map((answer, index) => {
+            const options = questions[index].options;
+            const selectedOption = answer.selectedOption ?? null;
+            if (selectedOption !== null && selectedOption >= options.length) {
+                throw new AppError(400, "AI_QUESTION_OPTION_INVALID", `第 ${index + 1} 个问题的选项编号无效`);
             }
-            selectedOption = payload.selectedOption;
-            selectedOptionLabel = options[selectedOption] ?? "";
-        }
-        if (selectedOption === null && !customAnswer)
-            throw new AppError(400, "AI_QUESTION_ANSWER_REQUIRED", "必须提供回答");
-        const isCustom = Boolean(customAnswer);
-        const storedAnswerText = customAnswer || selectedOptionLabel;
+            return { selectedOption, customAnswer: answer.customAnswer ?? "" };
+        });
+        const firstAnswer = answers[0];
+        const firstResolved = resolveQuestionAnswer(questions[0].options, firstAnswer.selectedOption, firstAnswer.customAnswer);
         const updated = this.database.run(`UPDATE ai_user_questions
-       SET status = 'answered', selected_option = ?, answer_text = ?, is_custom_answer = ?, decided_at = ?
-      WHERE id = ? AND status = 'pending'`, selectedOption, storedAnswerText, isCustom ? 1 : 0, now(), questionId);
+       SET status = 'answered', selected_option = ?, answer_text = ?, is_custom_answer = ?, answers_json = ?, decided_at = ?
+      WHERE id = ? AND status = 'pending'`, firstAnswer.selectedOption, firstAnswer.customAnswer || firstResolved.selectedOptionLabel || "", firstAnswer.customAnswer ? 1 : 0, JSON.stringify(answers), now(), questionId);
         if (updated.changes !== 1)
             throw new AppError(409, "AI_QUESTION_ALREADY_DECIDED", "该问题已被处理");
         this.store.audit(row.work_id, "ai.question.answered", "ai_user_question", questionId, {
             answeredBy: respondent?.userId ?? null,
-            isCustomAnswer: isCustom,
-            hasSelectedOption: selectedOption !== null
+            questionCount: questions.length,
+            customAnswerCount: answers.filter((answer) => answer.customAnswer).length,
+            selectedOptionCount: answers.filter((answer) => answer.selectedOption !== null).length
         });
         return this.getQuestion(questionId, workId, respondent);
     }
@@ -2101,7 +2180,8 @@ export class AiWritePlanManager {
         const claimed = this.database.run("UPDATE ai_user_questions SET resume_state = 'claimed', resumed_at = ? WHERE id = ? AND resume_state = 'pending' AND status IN ('answered', 'rejected', 'expired')", now(), questionId);
         if (claimed.changes !== 1)
             return null;
-        const answer = resolvedQuestionAnswer(row);
+        const questionView = this.toQuestionView(row);
+        const answer = firstQuestionAnswer(row);
         return {
             ...continuation,
             questionId,
@@ -2111,7 +2191,13 @@ export class AiWritePlanManager {
             selectedOptionLabel: answer.selectedOptionLabel,
             customAnswer: answer.customAnswer,
             toolCallId: row.tool_call_id,
-            questionView: this.toQuestionView(row)
+            answers: questionView.questions.map((question) => ({
+                question: question.question,
+                answer: question.answerText,
+                selectedOption: question.selectedOptionLabel,
+                supplementalAnswer: question.customAnswer || null
+            })),
+            questionView
         };
     }
     finishQuestionContinuation(questionId, result, failed = false) {
@@ -2151,21 +2237,23 @@ export class AiWritePlanManager {
         return this.database.get("SELECT * FROM ai_user_questions WHERE id = ?", row.id);
     }
     toQuestionView(row) {
-        const options = json(row.options_json, []);
-        const answer = resolvedQuestionAnswer(row);
+        const questions = resolvedQuestionAnswers(row);
+        const firstQuestion = questions[0];
         return {
             id: row.id,
             workId: row.work_id,
             conversationId: row.conversation_id,
             question: row.question,
+            questionCount: questions.length,
+            questions,
             status: row.status,
             statusLabel: aiQuestionStatusLabels[row.status] ?? row.status,
-            options: options.map((label, index) => ({ index, label, recommended: index === 0 })),
-            selectedOption: answer.selectedOption,
-            selectedOptionLabel: answer.selectedOptionLabel,
-            customAnswer: answer.customAnswer,
-            answerText: answer.answerText,
-            isCustomAnswer: row.is_custom_answer === 1,
+            options: firstQuestion.options,
+            selectedOption: firstQuestion.selectedOption,
+            selectedOptionLabel: firstQuestion.selectedOptionLabel,
+            customAnswer: firstQuestion.customAnswer,
+            answerText: combinedQuestionAnswerText(questions),
+            isCustomAnswer: firstQuestion.isCustomAnswer,
             createdAt: row.created_at,
             expiresAt: row.expires_at,
             decidedAt: row.decided_at,