@waterwx/dsh-novel-forge 1.1.0 → 1.1.2

package/lib/index.js

@@ -913,7 +913,12 @@ function parseJson(text, wantArray) {
 		return out;
 	};
 	for (const candidate of candidates) for (const attempt of [candidate, repair(candidate)]) try {
-		return JSON.parse(attempt);
+		const value = JSON.parse(attempt);
+		if (!wantArray || Array.isArray(value)) return value;
+		if (typeof value === "object" && value !== null) for (const key of Object.keys(value)) {
+			const inner = value[key];
+			if (Array.isArray(inner)) return inner;
+		}
 	} catch {}
 	const preview = text.length > 300 ? text.slice(0, 300) + "…" : text;
 	throw new Error(`模型输出中未找到 JSON 数据。模型原始输出：${preview}`);
@@ -1092,33 +1097,59 @@ function writeSystemPrompt(project) {
 /**
 * Plan chapters from an outline (optionally for one volume).
 */
-async function planChapters(ctx, config, project, chapterCount, volumeNo) {
+async function planChapters(ctx, config, project, chapterCount, volumeNo, outputDir) {
 	const volume = project.volumes?.find((v) => v.no === volumeNo);
+	const existing = project.chapters;
+	const startNo = existing.length === 0 ? 1 : Math.max(...existing.map((c) => c.no)) + 1;
+	const continuation = existing.length > 0;
+	const latestFacts = continuation && Array.isArray(project.facts) ? project.facts.slice(-15).map((f) => `[第${f.chapterNo}章] ${f.text.slice(0, 150)}`).join("\n") : "";
+	let prevTail = "";
+	if (continuation) {
+		const written = existing.filter((c) => c.status !== "pending");
+		const last = written[written.length - 1];
+		if (last !== void 0 && last.file !== void 0 && outputDir !== void 0) try {
+			prevTail = readFileSync(join(outputDir, last.file), "utf8").replace(/^#.*$/m, "").trim().slice(-600);
+		} catch {}
+	}
+	const outlineBlock = continuation ? (() => {
+		const cut = project.outline.indexOf("七、关键剧情桥段");
+		if (cut > 1500) return project.outline.slice(0, cut).trimEnd() + "\n（大纲后续剧情桥段与分卷细节从略；续写请以「上一章结尾原文」与「最新剧情状态」为剧情起点）";
+		return project.outline.slice(0, 3e3) + "\n…（大纲过长已节选）";
+	})() : project.outline;
 	const user = [
 		"请为下面这部小说规划章节。",
-		volume !== void 0 ? `本次只规划第 ${volume.no} 卷《${volume.title}》的章节：\n${volume.summary}` : "请规划全书开篇章节。",
-		`大纲如下：\n${project.outline}`,
+		volume !== void 0 ? `本次只规划第 ${volume.no} 卷《${volume.title}》的章节：\n${volume.summary}` : continuation ? `本书已有 ${existing.length} 章已规划/已写作（见下方「已有章节」）。请规划**后续**章节：从第 ${startNo} 章开始。` : "请规划全书开篇章节。",
+		continuation ? "【续写硬性要求】已有章节的剧情不得重写或重复，章节标题也不得与已有章节重复。以下情节均已在已有章节中发生过，后续章节**绝对不得再次出现**：穿越、暴雨送餐、滴血认主/古玉认主、首次进入墟境、用废铁淬炼首件法器、绝境肉身入鼎洗炼（该机缘已用尽）、杀死白袍弟子与灰衣随从、藏尸水沟。" : "",
+		prevTail !== "" ? `【上一章（第 ${startNo - 1} 章）结尾原文】第 ${startNo} 章必须紧接此状态继续，从新的事件写起，不得回顾重述：\n${prevTail}` : "",
+		latestFacts !== "" ? `【最新剧情状态（本书编年录，第 ${startNo - 1} 章结尾的事实）】规划续写时必须以此为起点，时间线、人物状态与地点衔接一致：\n${latestFacts}` : "",
+		continuation ? "已有章节：\n" + existing.map((c) => {
+			const sm = c.summary !== void 0 && c.summary !== "" ? `（${c.summary.slice(0, 120)}）` : "";
+			return `第${c.no}章《${c.title}》${sm}`;
+		}).join("\n") : "",
+		`全书大纲（设定参考，续写剧情不得与设定冲突）：\n${outlineBlock}`,
 		"",
 		`请规划 ${chapterCount} 章。输出 JSON 数组（不要输出其他文字）：`
 	].join("\n");
 	const parsed = parseJsonArray(await complete(ctx, config, {
-		system: planSystemPrompt(project.volumes),
+		system: planSystemPrompt(project.volumes) + (continuation ? "\n重要：本次是**续写规划**——已有章节的剧情不得重写或重复，新章节标题不得与已有章节标题相同，新章节的剧情必须从上一章结尾自然接续（人物状态、时间线、地点衔接一致）。" : ""),
 		user,
 		temperature: .7,
-		maxTokens: Math.max(config.maxTokens, 2e4)
+		maxTokens: Math.max(config.maxTokens, 4e4)
 	}));
 	const chapters = [];
-	const existing = new Set(project.chapters.map((c) => c.no));
-	const startNo = project.chapters.length + 1;
-	for (let i = 0; i < Math.min(parsed.length, chapterCount); i++) {
-		const item = parsed[i];
+	const existingNos = new Set(existing.map((c) => c.no));
+	const existingTitles = new Set(existing.map((c) => c.title));
+	let cursor = startNo;
+	for (const item of parsed) {
+		if (chapters.length >= chapterCount) break;
 		if (typeof item !== "object" || item === null) continue;
 		const entry = item;
 		const title = typeof entry.title === "string" ? entry.title.trim().slice(0, 30) : "";
 		const beats = typeof entry.beats === "string" ? entry.beats.trim() : "";
 		if (title === "" && beats === "") continue;
-		const no = startNo + i;
-		if (existing.has(no)) continue;
+		if (title !== "" && existingTitles.has(title)) continue;
+		while (existingNos.has(cursor)) cursor++;
+		const no = cursor++;
 		chapters.push({
 			no,
 			volume: volumeOf(no, project.volumes),
@@ -3026,7 +3057,7 @@ function makeRoutes(deps) {
 			}
 			try {
 				const next = project ?? createProject(outline);
-				const chapters = await planChapters(ctx, config, next, count, body?.volume);
+				const chapters = await planChapters(ctx, config, next, count, body?.volume, config.outputDir);
 				next.chapters.push(...chapters);
 				next.updatedAt = (/* @__PURE__ */ new Date()).toISOString();
 				saveProject(config.outputDir, next);