@allbluecn/web-app 0.12.0 → 0.12.2

package/src/server/__tests__/sandbox-entitlement.test.ts

@@ -0,0 +1,169 @@
+import { beforeEach, describe, expect, it, vi } from 'vitest'
+import { assertSandboxEntitled } from '../ai-gateway/sandbox/entitlement'
+import { FeatureForbiddenError } from '../entitlements'
+import { ResourceLimitError } from '@/lib/resource-errors'
+
+const proSvc = vi.hoisted(() => ({}) as any)
+
+const db = vi.hoisted(() => ({
+  subscription: { findUnique: vi.fn() },
+  plan: { findFirst: vi.fn(), findUnique: vi.fn() },
+  teamMembership: { count: vi.fn() },
+  teamInvitation: { count: vi.fn() },
+  workspace: { count: vi.fn(), findUnique: vi.fn(), findFirst: vi.fn() },
+  user: { findUnique: vi.fn() },
+  aiUsageLog: { count: vi.fn(), aggregate: vi.fn() },
+  userAgent: { count: vi.fn() },
+  userSkill: { count: vi.fn() },
+  generationRun: { count: vi.fn().mockResolvedValue(0) },
+}))
+
+vi.mock('@/server/pro-services', () => ({
+  getProService: (key: string) => proSvc[key] ?? null,
+  isProEdition: () => Object.keys(proSvc).length > 0,
+}))
+
+vi.mock('@/server/auth-helpers', () => ({
+  requireAuth: vi.fn(),
+}))
+
+vi.mock('@/server/entitlements-token-client', () => ({
+  readVerifiedEntitlements: vi.fn().mockResolvedValue(null),
+}))
+
+vi.mock('@allbluecn/database', () => ({ prisma: db }))
+
+import { requireAuth } from '@/server/auth-helpers'
+
+const mockRequireAuth = requireAuth as ReturnType<typeof vi.fn>
+
+const PRO_PLAN_LIMITS = {
+  maxSeats: 5,
+  maxGuestSeats: 10,
+  maxWorkspaces: 1,
+  aiQuotaMonthly: 100,
+  agentQuota: 1,
+  skillQuota: 3,
+  sandboxMinutesMonthly: 600,
+}
+
+describe('assertSandboxEntitled：feature + 分钟配额双闸门', () => {
+  beforeEach(() => {
+    vi.clearAllMocks()
+    vi.unstubAllEnvs()
+    vi.stubEnv('DEPLOY_TARGET', 'web')
+
+    proSvc.billing = {
+      getEffectivePlan: vi.fn().mockResolvedValue({
+        planName: 'pro',
+        features: { sandboxExecution: true },
+      }),
+    }
+    proSvc.hostedAi = null
+
+    // 走 subscription ACTIVE 分支，limits 直接来自订阅的 plan 快照
+    db.subscription.findUnique.mockResolvedValue({
+      status: 'ACTIVE',
+      addonSeats: 0,
+      plan: { ...PRO_PLAN_LIMITS },
+    })
+    db.teamMembership.count.mockResolvedValue(0)
+    db.teamInvitation.count.mockResolvedValue(0)
+    db.workspace.count.mockResolvedValue(1)
+    db.aiUsageLog.count.mockResolvedValue(0)
+    db.generationRun.count.mockResolvedValue(0)
+    db.userAgent.count.mockResolvedValue(0)
+    db.userSkill.count.mockResolvedValue(0)
+    db.aiUsageLog.aggregate.mockResolvedValue({ _sum: { durationMs: null } })
+    mockRequireAuth.mockResolvedValue({ user: { id: 'user-1' } })
+  })
+
+  it('granted 用户 + 有余量 → 放行，且用量查询带 activity=sandbox', async () => {
+    await expect(
+      assertSandboxEntitled(new Request('http://test/api/chat')),
+    ).resolves.toBeUndefined()
+
+    expect(db.aiUsageLog.aggregate).toHaveBeenCalledTimes(1)
+    expect(db.aiUsageLog.aggregate).toHaveBeenCalledWith({
+      _sum: { durationMs: true },
+      where: {
+        userId: 'user-1',
+        activity: 'sandbox',
+        createdAt: { gte: expect.any(Date) },
+      },
+    })
+  })
+
+  it('feature 未开通 → FeatureForbiddenError，且不做配额查询（先闸后量）', async () => {
+    proSvc.billing.getEffectivePlan.mockResolvedValue({
+      planName: 'free',
+      features: { sandboxExecution: false },
+    })
+
+    await expect(assertSandboxEntitled(undefined)).rejects.toThrow(
+      FeatureForbiddenError,
+    )
+    expect(db.aiUsageLog.aggregate).not.toHaveBeenCalled()
+  })
+
+  it('配额耗尽 → ResourceLimitError（resource=sandboxMinutesMonthly）', async () => {
+    // 已用满 600 分钟：预检 amount=1 使 usage+1 > limit，下一轮被挡住
+    db.aiUsageLog.aggregate.mockResolvedValue({ _sum: { durationMs: 600 * 60_000 } })
+
+    const err = await assertSandboxEntitled(undefined).catch((e) => e)
+    expect(err).toBeInstanceOf(ResourceLimitError)
+    expect((err as ResourceLimitError).resource).toBe('sandboxMinutesMonthly')
+    expect((err as ResourceLimitError).limit).toBe(600)
+    expect((err as ResourceLimitError).usage).toBe(600)
+  })
+
+  it('不足一分钟按一分钟计（向上取整，防止 30 秒×2 绕过配额）', async () => {
+    db.subscription.findUnique.mockResolvedValue({
+      status: 'ACTIVE',
+      addonSeats: 0,
+      plan: { ...PRO_PLAN_LIMITS, sandboxMinutesMonthly: 0 },
+    })
+    db.aiUsageLog.aggregate.mockResolvedValue({ _sum: { durationMs: 30_000 } })
+
+    await expect(assertSandboxEntitled(undefined)).rejects.toThrow(
+      ResourceLimitError,
+    )
+  })
+
+  it('限额为 null = 不限量：用量再大也不拦', async () => {
+    db.subscription.findUnique.mockResolvedValue({
+      status: 'ACTIVE',
+      addonSeats: 0,
+      plan: { ...PRO_PLAN_LIMITS, sandboxMinutesMonthly: null },
+    })
+    db.aiUsageLog.aggregate.mockResolvedValue({
+      _sum: { durationMs: 366 * 24 * 60 * 60_000 },
+    })
+
+    await expect(assertSandboxEntitled(undefined)).resolves.toBeUndefined()
+  })
+
+  it('desktop 部署 → 两道检查都跳过（本机沙箱无订阅体系）', async () => {
+    vi.stubEnv('DEPLOY_TARGET', 'desktop')
+    // 即便 feature 未开通 + 配额为 0，desktop 也直放
+    proSvc.billing.getEffectivePlan.mockResolvedValue({
+      planName: 'open-source',
+      features: { sandboxExecution: false },
+    })
+    db.subscription.findUnique.mockResolvedValue({
+      status: 'ACTIVE',
+      addonSeats: 0,
+      plan: { ...PRO_PLAN_LIMITS, sandboxMinutesMonthly: 0 },
+    })
+
+    await expect(assertSandboxEntitled(undefined)).resolves.toBeUndefined()
+    expect(db.aiUsageLog.aggregate).not.toHaveBeenCalled()
+  })
+
+  it('desktop 豁免不影响 feature 检查本身（requireFeature 仍被调用一次）', async () => {
+    vi.stubEnv('DEPLOY_TARGET', 'desktop')
+    proSvc.billing = null
+
+    await expect(assertSandboxEntitled(undefined)).resolves.toBeUndefined()
+  })
+})

package/src/server/ai-gateway/__tests__/adapter-temperature.test.ts

@@ -0,0 +1,104 @@
+// @vitest-environment node
+// 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, beforeEach } from "vitest"
+
+type ResolveUnifiedAiStreamArg = { userId?: string; model?: string; temperature?: number; messages?: unknown[]; systemPrompts?: string[]; workspaceId?: string }
+
+const resolveUnifiedAiStreamMock = vi.hoisted(() =>
+  vi.fn(async (req: ResolveUnifiedAiStreamArg) =>
+    (async function* () { yield { type: "text", delta: "ok" } })(),
+  ),
+)
+
+vi.mock("../unified-router", () => ({
+  resolveUnifiedAiStream: resolveUnifiedAiStreamMock,
+}))
+
+import { runChatStream, runChat, runChatWithModel } from "../adapter"
+
+beforeEach(() => {
+  vi.clearAllMocks()
+})
+
+describe("runChatStream temperature 透传", () => {
+  it("modelOptions.temperature 为 number 时透传给 resolveUnifiedAiStream", async () => {
+    await runChatStream(
+      [{ role: "user", content: "hi" }],
+      { userId: "user-1", modelOptions: { temperature: 0.3 } },
+    )
+
+    expect(resolveUnifiedAiStreamMock).toHaveBeenCalledTimes(1)
+    const req = resolveUnifiedAiStreamMock.mock.calls[0]![0]
+    expect(req.temperature).toBe(0.3)
+    expect(req.model).toBeUndefined()
+  })
+
+  it("modelOptions 无 temperature 时不注入该字段", async () => {
+    await runChatStream(
+      [{ role: "user", content: "hi" }],
+      { userId: "user-1", modelOptions: { maxTokens: 100 } },
+    )
+
+    expect(resolveUnifiedAiStreamMock).toHaveBeenCalledTimes(1)
+    const req = resolveUnifiedAiStreamMock.mock.calls[0]![0]
+    expect(req.temperature).toBeUndefined()
+  })
+
+  it("无 modelOptions 时不注入 temperature", async () => {
+    await runChatStream([{ role: "user", content: "hi" }], { userId: "user-1" })
+
+    const req = resolveUnifiedAiStreamMock.mock.calls[0]![0]
+    expect(req.temperature).toBeUndefined()
+  })
+
+  it("modelOptions.temperature 为非 number 类型时不注入", async () => {
+    await runChatStream(
+      [{ role: "user", content: "hi" }],
+      { userId: "user-1", modelOptions: { temperature: "0.3" } },
+    )
+
+    const req = resolveUnifiedAiStreamMock.mock.calls[0]![0]
+    expect(req.temperature).toBeUndefined()
+  })
+})
+
+describe("runChat temperature 透传", () => {
+  it("modelOptions.temperature 经 runChatStream 透传给 resolveUnifiedAiStream", async () => {
+    await runChat(
+      [{ role: "user", content: "hi" }],
+      { userId: "user-1", modelOptions: { temperature: 0.2 } },
+    )
+
+    const req = resolveUnifiedAiStreamMock.mock.calls[0]![0]
+    expect(req.temperature).toBe(0.2)
+  })
+})
+
+describe("runChatWithModel temperature 透传", () => {
+  it("modelOptions.temperature 经 runChatStream 透传给 resolveUnifiedAiStream", async () => {
+    await runChatWithModel(
+      [{ role: "user", content: "hi" }],
+      { userId: "user-1", model: "gpt-4o", modelOptions: { temperature: 0.7 } },
+    )
+
+    const req = resolveUnifiedAiStreamMock.mock.calls[0]![0]
+    expect(req.temperature).toBe(0.7)
+    expect(req.model).toBe("gpt-4o")
+  })
+})

package/src/server/ai-gateway/__tests__/agent-runtime.test.ts

@@ -0,0 +1,334 @@
+// @vitest-environment node
+// 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, beforeEach } from "vitest"
+
+const mocks = vi.hoisted(() => {
+  const bindingFindFirst = vi.fn()
+  const defaultAgentFindFirst = vi.fn()
+  return {
+    bindingFindFirst,
+    defaultAgentFindFirst,
+    prisma: {
+      businessAgentBinding: { findFirst: bindingFindFirst },
+      userAgent: { findFirst: defaultAgentFindFirst },
+    },
+  }
+})
+
+vi.mock("@allbluecn/database", () => ({ prisma: mocks.prisma }))
+
+vi.mock("../prompts/registry", () => ({
+  resolveSystemPrompt: vi.fn(async () => ["BASE_P1", "BASE_P2"]),
+}))
+
+vi.mock("../adapter", () => ({
+  runChat: vi.fn(async () => "runChat-OK"),
+  runChatWithModel: vi.fn(async () => "runChatWithModel-OK"),
+  runChatStream: vi.fn(async () => (async function* () {
+    yield { type: "text", delta: "stream-OK" }
+  })()),
+}))
+
+import { resolveSystemPrompt } from "../prompts/registry"
+import { runChat as _runChat, runChatWithModel as _runChatWithModel, runChatStream as _runChatStream } from "../adapter"
+import {
+  BIZ_AGENT_KEYS,
+  BIZ_KEY_LABELS,
+  BIZ_KEY_TO_PROMPT,
+} from "../runs/biz-keys"
+import { resolveAgentFor, runChatAsAgent, runChatStreamAsAgent } from "../runs/agent-runtime"
+
+const runChat = vi.mocked(_runChat)
+const runChatWithModel = vi.mocked(_runChatWithModel)
+const runChatStream = vi.mocked(_runChatStream)
+
+const USER_ID = "user-1"
+
+function agentRow(overrides: Record<string, unknown> = {}) {
+  return {
+    id: "agent-1",
+    systemPrompt: "你是板块专属 Agent",
+    model: null,
+    temperature: null,
+    allowedTools: [],
+    knowledgeBaseIds: [],
+    ...overrides,
+  }
+}
+
+beforeEach(() => {
+  vi.clearAllMocks()
+  mocks.bindingFindFirst.mockResolvedValue(null)
+  mocks.defaultAgentFindFirst.mockResolvedValue(null)
+})
+
+describe("biz-keys", () => {
+  it("BIZ_KEY_TO_PROMPT 覆盖全部 9 个 bizKey，每个值都是非空 string", () => {
+    expect(Object.keys(BIZ_KEY_TO_PROMPT)).toHaveLength(9)
+    for (const key of BIZ_AGENT_KEYS) {
+      expect(BIZ_KEY_TO_PROMPT[key]).toEqual(expect.any(String))
+      expect(BIZ_KEY_TO_PROMPT[key]!.length).toBeGreaterThan(0)
+      expect(BIZ_KEY_LABELS[key]).toEqual(expect.any(String))
+    }
+  })
+
+  it("BIZ_KEY_TO_PROMPT 映射到已注册的内置 prompt key", () => {
+    expect(BIZ_KEY_TO_PROMPT["harness.default"]).toBe("chat.default")
+    expect(BIZ_KEY_TO_PROMPT["story.pm"]).toBe("story.generate")
+    expect(BIZ_KEY_TO_PROMPT["story.split"]).toBe("story.generate")
+    expect(BIZ_KEY_TO_PROMPT["prd.annotation"]).toBe("prd.fill-annotation")
+    expect(BIZ_KEY_TO_PROMPT["prd.prototype"]).toBe("prd.prototype")
+    expect(BIZ_KEY_TO_PROMPT["translation.run"]).toBe("translation.run")
+    expect(BIZ_KEY_TO_PROMPT["knowledge.curator"]).toBe("chat.default")
+    expect(BIZ_KEY_TO_PROMPT["label.suggest"]).toBe("story.label-suggest")
+  })
+})
+
+describe("resolveAgentFor", () => {
+  it("① 板块绑定命中：返回该 agent 的 persona/model，且 findFirst 的 where 含 bizKey", async () => {
+    mocks.bindingFindFirst.mockResolvedValue({
+      agent: agentRow({ model: "biz-model", temperature: 0.4 }),
+    })
+
+    const result = await resolveAgentFor("story.pm", USER_ID)
+
+    expect(result).toMatchObject({
+      agentId: "agent-1",
+      persona: "你是板块专属 Agent",
+      model: "biz-model",
+      temperature: 0.4,
+    })
+    expect(mocks.bindingFindFirst).toHaveBeenCalledTimes(1)
+    const call = mocks.bindingFindFirst.mock.calls[0][0]
+    expect(call.where).toMatchObject({ userId: USER_ID, bizKey: "story.pm" })
+    expect(call.where.agent).toEqual({ isDeleted: false })
+    // 绑定命中时不查默认 agent
+    expect(mocks.defaultAgentFindFirst).not.toHaveBeenCalled()
+  })
+
+  it("② 无绑定：用户默认 agent（isDefault）生效", async () => {
+    mocks.defaultAgentFindFirst.mockResolvedValue(
+      agentRow({ id: "agent-default", systemPrompt: "默认 persona", model: "def-model" }),
+    )
+
+    const result = await resolveAgentFor("translation.run", USER_ID)
+
+    expect(result).toMatchObject({
+      agentId: "agent-default",
+      persona: "默认 persona",
+      model: "def-model",
+    })
+    expect(mocks.bindingFindFirst).toHaveBeenCalledTimes(1)
+    expect(mocks.defaultAgentFindFirst).toHaveBeenCalledTimes(1)
+    const defCall = mocks.defaultAgentFindFirst.mock.calls[0][0]
+    expect(defCall.where).toMatchObject({
+      userId: USER_ID,
+      isDefault: true,
+      isDeleted: false,
+    })
+  })
+
+  it("③ 均无：返回 null", async () => {
+    const result = await resolveAgentFor("knowledge.curator", USER_ID)
+    expect(result).toBeNull()
+    expect(mocks.bindingFindFirst).toHaveBeenCalledTimes(1)
+    expect(mocks.defaultAgentFindFirst).toHaveBeenCalledTimes(1)
+  })
+
+  it("绑定指向已删 agent（mock 返回 null）：降级到默认 agent 链", async () => {
+    mocks.bindingFindFirst.mockResolvedValue(null)
+    mocks.defaultAgentFindFirst.mockResolvedValue(
+      agentRow({ id: "agent-fallback" }),
+    )
+
+    const result = await resolveAgentFor("prd.annotation", USER_ID)
+
+    expect(result).toMatchObject({ agentId: "agent-fallback" })
+  })
+
+  it("allowedTools 只保留合法 tool key，非法值被过滤", async () => {
+    mocks.bindingFindFirst.mockResolvedValue({
+      agent: agentRow({
+        allowedTools: ["knowledge_search", "nonexistent_tool", "web_search"],
+        knowledgeBaseIds: ["kb-1", 42, "kb-2"],
+      }),
+    })
+
+    const result = await resolveAgentFor("story.split", USER_ID)
+
+    expect(result?.allowedTools).toEqual(["knowledge_search", "web_search"])
+    expect(result?.knowledgeBaseIds).toEqual(["kb-1", "kb-2"])
+  })
+
+  it("temperature 为 null 时返回 undefined 而非 null", async () => {
+    mocks.bindingFindFirst.mockResolvedValue({
+      agent: agentRow({ temperature: null }),
+    })
+
+    const result = await resolveAgentFor("story.pm", USER_ID)
+
+    expect(result).not.toBeNull()
+    expect(result!.temperature).toBeUndefined()
+  })
+
+  it("temperature 缺失（undefined）时返回 undefined", async () => {
+    mocks.bindingFindFirst.mockResolvedValue({
+      agent: agentRow({ temperature: undefined }),
+    })
+
+    const result = await resolveAgentFor("story.pm", USER_ID)
+
+    expect(result?.temperature).toBeUndefined()
+  })
+
+  it("非数组的 Json 字段降级为空数组", async () => {
+    mocks.bindingFindFirst.mockResolvedValue({
+      agent: agentRow({ allowedTools: { bad: "shape" }, knowledgeBaseIds: "not-an-array" }),
+    })
+
+    const result = await resolveAgentFor("story.pm", USER_ID)
+
+    expect(result).toEqual({
+      agentId: "agent-1",
+      persona: "你是板块专属 Agent",
+    })
+  })
+})
+
+describe("runChatAsAgent", () => {
+  it("无 agent 命中：systemPrompts 走板块 prompt，runChat 路径不带 temperature", async () => {
+    await runChatAsAgent(
+      "harness.default",
+      [{ role: "user", content: "hi" }],
+      { userId: USER_ID },
+    )
+
+    expect(resolveSystemPrompt).toHaveBeenCalledWith(
+      "chat.default",
+      {},
+      { userId: USER_ID, workspaceName: undefined },
+    )
+    expect(runChat).toHaveBeenCalledTimes(1)
+    expect(runChatWithModel).not.toHaveBeenCalled()
+    expect(runChat.mock.calls[0]![1]!).toMatchObject({
+      systemPrompts: ["BASE_P1", "BASE_P2"],
+      userId: USER_ID,
+    })
+  })
+
+  it("agent 有 model：走 runChatWithModel，persona 前置到 systemPrompts 首位", async () => {
+    mocks.bindingFindFirst.mockResolvedValue({
+      agent: agentRow({ model: "agent-model", temperature: 0.7 }),
+    })
+
+    const text = await runChatAsAgent(
+      "story.pm",
+      [{ role: "user", content: "hi" }],
+      { userId: USER_ID, promptVars: { workspaceName: "ws" }, workspaceName: "ws" },
+    )
+
+    expect(text).toBe("runChatWithModel-OK")
+    expect(runChatWithModel).toHaveBeenCalledTimes(1)
+    expect(runChat).not.toHaveBeenCalled()
+    const opts = runChatWithModel.mock.calls[0]![1]!
+    expect(opts.model).toBe("agent-model")
+    expect(opts.userId).toBe(USER_ID)
+    expect(opts.systemPrompts![0]).toBe("你是板块专属 Agent")
+    expect(opts.systemPrompts!.slice(1)).toEqual(["BASE_P1", "BASE_P2"])
+    expect(resolveSystemPrompt).toHaveBeenCalledWith(
+      "story.generate",
+      { workspaceName: "ws" },
+      { userId: USER_ID, workspaceName: "ws" },
+    )
+  })
+
+  it("agent 无 model 但 opts 有 model：走 runChatWithModel（opts.model）", async () => {
+    mocks.bindingFindFirst.mockResolvedValue({
+      agent: agentRow({ model: null }),
+    })
+
+    await runChatAsAgent("story.pm", [{ role: "user", content: "hi" }], {
+      userId: USER_ID,
+      model: "caller-model",
+    })
+
+    expect(runChatWithModel).toHaveBeenCalledTimes(1)
+    expect(runChatWithModel.mock.calls[0]![1]!.model).toBe("caller-model")
+    expect(runChat).not.toHaveBeenCalled()
+  })
+
+  it("agent 有 model 优先于 opts.model", async () => {
+    mocks.bindingFindFirst.mockResolvedValue({
+      agent: agentRow({ model: "agent-model" }),
+    })
+
+    await runChatAsAgent("story.pm", [{ role: "user", content: "hi" }], {
+      userId: USER_ID,
+      model: "caller-model",
+    })
+
+    expect(runChatWithModel.mock.calls[0]![1]!.model).toBe("agent-model")
+  })
+
+  it("无 model 且有 temperature：runChat 携带 modelOptions.temperature", async () => {
+    await runChatAsAgent("story.pm", [{ role: "user", content: "hi" }], {
+      userId: USER_ID,
+      temperature: 0.3,
+    })
+
+    expect(runChat).toHaveBeenCalledTimes(1)
+    expect(runChat.mock.calls[0]![1]!.modelOptions).toEqual({ temperature: 0.3 })
+  })
+
+  it("agent.temperature 优先于 opts.temperature", async () => {
+    mocks.bindingFindFirst.mockResolvedValue({
+      agent: agentRow({ temperature: 0.9 }),
+    })
+
+    await runChatAsAgent("story.pm", [{ role: "user", content: "hi" }], {
+      userId: USER_ID,
+      temperature: 0.1,
+    })
+
+    expect(runChat.mock.calls[0]![1]!.modelOptions).toEqual({ temperature: 0.9 })
+  })
+})
+
+describe("runChatStreamAsAgent", () => {
+  it("agent 有 model：runChatStream 携带 model + persona 前置", async () => {
+    mocks.bindingFindFirst.mockResolvedValue({
+      agent: agentRow({ model: "stream-model" }),
+    })
+
+    const stream = await runChatStreamAsAgent(
+      "story.pm",
+      [{ role: "user", content: "hi" }],
+      { userId: USER_ID },
+    )
+
+    expect(runChatStream).toHaveBeenCalledTimes(1)
+    const opts = runChatStream.mock.calls[0]![1]!
+    expect(opts.model).toBe("stream-model")
+    expect(opts.userId).toBe(USER_ID)
+    expect(opts.systemPrompts![0]).toBe("你是板块专属 Agent")
+    const chunks = []
+    for await (const chunk of stream) chunks.push(chunk)
+    expect(chunks).toHaveLength(1)
+  })
+})