agentworkshop 0.7.41 → 0.7.44

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

@@ -1,469 +0,0 @@
-/**
- * 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)
-})