feat(frontend): 编辑器内「续写」——读本章前文生成多候选,点选插入正文
Phase 1(写作工作台重构):编辑器「续写」按钮复用 continue 生成器(with_prior_chapter 自动读该章已写正文承接),每次产出一条候选并累加成多候选(useContinue,已单测), 「再来一个」可多生成几版;作者点某条「插入正文」才追加到章末(HITL,不静默写入)。 续写前 flush 自动保存,确保读到最新草稿。润色/续写结果卡互斥不并占。
This commit is contained in:
106
apps/web/lib/workbench/useContinue.test.ts
Normal file
106
apps/web/lib/workbench/useContinue.test.ts
Normal file
@@ -0,0 +1,106 @@
|
||||
// @vitest-environment jsdom
|
||||
import { act, renderHook } from "@testing-library/react";
|
||||
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
|
||||
|
||||
import { useContinue } from "./useContinue";
|
||||
|
||||
const post = vi.fn();
|
||||
const toast = vi.fn();
|
||||
vi.mock("@/lib/api/client", () => ({
|
||||
api: { POST: (...a: unknown[]) => post(...a) },
|
||||
}));
|
||||
vi.mock("@/components/Toast", () => ({ useToast: () => toast }));
|
||||
|
||||
const GEN_PATH = "/projects/{project_id}/skills/{tool_key}/generate";
|
||||
|
||||
describe("useContinue", () => {
|
||||
beforeEach(() => {
|
||||
post.mockReset();
|
||||
toast.mockReset();
|
||||
});
|
||||
afterEach(() => vi.clearAllMocks());
|
||||
|
||||
it("初始为 idle,无候选", () => {
|
||||
const { result } = renderHook(() => useContinue());
|
||||
expect(result.current.status).toBe("idle");
|
||||
expect(result.current.candidates).toEqual([]);
|
||||
});
|
||||
|
||||
it("续写:调 continue 生成器(读该章前文),产出一条候选,status=ready", async () => {
|
||||
post.mockResolvedValue({
|
||||
data: { output_kind: "ContinuationResult", preview: { text: "续写正文一" } },
|
||||
error: null,
|
||||
});
|
||||
|
||||
const { result } = renderHook(() => useContinue());
|
||||
await act(async () => {
|
||||
await result.current.generate("p1", 3);
|
||||
});
|
||||
|
||||
expect(post).toHaveBeenCalledWith(GEN_PATH, {
|
||||
params: { path: { project_id: "p1", tool_key: "continue" } },
|
||||
body: { brief: "", chapter_no: 3 },
|
||||
});
|
||||
expect(result.current.status).toBe("ready");
|
||||
expect(result.current.candidates).toEqual(["续写正文一"]);
|
||||
});
|
||||
|
||||
it("多候选:再次生成累加候选,不覆盖前一条", async () => {
|
||||
post
|
||||
.mockResolvedValueOnce({ data: { preview: { text: "候选一" } }, error: null })
|
||||
.mockResolvedValueOnce({ data: { preview: { text: "候选二" } }, error: null });
|
||||
|
||||
const { result } = renderHook(() => useContinue());
|
||||
await act(async () => {
|
||||
await result.current.generate("p1", 1);
|
||||
});
|
||||
await act(async () => {
|
||||
await result.current.generate("p1", 1);
|
||||
});
|
||||
|
||||
expect(result.current.candidates).toEqual(["候选一", "候选二"]);
|
||||
});
|
||||
|
||||
it("产出为空:不加候选、提示重试、status=ready", async () => {
|
||||
post.mockResolvedValue({ data: { preview: { text: "" } }, error: null });
|
||||
const { result } = renderHook(() => useContinue());
|
||||
await act(async () => {
|
||||
await result.current.generate("p1", 1);
|
||||
});
|
||||
expect(result.current.candidates).toEqual([]);
|
||||
expect(result.current.status).toBe("ready");
|
||||
expect(toast).toHaveBeenCalledWith(expect.any(String), "info");
|
||||
});
|
||||
|
||||
it("后端返回 error:status=error、弹错误 toast、候选不变", async () => {
|
||||
post.mockResolvedValue({ data: null, error: { detail: "配额不足" } });
|
||||
const { result } = renderHook(() => useContinue());
|
||||
await act(async () => {
|
||||
await result.current.generate("p1", 1);
|
||||
});
|
||||
expect(result.current.status).toBe("error");
|
||||
expect(result.current.candidates).toEqual([]);
|
||||
expect(toast).toHaveBeenCalledWith(expect.any(String), "error");
|
||||
});
|
||||
|
||||
it("请求抛异常:status=error、弹网络异常 toast", async () => {
|
||||
post.mockRejectedValue(new Error("network down"));
|
||||
const { result } = renderHook(() => useContinue());
|
||||
await act(async () => {
|
||||
await result.current.generate("p1", 1);
|
||||
});
|
||||
expect(result.current.status).toBe("error");
|
||||
expect(toast).toHaveBeenCalledWith("续写请求异常,请检查网络。", "error");
|
||||
});
|
||||
|
||||
it("reset 清回 idle 与空候选", async () => {
|
||||
post.mockResolvedValue({ data: { preview: { text: "候选" } }, error: null });
|
||||
const { result } = renderHook(() => useContinue());
|
||||
await act(async () => {
|
||||
await result.current.generate("p1", 1);
|
||||
});
|
||||
act(() => result.current.reset());
|
||||
expect(result.current.status).toBe("idle");
|
||||
expect(result.current.candidates).toEqual([]);
|
||||
});
|
||||
});
|
||||
72
apps/web/lib/workbench/useContinue.ts
Normal file
72
apps/web/lib/workbench/useContinue.ts
Normal file
@@ -0,0 +1,72 @@
|
||||
"use client";
|
||||
|
||||
import { useCallback, useState } from "react";
|
||||
|
||||
import { api } from "@/lib/api/client";
|
||||
import { useToast } from "@/components/Toast";
|
||||
import { generationErrorMessage } from "@/lib/generation/cards";
|
||||
|
||||
// 编辑器「续写」:复用 continue 生成器(with_prior_chapter,自动读该章已写正文承接),
|
||||
// 每次生成一条候选并累加成多候选,作者点选某条插入正文(HITL,不静默写入)。
|
||||
export type ContinueStatus = "idle" | "generating" | "ready" | "error";
|
||||
|
||||
export interface UseContinue {
|
||||
status: ContinueStatus;
|
||||
candidates: string[];
|
||||
// 生成一条续写候选(累加,不覆盖);chapterNo 指定承接哪一章。
|
||||
generate: (projectId: string, chapterNo: number) => Promise<void>;
|
||||
reset: () => void;
|
||||
}
|
||||
|
||||
function extractText(preview: unknown): string {
|
||||
if (preview && typeof preview === "object" && "text" in preview) {
|
||||
const text = (preview as { text: unknown }).text;
|
||||
return typeof text === "string" ? text : "";
|
||||
}
|
||||
return "";
|
||||
}
|
||||
|
||||
export function useContinue(): UseContinue {
|
||||
const [status, setStatus] = useState<ContinueStatus>("idle");
|
||||
const [candidates, setCandidates] = useState<string[]>([]);
|
||||
const toast = useToast();
|
||||
|
||||
const generate = useCallback<UseContinue["generate"]>(
|
||||
async (projectId, chapterNo) => {
|
||||
setStatus("generating");
|
||||
try {
|
||||
const { data, error } = await api.POST(
|
||||
"/projects/{project_id}/skills/{tool_key}/generate",
|
||||
{
|
||||
params: { path: { project_id: projectId, tool_key: "continue" } },
|
||||
body: { brief: "", chapter_no: chapterNo },
|
||||
},
|
||||
);
|
||||
if (error || !data) {
|
||||
setStatus("error");
|
||||
toast(generationErrorMessage(error), "error");
|
||||
return;
|
||||
}
|
||||
const text = extractText(data.preview).trim();
|
||||
if (text.length === 0) {
|
||||
setStatus("ready");
|
||||
toast("本次未生成有效续写,请再试一次。", "info");
|
||||
return;
|
||||
}
|
||||
setCandidates((prev) => [...prev, text]);
|
||||
setStatus("ready");
|
||||
} catch {
|
||||
setStatus("error");
|
||||
toast("续写请求异常,请检查网络。", "error");
|
||||
}
|
||||
},
|
||||
[toast],
|
||||
);
|
||||
|
||||
const reset = useCallback((): void => {
|
||||
setStatus("idle");
|
||||
setCandidates([]);
|
||||
}, []);
|
||||
|
||||
return { status, candidates, generate, reset };
|
||||
}
|
||||
Reference in New Issue
Block a user