session-models.host.spec.ts 27 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688
  1. /**
  2. * Session Controller model-directory and selection behavior: dynamic provider grouping,
  3. * provider-local catalog failures, logged-selection restoration without stale
  4. * catalog injection, advisory pass-through models, and the prompt-assembly
  5. * boundary for a running selection change.
  6. */
  7. import { describe, expect, it, vi } from 'vitest'
  8. import { Context } from '@deepseek-ai/cordis'
  9. import AgentRegistry, { agentEvents } from '@deepseek-ai/dsh-agent'
  10. import type { Agent } from '@deepseek-ai/dsh-agent'
  11. import AttachmentStore from '@deepseek-ai/dsh-attachment'
  12. import LlmRuntime, { LlmAdapter, ReasoningEffortId } from '@deepseek-ai/dsh-llm'
  13. import type {
  14. GenerateOptions, LlmCallConfig, LlmCallConfigAdapterDefaults, LlmModelInfo,
  15. LlmModelReasoningInfo, LlmProviderInfo, LlmResolvedModelInfo, StreamChunk,
  16. UserMessage,
  17. } from '@deepseek-ai/dsh-llm'
  18. import SessionStore from '@deepseek-ai/dsh-session'
  19. import type { SessionId } from '@deepseek-ai/dsh-session'
  20. import type { SessionPromptRequest, SessionRequestId } from '../src/types.ts'
  21. import { ApiSessionAgentController } from '../src/agent.ts'
  22. import { buildModelCatalog } from '../src/catalog.ts'
  23. import SystemPrompt from '@deepseek-ai/dsh-system-prompt'
  24. import { RemoteError } from '@deepseek-ai/dsh-typert-protocol'
  25. import { createSessionTestRemote } from './test-remote.ts'
  26. function request<P>(payload: P): P {
  27. return payload
  28. }
  29. let nextRequestId = 1
  30. function promptRequest(
  31. payload: Omit<SessionPromptRequest, 'requestId'>,
  32. ): SessionPromptRequest {
  33. return {
  34. ...payload,
  35. requestId: `models-${String(nextRequestId++)}` as SessionRequestId,
  36. }
  37. }
  38. class CatalogAdapter extends LlmAdapter {
  39. constructor(
  40. private readonly name: string,
  41. private readonly models: readonly LlmModelInfo[] | Error,
  42. private readonly reasoning?: LlmModelReasoningInfo,
  43. private readonly exactError?: Error,
  44. ) {
  45. super()
  46. }
  47. override providerInfo(provider: string): LlmProviderInfo {
  48. return { id: provider, name: this.name }
  49. }
  50. override listModels(): Promise<readonly LlmModelInfo[]> {
  51. return this.models instanceof Error
  52. ? Promise.reject(this.models)
  53. : Promise.resolve(this.models)
  54. }
  55. override resolveModel(provider: string, model: string): Promise<LlmResolvedModelInfo> {
  56. if (this.exactError !== undefined) return Promise.reject(this.exactError)
  57. return Promise.resolve({
  58. provider,
  59. id: model,
  60. name: model,
  61. ...this.reasoning === undefined ? {} : { reasoning: this.reasoning },
  62. })
  63. }
  64. override async *stream(_options: GenerateOptions): AsyncIterable<StreamChunk> {
  65. // Catalog tests never enter provider streaming.
  66. }
  67. }
  68. const REASONING: LlmModelReasoningInfo = {
  69. efforts: [
  70. { id: ReasoningEffortId('off'), name: 'Off' },
  71. { id: ReasoningEffortId('high'), name: 'High' },
  72. { id: ReasoningEffortId('max'), name: 'Max' },
  73. ],
  74. defaultEffort: ReasoningEffortId('high'),
  75. }
  76. async function harness(logged?: {
  77. provider: string
  78. model: string
  79. reasoningEffort?: ReasoningEffortId
  80. adapterDefaults?: LlmCallConfigAdapterDefaults
  81. }): Promise<{
  82. ctx: Context
  83. agent: Agent
  84. sessionId: SessionId
  85. }> {
  86. const ctx = new Context()
  87. await ctx.plugin(SessionStore)
  88. await ctx.plugin(SystemPrompt, { persona: '' })
  89. await ctx.plugin(LlmRuntime)
  90. await ctx.plugin(AgentRegistry)
  91. ctx.llm.registerAdapter(['deepseek-official'], new CatalogAdapter('DeepSeek', [
  92. { provider: 'deepseek-official', id: 'deepseek-chat', name: 'DeepSeek Chat' },
  93. { provider: 'deepseek-official', id: 'deepseek-reasoner', name: 'DeepSeek Reasoner', description: 'Reasoning model' },
  94. ], REASONING))
  95. ctx.llm.registerAdapter(['broken'], new CatalogAdapter('Broken Provider', new Error('catalog offline')))
  96. ctx.llm.registerAdapter(['metadata-broken'], new CatalogAdapter('Metadata Broken', [
  97. { provider: 'metadata-broken', id: 'listed', name: 'Listed' },
  98. ], undefined, new Error('reasoning metadata offline')))
  99. ctx.llm.registerAdapter(['remote-rejected'], new CatalogAdapter(
  100. 'Remote Rejected',
  101. [],
  102. undefined,
  103. new RemoteError('gateway/internal', 'fixture rejected the selection', {}),
  104. ))
  105. ctx.llm.registerAdapter(['empty'], new CatalogAdapter('Empty Provider', []))
  106. ctx.llm.registerAdapter(['duplicate'], new CatalogAdapter('Duplicate Provider', [
  107. { provider: 'duplicate', id: 'same', name: 'Same' },
  108. { provider: 'duplicate', id: 'same', name: 'Same Again' },
  109. ]))
  110. const session = ctx.sessions.create()
  111. if (logged !== undefined) {
  112. const { adapterDefaults, ...config } = logged
  113. session.append('request/header', {
  114. header: { config, ...adapterDefaults === undefined ? {} : { adapterDefaults } },
  115. reason: 'initial',
  116. })
  117. }
  118. const agent = {
  119. id: session.id,
  120. session,
  121. status: 'running',
  122. ctx,
  123. inbox: { nextTurn: [], nextStep: [] },
  124. } as unknown as Agent
  125. ctx.agents.register(agent)
  126. return { ctx, agent, sessionId: session.id }
  127. }
  128. function expectValue<T>(result: { ok: true; value: T } | { ok: false }): T {
  129. if (!result.ok) throw new Error('expected successful response')
  130. return result.value
  131. }
  132. function registerTextOnly(ctx: Context): void {
  133. ctx.llm.registerAdapter(['text-only'], new class extends CatalogAdapter {
  134. override resolveModel(provider: string, model: string): Promise<LlmResolvedModelInfo> {
  135. return Promise.resolve({ provider, id: model, name: model, inputModalities: ['text'] })
  136. }
  137. }('Text Only', []))
  138. }
  139. /** Resolve the Client-visible next selection from durable state and the Host default. */
  140. function currentSelection(ctx: Context, sessionId: SessionId) {
  141. const session = ctx.sessions.get(sessionId)
  142. if (session === undefined) throw new Error('expected a live test Session')
  143. return ctx.sessionProjections.snapshot(session).values.modelSelection?.next
  144. ?? ctx.agentDefaultModel.currentSelection()
  145. }
  146. describe('Web session model selection', () => {
  147. it('validates an ordered image batch before persisting any member', async () => {
  148. const { ctx, agent, sessionId } = await harness()
  149. const validateImage = vi.fn((_input: { data: Uint8Array }) => Promise.resolve())
  150. const saveImage = vi.fn((input: { data: Uint8Array; mediaType: 'image/png'; name?: string }) => Promise.resolve({
  151. attachmentId: `att-${String(input.data[0])}`,
  152. mediaType: input.mediaType,
  153. bytes: input.data.byteLength,
  154. width: 1,
  155. height: 1,
  156. ...input.name === undefined ? {} : { name: input.name },
  157. }))
  158. const attachments = {
  159. imageLimits: {
  160. maxImageBytes: 4,
  161. maxImagesPerMessage: 2,
  162. maxMessageImageBytes: 4,
  163. maxImagePixels: 4,
  164. maxImageDimension: 2000,
  165. mediaTypes: ['image/png'],
  166. },
  167. validateImage,
  168. saveImage,
  169. }
  170. ctx.provide('attachments', Object.setPrototypeOf(attachments, AttachmentStore.prototype) as never)
  171. const followup = vi.fn()
  172. Object.assign(agent, { followup })
  173. const remote = createSessionTestRemote(ctx, {
  174. defaultModelSelection: () => ({ provider: 'deepseek-official', model: 'deepseek-chat' }),
  175. cwd: '/tmp',
  176. })
  177. const result = await remote.prompt(promptRequest({
  178. sessionId,
  179. mode: 'queue' as const,
  180. content: [
  181. { type: 'image' as const, mediaType: 'image/png' as const, data: 'AQ==', name: 'first.png' },
  182. { type: 'text' as const, text: 'compare' },
  183. { type: 'image' as const, mediaType: 'image/png' as const, data: 'Ag==' },
  184. ],
  185. }))
  186. expect(result.ok).toBe(true)
  187. expect(validateImage.mock.calls.map(([input]) => [...input.data])).toEqual([[1], [2]])
  188. expect(saveImage.mock.calls.map(([input]) => [...input.data])).toEqual([[1], [2]])
  189. expect((followup.mock.calls[0]?.[0] as UserMessage).content).toEqual([
  190. {
  191. type: 'image',
  192. attachment: {
  193. attachmentId: 'att-1', mediaType: 'image/png', bytes: 1, width: 1, height: 1, name: 'first.png',
  194. },
  195. },
  196. { type: 'text', text: 'compare' },
  197. { type: 'image', attachment: { attachmentId: 'att-2', mediaType: 'image/png', bytes: 1, width: 1, height: 1 } },
  198. ])
  199. const denied = await remote.prompt(promptRequest({
  200. sessionId,
  201. mode: 'queue' as const,
  202. content: Array.from({ length: 3 }, () => ({
  203. type: 'image' as const, mediaType: 'image/png' as const, data: 'AQ==',
  204. })),
  205. }))
  206. expect(denied).toMatchObject({
  207. ok: false,
  208. error: { code: 'session/attachment-invalid', details: { reason: 'TOO_MANY_IMAGES' } },
  209. })
  210. expect(saveImage).toHaveBeenCalledTimes(2)
  211. await ctx.fiber.dispose()
  212. })
  213. it('allows a text-only selection while durable or pending images remain available for later models', async () => {
  214. const { ctx, agent, sessionId } = await harness()
  215. registerTextOnly(ctx)
  216. const remote = createSessionTestRemote(ctx, {
  217. defaultModelSelection: () => ({ provider: 'deepseek-official', model: 'deepseek-chat' }),
  218. cwd: '/tmp',
  219. })
  220. const image = {
  221. type: 'image' as const,
  222. attachment: { attachmentId: 'att-history', mediaType: 'image/png' as const, bytes: 1, width: 1, height: 1 },
  223. }
  224. const imageEvent = agent.session.append('user/message', {
  225. id: 'image-message', role: 'user', source: { kind: 'user' }, content: [image],
  226. } as never, { surfaceOp: 'append' })
  227. expect(expectValue(await remote.selectModel(request({
  228. sessionId, provider: 'text-only', model: 'plain',
  229. }))).selected).toEqual({ provider: 'text-only', model: 'plain' })
  230. agent.session.append('user/message', {
  231. id: 'summary', role: 'user', source: { kind: 'plugin', plugin: 'compact' },
  232. content: [{ type: 'text', text: 'image summarized' }],
  233. } as never, {
  234. surfaceOp: { op: 'replace', start: imageEvent.seq, end: imageEvent.seq },
  235. sourceEventSeqs: [imageEvent.seq],
  236. })
  237. ;(agent.inbox.nextTurn as UserMessage[]).push({
  238. id: 'pending-image', role: 'user', source: { kind: 'user' }, content: [image],
  239. } as never)
  240. expect(expectValue(await remote.selectModel(request({
  241. sessionId, provider: 'text-only', model: 'plain',
  242. }))).selected).toEqual({ provider: 'text-only', model: 'plain' })
  243. await ctx.fiber.dispose()
  244. })
  245. it('authorizes attachment bytes only when the session event stream references the id', async () => {
  246. const { ctx, agent, sessionId } = await harness()
  247. const ref = {
  248. attachmentId: 'att-authorized', mediaType: 'image/png' as const, bytes: 2, width: 1, height: 1,
  249. }
  250. const readImage = vi.fn(() => Promise.resolve({ ref, data: Uint8Array.of(1, 2) }))
  251. ctx.provide('attachments', { readImage } as never)
  252. const remote = createSessionTestRemote(ctx, {
  253. defaultModelSelection: () => ({ provider: 'deepseek-official', model: 'deepseek-chat' }),
  254. cwd: '/tmp',
  255. })
  256. agent.session.append('agent/inbox/spliced', {
  257. target: 'next-turn',
  258. start: 0,
  259. inserted: [{
  260. id: 'queued-image', role: 'user', source: { kind: 'user' },
  261. content: [{ type: 'image', attachment: ref }],
  262. }],
  263. } as never)
  264. const allowed = await remote.attachment(request({
  265. sessionId, attachmentId: 'att-authorized' as never,
  266. }))
  267. expect(allowed).toMatchObject({ ok: true, value: { attachment: ref, data: 'AQI=' } })
  268. const denied = await remote.attachment(request({
  269. sessionId, attachmentId: 'att-other' as never,
  270. }))
  271. expect(denied).toMatchObject({
  272. ok: false,
  273. error: { code: 'session/attachment-invalid', details: { reason: 'ATTACHMENT_NOT_REFERENCED' } },
  274. })
  275. expect(readImage).toHaveBeenCalledOnce()
  276. await ctx.fiber.dispose()
  277. })
  278. it('groups successful providers and leaves an unlisted current selection out of the catalog', async () => {
  279. const { ctx, sessionId } = await harness({
  280. provider: 'deepseek-official',
  281. model: 'private-preview',
  282. reasoningEffort: ReasoningEffortId('max'),
  283. })
  284. const remote = createSessionTestRemote(ctx, { defaultModelSelection: () => ({ provider: 'deepseek-official', model: 'deepseek-chat' }), cwd: '/tmp' })
  285. const catalog = expectValue(await remote.modelCatalog())
  286. expect(currentSelection(ctx, sessionId)).toEqual({
  287. provider: 'deepseek-official',
  288. model: 'private-preview',
  289. reasoningEffort: 'max',
  290. })
  291. expect(catalog.groups).toEqual([{
  292. id: 'deepseek-official',
  293. name: 'DeepSeek',
  294. models: [
  295. { id: 'deepseek-chat', name: 'DeepSeek Chat', reasoning: REASONING },
  296. {
  297. id: 'deepseek-reasoner',
  298. name: 'DeepSeek Reasoner',
  299. description: 'Reasoning model',
  300. reasoning: REASONING,
  301. },
  302. ],
  303. }])
  304. expect(catalog.failures).toEqual([
  305. { id: 'broken', name: 'Broken Provider', message: 'catalog offline' },
  306. { id: 'metadata-broken', name: 'Metadata Broken', message: 'reasoning metadata offline' },
  307. {
  308. id: 'duplicate',
  309. name: 'Duplicate Provider',
  310. message: 'adapter returned invalid or duplicate model metadata for provider "duplicate"',
  311. },
  312. ])
  313. await ctx.fiber.dispose()
  314. })
  315. it('preserves optional catalog metadata and string provider failures', async () => {
  316. const { ctx } = await harness()
  317. ctx.llm.registerAdapter(['plain'], new CatalogAdapter('Plain', [
  318. { provider: 'plain', id: 'plain-model', name: 'Plain Model' },
  319. ]))
  320. ctx.llm.registerAdapter(['described-reasoning'], new CatalogAdapter('Described Reasoning', [
  321. { provider: 'described-reasoning', id: 'reasoning-model', name: 'Reasoning Model' },
  322. ], {
  323. efforts: [{ id: ReasoningEffortId('high'), name: 'High', description: 'More thinking' }],
  324. }))
  325. ctx.llm.registerAdapter(['string-failure'], new class extends CatalogAdapter {
  326. override listModels(): Promise<readonly LlmModelInfo[]> {
  327. // oxlint-disable-next-line typescript/prefer-promise-reject-errors -- non-Error provider normalization is the scenario.
  328. return Promise.reject('string catalog failure')
  329. }
  330. }('String Failure', []))
  331. createSessionTestRemote(ctx, {
  332. defaultModelSelection: () => ({ provider: 'deepseek-official', model: 'deepseek-chat' }),
  333. cwd: '/tmp',
  334. })
  335. const catalog = await buildModelCatalog(ctx)
  336. expect(catalog.groups).toEqual(expect.arrayContaining([
  337. { id: 'plain', name: 'Plain', models: [{ id: 'plain-model', name: 'Plain Model' }] },
  338. {
  339. id: 'described-reasoning',
  340. name: 'Described Reasoning',
  341. models: [{
  342. id: 'reasoning-model',
  343. name: 'Reasoning Model',
  344. reasoning: {
  345. efforts: [{ id: 'high', name: 'High', description: 'More thinking' }],
  346. },
  347. }],
  348. },
  349. ]))
  350. expect(catalog.failures).toContainEqual({
  351. id: 'string-failure', name: 'String Failure', message: 'string catalog failure',
  352. })
  353. await ctx.fiber.dispose()
  354. })
  355. it('accepts an advisory-unlisted model, rejects an unavailable provider, and switches only after the next assembly', async () => {
  356. const { ctx, agent, sessionId } = await harness()
  357. const remote = createSessionTestRemote(ctx, { defaultModelSelection: () => ({ provider: 'deepseek-official', model: 'deepseek-chat' }), cwd: '/tmp' })
  358. const seed: LlmCallConfig = { provider: 'seed', model: 'seed', temperature: 0.2 }
  359. const signal = new AbortController().signal
  360. expect(currentSelection(ctx, sessionId))
  361. .toEqual({ provider: 'deepseek-official', model: 'deepseek-chat' })
  362. const selected = expectValue(await remote.selectModel(request({
  363. sessionId,
  364. provider: 'deepseek-official',
  365. model: 'private-preview',
  366. reasoningEffort: 'max',
  367. })))
  368. expect(selected.selected).toEqual({
  369. provider: 'deepseek-official',
  370. model: 'private-preview',
  371. reasoningEffort: 'max',
  372. })
  373. await expect(agentEvents(ctx, agent).waterfall(
  374. 'agent/request', { turn: 1, step: 0, signal }, () => Promise.resolve(seed),
  375. )).resolves.toEqual(seed)
  376. expect((await ctx.systemPrompt.assemble()).variables)
  377. .toMatchObject({ provider: 'deepseek-official', model: 'private-preview' })
  378. await expect(agentEvents(ctx, agent).waterfall(
  379. 'agent/request', { turn: 1, step: 1, signal }, () => Promise.resolve(seed),
  380. )).resolves.toMatchObject({
  381. provider: 'deepseek-official',
  382. model: 'private-preview',
  383. reasoningEffort: 'max',
  384. })
  385. const unsupported = await remote.selectModel(request({
  386. sessionId,
  387. provider: 'deepseek-official',
  388. model: 'private-preview',
  389. reasoningEffort: 'medium',
  390. }))
  391. expect(unsupported).toMatchObject({
  392. ok: false,
  393. error: {
  394. code: 'session/model-unavailable',
  395. message: 'provider "deepseek-official" model "private-preview" does not support reasoning effort "medium"',
  396. },
  397. })
  398. const rejected = await remote.selectModel(request({
  399. sessionId,
  400. provider: 'missing',
  401. model: 'model',
  402. }))
  403. expect(rejected).toMatchObject({
  404. ok: false,
  405. error: {
  406. code: 'session/model-unavailable',
  407. message: 'no adapter registered for provider "missing"',
  408. details: { provider: 'missing', model: 'model' },
  409. },
  410. })
  411. expect(await remote.selectModel(request({
  412. sessionId,
  413. provider: 'remote-rejected',
  414. model: 'model',
  415. }))).toMatchObject({
  416. ok: false,
  417. error: {
  418. code: 'gateway/internal',
  419. message: 'fixture rejected the selection',
  420. details: {},
  421. },
  422. })
  423. expect(currentSelection(ctx, sessionId))
  424. .toEqual({ provider: 'deepseek-official', model: 'private-preview', reasoningEffort: 'max' })
  425. await ctx.fiber.dispose()
  426. })
  427. it('reads the Agent default live for a session whose log names no selection', async () => {
  428. const { ctx, sessionId } = await harness()
  429. let stored = { provider: 'deepseek-official', model: 'deepseek-chat' }
  430. createSessionTestRemote(ctx, {
  431. defaultModelSelection: () => stored,
  432. cwd: '/tmp',
  433. })
  434. expect(currentSelection(ctx, sessionId))
  435. .toEqual({ provider: 'deepseek-official', model: 'deepseek-chat' })
  436. // The default moving after the session exists still reaches it: New
  437. // Session reuses a blank session rather than minting another, so a seed
  438. // captured at creation would show the superseded model there.
  439. stored = { provider: 'deepseek-official', model: 'deepseek-reasoner' }
  440. expect(currentSelection(ctx, sessionId))
  441. .toEqual({ provider: 'deepseek-official', model: 'deepseek-reasoner' })
  442. await ctx.fiber.dispose()
  443. })
  444. it('keeps a session on its logged selection when the Agent default differs', async () => {
  445. const { ctx, sessionId } = await harness({
  446. provider: 'deepseek-official',
  447. model: 'deepseek-chat',
  448. })
  449. let stored = { provider: 'deepseek-official', model: 'deepseek-chat' }
  450. createSessionTestRemote(ctx, {
  451. defaultModelSelection: () => stored,
  452. cwd: '/tmp',
  453. })
  454. stored = { provider: 'duplicate', model: 'same' }
  455. expect(currentSelection(ctx, sessionId))
  456. .toEqual({ provider: 'deepseek-official', model: 'deepseek-chat' })
  457. await ctx.fiber.dispose()
  458. })
  459. it('does not reinterpret an adapter-owned reasoning default as an explicit Web selection', async () => {
  460. const { ctx, agent } = await harness({
  461. provider: 'deepseek-official',
  462. model: 'deepseek-chat',
  463. reasoningEffort: ReasoningEffortId('high'),
  464. adapterDefaults: { reasoningEffort: true },
  465. })
  466. createSessionTestRemote(ctx, {
  467. defaultModelSelection: () => ({ provider: 'duplicate', model: 'same' }),
  468. cwd: '/tmp',
  469. })
  470. expect(new ApiSessionAgentController(ctx).selectionFor(agent).current)
  471. .toEqual({ provider: 'deepseek-official', model: 'deepseek-chat' })
  472. await ctx.fiber.dispose()
  473. })
  474. it('saves an accepted selection as the default and survives a storage failure', async () => {
  475. const { ctx, sessionId } = await harness()
  476. const saved: unknown[] = []
  477. let reject = false
  478. const remote = createSessionTestRemote(ctx, {
  479. defaultModelSelection: () => ({ provider: 'deepseek-official', model: 'deepseek-chat' }),
  480. saveDefaultModelSelection: (selection) => {
  481. saved.push(selection)
  482. return reject ? Promise.reject(new Error('read-only document')) : Promise.resolve()
  483. },
  484. cwd: '/tmp',
  485. })
  486. expectValue(await remote.selectModel(request({
  487. sessionId, provider: 'deepseek-official', model: 'deepseek-reasoner', reasoningEffort: 'max',
  488. })))
  489. expect(saved).toEqual([
  490. { provider: 'deepseek-official', model: 'deepseek-reasoner', reasoningEffort: 'max' },
  491. ])
  492. // A refused selection never becomes anyone's default.
  493. await remote.selectModel(request({ sessionId, provider: 'missing', model: 'model' }))
  494. expect(saved).toHaveLength(1)
  495. // Storage failing is not the selection failing: the switch already applies
  496. // to this session, so the call still succeeds.
  497. reject = true
  498. const stillAccepted = expectValue(await remote.selectModel(request({
  499. sessionId, provider: 'deepseek-official', model: 'deepseek-chat',
  500. })))
  501. expect(stillAccepted.selected).toEqual({ provider: 'deepseek-official', model: 'deepseek-chat', reasoningEffort: 'high' })
  502. expect(currentSelection(ctx, sessionId))
  503. .toEqual({ provider: 'deepseek-official', model: 'deepseek-chat', reasoningEffort: 'high' })
  504. await ctx.fiber.dispose()
  505. })
  506. it('refuses a prompt no adapter can route, and reports it on the directory', async () => {
  507. const { ctx, sessionId } = await harness()
  508. const remote = createSessionTestRemote(ctx, {
  509. defaultModelSelection: () => ({ provider: 'deleted-gateway', model: 'deleted-model' }),
  510. cwd: '/tmp',
  511. })
  512. // The client disabling its input is an affordance; this method stays
  513. // callable, so the refusal has to live here.
  514. const refused = await remote.prompt(promptRequest({
  515. sessionId, mode: 'queue' as const, content: [{ type: 'text' as const, text: 'hi' }],
  516. }))
  517. expect(refused).toMatchObject({
  518. ok: false,
  519. error: { code: 'session/model-unavailable', details: { provider: 'deleted-gateway', model: 'deleted-model' } },
  520. })
  521. const unavailableCatalog = await buildModelCatalog(ctx)
  522. expect(unavailableCatalog.routableProviders.includes(currentSelection(ctx, sessionId).provider)).toBe(false)
  523. // An advisory-unlisted model on a live route is NOT this: the route
  524. // serves it, so the prompt goes through and nothing blocks.
  525. expectValue(await remote.selectModel(request({
  526. sessionId, provider: 'deepseek-official', model: 'unlisted-but-served',
  527. })))
  528. const catalog = await buildModelCatalog(ctx)
  529. expect(catalog.routableProviders.includes(currentSelection(ctx, sessionId).provider)).toBe(true)
  530. expect(catalog.groups.flatMap(group => group.models.map(model => model.id)))
  531. .not.toContain('unlisted-but-served')
  532. await ctx.fiber.dispose()
  533. })
  534. it('serves a session and its catalog when the stored default names a route that is gone', async () => {
  535. const { ctx, sessionId } = await harness()
  536. createSessionTestRemote(ctx, {
  537. // What a Models-page removal leaves behind: the settings document still
  538. // names the route the user last picked, and nothing serves it.
  539. defaultModelSelection: () => ({ provider: 'deleted-gateway', model: 'deleted-model' }),
  540. cwd: '/tmp',
  541. })
  542. const catalog = await buildModelCatalog(ctx)
  543. // Passed through rather than repaired: matching no group is precisely what
  544. // makes the composer seat prompt for a selection instead of naming a model
  545. // the deployment cannot reach.
  546. expect(currentSelection(ctx, sessionId)).toEqual({ provider: 'deleted-gateway', model: 'deleted-model' })
  547. expect(catalog.groups.flatMap(group => group.models.map(model => `${group.id}/${model.id}`)))
  548. .not.toContain('deleted-gateway/deleted-model')
  549. await ctx.fiber.dispose()
  550. })
  551. it('maps image admission failures and accepts image-capable selections', async () => {
  552. const { ctx, agent, sessionId } = await harness()
  553. registerTextOnly(ctx)
  554. ctx.llm.registerAdapter(['image-capable'], new class extends CatalogAdapter {
  555. override resolveModel(provider: string, model: string): Promise<LlmResolvedModelInfo> {
  556. return Promise.resolve({
  557. provider, id: model, name: model, inputModalities: ['text', 'image'],
  558. })
  559. }
  560. }('Image Capable', []))
  561. ctx.llm.registerAdapter(['string-error'], new class extends CatalogAdapter {
  562. override resolveModel(): Promise<LlmResolvedModelInfo> {
  563. // oxlint-disable-next-line typescript/prefer-promise-reject-errors -- non-Error provider normalization is the scenario.
  564. return Promise.reject('string selection failure')
  565. }
  566. }('String Error', []))
  567. let saveMode: 'success' | 'error' | 'remote' = 'success'
  568. const savedRef = {
  569. attachmentId: 'saved-image', mediaType: 'image/png' as const, bytes: 1, width: 1, height: 1,
  570. }
  571. ctx.provide('attachments', {
  572. saveImages: () => {
  573. if (saveMode === 'error') return Promise.reject(new Error('image store offline'))
  574. if (saveMode === 'remote') {
  575. return Promise.reject(new RemoteError('gateway/internal', 'fixture rejected', {}))
  576. }
  577. return Promise.resolve([savedRef])
  578. },
  579. } as never)
  580. const followup = vi.fn()
  581. Object.assign(agent, { followup })
  582. const remote = createSessionTestRemote(ctx, {
  583. defaultModelSelection: () => ({ provider: 'deepseek-official', model: 'deepseek-chat' }),
  584. cwd: '/tmp',
  585. })
  586. const image = { type: 'image' as const, mediaType: 'image/png' as const, data: 'AQ==' }
  587. expectValue(await remote.selectModel(request({
  588. sessionId, provider: 'text-only', model: 'plain',
  589. })))
  590. expect(await remote.prompt(promptRequest({
  591. sessionId, mode: 'queue', content: [image],
  592. }))).toMatchObject({
  593. ok: false,
  594. error: { code: 'session/attachment-invalid', details: { reason: 'MODEL_DOES_NOT_SUPPORT_IMAGES' } },
  595. })
  596. expectValue(await remote.selectModel(request({
  597. sessionId, provider: 'image-capable', model: 'vision',
  598. })))
  599. expect(await remote.prompt(promptRequest({
  600. sessionId, mode: 'queue', content: [{ ...image, data: '' }],
  601. }))).toMatchObject({
  602. ok: false,
  603. error: { code: 'session/attachment-invalid', details: { reason: 'INVALID_IMAGE_BASE64' } },
  604. })
  605. saveMode = 'error'
  606. expect(await remote.prompt(promptRequest({
  607. sessionId, mode: 'queue', content: [image],
  608. }))).toMatchObject({ ok: false, error: { code: 'session/agent-busy' } })
  609. saveMode = 'remote'
  610. expect(await remote.prompt(promptRequest({
  611. sessionId, mode: 'queue', content: [image],
  612. }))).toMatchObject({ ok: false, error: { code: 'gateway/internal', message: 'fixture rejected' } })
  613. saveMode = 'success'
  614. expectValue(await remote.prompt(promptRequest({ sessionId, mode: 'queue', content: [image] })))
  615. expect(followup).toHaveBeenCalledOnce()
  616. ;(agent.inbox.nextTurn as UserMessage[]).push({
  617. id: 'pending-image', role: 'user', source: { kind: 'user' },
  618. content: [{ type: 'image', attachment: savedRef }],
  619. } as never)
  620. expectValue(await remote.selectModel(request({
  621. sessionId, provider: 'deepseek-official', model: 'deepseek-chat',
  622. })))
  623. expectValue(await remote.selectModel(request({
  624. sessionId, provider: 'image-capable', model: 'vision',
  625. })))
  626. expect(await remote.selectModel(request({
  627. sessionId, provider: 'metadata-broken', model: 'broken',
  628. }))).toMatchObject({
  629. ok: false, error: { code: 'session/model-unavailable', message: 'reasoning metadata offline' },
  630. })
  631. expect(await remote.selectModel(request({
  632. sessionId, provider: 'string-error', model: 'broken',
  633. }))).toMatchObject({
  634. ok: false,
  635. error: { code: 'session/model-unavailable', message: 'string selection failure' },
  636. })
  637. await ctx.fiber.dispose()
  638. })
  639. })