@musnows/scriverse 0.5.8 → 0.5.9

package/dist/app.js

@@ -16,6 +16,7 @@ import { Database } from "./database.js";
 import { assertSafeDocxArchive } from "./docx-security.js";
 import { TASK_TYPES } from "./domain.js";
 import { AppError } from "./errors.js";
+import { HYBRID_SEARCH_TYPES } from "./hybrid-search.js";
 import { applyImportFileHints, parseNovelText } from "./parser.js";
 import { Store, versionedEntityTypes } from "./store.js";
 import { parsePagination } from "./pagination.js";
@@ -30,7 +31,7 @@ import { currentRequestActor, runWithRequestActor } from "./request-context.js";
 import { APP_VERSION } from "./version.js";
 import { fullWorkModulePermissions, proseReplacementPermissionModules } from "./work-permissions.js";
 import { CollaborationPresence, entityEditorPageKey, presencePageKinds } from "./collaboration-presence.js";
-import { clearSessionCookie, createCliApiScopeMiddleware, createUserSessionMiddleware, createWorkAuthorizationMiddleware, relationshipAnalysisReadModules, setSessionCookie, UserAuthService } from "./user-auth.js";
+import { analysisTaskReadModules, clearSessionCookie, createCliApiScopeMiddleware, createUserSessionMiddleware, createWorkAuthorizationMiddleware, relationshipAnalysisReadModules, setSessionCookie, UserAuthService } from "./user-auth.js";
 const nonEmpty = z.string().trim().min(1);
 const identifier = z.string().trim().min(1).max(200);
 const optionalStrings = z.array(z.string()).optional();
@@ -239,6 +240,7 @@ const raceSchema = z.object({
     settingsSections: knowledgeSectionsSchema.optional(),
     memberIds: z.array(identifier).max(1000).optional()
 }).strict();
+const raceHierarchyScopeSchema = z.enum(["roots", "descendants"]).optional();
 const chapterOutlineSchema = z.object({
     goal: z.string().max(100_000).optional(),
     conflict: z.string().max(100_000).optional(),
@@ -355,6 +357,8 @@ const workAiSettingsSchema = z.object({
     autoRunEnabled: z.boolean().optional(),
     autoRunConcurrency: z.number().int().min(1).max(8).optional(),
     autoRunBatchLimit: z.number().int().min(1).max(200).optional(),
+    autoRunDailyTaskLimit: z.number().int().min(0).max(10_000).optional(),
+    autoRunFailureThreshold: z.number().int().min(1).max(10).optional(),
     bookSummaryContextPercent: z.number().int().min(1).max(90).optional(),
     contextCompactThreshold: z.number().int().min(50).max(90).optional(),
     agentTools: z.array(z.enum(["story_index", "read_chapters", "grep", "search_story_entities", "read_character_sections"])).max(5).optional()
@@ -626,11 +630,7 @@ export function createRuntime(options) {
     };
     const captcha = new ImageCaptchaService({ revealAnswer: options.revealCaptchaAnswer === true });
     const ai = new AiManager(store, new CredentialVault(options.masterSecret), options.fetchImpl ?? fetch, options.security ? (url) => assertSafeAiEndpoint(url, options.security?.allowPrivateAiEndpoints) : undefined, (task, actor) => {
-        if (task.taskType !== "relationship-analysis")
-            return;
-        const requiredModules = relationshipAnalysisReadModules(task.scope);
-        if (requiredModules.length === 0)
-            return;
+        const requiredModules = analysisTaskReadModules(task.taskType, task.scope);
         const creator = actor ? null : database.get("SELECT created_by_user_id FROM analysis_tasks WHERE id = ?", String(task.id));
         const userId = actor?.userId ?? (typeof creator?.created_by_user_id === "string" ? creator.created_by_user_id : null);
         if (!userId)
@@ -1280,9 +1280,19 @@ export function createRuntime(options) {
         noContent(response);
     });
     app.get("/api/works/:workId/races", (request, response) => {
+        const hierarchyScope = parse(raceHierarchyScopeSchema, request.query.scope);
         const pagination = parsePagination(request.query);
+        if (hierarchyScope && pagination) {
+            throw new AppError(400, "RACE_HIERARCHY_PAGINATION_CONFLICT", "分层种族请求不能同时使用分页参数");
+        }
         const includeMarkdown = request.query.includeContent === "true";
-        const races = pagination ? store.listRacesPage(request.params.workId, pagination, includeMarkdown) : store.listRaces(request.params.workId, includeMarkdown);
+        const races = hierarchyScope
+            ? hierarchyScope === "roots"
+                ? { items: store.listRacesByHierarchyScope(request.params.workId, hierarchyScope, includeMarkdown), total: store.countRaces(request.params.workId) }
+                : store.listRacesByHierarchyScope(request.params.workId, hierarchyScope, includeMarkdown)
+            : pagination
+                ? store.listRacesPage(request.params.workId, pagination, includeMarkdown)
+                : store.listRaces(request.params.workId, includeMarkdown);
         const permissions = requestPermissions(request, request.params.workId);
         data(response, mapRecords(races, (race) => redactRaceMembers(race, permissions)));
     });
@@ -1458,6 +1468,7 @@ export function createRuntime(options) {
         data(response, pagination ? store.listReviewItemsPage(request.params.workId, pagination, status) : store.listReviewItems(request.params.workId, status));
     });
     app.post("/api/works/:workId/reviews", (request, response) => data(response, store.createReviewItem(request.params.workId, parse(reviewSchema, request.body)), 201));
+    app.get("/api/reviews/:reviewId", (request, response) => data(response, store.getReviewItem(request.params.reviewId)));
     app.patch("/api/reviews/:reviewId", (request, response) => data(response, store.updateReviewItem(request.params.reviewId, parse(reviewSchema.partial(), request.body))));
     app.post("/api/reviews/:reviewId/character-resolution", (request, response) => {
         const input = parse(z.discriminatedUnion("action", [
@@ -1502,17 +1513,15 @@ export function createRuntime(options) {
         const permissions = requestPermissions(request, request.params.workId);
         for (const taskId of store.listOldestPendingTaskIds(request.params.workId, store.countPendingTasks(request.params.workId))) {
             const task = store.getTask(taskId);
-            if (task.taskType !== "relationship-analysis")
-                continue;
-            const deniedModules = relationshipAnalysisReadModules(task.scope).filter((module) => permissions[module] === "none");
+            const deniedModules = analysisTaskReadModules(task.taskType, task.scope).filter((module) => permissions[module] === "none");
             if (deniedModules.length > 0) {
-                throw new AppError(403, "WORK_MODULE_READ_DENIED", "你没有读取待运行定向人物关系分析所需资料模块的权限", {
+                throw new AppError(403, "WORK_MODULE_READ_DENIED", "你没有读取待运行分析任务所需资料模块的权限", {
                     taskId,
                     modules: deniedModules
                 });
             }
         }
-        data(response, ai.startAutoRunBatch(request.params.workId));
+        data(response, ai.resumeAutoRun(request.params.workId));
     });
     app.get("/api/tasks/:taskId/detail", (request, response) => data(response, redactTaskCharacterNames(store.getTaskDetail(request.params.taskId), requestPermissions(request))));
     app.get("/api/tasks/:taskId/result", (request, response) => {
@@ -1531,14 +1540,12 @@ export function createRuntime(options) {
     app.post("/api/tasks/:taskId/run", async (request, response) => {
         const input = parse(z.object({ modelId: identifier.optional() }), request.body ?? {});
         const task = store.getTask(request.params.taskId);
-        if (task.taskType === "relationship-analysis") {
-            const permissions = requestPermissions(request, String(task.workId));
-            const deniedModules = relationshipAnalysisReadModules(task.scope).filter((module) => permissions[module] === "none");
-            if (deniedModules.length > 0) {
-                throw new AppError(403, "WORK_MODULE_READ_DENIED", "你没有读取本次定向人物关系分析所需资料模块的权限", {
-                    modules: deniedModules
-                });
-            }
+        const permissions = requestPermissions(request, String(task.workId));
+        const deniedModules = analysisTaskReadModules(task.taskType, task.scope).filter((module) => permissions[module] === "none");
+        if (deniedModules.length > 0) {
+            throw new AppError(403, "WORK_MODULE_READ_DENIED", "你没有读取本次分析所需资料模块的权限", {
+                modules: deniedModules
+            });
         }
         data(response, redactTaskCharacterNames(await ai.runTask(request.params.taskId, input.modelId, request.authUser ? {
             userId: request.authUser.userId,
@@ -1548,14 +1555,12 @@ export function createRuntime(options) {
     app.post("/api/tasks/:taskId/rerun", (request, response) => {
         parse(z.object({}).strict(), request.body ?? {});
         const task = store.getTask(request.params.taskId);
-        if (task.taskType === "relationship-analysis") {
-            const permissions = requestPermissions(request, String(task.workId));
-            const deniedModules = relationshipAnalysisReadModules(task.scope).filter((module) => permissions[module] === "none");
-            if (deniedModules.length > 0) {
-                throw new AppError(403, "WORK_MODULE_READ_DENIED", "你没有读取本次定向人物关系分析所需资料模块的权限", {
-                    modules: deniedModules
-                });
-            }
+        const permissions = requestPermissions(request, String(task.workId));
+        const deniedModules = analysisTaskReadModules(task.taskType, task.scope).filter((module) => permissions[module] === "none");
+        if (deniedModules.length > 0) {
+            throw new AppError(403, "WORK_MODULE_READ_DENIED", "你没有读取本次分析所需资料模块的权限", {
+                modules: deniedModules
+            });
         }
         data(response, redactTaskCharacterNames(ai.rerunTask(request.params.taskId), requestPermissions(request)), 201);
     });
@@ -1615,12 +1620,14 @@ export function createRuntime(options) {
     });
     app.patch("/api/works/:workId/ai-settings", (request, response) => {
         const workId = request.params.workId;
+        const input = parse(workAiSettingsSchema, request.body);
         const before = store.getWorkAiSettings(workId);
-        const updated = store.updateWorkAiSettings(workId, parse(workAiSettingsSchema, request.body));
+        let updated = store.updateWorkAiSettings(workId, input);
         if (updated.autoRunEnabled) {
-            if (!before.autoRunEnabled)
-                ai.resetAutoRunBatch(workId);
-            ai.scheduleAutoRun(workId);
+            if (input.autoRunEnabled === true && !before.autoRunEnabled)
+                updated = ai.resumeAutoRun(workId);
+            else
+                ai.scheduleAutoRun(workId);
         }
         data(response, updated);
     });
@@ -1858,9 +1865,13 @@ export function createRuntime(options) {
         const pagination = parsePagination(request.query);
         data(response, pagination ? ai.listCallsPage(request.params.workId, pagination) : ai.listCalls(request.params.workId));
     });
-    app.get("/api/works/:workId/search", (request, response) => {
-        const query = parse(z.string().trim().min(1).max(500), request.query.q);
-        data(response, store.search(request.params.workId, query));
+    app.get("/api/works/:workId/search", async (request, response) => {
+        const query = parse(z.object({
+            q: z.string().trim().min(1).max(500),
+            type: z.enum(HYBRID_SEARCH_TYPES).optional(),
+            limit: z.coerce.number().int().min(1).max(100).optional()
+        }).strict(), request.query);
+        data(response, await ai.searchWork(request.params.workId, query.q, { type: query.type, limit: query.limit }));
     });
     app.get("/api/works/:workId/export", async (request, response) => {
         const format = parse(z.enum(["json", "txt", "markdown"]), request.query.format ?? "json");