api-proxy-cold.spec.ts 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659
  1. /**
  2. * Cold-session and degenerate-composition paths of the host ApiProxy:
  3. * metadata-only listing, Agent-free history reads, subagent ownership
  4. * isolation, and prompt failure mapping.
  5. */
  6. import { mkdtempSync, writeFileSync, utimesSync } from 'node:fs'
  7. import { tmpdir } from 'node:os'
  8. import { join } from 'node:path'
  9. import { describe, expect, it, vi } from 'vitest'
  10. import { Context } from '@deepseek-ai/cordis'
  11. import SessionStore from '@deepseek-ai/dsh-session'
  12. import AgentRegistry from '@deepseek-ai/dsh-agent'
  13. import { TypeRTLookupFailure } from '@deepseek-ai/dsh-type-meta'
  14. import TypertRegistry from '@deepseek-ai/dsh-typert-registry'
  15. import { MessageId } from '@deepseek-ai/dsh-llm'
  16. import type { Agent } from '@deepseek-ai/dsh-agent'
  17. import UserInteractionService from '@deepseek-ai/dsh-user-interaction'
  18. import type { SessionEvent, SessionHeader, SessionId } from '@deepseek-ai/dsh-session'
  19. import {
  20. PersistenceCoordinator,
  21. SessionPersistenceRevision,
  22. type PersistenceBackend,
  23. type StoredPrefix,
  24. } from '@deepseek-ai/dsh-session-persistence'
  25. import type { RpcRequest } from '@deepseek-ai/dsh-host-apiproxy/api/rpc'
  26. import { RpcId } from '@deepseek-ai/dsh-host-apiproxy/api/rpc'
  27. import { createApiProxy } from '@deepseek-ai/dsh-host-apiproxy'
  28. const sid = (id: string): SessionId => id as SessionId
  29. let nextRpc = 1
  30. function request<P>(payload: P): RpcRequest<P> {
  31. return { rpcId: RpcId(`cold-${String(nextRpc++)}`), payload }
  32. }
  33. function header(id: string, createdAt: number, extra: Partial<SessionHeader> = {}): SessionHeader {
  34. return { version: 0, id: sid(id), createdAt, cwd: '/proj', ...extra }
  35. }
  36. describe('sessions.list cold merge', () => {
  37. it('summarizes unattached sessions: log mtime, locate-less and vanished-log createdAt fallbacks, lineage', async () => {
  38. const ctx = new Context()
  39. await ctx.plugin(SessionStore)
  40. await ctx.plugin(UserInteractionService)
  41. const root = mkdtempSync(join(tmpdir(), 'dsh-cold-'))
  42. const logPath = join(root, 'a.log')
  43. writeFileSync(logPath, 'log-bytes')
  44. utimesSync(logPath, 5000, 5000) // mtime 5_000_000 ms — newer than every createdAt below
  45. const metas = [
  46. header('session-a', 1000),
  47. header('session-b', 2000, { parentSession: sid('session-parent'), origin: 'subagent' }),
  48. header('session-c', 1500),
  49. ]
  50. // Structural fake of the persistence face list() consumes: list + locate.
  51. // locate: a real per-session file (mtime wins), a backend without one
  52. // (SQLite shape → createdAt), and a path whose file vanished (stat ENOENT
  53. // → createdAt).
  54. ctx.provide('sessionPersistence', {
  55. list: () => Promise.resolve(metas),
  56. locate: (meta: SessionHeader) => {
  57. if (meta.id === sid('session-a')) return { kind: 'jsonl', path: logPath }
  58. if (meta.id === sid('session-c')) return { kind: 'jsonl', path: join(root, 'vanished.log') }
  59. return undefined
  60. },
  61. })
  62. const api = createApiProxy(ctx, { defaultModelSelection: () => ({ provider: 'p', model: 'm' }), cwd: '/tmp' })
  63. const response = await api.sessions.list(request({}))
  64. expect(response.result.ok).toBe(true)
  65. if (!response.result.ok) throw new Error('unreachable')
  66. const items = response.result.value.items
  67. expect(items.map(item => item.sessionId)).toEqual(['session-a', 'session-b', 'session-c'])
  68. const [a, b, c] = items
  69. expect(a?.updatedAt).toBeCloseTo(5_000_000, -3)
  70. expect(a?.running).toBe(false)
  71. // Cold summaries are never blank: lazy persistence keeps never-appended
  72. // sessions out of list(), so a listed session necessarily has events.
  73. expect(items.every(item => !item.blank)).toBe(true)
  74. expect(a?.cwd).toBe('/proj')
  75. expect(a?.parentSessionId).toBeUndefined()
  76. expect(b?.updatedAt).toBe(2000)
  77. expect(b?.parentSessionId).toBe('session-parent')
  78. expect(b?.origin).toBe('subagent')
  79. expect(c?.updatedAt).toBe(1500)
  80. })
  81. })
  82. describe('attached updatedAt excludes end-seed', () => {
  83. it('reports the last real work, not the pickup, so a resumed-untouched session does not float', async () => {
  84. const ctx = new Context()
  85. await ctx.plugin(SessionStore)
  86. await ctx.plugin(UserInteractionService)
  87. await ctx.plugin(AgentRegistry)
  88. const api = createApiProxy(ctx, { defaultModelSelection: () => ({ provider: 'p', model: 'm' }), cwd: '/tmp' })
  89. // Old work, resumed just now: the log tail would report the pickup.
  90. const worked = 1_000_000
  91. const resumed = ctx.sessions.create(sid('resumed-untouched'), {
  92. seed: [
  93. { type: 'turn/start', seq: 0, time: worked, data: { turn: 1 } },
  94. { type: 'turn/end', seq: 1, time: worked, data: { turn: 1, reason: { kind: 'completed' } } },
  95. ],
  96. meta: { cwd: '/proj', createdAt: 500 },
  97. })
  98. ctx.agents.register({ id: resumed.id, session: resumed, status: 'idle', ctx } as Agent)
  99. const boundary = resumed.events.at(-1)
  100. expect(boundary?.type).toBe('session/end-seed')
  101. expect(boundary?.time).toBeGreaterThan(worked)
  102. const listed = await api.sessions.list(request({}))
  103. if (!listed.result.ok) throw new Error('list failed')
  104. const summary = listed.result.value.items.find(item => item.sessionId === 'resumed-untouched')
  105. expect(summary?.updatedAt).toBe(worked)
  106. // Real work appended after end-seed does move it.
  107. resumed.append('turn/start', { turn: 2 })
  108. const after = await api.sessions.list(request({}))
  109. if (!after.result.ok) throw new Error('list failed')
  110. const moved = after.result.value.items.find(item => item.sessionId === 'resumed-untouched')
  111. expect(moved?.updatedAt).toBeGreaterThan(worked)
  112. })
  113. })
  114. describe('cold history recovery view', () => {
  115. it('shows in-memory interruption repair without activating the session', async () => {
  116. const ctx = new Context()
  117. await ctx.plugin(SessionStore)
  118. await ctx.plugin(UserInteractionService)
  119. const sessionId = sid('session-interrupted')
  120. const meta = header(sessionId, 1000)
  121. const stored: StoredPrefix<never> = {
  122. meta,
  123. events: [{ type: 'turn/start', seq: 0, time: 1, data: { turn: 1 } }],
  124. revision: SessionPersistenceRevision('history-recovery-test:1'),
  125. }
  126. const backend: PersistenceBackend<never> = {
  127. name: 'history-recovery-test',
  128. loadStored: id => Promise.resolve(id === sessionId ? structuredClone(stored) : undefined),
  129. readStoredRevision: id => Promise.resolve(
  130. id === sessionId ? SessionPersistenceRevision('history-recovery-test:1') : undefined,
  131. ),
  132. appendBatch: () => Promise.resolve(),
  133. commitRepair: () => Promise.resolve(),
  134. list: () => Promise.resolve([structuredClone(meta)]),
  135. }
  136. const coordinator = new PersistenceCoordinator(ctx, backend)
  137. ctx.provide('sessionPersistence', {
  138. list: (signal?: AbortSignal) => backend.list(signal),
  139. inspect: (id: SessionId, signal?: AbortSignal) => coordinator.inspect(id, signal),
  140. locate: () => undefined,
  141. } as never)
  142. const api = createApiProxy(ctx, { defaultModelSelection: () => ({ provider: 'p', model: 'm' }), cwd: '/tmp' })
  143. const history = await api.sessions.history(request({ sessionId, beforeSeq: 2, maxMessages: 10 }))
  144. if (!history.result.ok) throw new Error('history failed')
  145. expect(history.result.value.events.map(entry => entry.event)).toMatchInlineSnapshot(`
  146. [
  147. {
  148. "data": {
  149. "turn": 1,
  150. },
  151. "seq": 0,
  152. "time": 1,
  153. "type": "turn/start",
  154. },
  155. {
  156. "data": {
  157. "reason": {
  158. "kind": "interrupted",
  159. },
  160. "turn": 1,
  161. },
  162. "seq": 1,
  163. "time": 1,
  164. "type": "turn/end",
  165. },
  166. ]
  167. `)
  168. expect(ctx.sessions.get(sessionId)).toBeUndefined()
  169. await ctx.fiber.dispose()
  170. })
  171. })
  172. describe('Remote Agent and Session lookup policy', () => {
  173. it('deduplicates a cold resume across Agent and Session parameters', async () => {
  174. const ctx = new Context()
  175. await ctx.plugin(TypertRegistry)
  176. await ctx.plugin(SessionStore)
  177. await ctx.plugin(AgentRegistry)
  178. await ctx.plugin(UserInteractionService)
  179. const sessionId = sid('session-remote-cold')
  180. const meta = header(sessionId, 1000)
  181. const inspect = vi.fn(() => Promise.resolve({ meta, events: [] as SessionEvent[] }))
  182. ctx.provide('sessionPersistence', {
  183. list: () => Promise.resolve([meta]),
  184. inspect,
  185. locate: () => undefined,
  186. } as never)
  187. const resumedSession = { id: sessionId, header: meta, events: [] } as unknown as import('@deepseek-ai/dsh-session').Session
  188. const resumedAgent = { id: sessionId, session: resumedSession, status: 'idle', ctx } as Agent
  189. const release = Promise.withResolvers<undefined>()
  190. const resume = vi.spyOn(ctx.agents, 'resume').mockImplementation(async () => {
  191. await release.promise
  192. return { agent: resumedAgent, dispose: () => Promise.resolve() }
  193. })
  194. const defaultAgentLookup = ctx.typert.lookups.get('agent')
  195. const defaultSessionLookup = ctx.typert.lookups.get('session')
  196. createApiProxy(ctx, { defaultModelSelection: () => ({ provider: 'p', model: 'm' }), cwd: '/tmp' })
  197. await vi.waitFor(() => {
  198. expect(ctx.typert.lookups.get('agent')).not.toBe(defaultAgentLookup)
  199. expect(ctx.typert.lookups.get('session')).not.toBe(defaultSessionLookup)
  200. })
  201. const agentLookup = ctx.typert.lookups.get('agent')
  202. const sessionLookup = ctx.typert.lookups.get('session')
  203. if (agentLookup === undefined || sessionLookup === undefined) throw new Error('core lookup providers were not mounted')
  204. const resolvedAgent = Promise.resolve(agentLookup.resolve(sessionId))
  205. const resolvedSession = Promise.resolve(sessionLookup.resolve(sessionId))
  206. await vi.waitFor(() => { expect(resume).toHaveBeenCalledOnce() })
  207. release.resolve(undefined)
  208. await expect(resolvedAgent).resolves.toBe(resumedAgent)
  209. await expect(resolvedSession).resolves.toBe(resumedSession)
  210. expect(inspect).toHaveBeenCalledOnce()
  211. })
  212. it('preserves the subagent ownership fence for cold and live Remote lookups', async () => {
  213. const ctx = new Context()
  214. await ctx.plugin(TypertRegistry)
  215. await ctx.plugin(SessionStore)
  216. await ctx.plugin(AgentRegistry)
  217. await ctx.plugin(UserInteractionService)
  218. const coldId = sid('session-remote-cold-child')
  219. const coldMeta = header(coldId, 1000, {
  220. parentSession: sid('session-parent'),
  221. origin: 'subagent',
  222. })
  223. const inspect = vi.fn(() => Promise.resolve({ meta: coldMeta, events: [] as SessionEvent[] }))
  224. ctx.provide('sessionPersistence', {
  225. list: () => Promise.resolve([coldMeta]),
  226. inspect,
  227. locate: () => undefined,
  228. } as never)
  229. const liveSession = ctx.sessions.create(sid('session-remote-live-child'), {
  230. meta: { cwd: '/proj', parentSession: sid('session-parent'), origin: 'subagent' },
  231. })
  232. const liveAgent = { id: liveSession.id, session: liveSession, status: 'idle', ctx } as Agent
  233. ctx.agents.register(liveAgent)
  234. const resume = vi.spyOn(ctx.agents, 'resume')
  235. const defaultAgentLookup = ctx.typert.lookups.get('agent')
  236. const defaultSessionLookup = ctx.typert.lookups.get('session')
  237. createApiProxy(ctx, { defaultModelSelection: () => ({ provider: 'p', model: 'm' }), cwd: '/tmp' })
  238. await vi.waitFor(() => {
  239. expect(ctx.typert.lookups.get('agent')).not.toBe(defaultAgentLookup)
  240. expect(ctx.typert.lookups.get('session')).not.toBe(defaultSessionLookup)
  241. })
  242. const agentLookup = ctx.typert.lookups.get('agent')
  243. const sessionLookup = ctx.typert.lookups.get('session')
  244. if (agentLookup === undefined || sessionLookup === undefined) throw new Error('core lookup providers were not mounted')
  245. const ownershipFailure = {
  246. failure: {
  247. code: 'agent-busy',
  248. details: { reason: 'use subagent delivery for this child session' },
  249. },
  250. }
  251. const coldFailure = Promise.resolve(agentLookup.resolve(coldId))
  252. const liveFailure = Promise.resolve(sessionLookup.resolve(liveSession.id))
  253. await expect(coldFailure).rejects.toBeInstanceOf(TypeRTLookupFailure)
  254. await expect(coldFailure).rejects.toMatchObject(ownershipFailure)
  255. await expect(liveFailure).rejects.toBeInstanceOf(TypeRTLookupFailure)
  256. await expect(liveFailure).rejects.toMatchObject(ownershipFailure)
  257. expect(resume).not.toHaveBeenCalled()
  258. expect(inspect).toHaveBeenCalledOnce()
  259. })
  260. })
  261. describe('subagent ownership fence', () => {
  262. it('reads a cold child without an Agent and rejects generic resume or adoption', async () => {
  263. const ctx = new Context()
  264. await ctx.plugin(SessionStore)
  265. await ctx.plugin(AgentRegistry)
  266. await ctx.plugin(UserInteractionService)
  267. const sessionId = sid('session-child')
  268. const meta = header('session-child', 1000, {
  269. parentSession: sid('session-parent'),
  270. seedLength: 0,
  271. origin: 'subagent',
  272. })
  273. const events = [
  274. { type: 'turn/start', seq: 0, time: 1, data: { turn: 1, trigger: { kind: 'message', source: { kind: 'user' } } } },
  275. {
  276. type: 'user/message',
  277. seq: 1,
  278. time: 2,
  279. data: { content: [{ type: 'text', text: 'work' }], source: { kind: 'user' } },
  280. surfaceOp: 'append',
  281. },
  282. {
  283. type: 'subagent/descriptor',
  284. seq: 2,
  285. time: 3,
  286. data: { version: 2, mode: 'continuable', provider: 'spawn', label: 'child' },
  287. },
  288. { type: 'turn/end', seq: 3, time: 4, data: { turn: 1, reason: { kind: 'completed' } } },
  289. ] as SessionEvent[]
  290. const inspect = vi.fn(() => Promise.resolve({ meta, events }))
  291. ctx.provide('sessionPersistence', {
  292. list: () => Promise.resolve([meta]),
  293. inspect,
  294. locate: () => undefined,
  295. } as never)
  296. const resume = vi.spyOn(ctx.agents, 'resume')
  297. const api = createApiProxy(ctx, { defaultModelSelection: () => ({ provider: 'p', model: 'm' }), cwd: '/tmp' })
  298. const history = await api.sessions.history(request({ sessionId }))
  299. expect(history.result.ok).toBe(true)
  300. if (history.result.ok) {
  301. expect(history.result.value.events.map(entry => entry.event.type)).toEqual(events.map(event => event.type))
  302. }
  303. expect(ctx.agents.get(sessionId)).toBeUndefined()
  304. const prompt = await api.sessions.prompt(request({
  305. sessionId,
  306. mode: 'queue',
  307. content: [{ type: 'text', text: 'follow up' }],
  308. }))
  309. expect(prompt.result.ok).toBe(false)
  310. if (!prompt.result.ok) {
  311. expect(prompt.result.error).toMatchObject({
  312. code: 'agent-busy',
  313. details: { reason: 'use subagent delivery for this child session' },
  314. })
  315. }
  316. const create = await api.sessions.create(request({ sessionId, cwd: '/proj' }))
  317. expect(create.result.ok).toBe(false)
  318. if (!create.result.ok) expect(create.result.error.code).toBe('agent-busy')
  319. expect(resume).not.toHaveBeenCalled()
  320. expect(ctx.agents.get(sessionId)).toBeUndefined()
  321. expect(inspect).toHaveBeenCalledTimes(3)
  322. })
  323. it('no longer treats a descriptor-only cold child without origin as subagent-owned', async () => {
  324. const ctx = new Context()
  325. await ctx.plugin(SessionStore)
  326. await ctx.plugin(AgentRegistry)
  327. await ctx.plugin(UserInteractionService)
  328. const sessionId = sid('session-legacy-child')
  329. const meta = header('session-legacy-child', 1000, {
  330. parentSession: sid('session-parent'),
  331. seedLength: 0,
  332. })
  333. const events = [
  334. {
  335. type: 'subagent/descriptor',
  336. seq: 0,
  337. time: 1,
  338. data: { version: 2, mode: 'continuable', provider: 'spawn', label: 'child' },
  339. },
  340. ] as SessionEvent[]
  341. ctx.provide('sessionPersistence', {
  342. list: () => Promise.resolve([meta]),
  343. inspect: () => Promise.resolve({ meta, events }),
  344. locate: () => undefined,
  345. } as never)
  346. // Stores whose headers predate `origin` classify a child only through the
  347. // descriptor event; the pre-release decision stops recognizing them, so
  348. // the ownership fence lets generic resume reach the registry instead of
  349. // answering `agent-busy`.
  350. const resume = vi.spyOn(ctx.agents, 'resume')
  351. .mockRejectedValue(new Error('registry unavailable in this bench'))
  352. const api = createApiProxy(ctx, { defaultModelSelection: () => ({ provider: 'p', model: 'm' }), cwd: '/tmp' })
  353. const prompt = await api.sessions.prompt(request({
  354. sessionId,
  355. mode: 'queue',
  356. content: [{ type: 'text', text: 'follow up' }],
  357. }))
  358. expect(resume).toHaveBeenCalledTimes(1)
  359. expect(prompt.result.ok).toBe(false)
  360. if (!prompt.result.ok) expect(prompt.result.error.code).toBe('internal')
  361. })
  362. it('rejects origin-marked and runtime-owned live children from generic controls', async () => {
  363. const ctx = new Context()
  364. await ctx.plugin(SessionStore)
  365. await ctx.plugin(AgentRegistry)
  366. await ctx.plugin(UserInteractionService)
  367. const parentSession = ctx.sessions.create(sid('session-parent'), { meta: { cwd: '/proj' } })
  368. const parent = { id: parentSession.id, session: parentSession, status: 'idle', ctx } as Agent
  369. ctx.agents.register(parent)
  370. const originSession = ctx.sessions.create(sid('session-origin-child'), {
  371. meta: { cwd: '/proj', parentSession: parent.id, origin: 'subagent' },
  372. })
  373. const cancel = vi.fn()
  374. const updateInbox = vi.fn(() => 'applied' as const)
  375. const originChild = {
  376. id: originSession.id,
  377. session: originSession,
  378. status: 'idle',
  379. ctx,
  380. cancel,
  381. updateInbox,
  382. } as unknown as Agent
  383. ctx.agents.register(originChild)
  384. const startingSession = ctx.sessions.create(sid('session-starting-child'), {
  385. meta: { cwd: '/proj', parentSession: parent.id },
  386. })
  387. const startingChild = { id: startingSession.id, session: startingSession, status: 'idle', ctx } as Agent
  388. ctx.agents.enter(startingChild, parent)
  389. const api = createApiProxy(ctx, { defaultModelSelection: () => ({ provider: 'p', model: 'm' }), cwd: '/tmp' })
  390. const stopped = await api.sessions.cancel(request({ sessionId: originChild.id }))
  391. expect(stopped.result.ok).toBe(false)
  392. if (!stopped.result.ok) expect(stopped.result.error.code).toBe('agent-busy')
  393. expect(cancel).not.toHaveBeenCalled()
  394. const queued = await api.sessions.updateQueue(request({
  395. sessionId: originChild.id,
  396. itemId: MessageId('queued-item'),
  397. action: { kind: 'remove' },
  398. }))
  399. expect(queued.result.ok).toBe(false)
  400. if (!queued.result.ok) expect(queued.result.error.code).toBe('agent-busy')
  401. expect(updateInbox).not.toHaveBeenCalled()
  402. const models = await api.sessions.models(request({ sessionId: startingChild.id }))
  403. expect(models.result.ok).toBe(false)
  404. if (!models.result.ok) expect(models.result.error.code).toBe('agent-busy')
  405. const create = await api.sessions.create(request({ sessionId: originChild.id, cwd: '/proj' }))
  406. expect(create.result.ok).toBe(false)
  407. if (!create.result.ok) expect(create.result.error.code).toBe('agent-busy')
  408. const history = await api.sessions.history(request({ sessionId: originChild.id }))
  409. expect(history.result.ok).toBe(true)
  410. expect(ctx.agents.get(originChild.id)).toBe(originChild)
  411. })
  412. it('does not classify an ordinary fork from an inherited ancestor descriptor', async () => {
  413. const ctx = new Context()
  414. await ctx.plugin(SessionStore)
  415. await ctx.plugin(AgentRegistry)
  416. await ctx.plugin(UserInteractionService)
  417. const session = ctx.sessions.create(sid('session-ordinary-fork'), {
  418. seed: [{
  419. type: 'subagent/descriptor',
  420. seq: 0,
  421. time: 1,
  422. data: { version: 2, mode: 'continuable', provider: 'spawn', label: 'ancestor' },
  423. }],
  424. meta: { cwd: '/proj', parentSession: sid('session-source'), seedLength: 1 },
  425. })
  426. const followup = vi.fn()
  427. const agent = { id: session.id, session, status: 'idle', ctx, followup } as unknown as Agent
  428. ctx.agents.register(agent)
  429. const api = createApiProxy(ctx, { defaultModelSelection: () => ({ provider: 'p', model: 'm' }), cwd: '/tmp' })
  430. const response = await api.sessions.prompt(request({
  431. sessionId: agent.id,
  432. mode: 'queue',
  433. content: [{ type: 'text', text: 'ordinary work' }],
  434. }))
  435. expect(response.result.ok).toBe(true)
  436. expect(followup).toHaveBeenCalledOnce()
  437. })
  438. it('canonicalizes a supplied browser zone on the exact prompt and rejects invalid names', async () => {
  439. const ctx = new Context()
  440. await ctx.plugin(SessionStore)
  441. await ctx.plugin(AgentRegistry)
  442. await ctx.plugin(UserInteractionService)
  443. const session = ctx.sessions.create(sid('session-browser-zone'), { meta: { cwd: '/proj' } })
  444. const followup = vi.fn()
  445. const agent = { id: session.id, session, status: 'idle', ctx, followup } as unknown as Agent
  446. ctx.agents.register(agent)
  447. const api = createApiProxy(ctx, {
  448. defaultModelSelection: () => ({ provider: 'p', model: 'm' }),
  449. cwd: '/tmp',
  450. })
  451. const alias = 'US/Pacific'
  452. const canonical = new Intl.DateTimeFormat('en-US', { timeZone: alias })
  453. .resolvedOptions().timeZone
  454. const zonedRequest = request({
  455. sessionId: agent.id,
  456. mode: 'queue' as const,
  457. content: [{ type: 'text' as const, text: 'zoned work' }],
  458. clientTimeZone: alias,
  459. })
  460. await expect(api.sessions.prompt(zonedRequest)).resolves.toMatchObject({
  461. result: { ok: true },
  462. })
  463. expect(followup).toHaveBeenNthCalledWith(1, expect.objectContaining({
  464. source: { kind: 'user', rpcId: zonedRequest.rpcId, clientTimeZone: canonical },
  465. }))
  466. const utcRequest = request({
  467. sessionId: agent.id,
  468. mode: 'queue' as const,
  469. content: [{ type: 'text' as const, text: 'UTC work' }],
  470. clientTimeZone: 'UTC',
  471. })
  472. await expect(api.sessions.prompt(utcRequest)).resolves.toMatchObject({
  473. result: { ok: true },
  474. })
  475. expect(followup).toHaveBeenNthCalledWith(2, expect.objectContaining({
  476. source: { kind: 'user', rpcId: utcRequest.rpcId, clientTimeZone: 'UTC' },
  477. }))
  478. const unzonedRequest = request({
  479. sessionId: agent.id,
  480. mode: 'queue' as const,
  481. content: [{ type: 'text' as const, text: 'headless work' }],
  482. })
  483. await expect(api.sessions.prompt(unzonedRequest)).resolves.toMatchObject({
  484. result: { ok: true },
  485. })
  486. expect(followup).toHaveBeenNthCalledWith(3, expect.objectContaining({
  487. source: { kind: 'user', rpcId: unzonedRequest.rpcId },
  488. }))
  489. for (const clientTimeZone of ['', ' UTC', 'CST', 'Not/A_Real_Zone']) {
  490. const invalid = await api.sessions.prompt(request({
  491. sessionId: agent.id,
  492. mode: 'queue' as const,
  493. content: [{ type: 'text' as const, text: 'invalid zone' }],
  494. clientTimeZone,
  495. }))
  496. expect(invalid.result).toEqual({
  497. ok: false,
  498. error: {
  499. code: 'invalid-time-zone',
  500. message: 'clientTimeZone must be UTC or a valid IANA Area/Location name',
  501. details: { value: clientTimeZone },
  502. },
  503. })
  504. }
  505. expect(followup).toHaveBeenCalledTimes(3)
  506. })
  507. })
  508. describe('degenerate composition (no persistence, no factory)', () => {
  509. it('list skips the cold merge and history reports missing persistence as internal', async () => {
  510. const ctx = new Context()
  511. await ctx.plugin(SessionStore)
  512. await ctx.plugin(AgentRegistry)
  513. await ctx.plugin(UserInteractionService)
  514. const api = createApiProxy(ctx, { defaultModelSelection: () => ({ provider: 'p', model: 'm' }), cwd: '/tmp' })
  515. const listed = await api.sessions.list(request({}))
  516. expect(listed.result.ok).toBe(true)
  517. if (listed.result.ok) expect(listed.result.value.items).toEqual([])
  518. // No persistence means cold history cannot inspect a transcript.
  519. const response = await api.sessions.history(request({ sessionId: sid('session-ghost') }))
  520. expect(response.result.ok).toBe(false)
  521. if (!response.result.ok) {
  522. expect(response.result.error.code).toBe('internal')
  523. expect(response.result.error.message).toMatch(/history unavailable for session "session-ghost"/)
  524. }
  525. })
  526. it('maps a persistence catalog miss to session-not-found without inspection', async () => {
  527. const ctx = new Context()
  528. await ctx.plugin(SessionStore)
  529. await ctx.plugin(AgentRegistry)
  530. await ctx.plugin(UserInteractionService)
  531. const inspect = vi.fn()
  532. ctx.provide('sessionPersistence', {
  533. list: () => Promise.resolve([]),
  534. inspect,
  535. } as never)
  536. const api = createApiProxy(ctx, { defaultModelSelection: () => ({ provider: 'p', model: 'm' }), cwd: '/tmp' })
  537. const response = await api.sessions.history(request({ sessionId: sid('session-missing') }))
  538. expect(response.result.ok).toBe(false)
  539. if (!response.result.ok) expect(response.result.error.code).toBe('session-not-found')
  540. expect(inspect).not.toHaveBeenCalled()
  541. })
  542. })
  543. describe('sessions.prompt synchronous rejection', () => {
  544. it('maps a synchronous send throw (disposed/invalid input) to agent-busy with the reason attached', async () => {
  545. const ctx = new Context()
  546. await ctx.plugin(SessionStore)
  547. await ctx.plugin(AgentRegistry)
  548. await ctx.plugin(UserInteractionService)
  549. const session = ctx.sessions.create(sid('session-throwing'))
  550. // A live structural stub whose delivery verbs throw synchronously, the
  551. // shape a disposed loop presents at this gateway boundary.
  552. ctx.agents.register({
  553. id: session.id,
  554. session,
  555. status: 'idle',
  556. ctx,
  557. followup: () => { throw new Error('agent "session-throwing" lifecycle disposed') },
  558. steer: () => { throw new Error('agent "session-throwing" lifecycle disposed') },
  559. } as unknown as Agent)
  560. const api = createApiProxy(ctx, { defaultModelSelection: () => ({ provider: 'p', model: 'm' }), cwd: '/tmp' })
  561. for (const mode of ['queue', 'steer'] as const) {
  562. const response = await api.sessions.prompt(request({
  563. sessionId: session.id, mode, content: [{ type: 'text' as const, text: 'x' }],
  564. }))
  565. expect(response.result.ok).toBe(false)
  566. if (!response.result.ok) {
  567. expect(response.result.error.code).toBe('agent-busy')
  568. expect(response.result.error.message).toBe('prompt rejected')
  569. expect(response.result.error.details).toEqual({
  570. reason: 'Error: agent "session-throwing" lifecycle disposed',
  571. })
  572. }
  573. }
  574. })
  575. it('classifies a raced cold-resume ID collision as agent-busy', async () => {
  576. const ctx = new Context()
  577. await ctx.plugin(SessionStore)
  578. await ctx.plugin(AgentRegistry)
  579. await ctx.plugin(UserInteractionService)
  580. const sessionId = sid('race-resume')
  581. const meta: SessionHeader = header('race-resume', 1000)
  582. ctx.provide('sessionPersistence', {
  583. list: () => Promise.resolve([meta]),
  584. inspect: () => Promise.resolve({ meta, events: [] as SessionEvent[] }),
  585. locate: () => undefined,
  586. } as never)
  587. // The raced winner: a live parent-owned subagent publishes the identity
  588. // while the generic cold resume is in flight, so the resume collides.
  589. const parentSession = ctx.sessions.create(sid('race-parent'), { meta: { cwd: '/proj' } })
  590. const parent = { id: parentSession.id, session: parentSession, status: 'idle', ctx } as Agent
  591. ctx.agents.register(parent)
  592. const childSession = ctx.sessions.create(sessionId, {
  593. meta: { cwd: '/proj', parentSession: parent.id, origin: 'subagent' },
  594. })
  595. const child = { id: sessionId, session: childSession, status: 'idle', ctx } as unknown as Agent
  596. vi.spyOn(ctx.agents, 'resume').mockImplementationOnce(async () => {
  597. // The parent's `enter()` wins the identity between the pre-resume
  598. // re-check and publication; the generic resume then collides.
  599. ctx.agents.register(child)
  600. throw new Error('session id already published')
  601. })
  602. const api = createApiProxy(ctx, { defaultModelSelection: () => ({ provider: 'p', model: 'm' }), cwd: '/tmp' })
  603. const models = await api.sessions.models(request({ sessionId }))
  604. expect(models.result.ok).toBe(false)
  605. if (!models.result.ok) {
  606. expect(models.result.error).toMatchObject({
  607. code: 'agent-busy',
  608. details: { reason: 'use subagent delivery for this child session' },
  609. })
  610. }
  611. })
  612. })