@allbluecn/web-app 0.3.0

package/src/lib/ai/__tests__/story-ai-stream.test.ts

@@ -0,0 +1,280 @@
+// SPDX-License-Identifier: AGPL-3.0-or-later
+// AllBlue - 理想之海
+// Copyright (C) 2026 AllBlue Contributors
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program. If not, see <https://www.gnu.org/licenses/>.
+
+import { describe, it, expect, vi, afterEach } from "vitest"
+import { parseSseDelta, parseSseError, fetchStoryAiStream } from "../story-ai-stream"
+
+function sseResponse(lines: string[]): Response {
+  const body = new ReadableStream<Uint8Array>({
+    start(controller) {
+      for (const line of lines) {
+        controller.enqueue(new TextEncoder().encode(line))
+      }
+      controller.close()
+    },
+  })
+  return new Response(body, { status: 200, headers: { "Content-Type": "text/event-stream" } })
+}
+
+function sseLine(delta: string): string {
+  return `data: ${JSON.stringify({ type: "TEXT_MESSAGE_CONTENT", delta })}\n\n`
+}
+
+describe("parseSseDelta", () => {
+  it("TEXT_MESSAGE_CONTENT 事件返回 delta 文本", () => {
+    const line = `data: {"type":"TEXT_MESSAGE_CONTENT","delta":"{\\"name\\":\\"用户登录\\"}"}`
+    expect(parseSseDelta(line)).toBe('{"name":"用户登录"}')
+  })
+
+  it("RUN_COMPLETE 事件返回 null（不产生 delta）", () => {
+    const line = `data: {"type":"RUN_COMPLETE","reason":"done"}`
+    expect(parseSseDelta(line)).toBeNull()
+  })
+
+  it("RUN_ERROR 事件返回 null", () => {
+    const line = `data: {"type":"RUN_ERROR","error":{"message":"boom"}}`
+    expect(parseSseDelta(line)).toBeNull()
+  })
+
+  it("非 data 行返回 null", () => {
+    expect(parseSseDelta("event: text")).toBeNull()
+    expect(parseSseDelta("")).toBeNull()
+  })
+
+  it("data 内容非法 JSON 返回 null 不抛异常", () => {
+    expect(parseSseDelta("data: not-json")).toBeNull()
+  })
+
+  it("delta 非字符串类型返回 null", () => {
+    const line = `data: {"type":"TEXT_MESSAGE_CONTENT","delta":123}`
+    expect(parseSseDelta(line)).toBeNull()
+  })
+})
+
+describe("parseSseError", () => {
+  it("RUN_ERROR 事件返回错误消息文本", () => {
+    const line = `data: {"type":"RUN_ERROR","error":{"message":"AI 未配置：缺少 API Key"}}`
+    expect(parseSseError(line)).toBe("AI 未配置：缺少 API Key")
+  })
+
+  it("RUN_ERROR 顶层 message 字段也可解析（chat 内部错误事件路径）", () => {
+    const line = `data: {"type":"RUN_ERROR","model":"deepseek/deepseek-chat","timestamp":1,"message":"This request requires more credits","rawEvent":{"code":402}}`
+    expect(parseSseError(line)).toBe("This request requires more credits")
+  })
+
+  it("非 RUN_ERROR 事件返回 null", () => {
+    const line = `data: {"type":"TEXT_MESSAGE_CONTENT","delta":"x"}`
+    expect(parseSseError(line)).toBeNull()
+  })
+
+  it("RUN_ERROR 缺少 error.message 返回 null", () => {
+    expect(parseSseError(`data: {"type":"RUN_ERROR"}`)).toBeNull()
+    expect(parseSseError(`data: {"type":"RUN_ERROR","error":{}}`)).toBeNull()
+  })
+
+  it("非 data 行或非法 JSON 返回 null", () => {
+    expect(parseSseError("event: error")).toBeNull()
+    expect(parseSseError("data: not-json")).toBeNull()
+  })
+})
+
+describe("fetchStoryAiStream", () => {
+  const originalFetch = globalThis.fetch
+
+  afterEach(() => {
+    globalThis.fetch = originalFetch
+    vi.restoreAllMocks()
+  })
+
+  it("逐行解析 delta 并触发 onDelta，onDone 在流结束后触发", async () => {
+    const lines = [
+      sseLine('{"name":"第一条"}'),
+      sseLine('{"name":"第二条"}'),
+    ]
+    globalThis.fetch = vi.fn().mockResolvedValue(sseResponse(lines))
+
+    const deltas: string[] = []
+    let doneCalled = false
+    const controller = fetchStoryAiStream(
+      { mode: "create", prompt: "x" },
+      {
+        onDelta: (d) => deltas.push(d),
+        onDone: () => {
+          doneCalled = true
+        },
+      },
+    )
+
+    await vi.waitFor(() => expect(doneCalled).toBe(true))
+    expect(deltas).toEqual(['{"name":"第一条"}', '{"name":"第二条"}'])
+    expect(controller.signal.aborted).toBe(false)
+  })
+
+  it("流结束无换行的残留行也触发 onDelta（flush）", async () => {
+    // 最后一块不换行，模拟服务端输出末段后连接断开
+    const last = `data: ${JSON.stringify({ type: "TEXT_MESSAGE_CONTENT", delta: '{"name":"末段"' })}`
+    globalThis.fetch = vi.fn().mockResolvedValue(sseResponse([sseLine('{"name":"前段"}'), last]))
+
+    const deltas: string[] = []
+    const donePromise = new Promise<void>((resolve) => {
+      fetchStoryAiStream(
+        { mode: "create", prompt: "x" },
+        {
+          onDelta: (d) => deltas.push(d),
+          onDone: resolve,
+        },
+      )
+    })
+
+    await donePromise
+    expect(deltas).toEqual(['{"name":"前段"}', '{"name":"末段"'])
+  })
+
+  it("非 2xx 响应触发 onError", async () => {
+    globalThis.fetch = vi.fn().mockResolvedValue(new Response(null, { status: 500 }))
+
+    const errorPromise = new Promise<Error>((resolve) => {
+      fetchStoryAiStream(
+        { mode: "create", prompt: "x" },
+        {
+          onDelta: () => {},
+          onError: (err) => resolve(err),
+        },
+      )
+    })
+
+    const err = await errorPromise
+    expect(err.message).toContain("500")
+  })
+
+  it("流读取抛错触发 onError（非 abort 场景）", async () => {
+    globalThis.fetch = vi.fn().mockResolvedValue(
+      new Response(
+        new ReadableStream<Uint8Array>({
+          start(controller) {
+            controller.error(new Error("网络中断"))
+          },
+        }),
+        { status: 200 },
+      ),
+    )
+
+    const errorPromise = new Promise<Error>((resolve) => {
+      fetchStoryAiStream(
+        { mode: "create", prompt: "x" },
+        {
+          onDelta: () => {},
+          onError: (err) => resolve(err),
+        },
+      )
+    })
+
+    const err = await errorPromise
+    expect(err.message).toBe("网络中断")
+  })
+
+  it("abort 后不触发 onError", async () => {
+    globalThis.fetch = vi.fn().mockImplementation(
+      (_url: string, init?: RequestInit) =>
+        new Promise((_resolve, reject) => {
+          init?.signal?.addEventListener("abort", () =>
+            reject(new DOMException("Aborted", "AbortError")),
+          )
+        }),
+    )
+
+    let errorCalled = false
+    const controller = fetchStoryAiStream(
+      { mode: "create", prompt: "x" },
+      {
+        onDelta: () => {},
+        onError: () => {
+          errorCalled = true
+        },
+      },
+    )
+
+    controller.abort()
+    await vi.waitFor(() => expect(controller.signal.aborted).toBe(true))
+    await new Promise((r) => setTimeout(r, 10))
+    expect(errorCalled).toBe(false)
+  })
+
+  it("RUN_ERROR 事件触发 onError 而非 onDone", async () => {
+    const errLine = `data: ${JSON.stringify({ type: "RUN_ERROR", error: { message: "AI 未配置：缺少 API Key" } })}\n\n`
+    globalThis.fetch = vi.fn().mockResolvedValue(sseResponse([errLine]))
+
+    let doneCalled = false
+    const errorPromise = new Promise<Error>((resolve) => {
+      fetchStoryAiStream(
+        { mode: "create", prompt: "x" },
+        {
+          onDelta: () => {},
+          onDone: () => {
+            doneCalled = true
+          },
+          onError: (err) => resolve(err),
+        },
+      )
+    })
+
+    const err = await errorPromise
+    expect(err.message).toBe("AI 未配置：缺少 API Key")
+    expect(doneCalled).toBe(false)
+  })
+
+  it("RUN_ERROR 出现在残留行也触发 onError", async () => {
+    const last = `data: ${JSON.stringify({ type: "RUN_ERROR", error: { message: "boom" } })}`
+    globalThis.fetch = vi.fn().mockResolvedValue(sseResponse([last]))
+
+    const errorPromise = new Promise<Error>((resolve) => {
+      fetchStoryAiStream(
+        { mode: "create", prompt: "x" },
+        {
+          onDelta: () => {},
+          onError: (err) => resolve(err),
+        },
+      )
+    })
+
+    const err = await errorPromise
+    expect(err.message).toBe("boom")
+  })
+
+  it("顶层 message 结构的 RUN_ERROR 触发 onError 而非 onDone", async () => {
+    const errLine = `data: ${JSON.stringify({ type: "RUN_ERROR", model: "deepseek/deepseek-chat", message: "credits 不足", rawEvent: { code: 402 } })}\n\n`
+    globalThis.fetch = vi.fn().mockResolvedValue(sseResponse([errLine]))
+
+    let doneCalled = false
+    const errorPromise = new Promise<Error>((resolve) => {
+      fetchStoryAiStream(
+        { mode: "create", prompt: "x" },
+        {
+          onDelta: () => {},
+          onDone: () => {
+            doneCalled = true
+          },
+          onError: (err) => resolve(err),
+        },
+      )
+    })
+
+    const err = await errorPromise
+    expect(err.message).toBe("credits 不足")
+    expect(doneCalled).toBe(false)
+  })
+})

package/src/lib/ai/__tests__/story-jsonl.test.ts

@@ -0,0 +1,186 @@
+// SPDX-License-Identifier: AGPL-3.0-or-later
+// AllBlue - 理想之海
+// Copyright (C) 2026 AllBlue Contributors
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program. If not, see <https://www.gnu.org/licenses/>.
+
+import { describe, it, expect } from "vitest"
+import { parseJsonlBuffer, parseFinalBuffer } from "../story-jsonl"
+
+describe("parseJsonlBuffer 增量解析", () => {
+  it("完整多行解析出全部条目", () => {
+    const { items, rest, skippedCount } = parseJsonlBuffer(
+      '{"name":"用户登录","priority":"high"}\n{"name":"找回密码","description":"支持邮箱验证"}\n',
+    )
+    expect(items).toHaveLength(2)
+    expect(items[0]).toEqual({ name: "用户登录", priority: "high" })
+    expect(items[1]).toMatchObject({ name: "找回密码", description: "支持邮箱验证" })
+    expect(rest).toBe("")
+    expect(skippedCount).toBe(0)
+  })
+
+  it("最后一行无换行视为残尾返回 rest", () => {
+    const { items, rest, skippedCount } = parseJsonlBuffer(
+      '{"name":"已完成条"}\n{"name":"不完整条',
+    )
+    expect(items).toHaveLength(1)
+    expect(rest).toBe('{"name":"不完整条')
+    expect(skippedCount).toBe(0)
+  })
+
+  it("markdown 代码块包裹时仍能解析", () => {
+    const { items, skippedCount } = parseJsonlBuffer(
+      '```json\n{"name":"被代码块包裹"}\n{"name":"第二条"}\n```\n',
+    )
+    expect(items).toHaveLength(2)
+    expect(skippedCount).toBe(0)
+  })
+
+  it("夹杂解释文字的脏行被跳过并计数", () => {
+    const { items, skippedCount } = parseJsonlBuffer(
+      '下面是为您生成的 3 条需求：\n{"name":"A"}\n这是第二条\n{"name":"B"}\n',
+    )
+    expect(items.map((i) => i.name)).toEqual(["A", "B"])
+    expect(skippedCount).toBe(2)
+  })
+
+  it("单行数组包装剥离括号后逐条解析", () => {
+    const { items, skippedCount } = parseJsonlBuffer(
+      '[{"name":"数组一","priority":"medium"},{"name":"数组二"}]',
+    )
+    expect(items).toHaveLength(2)
+    expect(items[0].priority).toBe("medium")
+    expect(skippedCount).toBe(0)
+  })
+
+  it("空输入返回空结果", () => {
+    const { items, rest, skippedCount } = parseJsonlBuffer("")
+    expect(items).toHaveLength(0)
+    expect(rest).toBe("")
+    expect(skippedCount).toBe(0)
+  })
+
+  it("name 缺失或非法类型的条目被剔除并计数", () => {
+    const { items, skippedCount } = parseJsonlBuffer(
+      '{"priority":"high"}\n{"name":"","priority":"low"}\n{"name":"有效条"}\n',
+    )
+    expect(items).toHaveLength(1)
+    expect(items[0].name).toBe("有效条")
+    expect(skippedCount).toBe(2)
+  })
+
+  it("name 为纯空格字符串被剔除（避免 trim 后为空导致创建崩溃）", () => {
+    const { items, skippedCount } = parseJsonlBuffer(
+      '{"name":"   ","priority":"low"}\n{"name":"有效条"}\n',
+    )
+    expect(items).toHaveLength(1)
+    expect(items[0].name).toBe("有效条")
+    expect(skippedCount).toBe(1)
+  })
+
+  it("name 前后带空格被自动 trim", () => {
+    const { items } = parseJsonlBuffer('{"name":"  需求一  "}\n')
+    expect(items).toHaveLength(1)
+    expect(items[0].name).toBe("需求一")
+  })
+
+  it("priority 非法值导致整条剔除", () => {
+    const { items, skippedCount } = parseJsonlBuffer(
+      '{"name":"bad-priority","priority":"super"}\n',
+    )
+    expect(items).toHaveLength(0)
+    expect(skippedCount).toBe(1)
+  })
+
+  it("labelNames 数组被保留", () => {
+    const { items } = parseJsonlBuffer(
+      '{"name":"带标签条","labelNames":["登录","认证"]}\n',
+    )
+    expect(items[0].labelNames).toEqual(["登录", "认证"])
+  })
+
+  it("容忍 AI 输出的额外字段（passthrough）", () => {
+    const { items, skippedCount } = parseJsonlBuffer(
+      '{"name":"带额外字段条","id":123,"priority":"high"}\n',
+    )
+    expect(items).toHaveLength(1)
+    expect(items[0].name).toBe("带额外字段条")
+    expect(skippedCount).toBe(0)
+  })
+
+  it("labelNames 非法类型导致整条剔除并计数", () => {
+    const { items, skippedCount } = parseJsonlBuffer(
+      '{"name":"非法标签","labelNames":"foo"}\n{"name":"非法标签2","labelNames":[1,2]}\n',
+    )
+    expect(items).toHaveLength(0)
+    expect(skippedCount).toBe(2)
+  })
+
+  it("priority 合法枚举值全部通过", () => {
+    const { items, skippedCount } = parseJsonlBuffer(
+      '{"name":"urgent条","priority":"urgent"}\n{"name":"low条","priority":"low"}\n{"name":"none条","priority":"none"}\n',
+    )
+    expect(items).toHaveLength(3)
+    expect(items[0].priority).toBe("urgent")
+    expect(items[1].priority).toBe("low")
+    expect(items[2].priority).toBe("none")
+    expect(skippedCount).toBe(0)
+  })
+
+  it("字符串值含右花括号不误截断对象边界", () => {
+    const { items, skippedCount } = parseJsonlBuffer(
+      '{"name":"带}号需求","description":"描述含 } 与 { 符号"}\n',
+    )
+    expect(items).toHaveLength(1)
+    expect(items[0].name).toBe("带}号需求")
+    expect(items[0].description).toBe("描述含 } 与 { 符号")
+    expect(skippedCount).toBe(0)
+  })
+
+  it("嵌套对象整条剔除并计数", () => {
+    const { items, skippedCount } = parseJsonlBuffer(
+      '{"name":"外层","meta":{"k":"v"}}\n{"name":"正常条"}\n',
+    )
+    expect(items).toHaveLength(1)
+    expect(items[0].name).toBe("正常条")
+    expect(skippedCount).toBe(1)
+  })
+
+  it("残尾为完整对象+截断对象混合时整体留作 rest", () => {
+    const { items, rest } = parseJsonlBuffer(
+      '{"name":"已完成条"}\n{"name":"a"} {"name":"截断中',
+    )
+    expect(items).toHaveLength(1)
+    expect(rest).toBe('{"name":"a"} {"name":"截断中')
+  })
+})
+
+describe("parseFinalBuffer 残尾兜底", () => {
+  it("残尾为单个完整对象时并入", () => {
+    const { items, skippedCount } = parseFinalBuffer('{"name":"残尾完整条"}')
+    expect(items).toHaveLength(1)
+    expect(items[0].name).toBe("残尾完整条")
+    expect(skippedCount).toBe(0)
+  })
+
+  it("残尾为截断对象时丢弃且不计入成功", () => {
+    const { items } = parseFinalBuffer('{"name":"截断未完成')
+    expect(items).toHaveLength(0)
+  })
+
+  it("空残尾返回空", () => {
+    const { items } = parseFinalBuffer("")
+    expect(items).toHaveLength(0)
+  })
+})

package/src/lib/ai/__tests__/story-prompt.test.ts

@@ -0,0 +1,78 @@
+// SPDX-License-Identifier: AGPL-3.0-or-later
+// AllBlue - 理想之海
+// Copyright (C) 2026 AllBlue Contributors
+//
+// This program is free software: you can redistribute it and/or modify
+// it under the terms of the GNU Affero General Public License as published by
+// the Free Software Foundation, either version 3 of the License, or
+// (at your option) any later version.
+//
+// This program is distributed in the hope that it will be useful,
+// but WITHOUT ANY WARRANTY; without even the implied warranty of
+// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+// GNU Affero General Public License for more details.
+//
+// You should have received a copy of the GNU Affero General Public License
+// along with this program. If not, see <https://www.gnu.org/licenses/>.
+
+import { describe, it, expect } from "vitest"
+import { buildStorySystemPrompt } from '@allbluecn/ai-gateway'
+
+describe("buildStorySystemPrompt", () => {
+  it("基础 prompt 包含 JSONL 输出约束", () => {
+    const prompt = buildStorySystemPrompt({
+      taskType: "create",
+      requirement: "实现用户登录功能",
+    })
+    expect(prompt).toContain("JSONL")
+    expect(prompt).toContain("实现用户登录功能")
+  })
+
+  it("拆解任务时包含父需求描述", () => {
+    const prompt = buildStorySystemPrompt({
+      taskType: "breakdown",
+      requirement: "实现购物车结算",
+      parentStory: { name: "购物车结算", description: "支持优惠券抵扣" },
+    })
+    expect(prompt).toContain("购物车结算")
+    expect(prompt).toContain("支持优惠券抵扣")
+  })
+
+  it("包含优先级枚举值", () => {
+    const prompt = buildStorySystemPrompt({ taskType: "create", requirement: "x" })
+    expect(prompt).toContain("urgent")
+    expect(prompt).toContain("high")
+    expect(prompt).toContain("medium")
+    expect(prompt).toContain("low")
+    expect(prompt).toContain("none")
+  })
+
+  it("包含可用标签列表", () => {
+    const prompt = buildStorySystemPrompt({
+      taskType: "create",
+      requirement: "x",
+      availableLabels: ["前端", "后端", "登录"],
+    })
+    expect(prompt).toContain("前端")
+    expect(prompt).toContain("后端")
+    expect(prompt).toContain("登录")
+  })
+
+  it("输出字段仅含 name/description/priority/labelNames", () => {
+    const prompt = buildStorySystemPrompt({ taskType: "create", requirement: "x" })
+    expect(prompt).toContain("name")
+    expect(prompt).toContain("description")
+    expect(prompt).toContain("priority")
+    expect(prompt).toContain("labelNames")
+  })
+
+  it("拆解任务的字段约束说明覆盖全部四个字段", () => {
+    const prompt = buildStorySystemPrompt({ taskType: "breakdown", requirement: "x" })
+    const fieldLines = prompt.split("\n").filter((l) => l.includes("字段"))
+    expect(fieldLines.length).toBeGreaterThanOrEqual(2)
+    const joined = fieldLines.join(" ")
+    expect(joined).toContain("name")
+    expect(joined).toContain("priority")
+    expect(joined).toContain("labelNames")
+  })
+})