workspace.test.ts 106 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681
  1. // @vitest-environment node
  2. import { afterEach, beforeEach, describe, expect, it, vi } from 'vitest'
  3. import { mkdtempSync, rmSync } from 'node:fs'
  4. import { tmpdir } from 'node:os'
  5. import { join } from 'node:path'
  6. import { randomUUID } from 'node:crypto'
  7. import { WebSocketServer, type WebSocket } from 'ws'
  8. import { defaultAgentIdentity, encodeAgentAvatar, decodeAgentAvatar } from '../../src/shared/agentIdentity'
  9. import {readHostTools,saveHostTools} from '../../src/server/hostToolSettings'
  10. import {DesktopEnvironments} from '../../src/server/desktopEnvironments'
  11. import {HttpError} from '../../src/server/errors'
  12. import {WorkspaceAssets} from '../../src/server/workspaceAssets'
  13. import { WorkspaceStore } from '../../src/server/workspaceStore'
  14. import { WorkspaceRuntime, mentionedAgents } from '../../src/server/workspaceRuntime'
  15. import type { Work } from '../../src/server/workspaceScheduler'
  16. import { REPEATED_RELAY_NOTICE } from '../../src/server/workspaceRelay'
  17. import { WorkspaceNodes, WorkspaceGateway, type GatewayTarget } from '../../src/server/workspaceGateway'
  18. import { UploadStore } from '../../src/server/uploads'
  19. import type {
  20. WorkspaceConversation,
  21. WorkspaceRun,
  22. WorkspaceMessage,
  23. WorkspaceInteraction,
  24. } from '../../src/shared/workspace'
  25. let home: string,
  26. store: WorkspaceStore,
  27. uploads: UploadStore,
  28. runtime: WorkspaceRuntime,
  29. server: WebSocketServer
  30. let requests: Array<{ method: string; params: Record<string, any> }>,
  31. reply: (socket: WebSocket, params: Record<string, any>) => void
  32. let nodes: WorkspaceNodes
  33. let configuredCwds: Map<string, string>
  34. let rejectPrompt: string | undefined
  35. let rejectedInterrupts: number
  36. let recoveryHistory: Array<{ role: string; content: string; tool_calls?: unknown }>
  37. let storedByRuntime: Map<string, string>, recoveryByStored: Map<string, Array<{ role: string; content: string }>>
  38. const owner = 'first-user',
  39. other = 'second-user'
  40. beforeEach(async () => {
  41. home = mkdtempSync(join(tmpdir(), 'yaoyao-workspace-test-'))
  42. store = new WorkspaceStore(home)
  43. uploads = new UploadStore(home)
  44. requests = []
  45. configuredCwds = new Map()
  46. recoveryHistory = []
  47. storedByRuntime = new Map(); recoveryByStored = new Map()
  48. rejectPrompt = undefined
  49. rejectedInterrupts = 0
  50. server = new WebSocketServer({ port: 0, host: '127.0.0.1' })
  51. await new Promise<void>((resolve) => server.once('listening', resolve))
  52. const address = server.address() as { port: number }
  53. const target = {
  54. url: new URL(`http://127.0.0.1:${address.port}`),
  55. client: { directAgent: undefined },
  56. session: {
  57. webSocketCredential: async () => ({ name: 'ticket', value: 'test' }),
  58. request: async (path: string, options?: {search?: URLSearchParams}) => ({
  59. status: 200,
  60. body: Buffer.from(JSON.stringify(path === '/api/config'
  61. ? { terminal: { cwd: configuredCwds.get(options?.search?.get('profile') ?? '') } }
  62. : { messages: (() => { const all = recoveryByStored.get(decodeURIComponent(path.split('/')[3]!)) ?? recoveryHistory; const offset = Number(options?.search?.get('offset') ?? 0); return all.slice(Math.max(0,all.length-offset-500), all.length-offset) })() })),
  63. }),
  64. },
  65. }
  66. nodes = { requireSource: () => {}, target: () => target } as unknown as WorkspaceNodes
  67. runtime = new WorkspaceRuntime(store, nodes, uploads)
  68. reply = (socket, p) =>
  69. setTimeout(
  70. () =>
  71. socket.send(
  72. JSON.stringify({
  73. method: 'event',
  74. params: {
  75. type: 'message.complete',
  76. session_id: p.session_id,
  77. payload: { text: '完成', status: 'complete' },
  78. },
  79. }),
  80. ),
  81. 5,
  82. )
  83. server.on('connection', (socket) => {
  84. socket.send(JSON.stringify({ method: 'event', params: { type: 'gateway.ready' } }))
  85. socket.on('message', (data) => {
  86. const frame = JSON.parse(String(data))
  87. requests.push(frame)
  88. const respond = (result: unknown) => socket.send(JSON.stringify({ id: frame.id, result }))
  89. if (frame.method === 'session.create' || frame.method === 'session.resume') {
  90. const runtimeId = randomUUID(), storedId = frame.method === 'session.resume' ? frame.params.session_id : randomUUID()
  91. storedByRuntime.set(runtimeId, storedId)
  92. respond({ session_id: runtimeId, stored_session_id: storedId, running: false, info: { profile_name: frame.params.profile } })
  93. }
  94. else if (frame.method === 'session.cwd.set') respond({ cwd: frame.params.cwd })
  95. else if (frame.method === 'prompt.submit') {
  96. if (rejectPrompt) { socket.send(JSON.stringify({id:frame.id,error:{code:4006,message:rejectPrompt}})); return }
  97. respond({ status: 'streaming' })
  98. reply(socket, frame.params)
  99. } else if (frame.method === 'session.usage') respond({ context_used: 400, context_max: 1000 })
  100. else if (frame.method === 'session.interrupt' && rejectedInterrupts-- > 0) socket.send(JSON.stringify({id:frame.id,error:{code:500,message:'暂时不能中断'}}))
  101. else respond({ ok: true, status: 'interrupted' })
  102. })
  103. })
  104. })
  105. afterEach(async () => {
  106. runtime.close()
  107. await new Promise((resolve) => setTimeout(resolve, 10))
  108. for (const client of server.clients) client.terminate()
  109. await new Promise<void>((resolve) => server.close(() => resolve()))
  110. uploads.close()
  111. store.close()
  112. rmSync(home, { recursive: true, force: true })
  113. })
  114. function agent(name: string, user = owner) {
  115. return store.createAgent(user, { name, profile: 'default', instructions: `规则:你是${name}` })
  116. }
  117. it('executes a structured assignment on its actual worker and leaves acceptance to its coordinator', async () => {
  118. const lead = store.updateAgent(owner, agent('负责人').id, { canManageTeam: true })
  119. const worker = agent('研究成员')
  120. const source = direct(lead.id)
  121. const original = runtime.send(owner, source.id, { requestId: randomUUID(), content: '完成研究' })
  122. for (const work of store.list<any>(owner, 'turn').filter(w => w.runId === original.id))
  123. store.put(owner, 'turn', work.id, { ...work, status: 'complete', planned: true })
  124. original.status = 'complete';store.saveRun(owner, original)
  125. const team = store.createGroup(owner, { name: '研究团队', memberIds: [lead.id,worker.id], administratorId: lead.id })
  126. const goal = runtime.tasks.begin(owner, store.tasks(owner, team.id)[0]!, lead, '研究事实', { conversationId: source.id, runId: original.id, agentId: lead.id })
  127. const assignment = runtime.tasks.createAssignment(owner, lead.id, { requestId: randomUUID(), goalId: goal.id, agentId: worker.id, title: '核对事实', brief: '提交核对结果' })
  128. await vi.waitFor(() => expect(store.get<any>(owner,'assignment',assignment.id)?.status).toBe('review'), { timeout: 5000 })
  129. const submitted = requests.find(r => r.method === 'prompt.submit' && String(r.params.text).includes('核对事实'))!
  130. expect(submitted.params.text).toContain('你是 研究成员')
  131. expect(submitted.params.text).toContain('执行子任务')
  132. expect(store.get<any>(owner,'goal',goal.id)?.status).not.toBe('complete')
  133. })
  134. it('keeps structured goal coordination on the lead and does not relay mention text as duplicate work', async () => {
  135. const lead=store.updateAgent(owner,agent('交付负责人').id,{canManageTeam:true}),worker=agent('研究成员')
  136. const target=nodes.target(owner,'local'), request=target.session.request.bind(target.session)
  137. vi.spyOn(target.session,'request').mockImplementation(async(path,options)=>path.startsWith('/api/plugins/yaoyao-bot-bridge/')
  138. ? {status:200,body:Buffer.from(JSON.stringify({ok:true,version:1,ready:true,in_process:true,native_tools:true})),headers:new Headers()}
  139. : request(path,options))
  140. const team=store.createGroup(owner,{name:'自由协作交付',memberIds:[lead.id,worker.id],administratorId:lead.id,mode:'free',autoReplyIds:[worker.id]})
  141. reply=(socket,p)=>socket.send(JSON.stringify({method:'event',params:{type:'message.complete',session_id:p.session_id,payload:{text:'@研究成员 请核对资料',status:'complete'}}}))
  142. const run=runtime.send(owner,team.id,{requestId:randomUUID(),content:'交付研究报告',mode:'goal'})
  143. await vi.waitFor(()=>expect(['complete','failed']).toContain(store.require<WorkspaceRun>(owner,'run',run.id).status))
  144. const settled=store.require<WorkspaceRun>(owner,'run',run.id)
  145. expect(settled.status,settled.error).toBe('complete')
  146. const work=store.list<Work>(owner,'turn').filter(w=>w.runId===run.id)
  147. expect(work.map(w=>w.agentId)).toEqual([lead.id])
  148. expect(runtime.tasks.assignments(owner,run.goalId!)).toEqual([])
  149. expect(requests.filter(r=>r.method==='prompt.submit').every(r=>!String(r.params.text).startsWith('你是 研究成员'))).toBe(true)
  150. })
  151. function direct(id: string, user = owner) {
  152. return store
  153. .list<WorkspaceConversation>(user, 'conversation')
  154. .find((c) => c.kind === 'direct' && c.memberIds[0] === id)!
  155. }
  156. function defaultTask(conversationId: string, user = owner) {
  157. return store.tasks(user, conversationId)[0]!
  158. }
  159. async function finished(id: string) {
  160. await vi.waitFor(() =>
  161. expect(store.require<WorkspaceRun>(owner, 'run', id).status).toBe('complete'),
  162. )
  163. return store.require<WorkspaceRun>(owner, 'run', id)
  164. }
  165. describe('Web-owned workspace', () => {
  166. it('restores recent user and assistant context when moving a completed conversation to another runner',async()=>{
  167. const a=agent('移动成员'),c=direct(a.id)
  168. await finished(runtime.send(owner,c.id,{requestId:randomUUID(),content:'保存研究结论'}).id)
  169. const binding=store.list<any>(owner,'binding')[0]!
  170. store.put(owner,'binding',binding.id,{...binding,runnerId:'previous-runner'})
  171. const before=requests.length
  172. await finished(runtime.send(owner,c.id,{requestId:randomUUID(),content:'继续后续分析'}).id)
  173. const next=requests.slice(before)
  174. expect(next.some(r=>r.method==='session.resume')).toBe(false)
  175. const text=next.find(r=>r.method==='prompt.submit')!.params.text
  176. expect(text).toContain('执行节点已切换')
  177. expect(text).toContain('保存研究结论')
  178. expect(text).toContain('移动成员(complete):完成')
  179. expect(text).toContain('继续后续分析')
  180. })
  181. it('uses each base profile cwd and replaces historical cwd before the next Bot turn', async () => {
  182. configuredCwds.set('default', '/profile/first')
  183. configuredCwds.set('other', '/profile/other')
  184. const a = agent('主助手'), c = direct(a.id)
  185. await finished(runtime.send(owner, c.id, { requestId: randomUUID(), content: 'first' }).id)
  186. expect(requests.find(r => r.method === 'session.create')?.params.cwd).toBe('/profile/first')
  187. expect(requests.find(r => r.method === 'session.cwd.set')?.params.cwd).toBe('/profile/first')
  188. configuredCwds.set('default', '/profile/changed')
  189. const before = requests.length
  190. await finished(runtime.send(owner, c.id, { requestId: randomUUID(), content: 'second' }).id)
  191. // The prior turn's asynchronous usage refresh can arrive after completion.
  192. // This assertion concerns the next turn's execution RPCs, not telemetry.
  193. const next = requests.slice(before).filter(r => r.method !== 'session.usage')
  194. expect(next[0].method).toBe('session.resume')
  195. expect(next.find(r => r.method === 'session.cwd.set')?.params.cwd).toBe('/profile/changed')
  196. expect(next.findIndex(r => r.method === 'session.cwd.set')).toBeLessThan(next.findIndex(r => r.method === 'prompt.submit'))
  197. const b = store.createAgent(owner, { name: '其他 Profile', profile: 'other' })
  198. await finished(runtime.send(owner, direct(b.id).id, { requestId: randomUUID(), content: 'third' }).id)
  199. expect(requests.filter(r => r.method === 'session.create').at(-1)?.params.cwd).toBe('/profile/other')
  200. })
  201. it('flushes a paused delta before completion and cancels pending flush on finish', async () => {
  202. let upstream: { socket: WebSocket; params: Record<string, any> } | undefined
  203. reply = (socket, params) => {
  204. upstream = { socket, params }
  205. const event = (type: string, payload: Record<string, unknown>) => socket.send(JSON.stringify({
  206. method: 'event', params: { type, session_id: params.session_id, payload },
  207. }))
  208. event('message.start', {})
  209. event('message.delta', { text: '# 流式标题' })
  210. }
  211. const bot = agent('流式助手'), conversation = direct(bot.id)
  212. const run = runtime.send(owner, conversation.id, { requestId: randomUUID(), content: '验证暂停分片' })
  213. const assistant = () => store.list<WorkspaceMessage>(owner, 'message').find(m => m.runId === run.id && m.role === 'assistant')
  214. await vi.waitFor(() => expect(assistant()?.content).toBe('# 流式标题'))
  215. expect(assistant()?.status).toBe('streaming')
  216. expect(store.require<WorkspaceRun>(owner, 'run', run.id).status).not.toBe('complete')
  217. const { socket, params } = upstream!
  218. socket.send(JSON.stringify({ method: 'event', params: { type: 'message.delta', session_id: params.session_id, payload: { text: '\n\n最后一段' } } }))
  219. completeReply(socket, params, '# 最终正文')
  220. await finished(run.id)
  221. await new Promise(resolve => setTimeout(resolve, 150))
  222. expect(assistant()?.content).toBe('# 最终正文')
  223. expect(assistant()?.status).toBe('complete')
  224. })
  225. it('backfills legacy group history into the initial task instead of leaving a new empty task', () => {
  226. const legacyHome = mkdtempSync(join(tmpdir(), 'yaoyao-workspace-task-migration-'))
  227. let seededStore: WorkspaceStore | undefined = new WorkspaceStore(legacyHome)
  228. let migratedStore: WorkspaceStore | undefined
  229. try {
  230. const administrator = seededStore.createAgent(owner, { name: '历史管理员', profile: 'default' }),
  231. member = seededStore.createAgent(owner, { name: '历史成员', profile: 'default' }),
  232. conversation = seededStore.createGroup(owner, {
  233. name: '公众号文案团队',
  234. memberIds: [administrator.id, member.id],
  235. administratorId: administrator.id,
  236. }),
  237. initialTask = seededStore.tasks(owner, conversation.id)[0]!,
  238. userAt = 1_788_000_000_000,
  239. assistantAt = userAt + 60_000
  240. seededStore.put(owner, 'message', 'legacy-user', {
  241. id: 'legacy-user', conversationId: conversation.id, seq: 1, role: 'user',
  242. content: '请继续完成公众号文案', reasoning: '', status: 'complete',
  243. attachments: [], tools: [], createdAt: userAt,
  244. } satisfies WorkspaceMessage)
  245. seededStore.put(owner, 'message', 'legacy-assistant', {
  246. id: 'legacy-assistant', conversationId: conversation.id, seq: 2, role: 'assistant',
  247. agentId: administrator.id, agentName: '历史管理员', content: '旧内容已经完成。',
  248. reasoning: '', status: 'complete', attachments: [], tools: [], createdAt: assistantAt,
  249. } satisfies WorkspaceMessage)
  250. seededStore.put(owner, 'conversation', conversation.id, {
  251. ...conversation,
  252. readSeq: 1,
  253. lastSeq: 2,
  254. preview: '旧内容已经完成。',
  255. lastMessageAt: assistantAt,
  256. updatedAt: assistantAt,
  257. })
  258. seededStore.put(owner, 'run', 'legacy-run', {
  259. id: 'legacy-run', conversationId: conversation.id, messageId: 'legacy-user',
  260. mentionIds: [], status: 'complete', round: 1, createdAt: userAt, updatedAt: assistantAt,
  261. } satisfies WorkspaceRun)
  262. seededStore.put(owner, 'turn', 'legacy-turn', {
  263. id: 'legacy-turn', runId: 'legacy-run', conversationId: conversation.id,
  264. agentId: administrator.id, messageId: 'legacy-user', status: 'complete', createdAt: userAt,
  265. })
  266. seededStore.put(owner, 'interaction', 'legacy-interaction', {
  267. id: 'legacy-interaction', conversationId: conversation.id, runId: 'legacy-run',
  268. agentId: administrator.id, kind: 'clarification', message: '补充说明',
  269. choices: [], resolved: true,
  270. } satisfies WorkspaceInteraction)
  271. seededStore.put(owner, 'interaction-binding', 'legacy-interaction', {
  272. key: `${conversation.id}:${administrator.id}`,
  273. upstreamId: 'upstream-interaction', taskId: 'legacy-turn',
  274. })
  275. seededStore.put(owner, 'binding', `${conversation.id}:${administrator.id}`, {
  276. nodeId: 'local', profile: 'default', storedId: 'stored-legacy',
  277. runtimeId: 'runtime-legacy', aliases: [], runId: 'legacy-run',
  278. messageId: 'legacy-user', taskId: 'legacy-turn',
  279. })
  280. seededStore.put(owner, 'context', conversation.id, {
  281. usedTokens: 100, limitTokens: 1_000, observedAt: assistantAt,
  282. })
  283. seededStore.db.prepare('DELETE FROM workspace_migrations WHERE id=?')
  284. .run('conversation-task-history-v1')
  285. seededStore.close()
  286. seededStore = undefined
  287. migratedStore = new WorkspaceStore(legacyHome)
  288. const tasks = migratedStore.tasks(owner, conversation.id)
  289. expect(tasks).toHaveLength(1)
  290. expect(tasks[0]).toMatchObject({
  291. id: initialTask.id,
  292. title: '请继续完成公众号文案',
  293. titleSource: 'automatic',
  294. messageCount: 2,
  295. readSeq: 1,
  296. lastSeq: 2,
  297. unreadCount: 1,
  298. lastMessageAt: assistantAt,
  299. createdAt: userAt,
  300. updatedAt: assistantAt,
  301. })
  302. expect(
  303. migratedStore.messages(
  304. owner, conversation.id, Number.MAX_SAFE_INTEGER, 100, false, initialTask.id,
  305. ).map(message => message.id),
  306. ).toEqual(['legacy-user', 'legacy-assistant'])
  307. expect(migratedStore.require<WorkspaceRun>(owner, 'run', 'legacy-run').conversationTaskId)
  308. .toBe(initialTask.id)
  309. expect(migratedStore.require<{ conversationTaskId: string }>(owner, 'turn', 'legacy-turn').conversationTaskId)
  310. .toBe(initialTask.id)
  311. expect(migratedStore.require<WorkspaceInteraction>(owner, 'interaction', 'legacy-interaction').conversationTaskId)
  312. .toBe(initialTask.id)
  313. expect(migratedStore.get(owner, 'binding', `${conversation.id}:${administrator.id}`)).toBeUndefined()
  314. expect(migratedStore.get(owner, 'binding', `${conversation.id}:${initialTask.id}:${administrator.id}`))
  315. .toMatchObject({ conversationTaskId: initialTask.id })
  316. expect(migratedStore.get(owner, 'context', conversation.id)).toBeUndefined()
  317. expect(migratedStore.get(owner, 'context', initialTask.id))
  318. .toMatchObject({ conversationTaskId: initialTask.id })
  319. } finally {
  320. seededStore?.close()
  321. migratedStore?.close()
  322. rmSync(legacyHome, { recursive: true, force: true })
  323. }
  324. })
  325. it('runs one Agent across group tasks concurrently while keeping messages, sessions and context isolated', async () => {
  326. const administrator = store.createAgent(owner, { name: '多任务远程管理员', profile: 'remote-profile', nodeId: 'remote-node' }),
  327. member = agent('多任务成员'),
  328. conversation = store.createGroup(owner, {
  329. name: '多任务群聊',
  330. memberIds: [administrator.id, member.id],
  331. administratorId: administrator.id,
  332. }),
  333. firstTask = defaultTask(conversation.id),
  334. secondTask = store.createTask(owner, conversation.id, { title: '显式标题' })
  335. const pending: Array<{ socket: WebSocket; params: Record<string, any> }> = []
  336. reply = (socket, params) => {
  337. pending.push({ socket, params })
  338. }
  339. const complete = (item: typeof pending[number]) => {
  340. const text = item.params.text.includes('第一项工作') ? '第一项结果' : '第二项结果'
  341. setTimeout(() => item.socket.send(JSON.stringify({
  342. method: 'event',
  343. params: { type: 'message.complete', session_id: item.params.session_id, payload: { text, status: 'complete' } },
  344. })), 5)
  345. }
  346. const firstRun = runtime.send(owner, conversation.id, { requestId: randomUUID(), taskId: firstTask.id, content: '第一项工作' }),
  347. secondRun = runtime.send(owner, conversation.id, { requestId: randomUUID(), taskId: secondTask.id, content: '第二项工作' })
  348. await vi.waitFor(()=>expect(pending).toHaveLength(2))
  349. expect(store.require<WorkspaceRun>(owner,'run',firstRun.id).status).toBe('running')
  350. expect(store.require<WorkspaceRun>(owner,'run',secondRun.id).status).toBe('running')
  351. complete(pending.find(item => item.params.text.includes('第二项工作'))!)
  352. await finished(secondRun.id)
  353. expect(store.require<WorkspaceRun>(owner,'run',firstRun.id).status).toBe('running')
  354. complete(pending.find(item => item.params.text.includes('第一项工作'))!)
  355. await finished(firstRun.id)
  356. const firstMessages = store.messages(owner, conversation.id, Number.MAX_SAFE_INTEGER, 100, false, firstTask.id),
  357. secondMessages = store.messages(owner, conversation.id, Number.MAX_SAFE_INTEGER, 100, false, secondTask.id)
  358. expect(firstMessages.map(message => message.content)).toEqual(['第一项工作', '第一项结果'])
  359. expect(secondMessages.map(message => message.content)).toEqual(['第二项工作', '第二项结果'])
  360. expect(firstMessages.every(message => message.conversationTaskId === firstTask.id)).toBe(true)
  361. expect(secondMessages.every(message => message.conversationTaskId === secondTask.id)).toBe(true)
  362. const prompts = requests.filter(request => request.method === 'prompt.submit').map(request => String(request.params.text))
  363. expect(prompts.find(prompt => prompt.includes('第一项工作'))).not.toContain('第二项工作')
  364. expect(prompts.find(prompt => prompt.includes('第二项工作'))).not.toContain('第一项工作')
  365. const bindings = store.list<any>(owner, 'binding')
  366. expect(bindings.map(binding => binding.conversationTaskId).sort()).toEqual([firstTask.id, secondTask.id].sort())
  367. expect(bindings.every(binding => binding.nodeId === 'remote-node' && binding.profile === 'remote-profile')).toBe(true)
  368. expect(new Set(bindings.map(binding => binding.storedId)).size).toBe(2)
  369. expect(store.get<any>(owner, 'context', firstTask.id)).toMatchObject({ conversationTaskId: firstTask.id, percent: 40 })
  370. expect(store.get<any>(owner, 'context', secondTask.id)).toMatchObject({ conversationTaskId: secondTask.id, percent: 40 })
  371. const firstSummary = store.require<any>(owner, 'conversation-task', firstTask.id)
  372. expect(firstSummary).toMatchObject({ title: '第一项工作', titleSource: 'automatic', messageCount: 2, unread: true, unreadCount: 1 })
  373. expect(store.require<any>(owner, 'conversation-task', secondTask.id)).toMatchObject({ title: '显式标题', titleSource: 'user', messageCount: 2 })
  374. expect(store.markTaskRead(owner, conversation.id, firstTask.id, firstSummary.lastSeq).unreadCount).toBe(0)
  375. store.deleteTask(owner, conversation.id, firstTask.id)
  376. expect(store.list<WorkspaceMessage>(owner, 'message').some(message => message.conversationTaskId === firstTask.id)).toBe(false)
  377. expect(store.list<any>(owner, 'binding').some(binding => binding.conversationTaskId === firstTask.id)).toBe(false)
  378. expect(store.get(owner, 'context', firstTask.id)).toBeUndefined()
  379. expect(store.messages(owner, conversation.id, Number.MAX_SAFE_INTEGER, 100, false, secondTask.id).map(message => message.content))
  380. .toEqual(['第二项工作', '第二项结果'])
  381. })
  382. it('admits ten active conversation tasks for one Agent and rejects the eleventh without stopping them', async () => {
  383. const administrator = agent('并发任务管理员'),
  384. member = agent('并发任务成员'),
  385. conversation = store.createGroup(owner, {
  386. name: '并发任务群聊',
  387. memberIds: [administrator.id, member.id],
  388. administratorId: administrator.id,
  389. }),
  390. tasks = [defaultTask(conversation.id), ...Array.from({ length: 10 }, (_, index) => store.createTask(owner, conversation.id, { title: `任务 ${index + 2}` }))]
  391. reply = () => {}
  392. for (const [index, task] of tasks.slice(0, 10).entries())
  393. runtime.send(owner, conversation.id, { requestId: randomUUID(), taskId: task.id, content: `执行 ${index + 1}` })
  394. expect(store.activeTaskCount(owner, conversation.id)).toBe(10)
  395. await vi.waitFor(() => expect(requests.filter(r => r.method === 'prompt.submit')).toHaveLength(10))
  396. expect(new Set(requests.filter(r => r.method === 'prompt.submit').map(r => r.params.session_id)).size).toBe(10)
  397. expect(() => runtime.send(owner, conversation.id, { requestId: randomUUID(), taskId: tasks[10]!.id, content: '第十一项' }))
  398. .toThrow('最多同时运行 10 个任务')
  399. expect(requests.filter(r => r.method === 'session.interrupt')).toHaveLength(0)
  400. await Promise.all(tasks.slice(0, 10).map(task => runtime.stopTask(owner, conversation.id, task.id)))
  401. })
  402. it('keeps same-named upstream interactions scoped to their conversation task', async () => {
  403. const administrator = agent('交互隔离管理员'),
  404. member = agent('交互隔离成员'),
  405. conversation = store.createGroup(owner, {
  406. name: '交互隔离群聊',
  407. memberIds: [administrator.id, member.id],
  408. administratorId: administrator.id,
  409. }),
  410. firstTask = defaultTask(conversation.id),
  411. secondTask = store.createTask(owner, conversation.id, {})
  412. const pending: Array<{socket:WebSocket;params:Record<string,any>}> = []
  413. reply = (socket, params) => { pending.push({socket,params});setTimeout(() => socket.send(JSON.stringify({
  414. method: 'event',
  415. params: {
  416. type: 'approval.request',
  417. session_id: params.session_id,
  418. payload: { request_id: 'same-upstream-id', message: '允许执行吗?' },
  419. },
  420. })), 5) }
  421. const firstRun=runtime.send(owner, conversation.id, { requestId: randomUUID(), taskId: firstTask.id, content: '任务甲' })
  422. // Different tasks admitted in the same millisecond need not win a lane in
  423. // insertion order. Establish the active task before testing its isolation.
  424. await vi.waitFor(() => expect(pending).toHaveLength(1))
  425. const secondRun=runtime.send(owner, conversation.id, { requestId: randomUUID(), taskId: secondTask.id, content: '任务乙' })
  426. await vi.waitFor(() => expect(store.list<WorkspaceInteraction>(owner, 'interaction').filter(interaction => !interaction.resolved)).toHaveLength(2))
  427. const first=store.list<WorkspaceInteraction>(owner,'interaction').find(i=>i.conversationTaskId===firstTask.id)!
  428. expect(store.require<any>(owner, 'conversation-task', firstTask.id).activeRunStatus).toBe('waiting')
  429. expect(store.require<WorkspaceRun>(owner,'run',secondRun.id).status).toBe('waiting')
  430. await runtime.respond(owner, first.id, 'once')
  431. expect(requests.filter(r => r.method === 'approval.respond').map(r => r.params.session_id)).toEqual([pending[0]!.params.session_id])
  432. pending[0]!.socket.send(JSON.stringify({method:'event',params:{type:'message.complete',session_id:pending[0]!.params.session_id,payload:{text:'任务甲完成',status:'complete'}}}))
  433. await finished(firstRun.id)
  434. await vi.waitFor(()=>expect(store.list<WorkspaceInteraction>(owner,'interaction')).toHaveLength(2))
  435. const second=store.list<WorkspaceInteraction>(owner,'interaction').find(i=>i.conversationTaskId===secondTask.id)!
  436. expect(second.id).not.toBe(first.id)
  437. expect(store.require<any>(owner, 'conversation-task', secondTask.id).activeRunStatus).toBe('waiting')
  438. await runtime.respond(owner,first.id,'once')
  439. expect(store.require<WorkspaceInteraction>(owner, 'interaction', first.id).resolved).toBe(true)
  440. expect(store.require<WorkspaceInteraction>(owner, 'interaction', second.id).resolved).toBe(false)
  441. await Promise.all([runtime.stopTask(owner, conversation.id, firstTask.id), runtime.stopTask(owner, conversation.id, secondTask.id)])
  442. })
  443. it('persists a visible failure when a fixed member loses its source node',async()=>{
  444. const a=agent('不可用成员'),c=direct(a.id)
  445. runtime.close()
  446. runtime=new WorkspaceRuntime(store,{requireSource: () => {},target:()=>{throw new Error('基础节点不可用')}} as unknown as WorkspaceNodes,uploads)
  447. const run=runtime.send(owner,c.id,{requestId:randomUUID(),content:'执行任务'})
  448. await vi.waitFor(()=>expect(store.require<WorkspaceRun>(owner,'run',run.id).status).toBe('failed'))
  449. expect(store.messages(owner,c.id).at(-1)).toMatchObject({role:'system',status:'failed',content:'执行失败:基础节点不可用'})
  450. expect(store.require<WorkspaceConversation>(owner,'conversation',c.id).activeRunId).toBeUndefined()
  451. })
  452. it('interprets a reply against its launch-time roster when names or mode change mid-turn', async () => {
  453. const a = agent('管理员'),
  454. b = agent('执行者')
  455. const c = store.createGroup(owner, {
  456. name: '配置更新',
  457. memberIds: [a.id, b.id],
  458. administratorId: a.id,
  459. maxReplyRounds: 3,
  460. })
  461. let count = 0
  462. reply = (socket, p) => {
  463. let text = '已完成'
  464. if (count++ === 0) {
  465. store.updateAgent(owner, b.id, { name: '新名字', instructions: '新规则' })
  466. store.updateConversation(owner, c.id, { mode: 'free', administratorId: b.id })
  467. text = '请继续 @执行者'
  468. }
  469. setTimeout(
  470. () =>
  471. socket.send(
  472. JSON.stringify({
  473. method: 'event',
  474. params: { type: 'message.complete', session_id: p.session_id, payload: { text } },
  475. }),
  476. ),
  477. 5,
  478. )
  479. }
  480. await finished(runtime.send(owner, c.id, { requestId: randomUUID(), content: '开始' }).id)
  481. expect(
  482. store
  483. .messages(owner, c.id)
  484. .filter((m) => m.role === 'assistant')
  485. .map((m) => m.agentId),
  486. ).toEqual([a.id, b.id])
  487. expect(requests.filter((r) => r.method === 'prompt.submit').at(-1)!.params.text).toContain(
  488. '新规则',
  489. )
  490. })
  491. it('rejects unscoped and foreign-session completion events', async () => {
  492. const a = agent('会话隔离'),
  493. c = direct(a.id)
  494. reply = (socket, p) => {
  495. for (const session_id of [undefined, 'another-session'])
  496. socket.send(
  497. JSON.stringify({
  498. method: 'event',
  499. params: { type: 'message.complete', session_id, payload: { text: '别人的结果' } },
  500. }),
  501. )
  502. setTimeout(
  503. () =>
  504. socket.send(
  505. JSON.stringify({
  506. method: 'event',
  507. params: {
  508. type: 'message.complete',
  509. session_id: p.session_id,
  510. payload: { text: '本轮结果' },
  511. },
  512. }),
  513. ),
  514. 15,
  515. )
  516. }
  517. const run = runtime.send(owner, c.id, { requestId: randomUUID(), content: '执行' })
  518. await finished(run.id)
  519. expect(store.messages(owner, c.id).at(-1)?.content).toBe('本轮结果')
  520. expect(store.get<{ percent: number }>(owner, 'context', c.id)?.percent).toBe(40)
  521. })
  522. it('resumes a persisted uncertain run by inspecting history, without resending the prompt', async () => {
  523. const a = agent('恢复测试'),
  524. c = direct(a.id)
  525. reply = (socket, p) => {
  526. recoveryHistory = [
  527. { role: 'user', content: p.text },
  528. { role: 'assistant', content: '已经执行完成' },
  529. ]
  530. socket.terminate()
  531. }
  532. const run = runtime.send(owner, c.id, { requestId: randomUUID(), content: '修改一次' })
  533. await vi.waitFor(() =>
  534. expect(store.require<WorkspaceRun>(owner, 'run', run.id).status).toBe('uncertain'),
  535. )
  536. runtime.close()
  537. runtime = new WorkspaceRuntime(store, nodes, uploads)
  538. await runtime.reconcile(owner, run.id)
  539. await finished(run.id)
  540. expect(requests.filter((r) => r.method === 'prompt.submit')).toHaveLength(1)
  541. expect(store.messages(owner, c.id).map((m) => m.content)).toEqual(['修改一次', '已经执行完成'])
  542. })
  543. it('routes approvals and clarifications to the owning run and ignores optional push failures', async () => {
  544. const a = agent('交互测试'),
  545. c = direct(a.id)
  546. runtime.onNotify = () => {
  547. throw new Error('push storage unavailable')
  548. }
  549. let kind = 'approval'
  550. server.on('connection', (socket) =>
  551. socket.on('message', (data) => {
  552. const f = JSON.parse(String(data))
  553. if (['approval.respond', 'clarify.respond'].includes(f.method))
  554. socket.send(
  555. JSON.stringify({
  556. method: 'event',
  557. params: {
  558. type: 'message.complete',
  559. session_id: f.params.session_id,
  560. payload: { text: '确认后完成' },
  561. },
  562. }),
  563. )
  564. }),
  565. )
  566. reply = (socket, p) =>
  567. socket.send(
  568. JSON.stringify({
  569. method: 'event',
  570. params: {
  571. type: `${kind}.request`,
  572. session_id: p.session_id,
  573. payload: { request_id: 'upstream-interaction', message: '请确认' },
  574. },
  575. }),
  576. )
  577. for (kind of ['approval', 'clarify']) {
  578. const run = runtime.send(owner, c.id, { requestId: randomUUID(), content: '继续' })
  579. await vi.waitFor(() =>
  580. expect(store.require<WorkspaceRun>(owner, 'run', run.id).status).toBe('waiting'),
  581. )
  582. const interaction = store
  583. .list<WorkspaceInteraction>(owner, 'interaction')
  584. .find((i) => !i.resolved)!
  585. await expect(runtime.respond(other, interaction.id, 'once')).rejects.toThrow('记录不存在')
  586. await runtime.respond(owner, interaction.id, kind === 'approval' ? 'once' : '补充说明')
  587. await finished(run.id)
  588. expect(
  589. store.require<WorkspaceInteraction>(owner, 'interaction', interaction.id).resolved,
  590. ).toBe(true)
  591. }
  592. expect(
  593. requests
  594. .filter((r) => r.method === 'approval.respond' || r.method === 'clarify.respond')
  595. .map((r) => r.params.request_id),
  596. ).toEqual(['upstream-interaction', 'upstream-interaction'])
  597. })
  598. it('creates one direct chat without importing a native profile session', () => {
  599. const a = agent('编辑')
  600. expect(direct(a.id).kind).toBe('direct')
  601. expect(store.list(other, 'agent')).toEqual([])
  602. expect(() => store.require(other, 'conversation', direct(a.id).id)).toThrow('记录不存在')
  603. expect(requests).toEqual([])
  604. })
  605. it('protects the current administrator and validates member ownership', () => {
  606. const a = agent('甲'),
  607. b = agent('乙'),
  608. foreign = agent('外部', other)
  609. const c = store.createGroup(owner, {
  610. name: '团队',
  611. memberIds: [a.id, b.id],
  612. administratorId: a.id,
  613. })
  614. expect(() => store.updateConversation(owner, c.id, { memberIds: [b.id] })).toThrow('当前管理员不能移除')
  615. expect(() => store.updateConversation(owner, c.id, { memberIds: [b.id], administratorId: b.id })).toThrow('当前管理员不能移除')
  616. expect(() => store.updateConversation(owner, c.id, { memberIds: [a.id, foreign.id] })).toThrow('记录不存在')
  617. expect(() => store.updateConversation(owner, c.id, { memberIds: [a.id, a.id] })).toThrow('群成员或管理员无效')
  618. expect(() =>
  619. store.createGroup(owner, {
  620. name: '越权',
  621. memberIds: [a.id, foreign.id],
  622. administratorId: a.id,
  623. }),
  624. ).toThrow()
  625. store.updateAgent(owner, b.id, { archived: true })
  626. expect(store.require<WorkspaceConversation>(owner, 'conversation', c.id).memberIds).toEqual([
  627. a.id,
  628. b.id,
  629. ])
  630. expect(() =>
  631. store.createGroup(owner, { name: '新群', memberIds: [a.id, b.id], administratorId: a.id }),
  632. ).toThrow('已归档')
  633. expect(store.updateConversation(owner, c.id, { memberIds: [a.id] }).memberIds).toEqual([a.id])
  634. expect(() => store.updateConversation(owner, c.id, { memberIds: [a.id, b.id] })).toThrow('已归档')
  635. })
  636. it('adds and removes members, cleans automatic replies, and permits removal after administrator transfer', async () => {
  637. const a = agent('管理员'), b = agent('旧成员'), added = agent('新成员')
  638. const c = store.createGroup(owner, { name: '成员维护', memberIds: [a.id, b.id], administratorId: a.id, autoReplyIds: [b.id] })
  639. await finished(runtime.send(owner, c.id, { requestId: randomUUID(), content: '保留这条历史' }).id)
  640. const before = store.require<WorkspaceConversation>(owner, 'conversation', c.id)
  641. const history = store.messages(owner, c.id)
  642. const changed = store.updateConversation(owner, c.id, { memberIds: [a.id, added.id] })
  643. expect(changed.memberIds).toEqual([a.id, added.id])
  644. expect(changed.autoReplyIds).toEqual([])
  645. expect(changed.lastMessageAt).toBe(before.lastMessageAt)
  646. expect(store.messages(owner, c.id)).toEqual(history)
  647. expect(store.events(owner, 0).at(-1)).toMatchObject({ type: 'conversation.changed', data: { memberIds: [a.id, added.id] } })
  648. store.updateConversation(owner, c.id, { administratorId: added.id })
  649. expect(store.updateConversation(owner, c.id, { memberIds: [added.id] }).memberIds).toEqual([added.id])
  650. expect(() => store.updateConversation(owner, c.id, { memberIds: [a.id] })).toThrow('当前管理员不能移除')
  651. })
  652. it('skips removed queued members and accepts a newly added member on the next request', async () => {
  653. const a = agent('管理员'), b = agent('待回复'), added = agent('新成员')
  654. const c = store.createGroup(owner, { name: '队列维护', memberIds: [a.id, b.id], administratorId: a.id, mode: 'free', autoReplyIds: [a.id, b.id] })
  655. reply = (socket, p) => {
  656. socket.send(JSON.stringify({ method: 'event', params: { type: 'message.complete', session_id: p.session_id, payload: { text: '请@待回复继续' } } }))
  657. }
  658. const queued = runtime.send(owner, c.id, { requestId: randomUUID(), content: '开始' })
  659. store.updateConversation(owner, c.id, { memberIds: [a.id, added.id] })
  660. await finished(queued.id)
  661. expect(store.messages(owner, c.id).filter(m => m.role === 'assistant').map(m => m.agentId)).toEqual([a.id])
  662. await finished(runtime.send(owner, c.id, { requestId: randomUUID(), content: '请@新成员继续' }).id)
  663. expect(store.messages(owner, c.id).filter(m => m.role === 'assistant').map(m => m.agentId)).toEqual([a.id, added.id, a.id])
  664. })
  665. it('finishes an in-flight removed member reply without scheduling it again', async () => {
  666. const a = agent('管理员'), b = agent('正在回复')
  667. const c = store.createGroup(owner, { name: '回复中移除', memberIds: [a.id, b.id], administratorId: a.id })
  668. let turns = 0
  669. reply = (socket, p) => {
  670. if (turns++ === 1) store.updateConversation(owner, c.id, { memberIds: [a.id] })
  671. socket.send(JSON.stringify({ method: 'event', params: { type: 'message.complete', session_id: p.session_id, payload: { text: turns === 1 ? '请@正在回复继续' : '已完成' } } }))
  672. }
  673. await finished(runtime.send(owner, c.id, { requestId: randomUUID(), content: '@正在回复 开始' }).id)
  674. expect(store.messages(owner, c.id).filter(m => m.role === 'assistant').map(m => m.agentId)).toEqual([a.id, b.id, a.id])
  675. expect(() => runtime.send(owner, c.id, { requestId: randomUUID(), content: '继续', mentionIds: [b.id] })).toThrow('只能 @ 群内成员')
  676. })
  677. it('dispatches a request exactly once across retries and rejects payload reuse', async () => {
  678. const a = agent('编辑'),
  679. c = direct(a.id),
  680. requestId = randomUUID(),
  681. payload = { requestId, content: '检查代码' }
  682. const run = runtime.send(owner, c.id, payload)
  683. expect(runtime.send(owner, c.id, payload).id).toBe(run.id)
  684. expect(() => runtime.send(owner, c.id, { ...payload, content: '另一件事' })).toThrow('请求编号')
  685. await finished(run.id)
  686. expect(requests.filter((r) => r.method === 'prompt.submit')).toHaveLength(1)
  687. expect(store.messages(owner, c.id).map((m) => m.role)).toEqual(['user', 'assistant'])
  688. })
  689. it('keeps template responsibilities inside their group and follows Agent identity after rename', async () => {
  690. const a = agent('现有甲'), b = agent('现有乙')
  691. const before = store.require<WorkspaceAgent>(owner, 'agent', a.id)
  692. const c = store.createGroup(owner, { name: '模板群', memberIds: [a.id, b.id], administratorId: a.id, memberRoles: {
  693. [a.id]: { name: '调研负责人', description: '汇总可信结论' },
  694. [b.id]: { name: '事实核验', description: '交叉验证来源' },
  695. } })
  696. expect(store.require<WorkspaceAgent>(owner, 'agent', a.id)).toEqual(before)
  697. store.updateAgent(owner, a.id, { name: '改名后的甲' })
  698. await finished(runtime.send(owner, c.id, { requestId: randomUUID(), content: '开始调研' }).id)
  699. const groupPrompt = requests.filter(r => r.method === 'prompt.submit').at(-1)!.params.text
  700. expect(groupPrompt).toContain('@改名后的甲:调研负责人;汇总可信结论')
  701. expect(groupPrompt).toContain('@现有乙:事实核验;交叉验证来源')
  702. await finished(runtime.send(owner, direct(a.id).id, { requestId: randomUUID(), content: '单聊' }).id)
  703. expect(requests.filter(r => r.method === 'prompt.submit').at(-1)!.params.text).not.toContain('汇总可信结论')
  704. const changed = store.updateConversation(owner, c.id, { memberIds: [a.id] })
  705. expect(changed.memberRoles).toEqual({ [a.id]: { name: '调研负责人', description: '汇总可信结论' } })
  706. expect(() => store.updateConversation(owner, c.id, { memberRoles: { [b.id]: { name: '已移除', description: '' } } })).toThrow('群成员或管理员无效')
  707. })
  708. it('isolates roles over the same profile and applies edited rules on the next turn', async () => {
  709. const a = agent('编辑'),
  710. b = agent('审查'),
  711. ca = direct(a.id),
  712. cb = direct(b.id)
  713. const first = runtime.send(owner, ca.id, { requestId: randomUUID(), content: '编辑任务' })
  714. const second = runtime.send(owner, cb.id, { requestId: randomUUID(), content: '审查任务' })
  715. await Promise.all([finished(first.id), finished(second.id)])
  716. expect(requests.filter((r) => r.method === 'session.create')).toHaveLength(2)
  717. expect(store.messages(owner, ca.id).some((m) => m.content === '审查任务')).toBe(false)
  718. store.updateAgent(owner, a.id, { instructions: '最新规则:先给证据' })
  719. await finished(runtime.send(owner, ca.id, { requestId: randomUUID(), content: '继续' }).id)
  720. expect(requests.filter((r) => r.method === 'prompt.submit').at(-1)!.params.text).toContain(
  721. '最新规则:先给证据',
  722. )
  723. expect(requests.some((r) => r.method === 'profiles.configure')).toBe(false)
  724. expect(
  725. requests
  726. .filter((r) => r.method === 'session.create')
  727. .every((r) => r.params.hidden && r.params.source === 'yaoyao_workspace'),
  728. ).toBe(true)
  729. })
  730. it('executes bounded administrator delegation and keeps group sessions separate from direct chat', async () => {
  731. const a = agent('管理员'),
  732. b = agent('开发者'),
  733. c = store.createGroup(owner, {
  734. name: '开发',
  735. memberIds: [a.id, b.id],
  736. administratorId: a.id,
  737. maxReplyRounds: 3,
  738. })
  739. let turns = 0
  740. reply = (socket, p) => {
  741. const text = turns++ === 0 ? '我已整理需求,现在请@开发者继续实现。' : '任务完成'
  742. setTimeout(
  743. () =>
  744. socket.send(
  745. JSON.stringify({
  746. method: 'event',
  747. params: { type: 'message.complete', session_id: p.session_id, payload: { text } },
  748. }),
  749. ),
  750. 5,
  751. )
  752. }
  753. await finished(runtime.send(owner, c.id, { requestId: randomUUID(), content: '实现功能' }).id)
  754. expect(
  755. store
  756. .messages(owner, c.id)
  757. .filter((m) => m.role === 'assistant')
  758. .map((m) => m.agentId),
  759. ).toEqual([a.id, b.id, a.id])
  760. expect(store.messages(owner, direct(a.id).id)).toHaveLength(0)
  761. expect(turns).toBe(3)
  762. })
  763. it('does not trigger delegation from quoted text or code blocks', () => {
  764. const a = agent('甲'),
  765. b = agent('乙')
  766. expect(mentionedAgents('> @乙\n```\n@甲\n```\n`@all`', [a, b])).toEqual([])
  767. expect(mentionedAgents('请处理 @乙。', [a, b])).toEqual([b.id])
  768. })
  769. it('keeps uncertain submissions blocked instead of executing tools twice', async () => {
  770. const a = agent('执行者'),
  771. c = direct(a.id)
  772. reply = (socket) => socket.terminate()
  773. const run = runtime.send(owner, c.id, { requestId: randomUUID(), content: '修改文件' })
  774. await vi.waitFor(() =>
  775. expect(store.require<WorkspaceRun>(owner, 'run', run.id).status).toBe('uncertain'),
  776. )
  777. const queued = runtime.send(owner, c.id, { requestId: randomUUID(), content: '后续请求' })
  778. expect(queued.status).toBe('queued')
  779. expect(requests.filter((r) => r.method === 'prompt.submit')).toHaveLength(1)
  780. await runtime.stopConversation(owner, c.id)
  781. expect(
  782. store.require<WorkspaceConversation>(owner, 'conversation', c.id).activeRunId,
  783. ).toBeUndefined()
  784. })
  785. it('publishes durable ordered events only after commit and rolls back failed transactions', () => {
  786. const a = agent('甲'),
  787. c = direct(a.id),
  788. cursor = store.cursor(owner)
  789. expect(() =>
  790. store.atomic(() => {
  791. store.event(owner, 'test', {})
  792. throw new Error('rollback')
  793. }),
  794. ).toThrow('rollback')
  795. expect(store.cursor(owner)).toBe(cursor)
  796. store.updateAgent(owner, a.id, { name: '乙' })
  797. expect(store.events(owner, cursor).map((e) => e.type)).toEqual([
  798. 'agent.changed',
  799. 'conversation.changed',
  800. ])
  801. expect(store.events(other, 0)).toEqual([])
  802. expect(store.require<WorkspaceConversation>(owner, 'conversation', c.id).name).toBe('乙')
  803. })
  804. })
  805. it('pins a direct chat without injecting or clearing optional profile fields', () => {
  806. const avatar = encodeAgentAvatar({...defaultAgentIdentity('reviewer'),shape:'triangle',color:'#00b9ac',expression:'curious'})
  807. const agent = store.createAgent(owner, { name: '审查员', profile: 'default', avatar })
  808. const chat = store.list<WorkspaceConversation>(owner, 'conversation')[0]!
  809. expect(store.updateConversation(owner, chat.id, { pinned: true }).pinned).toBe(true)
  810. expect(store.updateConversation(owner, chat.id, { pinned: false }).pinned).toBe(false)
  811. expect(store.require<WorkspaceConversation>(owner, 'conversation', chat.id).avatar).toBe(avatar)
  812. expect(store.updateAgent(owner, agent.id, { instructions: '请检查边界' }).avatar).toBe(avatar)
  813. expect(decodeAgentAvatar(store.updateAgent(owner, agent.id, { avatar: '' }).avatar)).toMatchObject({shape:'circle',color:'#00c875'})
  814. })
  815. it('accepts the existing cross-client mascot and team avatar formats', () => {
  816. const a = store.createAgent(owner, { name: '设计', profile: 'default', avatar: 'yaoyao-mascot:v1:square:377fe6:friendly' })
  817. const b = store.createAgent(owner, { name: '开发', profile: 'default' })
  818. const group = store.createGroup(owner, { name: '产品团队', memberIds: [a.id, b.id], administratorId: a.id, avatar: 'builtin:team-animal:fox' })
  819. expect(group.avatar).toBe('builtin:team-animal:fox')
  820. expect(() => store.updateAgent(owner, a.id, { avatar: 'javascript:alert(1)' })).toThrow()
  821. })
  822. it('assigns a random v2 avatar only when a new Agent omits its avatar', () => {
  823. const random = vi.spyOn(Math, 'random').mockReturnValue(0)
  824. const created = store.createAgent(owner, { name: '随机头像', profile: 'default' })
  825. expect(decodeAgentAvatar(created.avatar)).toMatchObject({ shape: 'circle', color: '#000000', expression: 'idle' })
  826. expect(created.avatar).not.toBe(encodeAgentAvatar(defaultAgentIdentity(created.id, created.name)))
  827. random.mockRestore()
  828. })
  829. it('keeps the last message time independent of pinning and metadata edits', () => {
  830. const agent=store.createAgent(owner,{name:'时间验收',profile:'default'})
  831. const c=store.list<WorkspaceConversation>(owner,'conversation')[0]!
  832. const message: WorkspaceMessage={id:randomUUID(),conversationId:c.id,seq:0,role:'user',content:'历史消息',reasoning:'',status:'complete',attachments:[],tools:[],createdAt:Date.now()-600_000}
  833. store.saveMessage(owner,message)
  834. const pinned=store.updateConversation(owner,c.id,{pinned:true})
  835. expect(store.conversationSummary(owner,pinned).lastMessageAt).toBe(message.createdAt)
  836. store.updateAgent(owner,agent.id,{name:'改名之后'})
  837. expect(store.conversationSummary(owner,store.require(owner,'conversation',c.id)).lastMessageAt).toBe(message.createdAt)
  838. // Existing records without the summary field still use their transcript time.
  839. const previous={...pinned}; delete previous.lastMessageAt
  840. expect(store.conversationSummary(owner,previous).lastMessageAt).toBe(message.createdAt)
  841. })
  842. it('publishes only the executing member and clears avatar activity when stopped', async () => {
  843. const a=store.createAgent(owner,{name:'管理员头像',profile:'default'})
  844. const b=store.createAgent(owner,{name:'执行者头像',profile:'default'})
  845. const c=store.createGroup(owner,{name:'头像状态',memberIds:[a.id,b.id],administratorId:a.id})
  846. reply=(socket,p)=>socket.send(JSON.stringify({method:'event',params:{type:'approval.request',session_id:p.session_id,payload:{request_id:'approval-avatar',message:'需要确认'}}}))
  847. const run=runtime.send(owner,c.id,{requestId:randomUUID(),content:'处理任务',mentionIds:[b.id]})
  848. await vi.waitFor(()=>expect(store.require<WorkspaceRun>(owner,'run',run.id).status).toBe('waiting'))
  849. expect(store.require<WorkspaceConversation>(owner,'conversation',c.id)).toMatchObject({activeAgentId:a.id,activeRunStatus:'waiting'})
  850. await runtime.stop(owner,run.id)
  851. const stopped=store.require<WorkspaceConversation>(owner,'conversation',c.id)
  852. expect(stopped.activeAgentId).toBeUndefined()
  853. expect(stopped.activeRunStatus).toBeUndefined()
  854. })
  855. it('stores the expanded avatar shapes for cross-client role settings', () => {
  856. const agent=store.createAgent(owner,{name:'扩展头像验收',profile:'default'})
  857. for(const shape of ['ellipse','capsule','hexagon','cloud','droplet'] as const) {
  858. const avatar=encodeAgentAvatar({...defaultAgentIdentity('bot'),shape,color:'#f52ba5',expression:'curious'})
  859. expect(store.updateAgent(owner,agent.id,{avatar}).avatar).toBe(avatar)
  860. }
  861. })
  862. it('automatically reconciles lost completion without submitting the prompt again', async () => {
  863. const a=agent('自动恢复'),c=direct(a.id)
  864. reply=(socket,p)=>{recoveryHistory=[{role:'user',content:p.text},{role:'assistant',content:'已完成一次'}];socket.terminate()}
  865. const run=runtime.send(owner,c.id,{requestId:randomUUID(),content:'执行一次'})
  866. await vi.waitFor(()=>expect(store.require<WorkspaceRun>(owner,'run',run.id).status).toBe('complete'),{timeout:5000})
  867. expect(requests.filter(r=>r.method==='prompt.submit')).toHaveLength(1)
  868. expect(store.messages(owner,c.id).at(-1)).toMatchObject({content:'已完成一次',status:'complete'})
  869. })
  870. it('reports a rejected prompt as failed instead of leaving it uncertain', async () => {
  871. rejectPrompt='本轮请求被明确拒绝'
  872. const a=agent('拒绝测试'),c=direct(a.id)
  873. const run=runtime.send(owner,c.id,{requestId:randomUUID(),content:'执行'})
  874. await vi.waitFor(()=>expect(store.require<WorkspaceRun>(owner,'run',run.id).status).toBe('failed'))
  875. expect(store.require<WorkspaceConversation>(owner,'conversation',c.id).activeRunId).toBeUndefined()
  876. expect(store.messages(owner,c.id).some(m=>m.status==='uncertain')).toBe(false)
  877. })
  878. it('keeps an unconfirmed turn separate from a later reply and retries history without resubmitting', async () => {
  879. const a = agent('恢复边界'), c = direct(a.id)
  880. let prompt = ''
  881. reply = (socket, p) => {
  882. prompt = p.text
  883. recoveryHistory = [{ role: 'user', content: prompt }, { role: 'user', content: '另一个请求' }, { role: 'assistant', content: '另一轮回复' }]
  884. socket.terminate()
  885. }
  886. const run = runtime.send(owner, c.id, { requestId: randomUUID(), content: '只执行一次' })
  887. await vi.waitFor(() => expect(requests.filter(r => r.method === 'session.resume').length).toBeGreaterThanOrEqual(1), { timeout: 3000 })
  888. expect(store.require<WorkspaceRun>(owner, 'run', run.id).status).toBe('uncertain')
  889. expect(store.messages(owner, c.id).at(-1)?.content).not.toBe('另一轮回复')
  890. recoveryHistory = [{ role: 'user', content: prompt }, { role: 'assistant', content: '本轮结果' }]
  891. await vi.waitFor(() => expect(store.require<WorkspaceRun>(owner, 'run', run.id).status).toBe('complete'), { timeout: 4000 })
  892. expect(requests.filter(r => r.method === 'prompt.submit')).toHaveLength(1)
  893. expect(store.messages(owner, c.id).at(-1)).toMatchObject({ content: '本轮结果', status: 'complete' })
  894. })
  895. it('uses one generated avatar for the agent and its direct conversation',()=>{
  896. const a=agent('默认头像'),c=direct(a.id)
  897. expect(decodeAgentAvatar(a.avatar)).toMatchObject({version:2,avatarMode:'mascot'})
  898. expect(store.agentSummary(a).avatar).toBe(store.conversationSummary(owner,c).avatar)
  899. const changed=store.updateAgent(owner,a.id,{name:'换个名称'})
  900. expect(store.agentSummary(changed).avatar).toBe(store.agentSummary(a).avatar)
  901. })
  902. it('recognizes mentions inside Chinese sentences and ignores quoted code and emails',()=>{
  903. const a=agent('瑶儿'),b=agent('竹儿'),c=agent('瑶儿助手'),d=agent('Ann')
  904. const roster=[a,b,c,d]
  905. expect(mentionedAgents('先整理,再请@瑶儿帮忙,最后由(@竹儿)复核。',roster)).toEqual([a.id,b.id])
  906. expect(mentionedAgents('请**@瑶儿助手**继续',roster)).toEqual([c.id])
  907. expect(mentionedAgents('mail@Ann.com https://site/@竹儿 `@瑶儿` <quoted_message>@瑶儿</quoted_message>',roster)).toEqual([])
  908. expect(mentionedAgents('@Anna',roster)).toEqual([])
  909. })
  910. function completeReply(socket: WebSocket, p: Record<string, any>, text: string, status = 'complete') {
  911. socket.send(JSON.stringify({ method: 'event', params: { type: 'message.complete', session_id: p.session_id, payload: { text, status, ...(status === 'failed' ? { error: text } : {}) } } }))
  912. }
  913. const speaker = (p: Record<string, any>) => /你是 (.*?)。/.exec(p.text)?.[1]
  914. describe('automatic relay termination', () => {
  915. const receipt = '收到 @审核 终审维持通过。\n\n交付状态(不变):入草稿箱 Media ID `draft-123`,发布目录 `/work/final/`。\n等民哥群发指令。'
  916. const review = '@竹儿 维持通过,等民哥群发指令。无新事实、无新派工,本轮不重复核验。'
  917. const promptCount = () => requests.filter(r => r.method === 'prompt.submit').length
  918. async function idle(runId: string, conversationId: string) {
  919. await finished(runId)
  920. const count = promptCount()
  921. await new Promise(resolve => setTimeout(resolve, 80))
  922. expect(promptCount()).toBe(count)
  923. expect(store.list<Work>(owner, 'turn').filter(w => w.runId === runId && (!w.planned || !['complete', 'failed', 'interrupted'].includes(w.status)))).toEqual([])
  924. const summary = store.conversationSummary(owner, store.require(owner, 'conversation', conversationId))
  925. expect(summary.activeRunId).toBeUndefined()
  926. expect(summary.queuedMessageCount ?? 0).toBe(0)
  927. }
  928. it('finishes the observed host receipt cycle in exactly three calls and accepts a new user message', async () => {
  929. const a = agent('竹儿'), b = agent('审核')
  930. const g = store.createGroup(owner, { name: '收尾验收', memberIds: [a.id, b.id], administratorId: a.id, maxReplyRounds: -1 })
  931. let admin = 0
  932. reply = (socket, p) => completeReply(socket, p, speaker(p) === a.name ? admin++ === 0 ? '@审核 请审核文案' : receipt : review)
  933. const root = runtime.send(owner, g.id, { requestId: randomUUID(), content: '请完成文案' })
  934. await idle(root.id, g.id)
  935. expect(promptCount()).toBe(3)
  936. expect(store.messages(owner, g.id).map(m => m.content)).toEqual(['请完成文案', '@审核 请审核文案', review, receipt])
  937. expect(store.list<Work>(owner, 'turn').find(w => w.reviewOf)?.relay?.suppressed[b.id]).toBe('acknowledgement')
  938. const next = runtime.send(owner, g.id, { requestId: randomUUID(), content: '好的。' })
  939. await idle(next.id, g.id)
  940. expect(promptCount()).toBe(4)
  941. })
  942. it.each(['感谢 @审核,等待用户指令。', '收到,等待用户指令。'])('does not turn a free-mode receipt into an automatic broadcast: %s', content => {
  943. const a = agent('竹儿'), b = agent('审核')
  944. const g = store.createGroup(owner, { name: '自动收尾', memberIds: [a.id, b.id], administratorId: a.id, mode: 'free', autoReplyIds: [b.id], maxReplyRounds: -1 })
  945. reply = (socket, p) => completeReply(socket, p, content)
  946. return (async () => {
  947. const root = runtime.send(owner, g.id, { requestId: randomUUID(), content: '@竹儿 请处理' })
  948. await idle(root.id, g.id)
  949. expect(promptCount()).toBe(1)
  950. const next = runtime.send(owner, g.id, { requestId: randomUUID(), content: '感谢 @审核' })
  951. await idle(next.id, g.id)
  952. expect(speaker(requests.filter(r => r.method === 'prompt.submit').at(-1)!.params)).toBe(b.name)
  953. })()
  954. })
  955. it.each([
  956. ['host', 2, 5], ['host', 3, 7], ['free', 2, 4], ['free', 3, 5],
  957. ] as const)('bounds unchanged %s cooperation with %i members at %i calls', async (mode, size, expected) => {
  958. const members = [agent('甲'), agent('乙'), ...(size === 3 ? [agent('丙')] : [])]
  959. const g = store.createGroup(owner, { name: '重复往返', memberIds: members.map(a => a.id), administratorId: members[0]!.id, mode, maxReplyRounds: -1 })
  960. let count = 0
  961. reply = (socket, p) => {
  962. const index = members.findIndex(a => a.name === speaker(p)), whitespace = count++ % 2 ? ' ' : ' '
  963. const text = mode === 'host' ? index === 0 ? members.slice(1).map(a => `@${a.name}${whitespace}再检查一次`).join(',') : '检查结果相同'
  964. : `@${members[(index + 1) % members.length]!.name}${whitespace}再检查一次`
  965. completeReply(socket, p, count > 20 ? '完成' : text)
  966. completeReply(socket, p, count > 20 ? '完成' : text) // Duplicate completion must not double-plan.
  967. }
  968. const root = runtime.send(owner, g.id, { requestId: randomUUID(), content: '@甲 开始' })
  969. await idle(root.id, g.id)
  970. expect(count).toBe(expected)
  971. expect(store.messages(owner, g.id).filter(m => m.content === REPEATED_RELAY_NOTICE)).toHaveLength(1)
  972. expect(store.list<Work>(owner, 'turn').some(w => Object.values(w.relay?.suppressed ?? {}).includes('repeated'))).toBe(true)
  973. })
  974. it.each(['result', 'tool', 'attachment', 'interaction'] as const)('allows repeated instructions with new %s evidence beyond thirteen calls', async kind => {
  975. const a = agent('负责人'), b = agent('审核')
  976. const g = store.createGroup(owner, { name: '持续推进', memberIds: [a.id, b.id], administratorId: a.id, maxReplyRounds: -1 })
  977. let admin = 0, worker = 0
  978. reply = (socket, p) => {
  979. if (speaker(p) === a.name) completeReply(socket, p, admin++ < 7 ? '感谢 @审核,请重新检查第二段' : '完成')
  980. else { worker++; completeReply(socket, p, kind === 'result' ? `第 ${worker} 版结果` : '本次检查结果') }
  981. }
  982. runtime.onMessage = async (_owner, message) => {
  983. if (message.agentId !== b.id) return
  984. if (kind === 'tool') message.tools = [{ id: randomUUID(), name: 'verify', status: 'tool.complete', result: { revision: worker } }]
  985. if (kind === 'attachment') message.attachments = [{ id: randomUUID(), name: 'revision.txt', mimeType: 'text/plain', size: 10, createdAt: Date.now() }]
  986. if (kind === 'interaction') {
  987. const id = randomUUID()
  988. store.put(owner, 'interaction', id, { id, runId: message.runId, conversationId: g.id, conversationTaskId: message.conversationTaskId, agentId: b.id, kind: 'clarification', message: '继续重试?', answer: '继续', responseState: 'sent', resolved: true })
  989. store.put(owner, 'interaction-binding', id, { taskId: message.taskId })
  990. }
  991. store.saveMessage(owner, message)
  992. }
  993. const root = runtime.send(owner, g.id, { requestId: randomUUID(), content: '反复改进' })
  994. await idle(root.id, g.id)
  995. expect(promptCount()).toBe(15)
  996. expect(store.messages(owner, g.id).some(m => m.content === REPEATED_RELAY_NOTICE)).toBe(false)
  997. })
  998. it('ends only the repeating free-mode branch while an independent member finishes', async () => {
  999. const a = agent('甲'), b = agent('乙'), c = agent('独立成员')
  1000. const g = store.createGroup(owner, { name: '分支隔离', memberIds: [a.id, b.id, c.id], administratorId: a.id, mode: 'free', maxReplyRounds: -1 })
  1001. let pending: { socket: WebSocket; p: Record<string, any> } | undefined, loops = 0
  1002. reply = (socket, p) => {
  1003. if (speaker(p) === c.name) { pending = { socket, p }; return }
  1004. completeReply(socket, p, ++loops > 15 ? '完成' : speaker(p) === a.name ? '@乙 再检查' : '@甲 再检查')
  1005. }
  1006. const root = runtime.send(owner, g.id, { requestId: randomUUID(), content: '@甲 @独立成员 请分别处理' })
  1007. await vi.waitFor(() => expect(store.messages(owner, g.id).some(m => m.content === REPEATED_RELAY_NOTICE)).toBe(true))
  1008. expect(store.require<WorkspaceRun>(owner, 'run', root.id).status).toBe('running')
  1009. expect(store.list<Work>(owner, 'turn').filter(w => w.cancelRequested)).toEqual([])
  1010. expect(loops).toBe(4)
  1011. completeReply(pending!.socket, pending!.p, '完成')
  1012. await idle(root.id, g.id)
  1013. expect(store.messages(owner, g.id).find(m => m.agentId === c.id)).toMatchObject({ content: '完成', status: 'complete' })
  1014. })
  1015. it('checks each target independently when a new automatic member joins the same repeated reply', async () => {
  1016. const a = agent('甲'), b = agent('乙'), c = agent('丙')
  1017. const g = store.createGroup(owner, { name: '目标隔离', memberIds: [a.id, b.id, c.id], administratorId: a.id, mode: 'free', autoReplyIds: [a.id], maxReplyRounds: -1 })
  1018. let calls = 0, admin = 0
  1019. reply = (socket, p) => { calls++; completeReply(socket, p, calls > 15 ? '完成' : speaker(p) === a.name ? '@乙 再检查' : speaker(p) === b.name ? '继续检查' : '完成') }
  1020. runtime.onMessage = async (_owner, message) => {
  1021. if (message.agentId === b.id && ++admin === 2) store.updateConversation(owner, g.id, { autoReplyIds: [a.id, c.id] })
  1022. }
  1023. const root = runtime.send(owner, g.id, { requestId: randomUUID(), content: '@甲 开始' })
  1024. await idle(root.id, g.id)
  1025. expect(calls).toBe(5)
  1026. const blocked = store.list<Work>(owner, 'turn').find(w => w.relay?.suppressed[a.id] === 'repeated')!
  1027. expect(blocked.relay?.suppressed[c.id]).toBeUndefined()
  1028. expect(store.list<Work>(owner, 'turn').find(w => w.batchId === blocked.id && w.agentId === c.id)?.status).toBe('complete')
  1029. })
  1030. it('starts fresh for identical new requests in the same task and a different task', async () => {
  1031. const a = agent('甲'), b = agent('乙')
  1032. const g = store.createGroup(owner, { name: '任务隔离', memberIds: [a.id, b.id], administratorId: a.id, mode: 'free', maxReplyRounds: -1 })
  1033. let calls = 0
  1034. reply = (socket, p) => { calls++; completeReply(socket, p, calls > 20 ? '完成' : speaker(p) === a.name ? '@乙 再检查' : '@甲 再检查') }
  1035. const originalTask = defaultTask(g.id), secondTask = store.createTask(owner, g.id, { title: '另一个任务' })
  1036. for (const [index, task] of [originalTask, originalTask, secondTask].entries()) {
  1037. const root = runtime.send(owner, g.id, { requestId: randomUUID(), taskId: task.id, content: '@甲 开始' })
  1038. await idle(root.id, g.id)
  1039. expect(calls).toBe((index + 1) * 4)
  1040. expect(store.messages(owner, g.id, Number.MAX_SAFE_INTEGER, 100, false, task.id).filter(m => m.runId === root.id && m.content === REPEATED_RELAY_NOTICE)).toHaveLength(1)
  1041. }
  1042. })
  1043. it.each([false, true])('recovers repeat suppression across restart before planning (legacy=%s)', async legacy => {
  1044. const a = agent('负责人'), b = agent('成员')
  1045. const g = store.createGroup(owner, { name: '循环恢复', memberIds: [a.id, b.id], administratorId: a.id, maxReplyRounds: -1 })
  1046. let calls = 0
  1047. reply = (socket, p) => { calls++; completeReply(socket, p, calls > 15 ? '完成' : speaker(p) === a.name ? '@成员 再检查' : '结果相同') }
  1048. runtime.onMessage = async () => { if (calls === 5) runtime.close() }
  1049. const root = runtime.send(owner, g.id, { requestId: randomUUID(), content: '开始' })
  1050. await vi.waitFor(() => expect(store.list<Work>(owner, 'turn').filter(w => w.status === 'complete')).toHaveLength(5))
  1051. if (legacy) for (const work of store.list<Work>(owner, 'turn')) { delete work.relay; store.put(owner, 'turn', work.id, work) }
  1052. runtime = new WorkspaceRuntime(store, nodes, uploads); runtime.start()
  1053. await idle(root.id, g.id)
  1054. expect(calls).toBe(5)
  1055. expect(store.messages(owner, g.id).filter(m => m.content === REPEATED_RELAY_NOTICE)).toHaveLength(1)
  1056. runtime.close(); runtime = new WorkspaceRuntime(store, nodes, uploads); runtime.start()
  1057. await idle(root.id, g.id)
  1058. expect(calls).toBe(5)
  1059. expect(store.messages(owner, g.id).filter(m => m.content === REPEATED_RELAY_NOTICE)).toHaveLength(1)
  1060. })
  1061. })
  1062. it('starts with the administrator, executes a batch concurrently, and reviews failures once without spending a round', async () => {
  1063. const a = agent('管理员'), b = agent('成员乙'), c = agent('成员丙')
  1064. const group = store.createGroup(owner, { name: '并行验收', memberIds: [a.id,b.id,c.id], administratorId: a.id, maxReplyRounds: 2 })
  1065. const pending = new Map<string, { socket: WebSocket; p: Record<string, any> }>()
  1066. let admin = 0, review = ''
  1067. reply = (socket,p) => {
  1068. if (speaker(p) === a.name) { if (admin++ === 0) completeReply(socket,p,'请@成员乙检查接口,@成员丙检查数据'); else { review=p.text; completeReply(socket,p,'汇总完成') } }
  1069. else pending.set(speaker(p)!, {socket,p})
  1070. }
  1071. const root = runtime.send(owner,group.id,{requestId:randomUUID(),content:'@成员乙 请处理'})
  1072. await vi.waitFor(()=>expect(pending.size).toBe(2))
  1073. expect(admin).toBe(1)
  1074. expect(store.require<WorkspaceConversation>(owner,'conversation',group.id).activeAgentStates).toEqual({[b.id]:'running',[c.id]:'running'})
  1075. completeReply(pending.get(c.name)!.socket,pending.get(c.name)!.p,'数据检查失败','failed')
  1076. await vi.waitFor(()=>expect(store.messages(owner,group.id).some(m=>m.status==='failed')).toBe(true))
  1077. expect(admin).toBe(1)
  1078. completeReply(pending.get(b.name)!.socket,pending.get(b.name)!.p,'接口正常')
  1079. await finished(root.id)
  1080. expect(admin).toBe(2)
  1081. expect(review).toContain('数据检查失败')
  1082. expect(review).toContain('接口正常')
  1083. expect(store.require<WorkspaceRun>(owner,'run',root.id).round).toBe(1)
  1084. })
  1085. it('allows successive dependent delegations and does not charge administrator reviews against the limit', async () => {
  1086. const a=agent('负责人'),b=agent('先做'),c=agent('后做')
  1087. const g=store.createGroup(owner,{name:'串行依赖',memberIds:[a.id,b.id,c.id],administratorId:a.id,maxReplyRounds:3})
  1088. const order:string[]=[];let admin=0
  1089. reply=(socket,p)=>{const name=speaker(p)!;order.push(name);completeReply(socket,p,name===a.name ? ['@先做 第一步','@后做 第二步','完成'][admin++]! : '结果已提交')}
  1090. await finished(runtime.send(owner,g.id,{requestId:randomUUID(),content:'执行'}).id)
  1091. expect(order).toEqual([a.name,b.name,a.name,c.name,a.name])
  1092. })
  1093. it('continues free discussion automatically and hides irrelevant automatic replies', async () => {
  1094. const a=agent('公开管理员'),b=agent('自动成员')
  1095. const g=store.createGroup(owner,{name:'自由讨论',memberIds:[a.id,b.id],administratorId:a.id,mode:'free',autoReplyIds:[b.id],maxReplyRounds:3})
  1096. reply=(socket,p)=>completeReply(socket,p,speaker(p)===a.name?'我来回答':'[[YAOYAO_NO_REPLY_V1]]')
  1097. const root=runtime.send(owner,g.id,{requestId:randomUUID(),content:'一个问题'})
  1098. await finished(root.id)
  1099. expect(requests.filter(r=>r.method==='prompt.submit'&&speaker(r.params)===b.name)).toHaveLength(2)
  1100. expect(store.messages(owner,g.id).map(m=>m.content)).toEqual(['一个问题','我来回答'])
  1101. expect(store.conversationSummary(owner,store.require(owner,'conversation',g.id))).toMatchObject({unread:true, unreadCount:1})
  1102. expect(store.hiddenMessageIds(owner,g.id)).toHaveLength(2)
  1103. })
  1104. it('requires a public administrator reply when the model tries to remain silent', async () => {
  1105. const a=agent('必须答复'),b=agent('旁观者'),g=store.createGroup(owner,{name:'公开答复',memberIds:[a.id,b.id],administratorId:a.id})
  1106. reply=(socket,p)=>completeReply(socket,p,'[[YAOYAO_NO_REPLY_V1]]')
  1107. await finished(runtime.send(owner,g.id,{requestId:randomUUID(),content:'帮助我'}).id)
  1108. const answer=store.messages(owner,g.id).at(-1)!
  1109. expect(answer.content).toContain('请补充具体目标')
  1110. expect(answer.content).not.toContain('YAOYAO_')
  1111. })
  1112. it('queues later user requests durably without leaking them into the preceding collaboration', async () => {
  1113. const a=agent('排队管理员'),b=agent('排队执行者'),g=store.createGroup(owner,{name:'消息排队',memberIds:[a.id,b.id],administratorId:a.id})
  1114. let first: {socket:WebSocket;p:Record<string,any>}|undefined
  1115. let admin=0;const prompts:string[]=[]
  1116. reply=(socket,p)=>{prompts.push(p.text);if(speaker(p)===a.name&&admin++===0)first={socket,p};else completeReply(socket,p,'已完成')}
  1117. const one=runtime.send(owner,g.id,{requestId:randomUUID(),content:'第一条任务'})
  1118. await vi.waitFor(()=>expect(first).toBeDefined())
  1119. const two=runtime.send(owner,g.id,{requestId:randomUUID(),content:'之后才处理的秘密标记'})
  1120. expect(two.status).toBe('queued')
  1121. expect(store.require<WorkspaceConversation>(owner,'conversation',g.id).activeRunId).toBe(one.id)
  1122. completeReply(first!.socket,first!.p,'请@排队执行者继续')
  1123. await finished(one.id);await finished(two.id)
  1124. expect(prompts).toHaveLength(4)
  1125. expect(prompts.slice(0,3).every(p=>!p.includes('之后才处理的秘密标记'))).toBe(true)
  1126. expect(prompts[3]).toContain('之后才处理的秘密标记')
  1127. })
  1128. it('stops one member while keeping its sibling running and returns the interrupted result for review', async () => {
  1129. const a=agent('停止管理员'),b=agent('停止乙'),c=agent('继续丙'),g=store.createGroup(owner,{name:'独立停止',memberIds:[a.id,b.id,c.id],administratorId:a.id})
  1130. const pending=new Map<string,{socket:WebSocket;p:Record<string,any>}>();let admin=0,review=''
  1131. reply=(socket,p)=>{if(speaker(p)===a.name){if(admin++===0)completeReply(socket,p,'@停止乙 @继续丙');else {review=p.text;completeReply(socket,p,'已汇总')}}else pending.set(speaker(p)!,{socket,p})}
  1132. const root=runtime.send(owner,g.id,{requestId:randomUUID(),content:'开始'})
  1133. await vi.waitFor(()=>expect(pending.size).toBe(2))
  1134. await runtime.stopAgent(owner,g.id,b.id)
  1135. expect(admin).toBe(1)
  1136. expect(store.require<WorkspaceConversation>(owner,'conversation',g.id).activeAgentStates?.[c.id]).toBe('running')
  1137. completeReply(pending.get(c.name)!.socket,pending.get(c.name)!.p,'丙完成')
  1138. await finished(root.id)
  1139. expect(review).toContain('interrupted')
  1140. expect(review).toContain('丙完成')
  1141. })
  1142. it('recovers a partially completed parallel batch after restart without resubmitting members or duplicating review', async () => {
  1143. const a=agent('恢复管理员'),b=agent('恢复乙'),c=agent('恢复丙'),g=store.createGroup(owner,{name:'批次恢复',memberIds:[a.id,b.id,c.id],administratorId:a.id})
  1144. let admin=0;const pending=new Map<string,{socket:WebSocket;p:Record<string,any>}>()
  1145. reply=(socket,p)=>{if(speaker(p)===a.name)completeReply(socket,p,admin++===0?'@恢复乙 @恢复丙':'复核完成');else pending.set(speaker(p)!,{socket,p})}
  1146. const root=runtime.send(owner,g.id,{requestId:randomUUID(),content:'并行执行'})
  1147. await vi.waitFor(()=>expect(pending.size).toBe(2))
  1148. completeReply(pending.get(b.name)!.socket,pending.get(b.name)!.p,'乙已完成')
  1149. const last=pending.get(c.name)!
  1150. recoveryByStored.set(storedByRuntime.get(last.p.session_id)!,[{role:'user',content:last.p.text},{role:'assistant',content:'丙已完成'}])
  1151. last.socket.terminate()
  1152. await vi.waitFor(()=>expect(store.require<WorkspaceRun>(owner,'run',root.id).status).toBe('uncertain'))
  1153. runtime.close();runtime=new WorkspaceRuntime(store,nodes,uploads);runtime.start()
  1154. await finished(root.id)
  1155. expect(admin).toBe(2)
  1156. expect(requests.filter(r=>r.method==='prompt.submit')).toHaveLength(4)
  1157. expect(store.list<any>(owner,'turn').filter(t=>t.reviewOf)).toHaveLength(1)
  1158. })
  1159. it('persists the next delegation when restarted between reply completion and cascade planning', async () => {
  1160. const a=agent('记录管理员'),b=agent('记录成员'),g=store.createGroup(owner,{name:'计划恢复',memberIds:[a.id,b.id],administratorId:a.id})
  1161. let admin=0
  1162. reply=(socket,p)=>completeReply(socket,p,speaker(p)===a.name&&admin++===0?'@记录成员 执行':'完成')
  1163. runtime.onMessage=async()=>{runtime.close()}
  1164. const root=runtime.send(owner,g.id,{requestId:randomUUID(),content:'执行一次'})
  1165. await vi.waitFor(()=>expect(store.list<any>(owner,'turn')[0]?.status).toBe('complete'))
  1166. runtime=new WorkspaceRuntime(store,nodes,uploads);runtime.start()
  1167. await finished(root.id)
  1168. expect(requests.filter(r=>r.method==='prompt.submit')).toHaveLength(3)
  1169. })
  1170. it('bounds long context while preserving the task beginning and ending', async () => {
  1171. const a=agent('长上下文'),b=agent('辅助'),g=store.createGroup(owner,{name:'上下文边界',memberIds:[a.id,b.id],administratorId:a.id})
  1172. const text='任务开始标记'+ '文'.repeat(60_000)+'任务结束标记'
  1173. await finished(runtime.send(owner,g.id,{requestId:randomUUID(),content:text}).id)
  1174. const prompt=requests.find(r=>r.method==='prompt.submit')!.params.text
  1175. expect(prompt).toContain('任务开始标记');expect(prompt).toContain('任务结束标记')
  1176. expect(prompt).toContain('保留首尾片段');expect(prompt.length).toBeLessThan(32_000)
  1177. })
  1178. it('serializes a session and stops only that task while the same Agent keeps replying elsewhere', async () => {
  1179. const a=agent('同一机器人'), b=agent('旁观成员')
  1180. const c=store.createGroup(owner,{name:'会话隔离',memberIds:[a.id,b.id],administratorId:a.id})
  1181. const first=defaultTask(c.id), second=store.createTask(owner,c.id,{})
  1182. const pending:Array<{socket:WebSocket;p:Record<string,any>}>=[]
  1183. reply=(socket,p)=>pending.push({socket,p})
  1184. const one=runtime.send(owner,c.id,{requestId:randomUUID(),taskId:first.id,content:'第一话题开始'})
  1185. await vi.waitFor(()=>expect(pending).toHaveLength(1))
  1186. const queued=runtime.send(owner,c.id,{requestId:randomUUID(),taskId:first.id,content:'第一话题追问'})
  1187. const other=runtime.send(owner,c.id,{requestId:randomUUID(),taskId:second.id,content:'第二话题独立执行'})
  1188. await vi.waitFor(()=>expect(pending).toHaveLength(2))
  1189. expect(store.require<WorkspaceRun>(owner,'run',queued.id).status).toBe('queued')
  1190. completeReply(pending[0]!.socket,pending[0]!.p,'第一话题首轮结果')
  1191. await finished(one.id)
  1192. await vi.waitFor(()=>expect(pending).toHaveLength(3))
  1193. const firstBinding=store.get<any>(owner,'binding',`${c.id}:${first.id}:${a.id}`)!
  1194. expect(requests.filter(r=>r.method==='session.resume').map(r=>r.params.session_id)).toEqual([firstBinding.storedId])
  1195. expect(pending[2]!.p.text).not.toContain('第二话题独立执行')
  1196. await runtime.stopTask(owner,c.id,first.id)
  1197. await vi.waitFor(()=>expect(requests.filter(r=>r.method==='session.interrupt').map(r=>r.params.session_id)).toEqual([pending[2]!.p.session_id]))
  1198. expect(store.require<WorkspaceRun>(owner,'run',other.id).status).toBe('running')
  1199. expect(store.require<any>(owner,'conversation-task',second.id).activeRunStatus).toBe('running')
  1200. completeReply(pending[1]!.socket,pending[1]!.p,'第二话题独立完成')
  1201. await finished(other.id)
  1202. expect(store.messages(owner,c.id,Number.MAX_SAFE_INTEGER,100,false,second.id).map(m=>m.content)).toEqual(['第二话题独立执行','第二话题独立完成'])
  1203. expect(store.require<WorkspaceRun>(owner,'run',queued.id).status).toBe('interrupted')
  1204. })
  1205. it('recovers simultaneous sessions for one Agent after restart without replaying either prompt', async () => {
  1206. const a=agent('恢复机器人'),b=agent('恢复旁观者')
  1207. const c=store.createGroup(owner,{name:'并发恢复',memberIds:[a.id,b.id],administratorId:a.id})
  1208. const tasks=[defaultTask(c.id),store.createTask(owner,c.id,{})]
  1209. reply=(_socket,p)=>recoveryByStored.set(storedByRuntime.get(p.session_id)!,[
  1210. {role:'user',content:p.text},{role:'assistant',content:p.text.includes('恢复甲')?'甲的最终结果':'乙的最终结果'},
  1211. ])
  1212. const runs=tasks.map((task,i)=>runtime.send(owner,c.id,{requestId:randomUUID(),taskId:task.id,content:i===0?'恢复甲':'恢复乙'}))
  1213. await vi.waitFor(()=>expect(requests.filter(r=>r.method==='prompt.submit')).toHaveLength(2))
  1214. runtime.close();runtime=new WorkspaceRuntime(store,nodes,uploads);runtime.start()
  1215. await Promise.all(runs.map(r=>finished(r.id)))
  1216. expect(requests.filter(r=>r.method==='prompt.submit')).toHaveLength(2)
  1217. expect(requests.filter(r=>r.method==='session.interrupt')).toHaveLength(0)
  1218. expect(tasks.map(task=>store.messages(owner,c.id,Number.MAX_SAFE_INTEGER,100,false,task.id).at(-1)?.content)).toEqual(['甲的最终结果','乙的最终结果'])
  1219. })
  1220. it('enforces three turns per topic and ten globally while preserving queued work', async () => {
  1221. const agents=Array.from({length:12},(_,i)=>agent(`并发成员${i}`))
  1222. const group=store.createGroup(owner,{name:'并发限制',memberIds:agents.slice(0,4).map(a=>a.id),administratorId:agents[0]!.id,mode:'free',autoReplyIds:agents.slice(0,4).map(a=>a.id),maxReplyRounds:1})
  1223. const pending=new Map<string,{socket:WebSocket;p:Record<string,any>}>()
  1224. reply=(socket,p)=>pending.set(speaker(p)!,{socket,p})
  1225. const one=runtime.send(owner,group.id,{requestId:randomUUID(),content:'并发测试'})
  1226. await vi.waitFor(()=>expect(pending.size).toBe(3))
  1227. const others=agents.slice(4).map(a=>runtime.send(owner,direct(a.id).id,{requestId:randomUUID(),content:'另一群'}))
  1228. await vi.waitFor(()=>expect(pending.size).toBe(10))
  1229. expect(Object.keys(store.require<WorkspaceConversation>(owner,'conversation',group.id).activeAgentStates!)).toHaveLength(3)
  1230. expect(store.list<any>(owner,'turn').filter(t=>t.status==='running')).toHaveLength(10)
  1231. expect(store.list<any>(owner,'turn').filter(t=>t.status==='queued')).toHaveLength(2)
  1232. expect(requests.filter(r=>r.method==='session.interrupt')).toHaveLength(0)
  1233. const released=[...pending.values()]
  1234. for(const v of released) completeReply(v.socket,v.p,'完成')
  1235. await vi.waitFor(()=>expect(pending.size).toBe(12))
  1236. expect(store.list<any>(owner,'turn').filter(t=>t.status==='running').length).toBeLessThanOrEqual(10)
  1237. for(const v of pending.values()) if(!released.includes(v)) completeReply(v.socket,v.p,'完成')
  1238. await finished(one.id);await Promise.all(others.map(r=>finished(r.id)))
  1239. })
  1240. it('supports unlimited rounds beyond the finite limit and still permits stopping', async () => {
  1241. const a=agent('不限甲'),b=agent('不限乙'),g=store.createGroup(owner,{name:'不限轮次',memberIds:[a.id,b.id],administratorId:a.id,mode:'free',autoReplyIds:[a.id],maxReplyRounds:-1})
  1242. let count=0
  1243. reply=(socket,p)=>{
  1244. if(count++===0){const work=store.list<any>(owner,'turn').find(t=>t.status==='running')!;work.depth=100;store.put(owner,'turn',work.id,work);completeReply(socket,p,'继续讨论')}
  1245. else completeReply(socket,p,'[[YAOYAO_NO_REPLY_V1]]')
  1246. }
  1247. const root=runtime.send(owner,g.id,{requestId:randomUUID(),content:'@不限乙 开始'})
  1248. await finished(root.id)
  1249. expect(count).toBe(2)
  1250. expect(store.require<WorkspaceRun>(owner,'run',root.id).round).toBe(101)
  1251. expect(()=>store.updateConversation(owner,g.id,{maxReplyRounds:0})).toThrow()
  1252. })
  1253. it('applies one global approval policy to all Bots, ignoring legacy overrides and duplicate requests', async () => {
  1254. const bots=[agent('旧询问'),agent('旧允许'),agent('旧拒绝')]
  1255. bots.forEach((bot,index)=>store.put(owner,'agent',bot.id,{...bot,approvalPolicy:['ask','allow','deny'][index]}))
  1256. server.on('connection',socket=>socket.on('message',raw=>{const frame=JSON.parse(String(raw));if(frame.method==='approval.respond')completeReply(socket,frame.params,'审批已处理')}))
  1257. reply=(socket,params)=>{const frame=JSON.stringify({method:'event',params:{type:'approval.request',session_id:params.session_id,payload:{request_id:'same-id',message:'是否执行'}}});socket.send(frame);socket.send(frame)}
  1258. for(const policy of ['allow','deny'] as const){
  1259. saveHostTools(home,{approvalPolicy:policy})
  1260. const runs=bots.map(bot=>runtime.send(owner,direct(bot.id).id,{requestId:randomUUID(),content:'执行任务'}))
  1261. await Promise.all(runs.map(run=>finished(run.id)))
  1262. expect(requests.filter(f=>f.method==='approval.respond').slice(-3).map(f=>f.params.choice)).toEqual(Array(3).fill(policy==='allow'?'once':'deny'))
  1263. }
  1264. expect(requests.filter(f=>f.method==='approval.respond')).toHaveLength(6)
  1265. })
  1266. it('treats a legacy auto reply as one approval without altering global or other Bot policy', async () => {
  1267. const bot = agent('选择自动允许'), peer = agent('同一基础机器人')
  1268. reply = (socket, params) => socket.send(JSON.stringify({ method: 'event', params: { type: 'approval.request', session_id: params.session_id, payload: { request_id: 'first-approval', message: '生成图片' } } }))
  1269. server.on('connection', socket => socket.on('message', raw => {
  1270. const frame = JSON.parse(String(raw))
  1271. if (frame.method !== 'approval.respond') return
  1272. if (frame.params.request_id === 'first-approval') socket.send(JSON.stringify({ method: 'event', params: { type: 'approval.request', session_id: frame.params.session_id, payload: { request_id: 'next-approval', message: '保存图片' } } }))
  1273. else completeReply(socket, frame.params, '自动完成')
  1274. }))
  1275. const run = runtime.send(owner, direct(bot.id).id, { requestId: randomUUID(), content: '生成图片' })
  1276. await vi.waitFor(() => expect(store.list<WorkspaceInteraction>(owner, 'interaction')).toHaveLength(1))
  1277. const request = store.list<WorkspaceInteraction>(owner, 'interaction')[0]!
  1278. await expect(runtime.respond(other, request.id, 'auto')).rejects.toThrow('记录不存在')
  1279. expect(store.require<any>(owner, 'agent', bot.id).approvalPolicy).toBe('ask')
  1280. await runtime.respond(owner, request.id, 'auto')
  1281. await vi.waitFor(()=>expect(store.list<WorkspaceInteraction>(owner,'interaction').filter(i=>!i.resolved)).toHaveLength(1))
  1282. const next=store.list<WorkspaceInteraction>(owner,'interaction').find(i=>!i.resolved)!
  1283. expect(readHostTools(home).approvalPolicy).toBe('ask')
  1284. await runtime.respond(owner,next.id,'once')
  1285. await finished(run.id)
  1286. expect(requests.filter(frame => frame.method === 'approval.respond').map(frame => frame.params.choice)).toEqual(['once', 'once'])
  1287. expect(store.agentSummary(store.require<any>(owner, 'agent', bot.id)).approvalPolicy).toBe('ask')
  1288. expect(store.require<any>(owner, 'agent', peer.id).approvalPolicy).toBe('ask')
  1289. store.updateAgent(owner, bot.id, { approvalPolicy: 'ask' })
  1290. await runtime.respond(owner, request.id, 'auto')
  1291. expect(store.require<any>(owner, 'agent', bot.id).approvalPolicy).toBe('ask')
  1292. expect(requests.filter(frame => frame.method === 'approval.respond')).toHaveLength(2)
  1293. })
  1294. it('preserves current team grants for approval-only edits without reviving revoked grants', () => {
  1295. const bot = agent('有效工具授权')
  1296. store.put(owner, 'turn', 'valid-grant', { id: 'valid-grant', agentId: bot.id, status: 'waiting', teamManagementRevision: bot.revision })
  1297. store.put(owner, 'turn', 'revoked-grant', { id: 'revoked-grant', agentId: bot.id, status: 'running', teamManagementRevision: bot.revision - 1 })
  1298. const updated = store.updateAgent(owner, bot.id, { approvalPolicy: 'allow' })
  1299. expect(store.require<any>(owner, 'turn', 'valid-grant').teamManagementRevision).toBe(updated.revision)
  1300. expect(store.require<any>(owner, 'turn', 'revoked-grant').teamManagementRevision).toBe(bot.revision - 1)
  1301. store.updateAgent(owner, bot.id, { instructions: '新规则' })
  1302. expect(store.require<any>(owner, 'turn', 'valid-grant').teamManagementRevision).toBe(updated.revision)
  1303. })
  1304. it('keeps an explicit automatic preference but reports an unconfirmed reply without retrying it', async () => {
  1305. const bot = agent('自动设置与答复状态')
  1306. reply = (socket, params) => socket.send(JSON.stringify({ method: 'event', params: { type: 'approval.request', session_id: params.session_id, payload: { request_id: 'uncertain-auto', message: '需要确认' } } }))
  1307. runtime.send(owner, direct(bot.id).id, { requestId: randomUUID(), content: '继续' })
  1308. await vi.waitFor(() => expect(store.list<WorkspaceInteraction>(owner, 'interaction')).toHaveLength(1))
  1309. const interaction = store.list<WorkspaceInteraction>(owner, 'interaction')[0]!
  1310. const rpc = WorkspaceGateway.prototype.rpc
  1311. const spy = vi.spyOn(WorkspaceGateway.prototype, 'rpc').mockImplementation(function (this: WorkspaceGateway, method, params) {
  1312. return method === 'approval.respond' ? Promise.reject(new Error('connection lost')) : rpc.call(this, method, params)
  1313. })
  1314. try {
  1315. await expect(runtime.respond(owner, interaction.id, 'auto')).rejects.toMatchObject({ code: 'approval_auto_response_failed' })
  1316. expect(store.agentSummary(store.require<any>(owner, 'agent', bot.id)).approvalPolicy).toBe('ask')
  1317. expect(store.require<any>(owner, 'interaction', interaction.id)).toMatchObject({ resolved: false, responseState: 'uncertain', answer: 'auto' })
  1318. await expect(runtime.respond(owner, interaction.id, 'auto')).rejects.toMatchObject({ code: 'interaction_uncertain' })
  1319. expect(spy.mock.calls.filter(call => call[0] === 'approval.respond')).toHaveLength(1)
  1320. } finally { spy.mockRestore() }
  1321. })
  1322. it('leaves clarifications and already pending approvals for the user after a policy change', async () => {
  1323. const bot = agent('更改审批')
  1324. let kind = 'approval'
  1325. server.on('connection', socket => socket.on('message', raw => {
  1326. const frame = JSON.parse(String(raw))
  1327. if (['approval.respond', 'clarify.respond'].includes(frame.method)) completeReply(socket, frame.params, '已答复')
  1328. }))
  1329. reply = (socket, params) => socket.send(JSON.stringify({ method: 'event', params: { type: `${kind}.request`, session_id: params.session_id, payload: { request_id: 'pending', message: '请确认' } } }))
  1330. for (kind of ['approval', 'clarify']) {
  1331. const run = runtime.send(owner, direct(bot.id).id, { requestId: randomUUID(), content: '继续任务' })
  1332. await vi.waitFor(() => expect(store.require<WorkspaceRun>(owner, 'run', run.id).status).toBe('waiting'))
  1333. saveHostTools(home, { approvalPolicy: 'allow' })
  1334. const interaction = store.list<WorkspaceInteraction>(owner, 'interaction').find(item => !item.resolved)!
  1335. expect(requests.filter(frame => ['approval.respond', 'clarify.respond'].includes(frame.method))).toHaveLength(kind === 'approval' ? 0 : 1)
  1336. await runtime.respond(owner, interaction.id, kind === 'approval' ? 'once' : '下一步')
  1337. await finished(run.id)
  1338. }
  1339. })
  1340. it('defaults legacy and newly created Bots to manual approval and prevents autonomous grants', () => {
  1341. const parent = agent('创建者')
  1342. expect(parent.approvalPolicy).toBe('ask')
  1343. const child = store.createAgent(owner, { name: '子成员', profile: parent.profile, approvalPolicy: 'allow' }, { createdByAgentId: parent.id, createdFromRunId: randomUUID() })
  1344. expect(child.approvalPolicy).toBe('ask')
  1345. expect(() => store.updateAgent(owner, child.id, { approvalPolicy: 'always' })).toThrow('approvalPolicy')
  1346. store.put(owner, 'agent', child.id, { ...child, remoteAgentId: randomUUID() })
  1347. expect(store.agentSummary(store.updateAgent(owner, child.id, { approvalPolicy: 'deny' })).approvalPolicy).toBe('ask')
  1348. expect(() => store.updateAgent(owner, child.id, { name: '不能改远端资料' })).toThrow('远端管理')
  1349. })
  1350. it('keeps a failed automatic approval pending and does not retry a replayed request', async () => {
  1351. const bot = agent('审批失败')
  1352. saveHostTools(home,{approvalPolicy:'allow'})
  1353. const rpc = WorkspaceGateway.prototype.rpc
  1354. let attempts = 0
  1355. const spy = vi.spyOn(WorkspaceGateway.prototype, 'rpc').mockImplementation(function (this: WorkspaceGateway, method, params) {
  1356. if (method === 'approval.respond' && attempts++ === 0) return Promise.reject(new HttpError(409, '暂时拒绝答复', 'gateway_rejected'))
  1357. return rpc.call(this, method, params)
  1358. })
  1359. try {
  1360. const notify = vi.fn()
  1361. runtime.onNotify = notify
  1362. reply = (socket, params) => {
  1363. const frame = JSON.stringify({ method: 'event', params: { type: 'approval.request', session_id: params.session_id, payload: { request_id: 'failure', message: '确认操作' } } })
  1364. socket.send(frame); socket.send(frame)
  1365. }
  1366. server.on('connection', socket => socket.on('message', raw => {
  1367. const frame = JSON.parse(String(raw))
  1368. if (frame.method === 'approval.respond') completeReply(socket, frame.params, '手动处理完成')
  1369. }))
  1370. const run = runtime.send(owner, direct(bot.id).id, { requestId: randomUUID(), content: '继续' })
  1371. await vi.waitFor(() => expect(notify.mock.calls.some(call => call[4]?.kind === 'approval')).toBe(true))
  1372. expect(attempts).toBe(1)
  1373. const interaction = store.list<WorkspaceInteraction>(owner, 'interaction').find(item => !item.resolved)!
  1374. expect(store.require<WorkspaceRun>(owner, 'run', run.id).status).toBe('waiting')
  1375. await runtime.respond(owner, interaction.id, 'deny')
  1376. await finished(run.id)
  1377. } finally { spy.mockRestore() }
  1378. })
  1379. it('keeps parallel approval cards scoped to their member and deduplicates replayed requests and answers', async () => {
  1380. const a=agent('审批管理员'),b=agent('审批乙'),c=agent('审批丙'),g=store.createGroup(owner,{name:'并发审批',memberIds:[a.id,b.id,c.id],administratorId:a.id})
  1381. let admin=0
  1382. server.on('connection',socket=>socket.on('message',raw=>{const f=JSON.parse(String(raw));if(f.method==='approval.respond')completeReply(socket,f.params,'已确认')}))
  1383. reply=(socket,p)=>{
  1384. if(speaker(p)===a.name){completeReply(socket,p,admin++===0?'@审批乙 @审批丙':'汇总');return}
  1385. const frame=JSON.stringify({method:'event',params:{type:'approval.request',session_id:p.session_id,payload:{request_id:'same-request',message:'是否继续'}}})
  1386. socket.send(frame);socket.send(frame)
  1387. }
  1388. const root=runtime.send(owner,g.id,{requestId:randomUUID(),content:'开始'})
  1389. await vi.waitFor(()=>expect(store.list<WorkspaceInteraction>(owner,'interaction').filter(i=>!i.resolved)).toHaveLength(2))
  1390. const interactions=store.list<WorkspaceInteraction>(owner,'interaction')
  1391. expect(store.require<WorkspaceConversation>(owner,'conversation',g.id).activeAgentStates).toEqual({[b.id]:'waiting',[c.id]:'waiting'})
  1392. await runtime.respond(owner,interactions[0]!.id,'once')
  1393. await runtime.respond(owner,interactions[0]!.id,'once')
  1394. await expect(runtime.respond(owner,interactions[0]!.id,'deny')).rejects.toThrow('其他答复')
  1395. expect(admin).toBe(1)
  1396. await runtime.respond(owner,interactions[1]!.id,'once')
  1397. await finished(root.id)
  1398. expect(requests.filter(r=>r.method==='approval.respond')).toHaveLength(2)
  1399. expect(admin).toBe(2)
  1400. })
  1401. it('retries a rolled-back cascade plan without losing or duplicating delegation', async () => {
  1402. const a=agent('事务管理员'),b=agent('事务成员'),g=store.createGroup(owner,{name:'调度事务',memberIds:[a.id,b.id],administratorId:a.id})
  1403. const original=store.put.bind(store);let failed=false,admin=0
  1404. store.put=(user,kind,id,value)=>{if(!failed&&kind==='turn'&&(value as any).agentId===b.id){failed=true;throw new Error('模拟一次写入失败')}original(user,kind,id,value)}
  1405. reply=(socket,p)=>completeReply(socket,p,speaker(p)===a.name&&admin++===0?'@事务成员 继续':'完成')
  1406. const root=runtime.send(owner,g.id,{requestId:randomUUID(),content:'开始'})
  1407. await vi.waitFor(()=>expect(store.require<WorkspaceRun>(owner,'run',root.id).status).toBe('complete'),{timeout:4000})
  1408. expect(failed).toBe(true)
  1409. expect(store.list<any>(owner,'turn').filter(t=>t.agentId===b.id)).toHaveLength(1)
  1410. expect(requests.filter(r=>r.method==='prompt.submit')).toHaveLength(3)
  1411. })
  1412. it('upgrades an admitted v0.3.7 execution without submitting its prompt again', async () => {
  1413. const a=agent('升级执行'),c=direct(a.id)
  1414. reply=(socket,p)=>{recoveryHistory=[{role:'user',content:p.text},{role:'assistant',content:'升级前已完成'}];socket.terminate()}
  1415. const root=runtime.send(owner,c.id,{requestId:randomUUID(),content:'只执行一次'})
  1416. await vi.waitFor(()=>expect(store.require<WorkspaceRun>(owner,'run',root.id).status).toBe('uncertain'))
  1417. runtime.close()
  1418. const old=store.list<any>(owner,'turn')[0]!
  1419. for(const task of store.list<any>(owner,'turn'))store.remove(owner,'turn',task.id)
  1420. const legacy=store.require<any>(owner,'run',root.id)
  1421. Object.assign(legacy,{queue:[a.id],next:[],hostReturn:false,currentMessageId:old.currentMessageId,turnConfiguration:old.turnConfiguration})
  1422. store.put(owner,'run',root.id,legacy)
  1423. const binding=store.get<any>(owner,'binding',`${c.id}:${a.id}`)!
  1424. delete binding.taskId;store.put(owner,'binding',binding.id,binding)
  1425. runtime=new WorkspaceRuntime(store,nodes,uploads);runtime.start()
  1426. await finished(root.id)
  1427. expect(requests.filter(r=>r.method==='prompt.submit')).toHaveLength(1)
  1428. expect(store.messages(owner,c.id).at(-1)?.content).toBe('升级前已完成')
  1429. })
  1430. it('does not read unrelated or orphaned historical runs when projecting a group prompt', async () => {
  1431. const a=agent('上下文隔离'),b=agent('组内成员'),g=store.createGroup(owner,{name:'干净群',memberIds:[a.id,b.id],administratorId:a.id})
  1432. store.put(owner,'run','historical-orphan',{id:'historical-orphan',conversationId:'unrelated-conversation',messageId:'missing-history',mentionIds:[],status:'complete',round:0,createdAt:1,updatedAt:1})
  1433. await finished(runtime.send(owner,g.id,{requestId:randomUUID(),content:'当前群任务'}).id)
  1434. expect(requests.filter(r=>r.method==='prompt.submit')).toHaveLength(1)
  1435. })
  1436. it('pages backwards to find a long-running turn marker and never mistakes a tool call for a final answer', async () => {
  1437. const a=agent('分页恢复'),c=direct(a.id)
  1438. reply=(socket,p)=>{
  1439. recoveryHistory=[{role:'user',content:p.text},...Array.from({length:600},()=>({role:'tool',content:'工具结果'})),{role:'assistant',content:'最终结果'}]
  1440. socket.terminate()
  1441. }
  1442. await finished(runtime.send(owner,c.id,{requestId:randomUUID(),content:'长时间执行'}).id)
  1443. expect(store.messages(owner,c.id).at(-1)?.content).toBe('最终结果')
  1444. let submitted=''
  1445. reply=(socket,p)=>{submitted=p.text;recoveryHistory=[{role:'user',content:submitted},{role:'assistant',content:'准备调用工具',tool_calls:[{id:'tool'}]},{role:'tool',content:'工具输出'}];socket.terminate()}
  1446. const root=runtime.send(owner,c.id,{requestId:randomUUID(),content:'不要把过程当结论'})
  1447. await vi.waitFor(()=>expect(store.require<WorkspaceRun>(owner,'run',root.id).status).toBe('uncertain'))
  1448. await runtime.reconcile(owner,root.id)
  1449. await new Promise(resolve=>setTimeout(resolve,100))
  1450. expect(store.require<WorkspaceRun>(owner,'run',root.id).status).toBe('uncertain')
  1451. recoveryHistory.push({role:'assistant',content:'真正完成'})
  1452. await runtime.reconcile(owner,root.id)
  1453. await finished(root.id)
  1454. expect(requests.filter(r=>r.method==='prompt.submit')).toHaveLength(2)
  1455. })
  1456. it('keeps explicit replies visible and emits one notice when the configured cascade limit is reached', async () => {
  1457. const a=agent('轮数管理员'),b=agent('轮数成员'),g=store.createGroup(owner,{name:'有限轮次',memberIds:[a.id,b.id],administratorId:a.id,maxReplyRounds:2})
  1458. reply=(socket,p)=>completeReply(socket,p,speaker(p)===a.name?'@轮数成员 继续':'已完成本步')
  1459. const root=runtime.send(owner,g.id,{requestId:randomUUID(),content:'开始'})
  1460. await finished(root.id)
  1461. expect(requests.filter(r=>r.method==='prompt.submit')).toHaveLength(3)
  1462. expect(store.messages(owner,g.id).filter(m=>m.content.includes('轮数上限'))).toHaveLength(1)
  1463. store.updateConversation(owner,g.id,{mode:'free',autoReplyIds:[]})
  1464. reply=(socket,p)=>completeReply(socket,p,'[[YAOYAO_NO_REPLY_V1]]')
  1465. await finished(runtime.send(owner,g.id,{requestId:randomUUID(),content:'@轮数成员 显示这个字符串'}).id)
  1466. expect(store.messages(owner,g.id).at(-1)?.content).toBe('[[YAOYAO_NO_REPLY_V1]]')
  1467. })
  1468. it('retries a failed stop of a live task without falsely completing it or resubmitting work', async () => {
  1469. const a=agent('停止恢复'),c=direct(a.id)
  1470. reply=()=>{}
  1471. const root=runtime.send(owner,c.id,{requestId:randomUUID(),content:'等待停止'})
  1472. await vi.waitFor(()=>expect(requests.filter(r=>r.method==='prompt.submit')).toHaveLength(1))
  1473. rejectedInterrupts=1
  1474. await runtime.stop(owner,root.id)
  1475. expect(store.require<WorkspaceRun>(owner,'run',root.id).status).toBe('interrupted')
  1476. await vi.waitFor(()=>expect(requests.filter(r=>r.method==='session.interrupt')).toHaveLength(2),{timeout:3000})
  1477. expect(requests.filter(r=>r.method==='session.interrupt')).toHaveLength(2)
  1478. expect(requests.filter(r=>r.method==='prompt.submit')).toHaveLength(1)
  1479. })
  1480. it('marks a stopped administrator as interrupted without sending a false completion notification', async () => {
  1481. const a=agent('管理员停止'),b=agent('普通成员'),g=store.createGroup(owner,{name:'停止管理员',memberIds:[a.id,b.id],administratorId:a.id})
  1482. reply=()=>{}
  1483. const notifications: string[]=[]
  1484. runtime.onNotify=(_owner,_c,run)=>notifications.push(run.status)
  1485. const root=runtime.send(owner,g.id,{requestId:randomUUID(),content:'等待'})
  1486. await vi.waitFor(()=>expect(requests.filter(r=>r.method==='prompt.submit')).toHaveLength(1))
  1487. await runtime.stopAgent(owner,g.id,a.id)
  1488. await vi.waitFor(()=>expect(store.require<WorkspaceRun>(owner,'run',root.id).status).toBe('interrupted'))
  1489. expect(notifications).toEqual([])
  1490. })
  1491. it('randomizes old default avatars once while preserving photos, custom v2 settings and direct snapshots',()=>{
  1492. const a=store.createAgent(owner,{name:'旧头像',profile:'default',instructions:'保留规则'})
  1493. const photo='data:image/png;base64,aGVsbG8='
  1494. const b=store.createAgent(owner,{name:'照片',profile:'default',avatar:photo})
  1495. store.put(owner,'agent',a.id,{...a,avatar:'yaoyao-mascot:v1:triangle:ff0000:friendly'})
  1496. store.put(owner,'agent',b.id,{...b,avatar:photo})
  1497. store.db.prepare("DELETE FROM workspace_migrations WHERE id IN ('avatar-v2','avatar-random-default-v1')").run()
  1498. const random=vi.spyOn(Math,'random').mockReturnValue(0)
  1499. const migrated=new WorkspaceStore(home)
  1500. random.mockRestore()
  1501. const next=migrated.require<any>(owner,'agent',a.id)
  1502. expect(next.instructions).toBe('保留规则')
  1503. expect(decodeAgentAvatar(next.avatar)).toMatchObject({avatarMode:'mascot'})
  1504. expect(next.avatar).not.toBe(encodeAgentAvatar(defaultAgentIdentity(a.id,a.name)))
  1505. const image=migrated.require<any>(owner,'agent',b.id)
  1506. expect(decodeAgentAvatar(image.avatar)).toMatchObject({avatarMode:'image',imageDataURL:photo,imageCrop:'rounded'})
  1507. for(const c of migrated.list<WorkspaceConversation>(owner,'conversation')) expect(c.avatar).toBe(migrated.require<any>(owner,'agent',c.memberIds[0]!).avatar)
  1508. const customized=encodeAgentAvatar({...defaultAgentIdentity('bot'),bodyId:'star',expression:'proud',color:'#1488ff'})
  1509. migrated.updateAgent(owner,a.id,{avatar:customized})
  1510. migrated.close()
  1511. const reopened=new WorkspaceStore(home)
  1512. expect(reopened.require<any>(owner,'agent',a.id).avatar).toBe(customized)
  1513. expect(reopened.db.prepare("SELECT count(*) AS count FROM workspace_migrations WHERE id='avatar-random-default-v1'").get()!.count).toBe(1)
  1514. reopened.close()
  1515. })
  1516. // A member can have a queued follow-up while their current task needs approval.
  1517. it('keeps active avatar state ahead of queued work and publishes only recent outcomes', () => {
  1518. const a=agent('状态优先级'), c=direct(a.id), now=Date.now()
  1519. const run: WorkspaceRun={id:'avatar-run',conversationId:c.id,messageId:'unused',mentionIds:[],status:'complete',round:1,createdAt:now,updatedAt:now}
  1520. for (const [id,status] of [['current','waiting'],['next','queued']] as const)
  1521. store.put(owner,'turn',id,{id,conversationId:c.id,agentId:a.id,status,planned:true,updatedAt:now})
  1522. store.put(owner,'turn','result',{id:'result',conversationId:c.id,agentId:a.id,status:'failed',updatedAt:now})
  1523. store.saveRun(owner,run)
  1524. let summary=store.require<WorkspaceConversation>(owner,'conversation',c.id)
  1525. expect(summary.activeAgentStates).toEqual({[a.id]:'waiting'})
  1526. expect(summary.avatarSignals?.[a.id]).toMatchObject({id:'result',state:'failure'})
  1527. store.put(owner,'turn','result',{id:'result',conversationId:c.id,agentId:a.id,status:'complete',updatedAt:now-3000})
  1528. store.saveRun(owner,run)
  1529. expect(store.require<WorkspaceConversation>(owner,'conversation',c.id).avatarSignals).toEqual({})
  1530. })
  1531. it('queues an unsubmitted turn while computer capacity is unavailable and submits it once after recovery',async()=>{
  1532. const a=agent('等候电脑'),c=direct(a.id),target=nodes.target(owner,'local')
  1533. vi.spyOn(target.session,'request').mockRejectedValueOnce(new HttpError(429,'资源不足','computer_quota'))
  1534. const run=runtime.send(owner,c.id,{requestId:randomUUID(),content:'等待资源后执行'})
  1535. await vi.waitFor(()=>expect(store.list<any>(owner,'turn').find(work=>work.runId===run.id)?.resourceWait).toBe(true))
  1536. expect(store.require<WorkspaceRun>(owner,'run',run.id).status).toBe('queued')
  1537. await vi.waitFor(()=>expect(store.require<WorkspaceRun>(owner,'run',run.id).status).toBe('complete'),{timeout:5000})
  1538. expect(requests.filter(request=>request.method==='prompt.submit')).toHaveLength(1)
  1539. })
  1540. it('keeps legacy VM Bots chatting on the server while their Runner is offline',async()=>{
  1541. const a=store.updateAgent(owner,agent('离线电脑').id,{execution:'computer',computer:'vm'}),c=direct(a.id)
  1542. nodes.targetForAgent=vi.fn(()=>{throw new HttpError(503,'节点未连接','runner_offline')})
  1543. const run=runtime.send(owner,c.id,{requestId:randomUUID(),content:'普通聊天不需要虚拟机'})
  1544. await finished(run.id)
  1545. expect(nodes.targetForAgent).not.toHaveBeenCalled()
  1546. expect(store.get<any>(owner,'binding',`${c.id}:${a.id}`)).toMatchObject({execution:'profile'})
  1547. })
  1548. it('can stop an unsubmitted turn that is waiting for computer capacity',async()=>{
  1549. const a=store.updateAgent(owner,agent('可停止等候').id,{execution:'computer',computer:'vm'}),c=direct(a.id)
  1550. vi.spyOn(nodes.target(owner,'local').session,'request').mockRejectedValue(new HttpError(429,'资源不足','computer_quota'))
  1551. const run=runtime.send(owner,c.id,{requestId:randomUUID(),content:'等待中停止'})
  1552. const work=await vi.waitFor(()=>{
  1553. const current=store.list<Work>(owner,'turn').find(work=>work.runId===run.id)
  1554. expect(current?.resourceWait).toBe(true)
  1555. return current!
  1556. })
  1557. await runtime.stop(owner,run.id)
  1558. await vi.waitFor(()=>expect(store.require<Work>(owner,'turn',work.id).status).toBe('interrupted'))
  1559. })
  1560. it('forwards a generated artifact to a new task without changing its original provenance',async()=>{
  1561. const a=agent('产物接收者'),c=direct(a.id)
  1562. const first=runtime.send(owner,c.id,{requestId:randomUUID(),content:'first'})
  1563. await finished(first.id)
  1564. const message=store.messages(owner,c.id).find(message=>message.role==='assistant')!
  1565. const assets=new WorkspaceAssets(store,nodes,home)
  1566. const file=assets.publish(owner,message,a,'review.txt',Buffer.from('verified artifact'))
  1567. const sent=runtime.dispatch(owner,c.id,{requestId:randomUUID(),content:'复核附件',fileIds:[file.id]},{agentId:a.id,kind:'task_review'})
  1568. await finished(sent.id)
  1569. const attached=requests.find(request=>request.method==='file.attach')!
  1570. expect(Buffer.from(attached.params.data_url.split(',')[1],'base64').toString()).toBe('verified artifact')
  1571. expect(store.require<any>(owner,'file',file.id).messageId).toBe(message.id)
  1572. expect(store.require<WorkspaceMessage>(owner,'message',sent.messageId).attachments.map(file=>file.id)).toEqual([file.id])
  1573. assets.close()
  1574. })
  1575. it('migrates a stored Worker binding to server Hermes once and preserves prior conversation',async()=>{
  1576. const a=store.updateAgent(owner,agent('迁移测试').id,{execution:'computer',computer:'vm',vmExecution:'profile'}),c=direct(a.id)
  1577. const first=runtime.send(owner,c.id,{requestId:randomUUID(),content:'已确认的原会话事实'})
  1578. await finished(first.id)
  1579. const key=`${c.id}:${a.id}`,old=store.get<any>(owner,'binding',key)!
  1580. store.put(owner,'binding',key,{...old,execution:'computer',runnerId:'old-runner',hermesComputer:true})
  1581. nodes.targetForAgent=vi.fn(()=>{throw new Error('旧 Runner 不应打开')})
  1582. const second=runtime.send(owner,c.id,{requestId:randomUUID(),content:'继续这次任务'})
  1583. await finished(second.id)
  1584. const migrated=store.get<any>(owner,'binding',key)!
  1585. expect(migrated.execution).toBe('profile');expect(migrated.storedId).not.toBe(old.storedId)
  1586. expect(requests.filter(r=>r.method==='session.create')).toHaveLength(2)
  1587. expect(requests.filter(r=>r.method==='prompt.submit')[1].params.text).toContain('已确认的原会话事实')
  1588. const third=runtime.send(owner,c.id,{requestId:randomUUID(),content:'再次继续'})
  1589. await finished(third.id)
  1590. expect(requests.filter(r=>r.method==='session.resume')).toHaveLength(1)
  1591. expect(nodes.targetForAgent).not.toHaveBeenCalled()
  1592. })
  1593. it('keeps memory flags out of session.create and grants scoped memory through the bridge',async()=>{
  1594. const a=agent('独立记忆'),c=direct(a.id),target=nodes.target(owner,'local'),request=target.session.request.bind(target.session),bindings:any[]=[]
  1595. vi.spyOn(target.session,'request').mockImplementation(async(path,options)=>{
  1596. if(path.endsWith('/capabilities'))return {status:200,headers:new Headers(),body:Buffer.from(JSON.stringify({ready:true,native_tools:true,in_process:true,memory_isolation:true}))}
  1597. if(path.endsWith('/bind')){bindings.push(options?.body);return {status:200,headers:new Headers(),body:Buffer.from(JSON.stringify({ok:true,native_tools:true,workspace_memory:true}))}}
  1598. if(path.endsWith('/unbind'))return {status:200,headers:new Headers(),body:Buffer.from('{"ok":true}')}
  1599. return request(path,options)
  1600. })
  1601. await finished(runtime.send(owner,c.id,{requestId:randomUUID(),content:'验证完整创建链路'}).id)
  1602. const create=requests.find(r=>r.method==='session.create')!
  1603. expect(create.params).not.toHaveProperty('skip_memory')
  1604. expect(create.params).not.toHaveProperty('workspace_memory')
  1605. expect(bindings).toContainEqual(expect.objectContaining({workspace_memory:true}))
  1606. })
  1607. it('snapshots each message origin and uses inherited provenance for internal dispatch only',async()=>{
  1608. const bot=agent('来源上下文'),c=direct(bot.id),a=randomUUID(),b=randomUUID()
  1609. runtime.desktopEnvironments=new DesktopEnvironments(store,{} as any,{localNodeID:'fixture',requireSource:()=>{}} as any)
  1610. const first=runtime.send(owner,c.id,{requestId:randomUUID(),content:'在本机工作',deviceHost:a})
  1611. await finished(first.id)
  1612. const second=runtime.send(owner,c.id,{requestId:randomUUID(),content:'切换设备',deviceHost:b})
  1613. await finished(second.id)
  1614. const delegated=runtime.dispatch(owner,c.id,{requestId:randomUUID(),content:'继续子任务',deviceHost:b},{agentId:bot.id,kind:'assignment',deviceHost:a})
  1615. await finished(delegated.id)
  1616. const unknown=runtime.send(owner,c.id,{requestId:randomUUID(),content:'浏览器消息',deviceHost:null})
  1617. await finished(unknown.id)
  1618. expect([first.deviceHost,second.deviceHost,delegated.deviceHost,unknown.deviceHost]).toEqual([a,b,a,undefined])
  1619. const prompts=requests.filter(r=>r.method==='prompt.submit').map(r=>r.params.text)
  1620. expect(prompts[0]).toContain(a);expect(prompts[1]).toContain(b);expect(prompts[2]).toContain(a)
  1621. expect(prompts[3]).toContain('历史消息的设备来源不适用于本轮')
  1622. expect(store.require<any>(owner,'message',delegated.messageId).deviceHost).toBe(a)
  1623. })