agentworkshop 0.7.34 → 0.7.38

package/scripts/.mimosa/hook-state/sess_aeaecb58-8833-4b83-b93c-050467d21e03.mtxwtxme-61028-1f4519fbc4.baseline/7bbb789d3a527817118cf8399ed1861d6a962ade10878fd07223435d333a7089.source

@@ -0,0 +1,469 @@
+/**
+ * SchedulerLoop 调度循环测试(node + tsx 直跑,无浏览器)。
+ *
+ * 覆盖:
+ *  1. 定时 tick:SUBMITTED 任务自动 dispatch 给 worker → worker 自动接取完成 → child-completed → 父任务完成(端到端)
+ *  2. wake() 立即触发一轮(大 tickMs 下无定时 tick 仍能 dispatch)
+ *  3. supervise 抛错 → 内置规则引擎兜底 dispatch
+ *  4. stop() 后不再调度
+ *  5. createAgentImpl 工厂(mock/claude/omp/未知 harness)
+ *
+ * 装配:真实 :memory: repo + 真实 TaskEngine + 真实 ChannelRuntime + 真实 Mailbox
+ *      + 真实 AgentRuntime(MockAgentImpl lead/worker)+ 真实 SchedulerLoop。
+ */
+import { randomUUID } from 'node:crypto'
+import { openWorkshopDb } from '../server/services/workshop/db/database'
+import { createChannelRepo } from '../server/services/workshop/db/channel.repo'
+import { createAgentRepo } from '../server/services/workshop/db/agent.repo'
+import { createChannelAgentRepo } from '../server/services/workshop/db/channel-agent.repo'
+import { createTaskRepo } from '../server/services/workshop/db/task.repo'
+import type { TaskPatch } from '../server/services/workshop/db/task.repo'
+import { createMessageRepo } from '../server/services/workshop/db/message.repo'
+import { createSubscriptionRepo } from '../server/services/workshop/db/subscription.repo'
+import { TaskEngine } from '../server/services/workshop/runtime/task-engine'
+import type { WorkspaceTask } from '../server/services/workshop/types/task'
+import { ChannelRuntime } from '../server/services/workshop/runtime/channel-runtime'
+import { AgentRuntime } from '../server/services/workshop/runtime/agent-runtime'
+import type { ChannelBus } from '../server/services/workshop/runtime/agent-runtime'
+import { Mailbox, rowToMessage } from '../server/services/workshop/runtime/mailbox'
+import { SchedulerLoop } from '../server/services/workshop/runtime/scheduler-loop'
+import { MockAgentImpl } from '../server/services/workshop/agents/mock-agent'
+import { ClaudeSdkAgentImpl } from '../server/services/workshop/agents/claude-agent'
+import { createAgentImpl } from '../server/services/workshop/agents/factory'
+import { encodeTaskMode, type ModeConfig } from '../server/services/workshop/runtime/execution-mode'
+import type {
+  AgentEvent,
+  AgentInfo,
+  AgentInterface,
+  AgentWorkspace,
+  SupervisionDecision,
+} from '../server/services/workshop/agents/agent-interface'
+import type { A2AMessage, ChannelMail } from '../server/services/workshop/types/a2a'
+import { AppError } from '../server/utils/errors'
+
+let failures = 0
+function check(name: string, ok: boolean, detail = ''): void {
+  console.log(`${ok ? 'PASS' : 'FAIL'}  ${name}${detail ? ` — ${detail}` : ''}`)
+  if (!ok) failures += 1
+}
+
+function sleep(ms: number): Promise<void> {
+  return new Promise(r => setTimeout(r, ms))
+}
+
+async function waitUntil(cond: () => boolean, timeoutMs = 5000): Promise<boolean> {
+  const start = Date.now()
+  for (;;) {
+    if (cond()) return true
+    if (Date.now() - start >= timeoutMs) return cond()
+    await sleep(10)
+  }
+}
+
+/** AgentRow(模板)与 ChannelAgentRow(实例)共有字段;实例 id 才是运行时身份 */
+type RowWithConfig = { id: string, name: string, harness: string, configJson: string }
+
+function rowToAgentInfo(row: RowWithConfig, channelId: string, role: 'lead' | 'worker'): AgentInfo {
+  return {
+    id: row.id,
+    channelId,
+    name: row.name,
+    harness: row.harness,
+    role,
+    config: JSON.parse(row.configJson) as Record<string, unknown>,
+  }
+}
+
+interface WorkspaceDeps {
+  engine: TaskEngine
+  cr: ChannelRuntime
+  tasks: ReturnType<typeof createTaskRepo>
+  messages: ReturnType<typeof createMessageRepo>
+}
+
+/** 进程内 AgentWorkspace(委托真实 TaskEngine/ChannelRuntime/repo,模拟 manager 的最小能力面) */
+function buildWorkspace(agent: AgentInfo, deps: WorkspaceDeps): AgentWorkspace {
+  const { engine, cr, tasks, messages } = deps
+  return {
+    listAgents: async () => [],
+    dispatchTask: async () => {
+      throw new AppError(400, 'BAD_REQUEST', '测试 workspace 不使用 dispatchTask')
+    },
+    listTasks: async () => engine.list(agent.channelId),
+    getTask: async (taskId) => {
+      const task = engine.get(taskId)
+      if (!task) throw new AppError(404, 'NOT_FOUND', `任务不存在: ${taskId}`)
+      return task
+    },
+    reportTask: async ({ taskId, progress, artifact, message }) => {
+      const task = engine.get(taskId)
+      if (!task) throw new AppError(404, 'NOT_FOUND', `任务不存在: ${taskId}`)
+      const patch: TaskPatch = {}
+      if (progress !== undefined) patch.progress = progress
+      if (artifact) patch.artifacts = [...task.artifacts, artifact]
+      if (message) {
+        patch.history = [
+          ...task.history,
+          { messageId: randomUUID(), contextId: task.channelId, role: 'ROLE_AGENT' as const, parts: [{ text: message }] },
+        ]
+      }
+      tasks.update(taskId, patch)
+      return engine.get(taskId)!
+    },
+    completeTask: async (taskId, artifacts) => {
+      const completed = engine.complete(taskId, artifacts)
+      if (completed.parentId) {
+        engine.onChildCompleted(completed)
+        const parent = engine.get(completed.parentId)
+        if (parent) cr.getAgents().find(a => a.agentId === parent.assigneeId)?.wakeMailbox()
+      }
+      return completed
+    },
+    cancelTask: async taskId => engine.cancel(taskId, agent.id),
+    myQueue: async () => engine.queueViewOf(agent.channelId, agent.id),
+    queueOverview: async () => [],
+    updateTask: async (taskId, patch) => engine.updateTask(taskId, patch, agent.id),
+    reassignTask: async (taskId, toAgentId) => engine.reassign(taskId, toAgentId),
+    sendMessage: async ({ toAgentId, parts, metadata }) => {
+      const message: A2AMessage = {
+        messageId: randomUUID(),
+        contextId: agent.channelId,
+        role: 'ROLE_AGENT',
+        parts,
+        metadata: { ...(metadata ?? {}), 'x-aw-target-agent': toAgentId, 'x-aw-from-agent': agent.id },
+      }
+      cr.route(message)
+      return message
+    },
+    pollMailbox: async (limit = 100) =>
+      messages.listPendingByChannelAgent(agent.channelId, agent.id).slice(0, limit).map(rowToMessage),
+    listMail: async (opts) => {
+      const limit = Math.max(1, Math.min(500, opts?.limit ?? 200))
+      const rows = messages.listRecentByChannel(agent.channelId, limit)
+      const mails: ChannelMail[] = rows.map(r => ({
+        messageId: r.id,
+        taskId: r.taskId,
+        fromAgentId: r.fromAgentId,
+        toAgentId: r.toAgentId,
+        role: r.role as 'ROLE_USER' | 'ROLE_AGENT',
+        parts: (JSON.parse(r.partsJson) ?? []) as ChannelMail['parts'],
+        metadata: (JSON.parse(r.metadataJson) ?? {}) as Record<string, unknown>,
+        state: r.state,
+        createdAt: r.createdAt,
+        consumedAt: r.consumedAt,
+      }))
+      return opts?.agentId
+        ? mails.filter(m => m.fromAgentId === opts.agentId || m.toAgentId === opts.agentId)
+        : mails
+    },
+    subscribe: async () => {},
+    recallMemory: async () => [],
+    saveMemory: async () => { throw new Error('unused') },
+  }
+}
+
+interface Setup {
+  engine: TaskEngine
+  cr: ChannelRuntime
+  lead: AgentRuntime
+  worker: AgentRuntime
+  channelId: string
+  loop: SchedulerLoop
+}
+
+interface SetupOptions {
+  tickMs?: number
+  leadImpl?: AgentInterface
+  workerImpl?: AgentInterface
+}
+
+function setup(opts: SetupOptions = {}): Setup {
+  const db = openWorkshopDb(':memory:')
+  const channels = createChannelRepo(db)
+  const agents = createAgentRepo(db)
+  const tasks = createTaskRepo(db)
+  const messages = createMessageRepo(db)
+  const subscriptions = createSubscriptionRepo(db)
+  const engine = new TaskEngine({ tasks, messages })
+  const channel = channels.create({ name: 'test-channel' })
+
+  const channelAgents = createChannelAgentRepo(db)
+  const leadRow = agents.create({ name: 'lead', harness: 'mock', config: { delayMs: 0 } })
+  const workerRow = agents.create({ name: 'worker', harness: 'mock', config: { delayMs: 0 } })
+  // 实例行(独立身份 id)才是运行时身份:lead 派发时 assignee 用的是实例 id
+  const leadInst = channelAgents.create({ channelId: channel.id, templateId: leadRow.id, name: leadRow.name, harness: leadRow.harness, config: { delayMs: 0 }, role: 'lead' })
+  const workerInst = channelAgents.create({ channelId: channel.id, templateId: workerRow.id, name: workerRow.name, harness: workerRow.harness, config: { delayMs: 0 }, role: 'worker' })
+  const leadInfo = rowToAgentInfo(leadInst, channel.id, 'lead')
+  const workerInfo = rowToAgentInfo(workerInst, channel.id, 'worker')
+
+  const cr = new ChannelRuntime(channel.id, { taskEngine: engine, subscriptionRepo: subscriptions, channelAgents })
+  const bus: ChannelBus = { emit: () => {}, onEvent: () => () => {}, notifyTask: () => {}, notifyAgent: () => {}, onAgentStatus: () => {}, onTaskEvent: () => {}, wakeScheduler: () => {} }
+
+  const lead = new AgentRuntime(leadInfo, opts.leadImpl ?? new MockAgentImpl(leadInfo.config), {
+    mailbox: new Mailbox(messages, leadInfo.channelId, leadInfo.id, () => cr.wakeScheduler()),
+    taskEngine: engine,
+    bus,
+    workspace: buildWorkspace(leadInfo, { engine, cr, tasks, messages }),
+  })
+  const worker = new AgentRuntime(workerInfo, opts.workerImpl ?? new MockAgentImpl(workerInfo.config), {
+    mailbox: new Mailbox(messages, workerInfo.channelId, workerInfo.id, () => cr.wakeScheduler()),
+    taskEngine: engine,
+    bus,
+    workspace: buildWorkspace(workerInfo, { engine, cr, tasks, messages }),
+  })
+
+  cr.addAgent(lead)
+  cr.addAgent(worker)
+  lead.start()
+  worker.start()
+
+  const loop = new SchedulerLoop(cr, lead, { tickMs: opts.tickMs ?? 10 })
+  cr.scheduler = loop
+
+  return { engine, cr, lead, worker, channelId: channel.id, loop }
+}
+
+async function teardown(s: Setup): Promise<void> {
+  s.loop.stop()
+  await Promise.all([s.lead.stop(), s.worker.stop()])
+}
+
+function submitTask(s: Setup, title: string) {
+  return s.engine.create({
+    channelId: s.channelId,
+    creatorId: '',
+    assigneeId: s.lead.agentId,
+    title,
+    description: '统筹交付',
+  })
+}
+
+/** 提交 loop 模式主任务(描述按 [mode:loop][interval:..][max:..] 编码,与 manager.submitChannelTask 同构) */
+function submitLoopTask(s: Setup, title: string, config: ModeConfig) {
+  return s.engine.create({
+    channelId: s.channelId,
+    creatorId: '',
+    assigneeId: s.lead.agentId,
+    title,
+    description: encodeTaskMode('loop', config, '循环测试任务'),
+  })
+}
+
+/** 模拟 manager 的 loop 重提交回调:原样创建新一轮主任务(经调度循环 dispatch 给 worker) */
+function wireLoopResubmit(s: Setup): void {
+  s.loop.setLoopResubmitCallback((title, description) => {
+    s.engine.create({
+      channelId: s.channelId,
+      creatorId: '',
+      assigneeId: s.lead.agentId,
+      title,
+      description,
+    })
+  })
+}
+
+// ===== 场景 =====
+
+async function testAutoDispatchAndComplete(): Promise<void> {
+  console.log('\n--- 1. 定时 tick 自动 dispatch + worker 完成 + 父完成 ---')
+  const s = setup({ tickMs: 10 })
+  s.loop.start()
+  const parent = submitTask(s, '主任务')
+
+  const dispatched = await waitUntil(() =>
+    s.engine.list(s.channelId).some(t => t.parentId === parent.id && t.assigneeId === s.worker.agentId),
+  )
+  check('定时 tick dispatch 子任务给 worker', dispatched)
+
+  const completed = await waitUntil(() => s.engine.get(parent.id)?.state === 'COMPLETED')
+  check('worker 完成子任务 → child-completed → 父任务完成', completed, `state=${s.engine.get(parent.id)?.state}`)
+
+  const child = s.engine.list(s.channelId).find(t => t.parentId === parent.id)
+  check('子任务 COMPLETED 且进度 100', child?.state === 'COMPLETED' && child?.progress === 100, `state=${child?.state} progress=${child?.progress}`)
+
+  await teardown(s)
+}
+
+async function testWakeTriggersRound(): Promise<void> {
+  console.log('\n--- 2. wake 立即触发一轮(大 tickMs,无定时 tick) ---')
+  const s = setup({ tickMs: 100000 })
+  s.loop.start()
+  const parent = submitTask(s, '主任务')
+  s.loop.wake()
+
+  const dispatched = await waitUntil(
+    () => s.engine.list(s.channelId).some(t => t.parentId === parent.id && t.assigneeId === s.worker.agentId),
+    2000,
+  )
+  check('wake 立即触发 dispatch(无定时 tick)', dispatched)
+
+  await teardown(s)
+}
+
+class ThrowingSuperviseImpl implements AgentInterface {
+  async* run(): AsyncIterable<AgentEvent> {
+    // 无消息处理(lead 仅测试 supervise 抛错)
+  }
+
+  async supervise(): Promise<SupervisionDecision[]> {
+    throw new Error('supervise boom')
+  }
+}
+
+async function testSuperviseThrowFallsBackToRules(): Promise<void> {
+  console.log('\n--- 3. supervise 抛错 → 规则引擎兜底 dispatch ---')
+  const s = setup({ tickMs: 10, leadImpl: new ThrowingSuperviseImpl() })
+  s.loop.start()
+  const parent = submitTask(s, '主任务')
+
+  const dispatched = await waitUntil(() =>
+    s.engine.list(s.channelId).some(t => t.parentId === parent.id && t.assigneeId === s.worker.agentId),
+  )
+  check('supervise 抛错后规则引擎兜底 dispatch', dispatched)
+
+  await teardown(s)
+}
+
+class FlakyWorkerImpl implements AgentInterface {
+  private attempts = 0
+  private readonly fallback = new MockAgentImpl({ delayMs: 100 })
+
+  async* run(request: Parameters<NonNullable<AgentInterface['run']>>[0], ctx: Parameters<NonNullable<AgentInterface['run']>>[1]): AsyncIterable<AgentEvent> {
+    if (request.message.metadata?.['x-aw-task-kind'] === 'assign' && ctx.role === 'worker' && this.attempts++ === 0) {
+      yield { kind: 'error', error: { code: 'TEST_TRANSIENT', message: 'transient test failure' } }
+      return
+    }
+    yield* this.fallback.run(request, ctx)
+  }
+}
+
+async function testRetryKeepsParentAlive(): Promise<void> {
+  console.log('\n--- 4. 可重试子任务失败时不提前取消父任务 ---')
+  const s = setup({ tickMs: 10, workerImpl: new FlakyWorkerImpl() })
+  s.loop.start()
+  const parent = submitTask(s, '可重试主任务')
+
+  const retried = await waitUntil(() => {
+    const child = s.engine.list(s.channelId).find(t => t.parentId === parent.id)
+    return child?.retryCount === 1
+  })
+  const retriedChild = s.engine.list(s.channelId).find(t => t.parentId === parent.id)
+  check('失败子任务被重派', retried, `parent=${s.engine.get(parent.id)?.state}`)
+  check(
+    '重试期间父任务保持 WAITING',
+    s.engine.get(parent.id)?.state === 'WAITING'
+    && (retriedChild?.state === 'ASSIGNED' || retriedChild?.state === 'WORKING'),
+    `parent=${s.engine.get(parent.id)?.state} child=${retriedChild?.state}`,
+  )
+
+  const completed = await waitUntil(() => s.engine.get(parent.id)?.state === 'COMPLETED')
+  check('重试成功后父任务完成', completed, `state=${s.engine.get(parent.id)?.state}`)
+
+  await teardown(s)
+}
+
+async function testStopStopsScheduling(): Promise<void> {
+  console.log('\n--- 5. stop 后不再调度 ---')
+  const s = setup({ tickMs: 10 })
+  s.loop.start()
+  const first = submitTask(s, '任务一')
+  const firstDone = await waitUntil(() => s.engine.get(first.id)?.state === 'COMPLETED')
+  check('stop 前任务完整完成', firstDone)
+
+  s.loop.stop()
+
+  const second = submitTask(s, '任务二')
+  await sleep(200)
+  const state = s.engine.get(second.id)?.state
+  const hasChild = s.engine.list(s.channelId).some(t => t.parentId === second.id)
+  check('stop 后新任务不再被 dispatch', state === 'SUBMITTED' && !hasChild, `state=${state} hasChild=${hasChild}`)
+
+  await teardown(s)
+}
+
+async function testLoopIntervalAndMaxIterations(): Promise<void> {
+  console.log('\n--- 6. loop 模式:间隔生效 + maxIterations 后停止 ---')
+  const s = setup({ tickMs: 10 })
+  wireLoopResubmit(s)
+  s.loop.start()
+
+  const title = '循环任务'
+  const intervalMs = 300
+  const maxIterations = 2
+  const first = submitLoopTask(s, title, { intervalMs, maxIterations })
+  // mock lead dispatch 的子任务会继承父任务标题,统计主任务需排除子任务(parentId 存在)
+  const parents = (): WorkspaceTask[] =>
+    s.engine.list(s.channelId).filter(t => t.title === title && !t.parentId)
+
+  // 首轮执行完成(此时才开始等待间隔)
+  await waitUntil(() => s.engine.get(first.id)?.state === 'COMPLETED')
+  check('loop 首轮执行完成', s.engine.get(first.id)?.state === 'COMPLETED')
+
+  // 间隔后重提交:出现第 2 轮主任务,且间隔 ≥ intervalMs(从首轮完成时刻起算)
+  const firstDoneAt = Date.now()
+  const resubmitted = await waitUntil(() => {
+    const same = parents()
+    return same.length >= 2 && same[1]!.state === 'COMPLETED'
+  }, 5000)
+  const waitMs = Date.now() - firstDoneAt
+  check('间隔后重提交并完成第 2 轮', resubmitted, `主任务数=${parents().length}`)
+  check('等待时长 ≥ 配置间隔', waitMs >= intervalMs, `wait=${waitMs}ms interval=${intervalMs}ms`)
+
+  // 达到 maxIterations=2 后不再重提交:再等数个间隔,主任务数仍为 2
+  await sleep(intervalMs * 3)
+  const finalCount = parents().length
+  check('达到 maxIterations 后停止重提交', finalCount === maxIterations, `主任务数=${finalCount}(期望 ${maxIterations})`)
+
+  // 全部主任务均完成(无残留半截循环)
+  const loopTasks = parents()
+  check('所有轮次主任务均完成', loopTasks.every(t => t.state === 'COMPLETED'), loopTasks.map(t => t.state).join(','))
+
+  await teardown(s)
+}
+
+function testFactory(): void {
+  console.log('\n--- 6. createAgentImpl 工厂 ---')
+  const mk = (harness: string): AgentInfo => ({
+    id: randomUUID(),
+    channelId: 'ch',
+    name: 'a',
+    harness,
+    role: 'worker',
+    config: {},
+  })
+
+  check('mock → MockAgentImpl 实例', createAgentImpl(mk('mock')) instanceof MockAgentImpl)
+
+  check('claude → ClaudeSdkAgentImpl 实例', createAgentImpl(mk('claude')) instanceof ClaudeSdkAgentImpl)
+
+  try {
+    createAgentImpl(mk('unknown'))
+    check('未知 harness 抛 UNKNOWN_HARNESS', false, '未抛异常')
+  }
+  catch (e) {
+    const err = e as AppError
+    check(
+      '未知 harness 抛 UNKNOWN_HARNESS',
+      err instanceof AppError && err.code === 'UNKNOWN_HARNESS' && err.status === 400,
+      `code=${err?.code} status=${err?.status}`,
+    )
+  }
+}
+
+async function main(): Promise<void> {
+  await testAutoDispatchAndComplete()
+  await testWakeTriggersRound()
+  await testSuperviseThrowFallsBackToRules()
+  await testRetryKeepsParentAlive()
+  await testStopStopsScheduling()
+  await testLoopIntervalAndMaxIterations()
+  testFactory()
+
+  console.log(`\n${failures === 0 ? 'ALL PASS' : `${failures} FAILED`}`)
+  process.exit(failures === 0 ? 0 : 1)
+}
+
+main().catch((e) => {
+  console.error('测试异常:', e)
+  process.exit(1)
+})