fix: 回合活跃判定放宽——压缩/通知行尾随不再误收起

2026-09-10 02:16:26 +08:00
parent 367f3f5c5b
commit 46813c23cd
2 changed files with 92 additions and 3 deletions
@@ -0,0 +1,79 @@
+/**
+ * 回合事件时序冒烟：验证 agent_settled 是否会在多回合工作（含 >600ms 慢工具）中途发出。
+ * 背景：渲染层回合折叠绑定 agent_settled（54b99a1），若 settled 在回合间被发出，
+ * 折叠会随 live 抖动反复展开/收起。本脚本在隔离的临时 cwd 下跑一轮真实多回合工作。
+ * 运行：node scripts/smoke-turn-events.mjs
+ */
+import { mkdtempSync } from 'node:fs'
+import { tmpdir } from 'node:os'
+import { join } from 'node:path'
+import {
+  createAgentSessionServices,
+  createAgentSessionFromServices,
+  defineTool,
+  getAgentDir,
+  SessionManager
+} from '@earendil-works/pi-coding-agent'
+import { Type } from '@earendil-works/pi-ai'
+
+const cwd = mkdtempSync(join(tmpdir(), 'aso-smoke-'))
+const t0 = Date.now()
+const ts = () => `${((Date.now() - t0) / 1000).toFixed(2).padStart(7)}s`
+
+const slowTool = defineTool({
+  name: 'slow_probe',
+  label: '慢探针',
+  description: '等待 1500ms 后返回 done（模拟浏览器工具耗时，测试回合折叠时序）',
+  promptSnippet: 'slow_probe: 测试用慢工具',
+  parameters: Type.Object({}),
+  async execute() {
+    await new Promise((r) => setTimeout(r, 1500))
+    return { content: [{ type: 'text', text: 'done' }] }
+  }
+})
+
+const loggingExtension = {
+  name: 'smoke-log',
+  factory: (pi) => {
+    for (const type of ['session_start', 'agent_start', 'agent_end', 'agent_settled', 'message_start', 'message_end', 'tool_call']) {
+      pi.on(type, (event) => console.log(ts(), `[ext] ${type}${event?.toolName ? ` (${event.toolName})` : ''}`))
+    }
+  }
+}
+
+const toolExtension = {
+  name: 'smoke-tools',
+  factory: (pi) => {
+    pi.registerTool(slowTool)
+  }
+}
+
+const services = await createAgentSessionServices({
+  cwd,
+  agentDir: getAgentDir(),
+  resourceLoaderOptions: { extensionFactories: [loggingExtension, toolExtension] }
+})
+
+const sessionManager = SessionManager.continueRecent(cwd)
+const { session } = await createAgentSessionFromServices({ services, sessionManager })
+
+const unsubscribe = session.subscribe((event) => {
+  const message = event.message ?? {}
+  const detail =
+    event.type === 'tool_execution_start' || event.type === 'tool_execution_end'
+      ? ` (${event.toolName})`
+      : event.type === 'message_start' || event.type === 'message_end'
+        ? ` (role=${message.role ?? '?'})`
+        : ''
+  console.log(ts(), `[session] ${event.type}${detail}`)
+})
+
+console.log(ts(), '--- prompt 开始：两次串行 slow_probe + 总结 ---')
+try {
+  await session.prompt('调用 slow_probe 工具两次（串行调用，不要并行），然后只回复"完成"两个字。')
+  console.log(ts(), '--- prompt 返回（整个工作结束） ---')
+} finally {
+  unsubscribe()
+  await session.abort?.().catch(() => undefined)
+  process.exit(0)
+}