mccode-ai 1.18.21

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (766) hide show
  1. package/AGENTS.md +131 -0
  2. package/Dockerfile +18 -0
  3. package/README.md +15 -0
  4. package/bin/mccode +199 -0
  5. package/bunfig.toml +7 -0
  6. package/git +0 -0
  7. package/migration/20260511173437_session-metadata/migration.sql +1 -0
  8. package/migration/20260511173437_session-metadata/snapshot.json +1500 -0
  9. package/package.json +171 -0
  10. package/parsers-config.ts +1 -0
  11. package/script/bench-search.ts +94 -0
  12. package/script/bench-test-suite.ts +52 -0
  13. package/script/build.ts +246 -0
  14. package/script/generate.ts +14 -0
  15. package/script/httpapi-exercise.ts +1 -0
  16. package/script/postinstall.mjs +189 -0
  17. package/script/profile-test-files.ts +42 -0
  18. package/script/publish.ts +213 -0
  19. package/script/run-workspace-server +106 -0
  20. package/script/schema.ts +77 -0
  21. package/script/time.ts +6 -0
  22. package/script/trace-imports.ts +153 -0
  23. package/specs/effect/error-boundaries-plan.md +235 -0
  24. package/specs/effect/errors.md +207 -0
  25. package/specs/effect/facades.md +218 -0
  26. package/specs/effect/guide.md +247 -0
  27. package/specs/effect/instance-context.md +13 -0
  28. package/specs/effect/loose-ends.md +30 -0
  29. package/specs/effect/migration.md +62 -0
  30. package/specs/effect/routes.md +61 -0
  31. package/specs/effect/schema.md +88 -0
  32. package/specs/effect/server-package.md +58 -0
  33. package/specs/effect/todo.md +241 -0
  34. package/specs/effect/tools.md +88 -0
  35. package/specs/openapi-translation-cleanup.md +204 -0
  36. package/specs/tui-plugins.md +544 -0
  37. package/specs/v2/api.ts +67 -0
  38. package/specs/v2/message-shape.md +136 -0
  39. package/specs/v2/notifications.md +13 -0
  40. package/specs/v2/tui-command-shim.md +67 -0
  41. package/src/account/account.ts +481 -0
  42. package/src/account/repo.ts +171 -0
  43. package/src/account/schema.ts +99 -0
  44. package/src/account/url.ts +8 -0
  45. package/src/acp/agent.ts +95 -0
  46. package/src/acp/config-option.ts +203 -0
  47. package/src/acp/content.ts +269 -0
  48. package/src/acp/directory.ts +212 -0
  49. package/src/acp/error.ts +97 -0
  50. package/src/acp/event.ts +421 -0
  51. package/src/acp/permission.ts +254 -0
  52. package/src/acp/profile.ts +42 -0
  53. package/src/acp/service.ts +1105 -0
  54. package/src/acp/session.ts +232 -0
  55. package/src/acp/tool.ts +364 -0
  56. package/src/acp/usage.ts +243 -0
  57. package/src/agent/agent.ts +476 -0
  58. package/src/agent/generate.txt +75 -0
  59. package/src/agent/prompt/compaction.txt +5 -0
  60. package/src/agent/prompt/explore.txt +18 -0
  61. package/src/agent/prompt/mc.txt +52 -0
  62. package/src/agent/prompt/summary.txt +11 -0
  63. package/src/agent/prompt/title.txt +44 -0
  64. package/src/agent/subagent-permissions.ts +27 -0
  65. package/src/audio.d.ts +14 -0
  66. package/src/auth/index.ts +97 -0
  67. package/src/background/job.ts +37 -0
  68. package/src/bus/global.ts +22 -0
  69. package/src/cli/bootstrap.ts +11 -0
  70. package/src/cli/cmd/account.ts +264 -0
  71. package/src/cli/cmd/acp.ts +73 -0
  72. package/src/cli/cmd/agent.ts +259 -0
  73. package/src/cli/cmd/attach.ts +148 -0
  74. package/src/cli/cmd/cmd.ts +7 -0
  75. package/src/cli/cmd/db.ts +62 -0
  76. package/src/cli/cmd/debug/agent.handler.ts +193 -0
  77. package/src/cli/cmd/debug/agent.ts +27 -0
  78. package/src/cli/cmd/debug/config.ts +14 -0
  79. package/src/cli/cmd/debug/file.ts +73 -0
  80. package/src/cli/cmd/debug/index.ts +87 -0
  81. package/src/cli/cmd/debug/lsp.ts +50 -0
  82. package/src/cli/cmd/debug/ripgrep.ts +79 -0
  83. package/src/cli/cmd/debug/scrap.ts +16 -0
  84. package/src/cli/cmd/debug/skill.ts +15 -0
  85. package/src/cli/cmd/debug/snapshot.ts +50 -0
  86. package/src/cli/cmd/debug/startup.ts +11 -0
  87. package/src/cli/cmd/debug/v2.ts +42 -0
  88. package/src/cli/cmd/export.ts +292 -0
  89. package/src/cli/cmd/generate.ts +54 -0
  90. package/src/cli/cmd/github.handler.ts +1599 -0
  91. package/src/cli/cmd/github.shared.ts +30 -0
  92. package/src/cli/cmd/github.ts +42 -0
  93. package/src/cli/cmd/import.ts +230 -0
  94. package/src/cli/cmd/mcp.ts +840 -0
  95. package/src/cli/cmd/models.ts +66 -0
  96. package/src/cli/cmd/plug.ts +230 -0
  97. package/src/cli/cmd/pr.ts +115 -0
  98. package/src/cli/cmd/prompt-display.ts +1 -0
  99. package/src/cli/cmd/providers.ts +534 -0
  100. package/src/cli/cmd/run/demo.ts +1274 -0
  101. package/src/cli/cmd/run/entry.body.ts +205 -0
  102. package/src/cli/cmd/run/footer.command.tsx +1070 -0
  103. package/src/cli/cmd/run/footer.menu.tsx +351 -0
  104. package/src/cli/cmd/run/footer.permission.tsx +472 -0
  105. package/src/cli/cmd/run/footer.prompt.tsx +1306 -0
  106. package/src/cli/cmd/run/footer.question.tsx +573 -0
  107. package/src/cli/cmd/run/footer.subagent.tsx +175 -0
  108. package/src/cli/cmd/run/footer.ts +1129 -0
  109. package/src/cli/cmd/run/footer.view.tsx +945 -0
  110. package/src/cli/cmd/run/footer.width.ts +27 -0
  111. package/src/cli/cmd/run/permission.shared.ts +256 -0
  112. package/src/cli/cmd/run/prompt.editor.ts +157 -0
  113. package/src/cli/cmd/run/prompt.shared.ts +153 -0
  114. package/src/cli/cmd/run/question.shared.ts +340 -0
  115. package/src/cli/cmd/run/runtime.boot.ts +202 -0
  116. package/src/cli/cmd/run/runtime.lifecycle.ts +406 -0
  117. package/src/cli/cmd/run/runtime.queue.ts +349 -0
  118. package/src/cli/cmd/run/runtime.shared.ts +17 -0
  119. package/src/cli/cmd/run/runtime.stdin.ts +37 -0
  120. package/src/cli/cmd/run/runtime.ts +814 -0
  121. package/src/cli/cmd/run/scrollback.shared.ts +92 -0
  122. package/src/cli/cmd/run/scrollback.surface.ts +431 -0
  123. package/src/cli/cmd/run/scrollback.writer.tsx +352 -0
  124. package/src/cli/cmd/run/session-data.ts +1113 -0
  125. package/src/cli/cmd/run/session-replay.ts +374 -0
  126. package/src/cli/cmd/run/session.shared.ts +196 -0
  127. package/src/cli/cmd/run/splash.ts +280 -0
  128. package/src/cli/cmd/run/stream.transport.ts +1462 -0
  129. package/src/cli/cmd/run/stream.ts +175 -0
  130. package/src/cli/cmd/run/subagent-data.ts +876 -0
  131. package/src/cli/cmd/run/theme.ts +690 -0
  132. package/src/cli/cmd/run/tool.ts +1486 -0
  133. package/src/cli/cmd/run/trace.ts +94 -0
  134. package/src/cli/cmd/run/turn-summary.ts +47 -0
  135. package/src/cli/cmd/run/types.ts +350 -0
  136. package/src/cli/cmd/run/variant.shared.ts +216 -0
  137. package/src/cli/cmd/run.ts +1016 -0
  138. package/src/cli/cmd/serve.ts +24 -0
  139. package/src/cli/cmd/session.ts +147 -0
  140. package/src/cli/cmd/stats.ts +393 -0
  141. package/src/cli/cmd/tui.ts +309 -0
  142. package/src/cli/cmd/uninstall.ts +353 -0
  143. package/src/cli/cmd/upgrade.ts +74 -0
  144. package/src/cli/cmd/web.ts +84 -0
  145. package/src/cli/effect/prompt.ts +37 -0
  146. package/src/cli/effect-cmd.ts +96 -0
  147. package/src/cli/error.ts +130 -0
  148. package/src/cli/heap.ts +45 -0
  149. package/src/cli/logo.ts +1 -0
  150. package/src/cli/network.ts +80 -0
  151. package/src/cli/tui/layer.ts +8 -0
  152. package/src/cli/tui/validate-session.ts +29 -0
  153. package/src/cli/tui/worker.ts +80 -0
  154. package/src/cli/ui.ts +127 -0
  155. package/src/cli/upgrade.ts +53 -0
  156. package/src/command/index.ts +201 -0
  157. package/src/command/template/initialize.txt +66 -0
  158. package/src/command/template/mcstatus.txt +1 -0
  159. package/src/command/template/new.txt +5 -0
  160. package/src/command/template/review.txt +101 -0
  161. package/src/config/agent.ts +59 -0
  162. package/src/config/command.ts +39 -0
  163. package/src/config/config.ts +681 -0
  164. package/src/config/entry-name.ts +19 -0
  165. package/src/config/managed.ts +69 -0
  166. package/src/config/markdown.ts +36 -0
  167. package/src/config/parse.ts +61 -0
  168. package/src/config/paths.ts +45 -0
  169. package/src/config/plugin.ts +79 -0
  170. package/src/config/tui-cwd.ts +5 -0
  171. package/src/config/tui-host-attention.ts +21 -0
  172. package/src/config/tui-migrate.ts +132 -0
  173. package/src/config/tui.ts +276 -0
  174. package/src/config/variable.ts +91 -0
  175. package/src/control-plane/adapters/index.ts +41 -0
  176. package/src/control-plane/adapters/worktree.ts +96 -0
  177. package/src/control-plane/dev/README.md +19 -0
  178. package/src/control-plane/dev/debug-workspace-plugin.ts +73 -0
  179. package/src/control-plane/types.ts +59 -0
  180. package/src/control-plane/util.ts +39 -0
  181. package/src/control-plane/workspace-adapter-runtime.ts +51 -0
  182. package/src/control-plane/workspace-context.ts +26 -0
  183. package/src/control-plane/workspace.ts +966 -0
  184. package/src/effect/app-node-builder-v1.ts +12 -0
  185. package/src/effect/app-runtime.ts +135 -0
  186. package/src/effect/bootstrap-runtime.ts +19 -0
  187. package/src/effect/bridge.ts +84 -0
  188. package/src/effect/config-service.ts +67 -0
  189. package/src/effect/instance-ref.ts +11 -0
  190. package/src/effect/instance-registry.ts +12 -0
  191. package/src/effect/instance-state.ts +69 -0
  192. package/src/effect/promise.ts +17 -0
  193. package/src/effect/run-service.ts +47 -0
  194. package/src/effect/runner.ts +217 -0
  195. package/src/effect/runtime-flags.ts +78 -0
  196. package/src/env/index.ts +41 -0
  197. package/src/event-manifest.ts +3 -0
  198. package/src/event-v2-bridge.ts +71 -0
  199. package/src/format/formatter.ts +404 -0
  200. package/src/format/index.ts +203 -0
  201. package/src/git/index.ts +348 -0
  202. package/src/id/id.ts +80 -0
  203. package/src/ide/index.ts +54 -0
  204. package/src/image/image.ts +172 -0
  205. package/src/index.ts +142 -0
  206. package/src/installation/index.ts +336 -0
  207. package/src/lsp/client.ts +650 -0
  208. package/src/lsp/diagnostic.ts +29 -0
  209. package/src/lsp/language.ts +121 -0
  210. package/src/lsp/launch.ts +21 -0
  211. package/src/lsp/lsp.ts +507 -0
  212. package/src/lsp/server.ts +1983 -0
  213. package/src/markdown.d.ts +4 -0
  214. package/src/mcp/auth.ts +163 -0
  215. package/src/mcp/browser.ts +37 -0
  216. package/src/mcp/catalog.ts +170 -0
  217. package/src/mcp/index.ts +1004 -0
  218. package/src/mcp/oauth-callback.ts +194 -0
  219. package/src/mcp/oauth-provider.ts +259 -0
  220. package/src/node.ts +4 -0
  221. package/src/patch/index.ts +686 -0
  222. package/src/permission/arity.ts +163 -0
  223. package/src/permission/evaluate.ts +1 -0
  224. package/src/permission/index.ts +223 -0
  225. package/src/plugin/azure.ts +26 -0
  226. package/src/plugin/cerebras.ts +11 -0
  227. package/src/plugin/cloudflare.ts +65 -0
  228. package/src/plugin/digitalocean.ts +325 -0
  229. package/src/plugin/github-copilot/copilot.ts +414 -0
  230. package/src/plugin/github-copilot/models.ts +261 -0
  231. package/src/plugin/index.ts +318 -0
  232. package/src/plugin/install.ts +439 -0
  233. package/src/plugin/loader.ts +237 -0
  234. package/src/plugin/meta.ts +188 -0
  235. package/src/plugin/modal/modal.ts +17 -0
  236. package/src/plugin/modal/models.ts +133 -0
  237. package/src/plugin/openai/README.md +31 -0
  238. package/src/plugin/openai/codex.ts +572 -0
  239. package/src/plugin/openai/ws-pool.ts +271 -0
  240. package/src/plugin/openai/ws.ts +383 -0
  241. package/src/plugin/pty-environment.ts +24 -0
  242. package/src/plugin/shared.ts +323 -0
  243. package/src/plugin/snowflake-cortex.ts +507 -0
  244. package/src/plugin/tui/internal.ts +10 -0
  245. package/src/plugin/tui/runtime.ts +1131 -0
  246. package/src/plugin/xai.ts +339 -0
  247. package/src/project/bootstrap-service.ts +9 -0
  248. package/src/project/bootstrap.ts +58 -0
  249. package/src/project/instance-context.ts +24 -0
  250. package/src/project/instance-runtime.ts +16 -0
  251. package/src/project/instance-store.ts +213 -0
  252. package/src/project/project.ts +483 -0
  253. package/src/project/vcs.ts +423 -0
  254. package/src/provider/auth.ts +229 -0
  255. package/src/provider/error.ts +195 -0
  256. package/src/provider/model-status.ts +8 -0
  257. package/src/provider/provider.ts +2047 -0
  258. package/src/provider/transform.ts +1856 -0
  259. package/src/question/index.ts +161 -0
  260. package/src/question/schema.ts +4 -0
  261. package/src/server/auth.ts +48 -0
  262. package/src/server/event.ts +10 -0
  263. package/src/server/global-lifecycle.ts +28 -0
  264. package/src/server/init-projectors.ts +3 -0
  265. package/src/server/mdns.ts +47 -0
  266. package/src/server/projectors.ts +1 -0
  267. package/src/server/proxy-util.ts +48 -0
  268. package/src/server/routes/instance/httpapi/AGENTS.md +39 -0
  269. package/src/server/routes/instance/httpapi/api.ts +97 -0
  270. package/src/server/routes/instance/httpapi/errors.ts +193 -0
  271. package/src/server/routes/instance/httpapi/groups/config.ts +65 -0
  272. package/src/server/routes/instance/httpapi/groups/control-plane.ts +35 -0
  273. package/src/server/routes/instance/httpapi/groups/control.ts +76 -0
  274. package/src/server/routes/instance/httpapi/groups/event.ts +29 -0
  275. package/src/server/routes/instance/httpapi/groups/experimental.ts +275 -0
  276. package/src/server/routes/instance/httpapi/groups/file.ts +185 -0
  277. package/src/server/routes/instance/httpapi/groups/global.ts +136 -0
  278. package/src/server/routes/instance/httpapi/groups/instance.ts +206 -0
  279. package/src/server/routes/instance/httpapi/groups/mcp.ts +156 -0
  280. package/src/server/routes/instance/httpapi/groups/metadata.ts +18 -0
  281. package/src/server/routes/instance/httpapi/groups/permission.ts +61 -0
  282. package/src/server/routes/instance/httpapi/groups/project-copy.ts +32 -0
  283. package/src/server/routes/instance/httpapi/groups/project.ts +93 -0
  284. package/src/server/routes/instance/httpapi/groups/provider.ts +101 -0
  285. package/src/server/routes/instance/httpapi/groups/pty.ts +172 -0
  286. package/src/server/routes/instance/httpapi/groups/query.ts +12 -0
  287. package/src/server/routes/instance/httpapi/groups/question.ts +74 -0
  288. package/src/server/routes/instance/httpapi/groups/session.ts +462 -0
  289. package/src/server/routes/instance/httpapi/groups/sync.ts +113 -0
  290. package/src/server/routes/instance/httpapi/groups/tui.ts +208 -0
  291. package/src/server/routes/instance/httpapi/groups/workspace.ts +141 -0
  292. package/src/server/routes/instance/httpapi/handlers/config.ts +34 -0
  293. package/src/server/routes/instance/httpapi/handlers/control-plane.ts +37 -0
  294. package/src/server/routes/instance/httpapi/handlers/control.ts +43 -0
  295. package/src/server/routes/instance/httpapi/handlers/event.ts +99 -0
  296. package/src/server/routes/instance/httpapi/handlers/experimental.ts +193 -0
  297. package/src/server/routes/instance/httpapi/handlers/file.ts +139 -0
  298. package/src/server/routes/instance/httpapi/handlers/global.ts +156 -0
  299. package/src/server/routes/instance/httpapi/handlers/instance.ts +110 -0
  300. package/src/server/routes/instance/httpapi/handlers/mcp.ts +111 -0
  301. package/src/server/routes/instance/httpapi/handlers/permission.ts +41 -0
  302. package/src/server/routes/instance/httpapi/handlers/project-copy.ts +83 -0
  303. package/src/server/routes/instance/httpapi/handlers/project.ts +63 -0
  304. package/src/server/routes/instance/httpapi/handlers/provider.ts +116 -0
  305. package/src/server/routes/instance/httpapi/handlers/pty.ts +273 -0
  306. package/src/server/routes/instance/httpapi/handlers/question.ts +54 -0
  307. package/src/server/routes/instance/httpapi/handlers/session-errors.ts +21 -0
  308. package/src/server/routes/instance/httpapi/handlers/session.ts +442 -0
  309. package/src/server/routes/instance/httpapi/handlers/sync.ts +89 -0
  310. package/src/server/routes/instance/httpapi/handlers/tui.ts +131 -0
  311. package/src/server/routes/instance/httpapi/handlers/workspace.ts +102 -0
  312. package/src/server/routes/instance/httpapi/lifecycle.ts +54 -0
  313. package/src/server/routes/instance/httpapi/middleware/authorization.ts +150 -0
  314. package/src/server/routes/instance/httpapi/middleware/compression.ts +64 -0
  315. package/src/server/routes/instance/httpapi/middleware/cors-vary.ts +29 -0
  316. package/src/server/routes/instance/httpapi/middleware/error.ts +43 -0
  317. package/src/server/routes/instance/httpapi/middleware/fence.ts +25 -0
  318. package/src/server/routes/instance/httpapi/middleware/instance-context.ts +43 -0
  319. package/src/server/routes/instance/httpapi/middleware/proxy.ts +131 -0
  320. package/src/server/routes/instance/httpapi/middleware/schema-error.ts +41 -0
  321. package/src/server/routes/instance/httpapi/middleware/workspace-routing.ts +250 -0
  322. package/src/server/routes/instance/httpapi/public.ts +537 -0
  323. package/src/server/routes/instance/httpapi/server.ts +325 -0
  324. package/src/server/routes/instance/httpapi/websocket-tracker.ts +60 -0
  325. package/src/server/server.ts +226 -0
  326. package/src/server/shared/fence.ts +60 -0
  327. package/src/server/shared/pty-ticket.ts +15 -0
  328. package/src/server/shared/public-ui.ts +12 -0
  329. package/src/server/shared/tui-control.ts +28 -0
  330. package/src/server/shared/ui.ts +108 -0
  331. package/src/server/shared/workspace-routing.ts +45 -0
  332. package/src/server/tui-event.ts +1 -0
  333. package/src/session/compaction.ts +608 -0
  334. package/src/session/instruction.ts +237 -0
  335. package/src/session/llm/AGENTS.md +90 -0
  336. package/src/session/llm/ai-sdk.ts +291 -0
  337. package/src/session/llm/native-request.ts +196 -0
  338. package/src/session/llm/native-runtime.ts +195 -0
  339. package/src/session/llm/request.ts +226 -0
  340. package/src/session/llm.ts +404 -0
  341. package/src/session/message-error.ts +14 -0
  342. package/src/session/message-v2.ts +737 -0
  343. package/src/session/message.ts +148 -0
  344. package/src/session/overflow.ts +34 -0
  345. package/src/session/processor.ts +718 -0
  346. package/src/session/prompt/anthropic.txt +110 -0
  347. package/src/session/prompt/beast.txt +152 -0
  348. package/src/session/prompt/build-switch.txt +5 -0
  349. package/src/session/prompt/codex.txt +84 -0
  350. package/src/session/prompt/copilot-gpt-5.txt +143 -0
  351. package/src/session/prompt/default.txt +100 -0
  352. package/src/session/prompt/gemini.txt +160 -0
  353. package/src/session/prompt/gpt.txt +112 -0
  354. package/src/session/prompt/kimi.txt +100 -0
  355. package/src/session/prompt/meta.txt +70 -0
  356. package/src/session/prompt/plan-mode.txt +70 -0
  357. package/src/session/prompt/plan-reminder-anthropic.txt +67 -0
  358. package/src/session/prompt/plan.txt +26 -0
  359. package/src/session/prompt/trinity.txt +102 -0
  360. package/src/session/prompt.ts +1631 -0
  361. package/src/session/reminders.ts +92 -0
  362. package/src/session/retry.ts +209 -0
  363. package/src/session/revert.ts +136 -0
  364. package/src/session/run-state.ts +151 -0
  365. package/src/session/schema.ts +26 -0
  366. package/src/session/session.ts +1016 -0
  367. package/src/session/status.ts +56 -0
  368. package/src/session/summary.ts +160 -0
  369. package/src/session/system.ts +152 -0
  370. package/src/session/todo.ts +74 -0
  371. package/src/session/tools.ts +590 -0
  372. package/src/share/session.ts +58 -0
  373. package/src/share/share-next.ts +371 -0
  374. package/src/skill/discovery.ts +140 -0
  375. package/src/skill/index.ts +354 -0
  376. package/src/snapshot/index.ts +807 -0
  377. package/src/sql.d.ts +4 -0
  378. package/src/storage/schema.ts +5 -0
  379. package/src/storage/storage.ts +327 -0
  380. package/src/sync/README.md +179 -0
  381. package/src/sync/schema.ts +11 -0
  382. package/src/temporary.ts +31 -0
  383. package/src/tool/apply_patch.ts +313 -0
  384. package/src/tool/apply_patch.txt +33 -0
  385. package/src/tool/code-mode.ts +310 -0
  386. package/src/tool/edit.ts +737 -0
  387. package/src/tool/edit.txt +10 -0
  388. package/src/tool/external-directory.ts +49 -0
  389. package/src/tool/glob.ts +76 -0
  390. package/src/tool/glob.txt +6 -0
  391. package/src/tool/grep.ts +115 -0
  392. package/src/tool/grep.txt +8 -0
  393. package/src/tool/invalid.ts +21 -0
  394. package/src/tool/json-schema.ts +164 -0
  395. package/src/tool/lsp.ts +113 -0
  396. package/src/tool/lsp.txt +24 -0
  397. package/src/tool/mcp-websearch.ts +96 -0
  398. package/src/tool/plan-enter.txt +14 -0
  399. package/src/tool/plan-exit.txt +13 -0
  400. package/src/tool/plan.ts +79 -0
  401. package/src/tool/question.ts +44 -0
  402. package/src/tool/question.txt +10 -0
  403. package/src/tool/read.ts +386 -0
  404. package/src/tool/read.txt +14 -0
  405. package/src/tool/registry.ts +455 -0
  406. package/src/tool/schema.ts +14 -0
  407. package/src/tool/shell/id.ts +19 -0
  408. package/src/tool/shell/prompt.ts +293 -0
  409. package/src/tool/shell/shell.txt +21 -0
  410. package/src/tool/shell.ts +645 -0
  411. package/src/tool/skill.ts +70 -0
  412. package/src/tool/skill.txt +5 -0
  413. package/src/tool/task.ts +371 -0
  414. package/src/tool/task.txt +19 -0
  415. package/src/tool/todo.ts +46 -0
  416. package/src/tool/todowrite.txt +44 -0
  417. package/src/tool/tool.ts +183 -0
  418. package/src/tool/truncate.ts +156 -0
  419. package/src/tool/truncation-dir.ts +4 -0
  420. package/src/tool/webfetch.ts +192 -0
  421. package/src/tool/webfetch.txt +13 -0
  422. package/src/tool/websearch.ts +143 -0
  423. package/src/tool/websearch.txt +14 -0
  424. package/src/tool/write.ts +104 -0
  425. package/src/tool/write.txt +8 -0
  426. package/src/util/archive.ts +17 -0
  427. package/src/util/bom.ts +27 -0
  428. package/src/util/data-url.ts +9 -0
  429. package/src/util/defer.ts +10 -0
  430. package/src/util/effect-http-client.ts +11 -0
  431. package/src/util/error.ts +1 -0
  432. package/src/util/filesystem.ts +251 -0
  433. package/src/util/html.ts +8 -0
  434. package/src/util/iife.ts +3 -0
  435. package/src/util/lazy.ts +20 -0
  436. package/src/util/local-context.ts +25 -0
  437. package/src/util/locale.ts +2 -0
  438. package/src/util/media.ts +26 -0
  439. package/src/util/process.ts +177 -0
  440. package/src/util/proxy-env.ts +72 -0
  441. package/src/util/queue.ts +32 -0
  442. package/src/util/record.ts +1 -0
  443. package/src/util/repository.ts +232 -0
  444. package/src/util/rpc.ts +66 -0
  445. package/src/util/signal.ts +12 -0
  446. package/src/util/timeout.ts +13 -0
  447. package/src/util/token.ts +1 -0
  448. package/src/util/wildcard.ts +59 -0
  449. package/src/worktree/index.ts +623 -0
  450. package/sst-env.d.ts +10 -0
  451. package/test/AGENTS.md +204 -0
  452. package/test/EFFECT_TEST_MIGRATION.md +169 -0
  453. package/test/account/repo.test.ts +355 -0
  454. package/test/account/service.test.ts +524 -0
  455. package/test/acp/config-option.test.ts +229 -0
  456. package/test/acp/content.test.ts +235 -0
  457. package/test/acp/directory.test.ts +188 -0
  458. package/test/acp/error.test.ts +67 -0
  459. package/test/acp/event.test.ts +751 -0
  460. package/test/acp/permission.test.ts +401 -0
  461. package/test/acp/service-session.test.ts +1374 -0
  462. package/test/acp/session.test.ts +201 -0
  463. package/test/acp/tool.test.ts +298 -0
  464. package/test/acp/usage.test.ts +318 -0
  465. package/test/agent/agent.test.ts +755 -0
  466. package/test/agent/plan-mode-subagent-bypass.test.ts +160 -0
  467. package/test/agent/plugin-agent-regression.test.ts +51 -0
  468. package/test/auth/auth.test.ts +75 -0
  469. package/test/background/job.test.ts +244 -0
  470. package/test/cli/account.test.ts +30 -0
  471. package/test/cli/acp/acp-test-client.ts +97 -0
  472. package/test/cli/acp/config-options.test.ts +103 -0
  473. package/test/cli/acp/helpers.ts +96 -0
  474. package/test/cli/acp/initialize-auth.test.ts +61 -0
  475. package/test/cli/acp/lifecycle.test.ts +118 -0
  476. package/test/cli/acp/prompt-content.test.ts +97 -0
  477. package/test/cli/acp/skills.test.ts +38 -0
  478. package/test/cli/cmd/tui/attention.test.ts +484 -0
  479. package/test/cli/effect-cmd-instance-als.test.ts +40 -0
  480. package/test/cli/error.test.ts +95 -0
  481. package/test/cli/github-action.test.ts +199 -0
  482. package/test/cli/github-remote.test.ts +90 -0
  483. package/test/cli/help/__snapshots__/help-snapshots.test.ts.snap +623 -0
  484. package/test/cli/help/help-snapshots.test.ts +140 -0
  485. package/test/cli/import.test.ts +90 -0
  486. package/test/cli/mcp-add.test.ts +74 -0
  487. package/test/cli/plugin-auth-picker.test.ts +120 -0
  488. package/test/cli/run/entry.body.test.ts +536 -0
  489. package/test/cli/run/footer.menu.test.ts +43 -0
  490. package/test/cli/run/footer.view.test.tsx +1411 -0
  491. package/test/cli/run/footer.width.test.ts +35 -0
  492. package/test/cli/run/permission.shared.test.ts +144 -0
  493. package/test/cli/run/prompt.editor.test.ts +101 -0
  494. package/test/cli/run/prompt.shared.test.ts +101 -0
  495. package/test/cli/run/question.shared.test.ts +115 -0
  496. package/test/cli/run/run-process.test.ts +338 -0
  497. package/test/cli/run/runtime.boot.test.ts +283 -0
  498. package/test/cli/run/runtime.queue.test.ts +481 -0
  499. package/test/cli/run/runtime.stdin.test.ts +71 -0
  500. package/test/cli/run/runtime.test.ts +238 -0
  501. package/test/cli/run/scrollback.surface.test.ts +1092 -0
  502. package/test/cli/run/session-data.test.ts +593 -0
  503. package/test/cli/run/session-replay.test.ts +691 -0
  504. package/test/cli/run/session.shared.test.ts +247 -0
  505. package/test/cli/run/stream.test.ts +56 -0
  506. package/test/cli/run/stream.transport.test.ts +2363 -0
  507. package/test/cli/run/subagent-data.test.ts +547 -0
  508. package/test/cli/run/theme.test.ts +177 -0
  509. package/test/cli/run/variant.shared.test.ts +218 -0
  510. package/test/cli/serve/serve-process.test.ts +61 -0
  511. package/test/cli/smokes/read-only.test.ts +115 -0
  512. package/test/cli/tui/attach.test.ts +11 -0
  513. package/test/cli/tui/editor-context-zed.test.ts +379 -0
  514. package/test/cli/tui/editor-context.test.tsx +297 -0
  515. package/test/cli/tui/plugin-add.test.ts +110 -0
  516. package/test/cli/tui/plugin-install.test.ts +87 -0
  517. package/test/cli/tui/plugin-lifecycle.test.ts +224 -0
  518. package/test/cli/tui/plugin-loader-entrypoint.test.ts +485 -0
  519. package/test/cli/tui/plugin-loader-pure.test.ts +72 -0
  520. package/test/cli/tui/plugin-loader.test.ts +1332 -0
  521. package/test/cli/tui/plugin-toggle.test.ts +264 -0
  522. package/test/cli/tui/thread.test.ts +101 -0
  523. package/test/config/agent-color.test.ts +47 -0
  524. package/test/config/config.test.ts +2047 -0
  525. package/test/config/entry-name.test.ts +57 -0
  526. package/test/config/fixtures/empty-frontmatter.md +4 -0
  527. package/test/config/fixtures/frontmatter.md +28 -0
  528. package/test/config/fixtures/markdown-header.md +11 -0
  529. package/test/config/fixtures/no-frontmatter.md +1 -0
  530. package/test/config/fixtures/weird-model-id.md +13 -0
  531. package/test/config/lsp.test.ts +69 -0
  532. package/test/config/markdown.test.ts +228 -0
  533. package/test/config/plugin.test.ts +0 -0
  534. package/test/config/tui.test.ts +894 -0
  535. package/test/control-plane/adapters.test.ts +71 -0
  536. package/test/control-plane/workspace.test.ts +1717 -0
  537. package/test/effect/app-runtime-logger.test.ts +101 -0
  538. package/test/effect/config-service.test.ts +65 -0
  539. package/test/effect/instance-state.test.ts +392 -0
  540. package/test/effect/run-service.test.ts +89 -0
  541. package/test/effect/runner.test.ts +514 -0
  542. package/test/effect/runtime-flags.test.ts +374 -0
  543. package/test/event-manifest.test.ts +24 -0
  544. package/test/fake/account.ts +9 -0
  545. package/test/fake/auth.ts +8 -0
  546. package/test/fake/npm.ts +8 -0
  547. package/test/fake/provider.ts +82 -0
  548. package/test/fake/skill.ts +8 -0
  549. package/test/filesystem/filesystem.test.ts +318 -0
  550. package/test/fixture/agent-plugin.constants.ts +6 -0
  551. package/test/fixture/agent-plugin.ts +12 -0
  552. package/test/fixture/config.ts +23 -0
  553. package/test/fixture/db.ts +11 -0
  554. package/test/fixture/fixture.test.ts +26 -0
  555. package/test/fixture/fixture.ts +228 -0
  556. package/test/fixture/flag.ts +20 -0
  557. package/test/fixture/flock-worker.ts +72 -0
  558. package/test/fixture/lsp/fake-lsp-server.js +249 -0
  559. package/test/fixture/mcp-lifecycle-stdio.ts +26 -0
  560. package/test/fixture/mcp-session-recovery.ts +50 -0
  561. package/test/fixture/plug-worker.ts +93 -0
  562. package/test/fixture/plugin-meta-worker.ts +19 -0
  563. package/test/fixture/plugin.ts +10 -0
  564. package/test/fixture/skills/agents-sdk/SKILL.md +152 -0
  565. package/test/fixture/skills/agents-sdk/references/callable.md +92 -0
  566. package/test/fixture/skills/cloudflare/SKILL.md +211 -0
  567. package/test/fixture/skills/index.json +6 -0
  568. package/test/fixture/tui-environment.tsx +32 -0
  569. package/test/fixture/tui-plugin.ts +355 -0
  570. package/test/fixture/tui-runtime.ts +56 -0
  571. package/test/fixture/tui-sdk.ts +82 -0
  572. package/test/fixture/workspace.ts +34 -0
  573. package/test/fixtures/recordings/session/native-anthropic-tool-loop.json +49 -0
  574. package/test/fixtures/recordings/session/native-openai-oauth-tool-loop.json +45 -0
  575. package/test/fixtures/recordings/session/native-zen-tool-loop.json +49 -0
  576. package/test/format/format.test.ts +235 -0
  577. package/test/git/git.test.ts +179 -0
  578. package/test/ide/ide.test.ts +82 -0
  579. package/test/image/fixtures/picture-5mb-base64.png +0 -0
  580. package/test/image/image.test.ts +121 -0
  581. package/test/installation/installation.test.ts +240 -0
  582. package/test/lib/cli-process.ts +535 -0
  583. package/test/lib/effect.ts +177 -0
  584. package/test/lib/filesystem.ts +10 -0
  585. package/test/lib/llm-server.ts +779 -0
  586. package/test/lib/snapshot.ts +73 -0
  587. package/test/lib/test-provider.ts +37 -0
  588. package/test/lib/websocket.ts +46 -0
  589. package/test/lsp/client.test.ts +488 -0
  590. package/test/lsp/index.test.ts +231 -0
  591. package/test/lsp/jdtls-root.test.ts +459 -0
  592. package/test/lsp/launch.test.ts +22 -0
  593. package/test/lsp/lifecycle.test.ts +160 -0
  594. package/test/mcp/auth.test.ts +78 -0
  595. package/test/mcp/catalog.test.ts +107 -0
  596. package/test/mcp/headers.test.ts +102 -0
  597. package/test/mcp/lifecycle.test.ts +560 -0
  598. package/test/mcp/oauth-auto-connect.test.ts +300 -0
  599. package/test/mcp/oauth-browser.test.ts +225 -0
  600. package/test/mcp/oauth-callback.test.ts +114 -0
  601. package/test/mcp/oauth-provider.test.ts +102 -0
  602. package/test/mcp/session-recovery.test.ts +27 -0
  603. package/test/mcp/transport.test.ts +38 -0
  604. package/test/patch/patch.test.ts +384 -0
  605. package/test/permission/arity.test.ts +33 -0
  606. package/test/permission/next.test.ts +1174 -0
  607. package/test/permission-task.test.ts +319 -0
  608. package/test/plugin/auth-override.test.ts +101 -0
  609. package/test/plugin/cerebras.test.ts +47 -0
  610. package/test/plugin/cloudflare.test.ts +25 -0
  611. package/test/plugin/codex.test.ts +463 -0
  612. package/test/plugin/github-copilot-models.test.ts +492 -0
  613. package/test/plugin/install-concurrency.test.ts +140 -0
  614. package/test/plugin/install.test.ts +570 -0
  615. package/test/plugin/loader-shared.test.ts +1306 -0
  616. package/test/plugin/meta.test.ts +137 -0
  617. package/test/plugin/modal-models.test.ts +198 -0
  618. package/test/plugin/openai-rollout.test.ts +17 -0
  619. package/test/plugin/openai-ws.test.ts +909 -0
  620. package/test/plugin/shared.test.ts +88 -0
  621. package/test/plugin/snowflake-cortex.test.ts +278 -0
  622. package/test/plugin/trigger.test.ts +108 -0
  623. package/test/plugin/workspace-adapter.test.ts +111 -0
  624. package/test/plugin/xai.test.ts +585 -0
  625. package/test/preload.ts +92 -0
  626. package/test/project/instance-bootstrap.test.ts +115 -0
  627. package/test/project/instance.test.ts +248 -0
  628. package/test/project/migrate-global.test.ts +168 -0
  629. package/test/project/project-directory.test.ts +202 -0
  630. package/test/project/project.test.ts +808 -0
  631. package/test/project/vcs.test.ts +335 -0
  632. package/test/project/worktree-remove.test.ts +128 -0
  633. package/test/project/worktree.test.ts +324 -0
  634. package/test/provider/amazon-bedrock.test.ts +361 -0
  635. package/test/provider/cf-ai-gateway-e2e.test.ts +308 -0
  636. package/test/provider/digitalocean.test.ts +124 -0
  637. package/test/provider/error.test.ts +24 -0
  638. package/test/provider/gitlab-duo.test.ts +412 -0
  639. package/test/provider/header-timeout.test.ts +234 -0
  640. package/test/provider/model-status.test.ts +61 -0
  641. package/test/provider/provider.test.ts +2118 -0
  642. package/test/provider/transform.test.ts +5682 -0
  643. package/test/question/question.test.ts +459 -0
  644. package/test/server/AGENTS.md +15 -0
  645. package/test/server/auth.test.ts +59 -0
  646. package/test/server/global-bus.ts +31 -0
  647. package/test/server/global-session-list.test.ts +108 -0
  648. package/test/server/httpapi-authorization.test.ts +174 -0
  649. package/test/server/httpapi-compression.test.ts +151 -0
  650. package/test/server/httpapi-config.test.ts +110 -0
  651. package/test/server/httpapi-control-plane.test.ts +63 -0
  652. package/test/server/httpapi-cors-vary.test.ts +63 -0
  653. package/test/server/httpapi-cors.test.ts +122 -0
  654. package/test/server/httpapi-error-middleware.test.ts +101 -0
  655. package/test/server/httpapi-event.test.ts +94 -0
  656. package/test/server/httpapi-exercise/assertions.ts +64 -0
  657. package/test/server/httpapi-exercise/backend.ts +144 -0
  658. package/test/server/httpapi-exercise/dsl.ts +210 -0
  659. package/test/server/httpapi-exercise/environment.ts +40 -0
  660. package/test/server/httpapi-exercise/index.ts +1802 -0
  661. package/test/server/httpapi-exercise/report.ts +66 -0
  662. package/test/server/httpapi-exercise/routing.ts +96 -0
  663. package/test/server/httpapi-exercise/runner.ts +267 -0
  664. package/test/server/httpapi-exercise/runtime.ts +52 -0
  665. package/test/server/httpapi-exercise/types.ts +127 -0
  666. package/test/server/httpapi-experimental.test.ts +298 -0
  667. package/test/server/httpapi-file.test.ts +83 -0
  668. package/test/server/httpapi-global.test.ts +66 -0
  669. package/test/server/httpapi-instance-context.test.ts +352 -0
  670. package/test/server/httpapi-instance-route-auth.test.ts +81 -0
  671. package/test/server/httpapi-instance.test.ts +265 -0
  672. package/test/server/httpapi-layer.ts +33 -0
  673. package/test/server/httpapi-listen.test.ts +465 -0
  674. package/test/server/httpapi-mcp-oauth.test.ts +73 -0
  675. package/test/server/httpapi-mcp.test.ts +223 -0
  676. package/test/server/httpapi-mdns.test.ts +79 -0
  677. package/test/server/httpapi-promptasync-context.test.ts +212 -0
  678. package/test/server/httpapi-provider.test.ts +401 -0
  679. package/test/server/httpapi-pty.test.ts +299 -0
  680. package/test/server/httpapi-public-openapi.test.ts +350 -0
  681. package/test/server/httpapi-query-schema-drift.test.ts +330 -0
  682. package/test/server/httpapi-reference.test.ts +63 -0
  683. package/test/server/httpapi-schema-error-body.test.ts +166 -0
  684. package/test/server/httpapi-sdk.test.ts +913 -0
  685. package/test/server/httpapi-session.test.ts +1090 -0
  686. package/test/server/httpapi-sync.test.ts +149 -0
  687. package/test/server/httpapi-ui.test.ts +457 -0
  688. package/test/server/httpapi-v2-location.test.ts +126 -0
  689. package/test/server/httpapi-v2-pty.test.ts +250 -0
  690. package/test/server/httpapi-workspace-routing.test.ts +552 -0
  691. package/test/server/httpapi-workspace.test.ts +506 -0
  692. package/test/server/negative-tokens-regression.test.ts +84 -0
  693. package/test/server/project-copy.test.ts +127 -0
  694. package/test/server/project-init-git.test.ts +118 -0
  695. package/test/server/proxy-util.test.ts +113 -0
  696. package/test/server/sdk-error-shape.test.ts +81 -0
  697. package/test/server/sdk-v1-smoke.test.ts +57 -0
  698. package/test/server/session-actions.test.ts +110 -0
  699. package/test/server/session-diff-missing-patch.test.ts +97 -0
  700. package/test/server/session-list.test.ts +302 -0
  701. package/test/server/session-messages.test.ts +180 -0
  702. package/test/server/session-select.test.ts +67 -0
  703. package/test/server/workspace-proxy.test.ts +181 -0
  704. package/test/server/workspace-routing.test.ts +101 -0
  705. package/test/server/worktree-endpoint-repro.test.ts +307 -0
  706. package/test/session/compaction.test.ts +1975 -0
  707. package/test/session/instruction.test.ts +264 -0
  708. package/test/session/llm-native-recorded.test.ts +413 -0
  709. package/test/session/llm-native.test.ts +761 -0
  710. package/test/session/llm.test.ts +2180 -0
  711. package/test/session/message-v2.test.ts +1729 -0
  712. package/test/session/messages-pagination.test.ts +1057 -0
  713. package/test/session/processor-effect.test.ts +1171 -0
  714. package/test/session/prompt.test.ts +2470 -0
  715. package/test/session/retry.test.ts +521 -0
  716. package/test/session/revert-compact.test.ts +683 -0
  717. package/test/session/schema-decoding.test.ts +313 -0
  718. package/test/session/session-schema.test.ts +78 -0
  719. package/test/session/session.test.ts +285 -0
  720. package/test/session/snapshot-tool-race.test.ts +189 -0
  721. package/test/session/structured-output-integration.test.ts +235 -0
  722. package/test/session/structured-output.test.ts +387 -0
  723. package/test/session/system.test.ts +168 -0
  724. package/test/share/share-next.test.ts +324 -0
  725. package/test/skill/discovery.test.ts +186 -0
  726. package/test/skill/skill.test.ts +585 -0
  727. package/test/snapshot/snapshot.test.ts +1218 -0
  728. package/test/storage/storage.test.ts +297 -0
  729. package/test/tool/__snapshots__/parameters.test.ts.snap +466 -0
  730. package/test/tool/__snapshots__/tool.test.ts.snap +9 -0
  731. package/test/tool/apply_patch.test.ts +529 -0
  732. package/test/tool/code-mode-integration.test.ts +331 -0
  733. package/test/tool/code-mode.test.ts +730 -0
  734. package/test/tool/edit.test.ts +574 -0
  735. package/test/tool/external-directory.test.ts +156 -0
  736. package/test/tool/fixtures/large-image.png +0 -0
  737. package/test/tool/fixtures/models-api.json +172692 -0
  738. package/test/tool/glob.test.ts +132 -0
  739. package/test/tool/grep.test.ts +223 -0
  740. package/test/tool/lsp.test.ts +184 -0
  741. package/test/tool/parameters.test.ts +290 -0
  742. package/test/tool/question.test.ts +133 -0
  743. package/test/tool/read.test.ts +608 -0
  744. package/test/tool/registry.test.ts +572 -0
  745. package/test/tool/shell.test.ts +1199 -0
  746. package/test/tool/skill.test.ts +134 -0
  747. package/test/tool/task.test.ts +1101 -0
  748. package/test/tool/tool-define.test.ts +154 -0
  749. package/test/tool/truncation.test.ts +266 -0
  750. package/test/tool/webfetch.test.ts +119 -0
  751. package/test/tool/websearch.test.ts +100 -0
  752. package/test/tool/write.test.ts +279 -0
  753. package/test/util/data-url.test.ts +14 -0
  754. package/test/util/error.test.ts +16 -0
  755. package/test/util/filesystem.test.ts +656 -0
  756. package/test/util/glob.test.ts +164 -0
  757. package/test/util/html.test.ts +15 -0
  758. package/test/util/iife.test.ts +36 -0
  759. package/test/util/lazy.test.ts +50 -0
  760. package/test/util/module.test.ts +59 -0
  761. package/test/util/process.test.ts +128 -0
  762. package/test/util/repository.test.ts +93 -0
  763. package/test/util/timeout.test.ts +21 -0
  764. package/test/util/wildcard.test.ts +90 -0
  765. package/test/v2/session-message-updater.test.ts +269 -0
  766. package/tsconfig.json +16 -0
@@ -0,0 +1,2180 @@
1
+ import { PermissionV1 } from "@opencode-ai/core/v1/permission"
2
+ import { ConfigV1 } from "@opencode-ai/core/v1/config/config"
3
+ import { afterAll, beforeAll, beforeEach, describe, expect, test } from "bun:test"
4
+ import { SessionV1 } from "@opencode-ai/core/v1/session"
5
+ import path from "path"
6
+ import { tool, type ModelMessage } from "ai"
7
+ import { Cause, Effect, Exit, Fiber, Layer, Stream } from "effect"
8
+ import { InstanceRef } from "../../src/effect/instance-ref"
9
+ import { HttpClientRequest, HttpClientResponse } from "effect/unstable/http"
10
+ import z from "zod"
11
+ import { LLM } from "../../src/session/llm"
12
+ import { LLMClient, RequestExecutor } from "@opencode-ai/llm/route"
13
+ import { Provider } from "@/provider/provider"
14
+ import { ProviderTransform } from "@/provider/transform"
15
+ import { ModelsDev } from "@opencode-ai/core/models-dev"
16
+
17
+ import { testEffect } from "../lib/effect"
18
+ import type { Agent } from "../../src/agent/agent"
19
+ import { MessageV2 } from "../../src/session/message-v2"
20
+ import { SessionID, MessageID } from "../../src/session/schema"
21
+ import { RuntimeFlags } from "@/effect/runtime-flags"
22
+ import { Permission } from "@/permission"
23
+ import { LLMAISDK } from "@/session/llm/ai-sdk"
24
+ import { Session as SessionNs } from "@/session/session"
25
+ import { ProviderV2 } from "@opencode-ai/core/provider"
26
+ import { ModelV2 } from "@opencode-ai/core/model"
27
+ import { AppNodeBuilder } from "@opencode-ai/core/effect/app-node-builder"
28
+ import { LayerNode } from "@opencode-ai/core/effect/layer-node"
29
+ import { LayerNodePlatform } from "@opencode-ai/core/effect/app-node-platform"
30
+ import { ProviderError } from "@/provider/error"
31
+
32
+ type ConfigModel = NonNullable<NonNullable<ConfigV1.Info["provider"]>[string]["models"]>[string]
33
+
34
+ const openAIConfig = (model: ModelsDev.Provider["models"][string], baseURL: string): Partial<ConfigV1.Info> => {
35
+ const { experimental: _experimental, ...configModel } = model
36
+ return {
37
+ enabled_providers: ["openai"],
38
+ provider: {
39
+ openai: {
40
+ name: "OpenAI",
41
+ env: ["OPENAI_API_KEY"],
42
+ npm: "@ai-sdk/openai",
43
+ api: "https://api.openai.com/v1",
44
+ models: {
45
+ [model.id]: JSON.parse(JSON.stringify(configModel)) as ConfigModel,
46
+ },
47
+ options: {
48
+ apiKey: "test-openai-key",
49
+ baseURL,
50
+ },
51
+ },
52
+ },
53
+ }
54
+ }
55
+
56
+ const it = testEffect(AppNodeBuilder.build(LayerNode.group([LLM.node, Provider.node])))
57
+
58
+ // LLM.stream returns a Stream, not an Effect, so we can't use the serviceUse proxy.
59
+ const drain = (input: LLM.StreamInput) => LLM.Service.use((svc) => svc.stream(input).pipe(Stream.runDrain))
60
+
61
+ // drainWith builds an isolated runtime so custom replacements fully own LLM and
62
+ // its transitive deps.
63
+ const drainWith = (layer: Layer.Layer<LLM.Service>, input: LLM.StreamInput) =>
64
+ Effect.gen(function* () {
65
+ const ctx = yield* InstanceRef
66
+ if (!ctx) return yield* Effect.die("InstanceRef not provided")
67
+ return yield* Effect.promise(() =>
68
+ Effect.runPromise(
69
+ LLM.Service.use((svc) => svc.stream(input).pipe(Stream.runDrain)).pipe(
70
+ Effect.provide(layer),
71
+ Effect.provideService(InstanceRef, ctx),
72
+ ),
73
+ ),
74
+ )
75
+ })
76
+
77
+ function llmLayerWithExecutor(
78
+ options: {
79
+ executor?: Layer.Layer<RequestExecutor.Service>
80
+ flags?: Partial<RuntimeFlags.Info>
81
+ } = {},
82
+ ) {
83
+ return AppNodeBuilder.build(LLM.node, [
84
+ [RuntimeFlags.node, RuntimeFlags.layer(options.flags)],
85
+ ...(options.executor ? ([[LayerNodePlatform.requestExecutor, options.executor]] as const) : []),
86
+ ])
87
+ }
88
+
89
+ describe("session.llm.hasToolCalls", () => {
90
+ test("returns false for empty messages array", () => {
91
+ expect(LLM.hasToolCalls([])).toBe(false)
92
+ })
93
+
94
+ test("returns false for messages with only text content", () => {
95
+ const messages: ModelMessage[] = [
96
+ {
97
+ role: "user",
98
+ content: [{ type: "text", text: "Hello" }],
99
+ },
100
+ {
101
+ role: "assistant",
102
+ content: [{ type: "text", text: "Hi there" }],
103
+ },
104
+ ]
105
+ expect(LLM.hasToolCalls(messages)).toBe(false)
106
+ })
107
+
108
+ test("returns true when messages contain tool-call", () => {
109
+ const messages = [
110
+ {
111
+ role: "user",
112
+ content: [{ type: "text", text: "Run a command" }],
113
+ },
114
+ {
115
+ role: "assistant",
116
+ content: [
117
+ {
118
+ type: "tool-call",
119
+ toolCallId: "call-123",
120
+ toolName: "bash",
121
+ },
122
+ ],
123
+ },
124
+ ] as ModelMessage[]
125
+ expect(LLM.hasToolCalls(messages)).toBe(true)
126
+ })
127
+
128
+ test("returns true when messages contain tool-result", () => {
129
+ const messages = [
130
+ {
131
+ role: "tool",
132
+ content: [
133
+ {
134
+ type: "tool-result",
135
+ toolCallId: "call-123",
136
+ toolName: "bash",
137
+ },
138
+ ],
139
+ },
140
+ ] as ModelMessage[]
141
+ expect(LLM.hasToolCalls(messages)).toBe(true)
142
+ })
143
+
144
+ test("returns false for messages with string content", () => {
145
+ const messages: ModelMessage[] = [
146
+ {
147
+ role: "user",
148
+ content: "Hello world",
149
+ },
150
+ {
151
+ role: "assistant",
152
+ content: "Hi there",
153
+ },
154
+ ]
155
+ expect(LLM.hasToolCalls(messages)).toBe(false)
156
+ })
157
+
158
+ test("returns true when tool-call is mixed with text content", () => {
159
+ const messages = [
160
+ {
161
+ role: "assistant",
162
+ content: [
163
+ { type: "text", text: "Let me run that command" },
164
+ {
165
+ type: "tool-call",
166
+ toolCallId: "call-456",
167
+ toolName: "read",
168
+ },
169
+ ],
170
+ },
171
+ ] as ModelMessage[]
172
+ expect(LLM.hasToolCalls(messages)).toBe(true)
173
+ })
174
+ })
175
+
176
+ describe("session.llm.ai-sdk adapter", () => {
177
+ type AISDKAdapterEvent = Parameters<typeof LLMAISDK.toLLMEvents>[1]
178
+
179
+ const adapt = (events: ReadonlyArray<AISDKAdapterEvent>) => {
180
+ const state = LLMAISDK.adapterState()
181
+ return Effect.runPromise(
182
+ Effect.forEach(events, (event) => LLMAISDK.toLLMEvents(state, event)).pipe(Effect.map((items) => items.flat())),
183
+ )
184
+ }
185
+ // oxlint-disable-next-line typescript-eslint/no-unsafe-type-assertion -- tests defensive adapter branches outside AI SDK's current typed surface
186
+ const uncheckedAdapterEvent = (input: unknown) => input as AISDKAdapterEvent
187
+
188
+ test("maps AI SDK stream chunks without losing session-visible fields", async () => {
189
+ const metadata = { openai: { itemID: "item-1" } }
190
+ const events = await adapt([
191
+ { type: "start" },
192
+ { type: "start-step", request: {}, warnings: [] },
193
+ { type: "text-start", id: "text-1", providerMetadata: metadata },
194
+ { type: "text-delta", id: "text-1", text: "Hel", providerMetadata: { openai: { delta: 1 } } },
195
+ { type: "text-delta", id: "text-1", text: "lo", providerMetadata: { openai: { delta: 2 } } },
196
+ { type: "text-end", id: "text-1", providerMetadata: { openai: { done: true } } },
197
+ { type: "reasoning-start", id: "reasoning-1", providerMetadata: metadata },
198
+ { type: "reasoning-delta", id: "reasoning-1", text: "Think", providerMetadata: { openai: { delta: 3 } } },
199
+ { type: "reasoning-end", id: "reasoning-1", providerMetadata: { openai: { done: true } } },
200
+ { type: "tool-input-start", id: "call-1", toolName: "lookup", providerMetadata: metadata },
201
+ { type: "tool-input-delta", id: "call-1", delta: '{"query":' },
202
+ { type: "tool-input-delta", id: "call-1", delta: '"weather"}' },
203
+ { type: "tool-input-end", id: "call-1", providerMetadata: { openai: { inputDone: true } } },
204
+ {
205
+ type: "tool-call",
206
+ toolCallId: "call-1",
207
+ toolName: "lookup",
208
+ input: { query: "weather" },
209
+ providerExecuted: true,
210
+ providerMetadata: { openai: { called: true } },
211
+ },
212
+ {
213
+ type: "tool-result",
214
+ toolCallId: "call-1",
215
+ toolName: "lookup",
216
+ input: { query: "weather" },
217
+ output: { title: "Lookup", output: "sunny", metadata: { ok: true } },
218
+ providerExecuted: true,
219
+ providerMetadata: { openai: { result: true } },
220
+ },
221
+ {
222
+ type: "finish-step",
223
+ response: { id: "response-1", timestamp: new Date(0), modelId: "gpt-test" },
224
+ finishReason: "other",
225
+ rawFinishReason: "other",
226
+ usage: {
227
+ inputTokens: 10,
228
+ outputTokens: 5,
229
+ totalTokens: 15,
230
+ inputTokenDetails: { noCacheTokens: 5, cacheReadTokens: 3, cacheWriteTokens: 2 },
231
+ outputTokenDetails: { textTokens: 4, reasoningTokens: 1 },
232
+ },
233
+ providerMetadata: { openai: { step: true } },
234
+ },
235
+ {
236
+ type: "finish",
237
+ finishReason: "other",
238
+ rawFinishReason: "other",
239
+ totalUsage: {
240
+ inputTokens: 11,
241
+ outputTokens: 6,
242
+ totalTokens: 17,
243
+ cachedInputTokens: 4,
244
+ reasoningTokens: 2,
245
+ inputTokenDetails: { noCacheTokens: 7, cacheReadTokens: 4, cacheWriteTokens: undefined },
246
+ outputTokenDetails: { textTokens: 4, reasoningTokens: 2 },
247
+ },
248
+ },
249
+ ])
250
+
251
+ expect(events).toMatchObject([
252
+ { type: "step-start", index: 0 },
253
+ { type: "text-start", id: "text-1", providerMetadata: metadata },
254
+ { type: "text-delta", id: "text-1", text: "Hel", providerMetadata: { openai: { delta: 1 } } },
255
+ { type: "text-delta", id: "text-1", text: "lo", providerMetadata: { openai: { delta: 2 } } },
256
+ { type: "text-end", id: "text-1", providerMetadata: { openai: { done: true } } },
257
+ { type: "reasoning-start", id: "reasoning-1", providerMetadata: metadata },
258
+ { type: "reasoning-delta", id: "reasoning-1", text: "Think", providerMetadata: { openai: { delta: 3 } } },
259
+ { type: "reasoning-end", id: "reasoning-1", providerMetadata: { openai: { done: true } } },
260
+ { type: "tool-input-start", id: "call-1", name: "lookup", providerMetadata: metadata },
261
+ { type: "tool-input-delta", id: "call-1", name: "lookup", text: '{"query":' },
262
+ { type: "tool-input-delta", id: "call-1", name: "lookup", text: '"weather"}' },
263
+ { type: "tool-input-end", id: "call-1", name: "lookup", providerMetadata: { openai: { inputDone: true } } },
264
+ {
265
+ type: "tool-call",
266
+ id: "call-1",
267
+ name: "lookup",
268
+ input: { query: "weather" },
269
+ providerExecuted: true,
270
+ providerMetadata: { openai: { called: true } },
271
+ },
272
+ {
273
+ type: "tool-result",
274
+ id: "call-1",
275
+ name: "lookup",
276
+ result: { type: "json", value: { title: "Lookup", output: "sunny", metadata: { ok: true } } },
277
+ providerExecuted: true,
278
+ providerMetadata: { openai: { result: true } },
279
+ },
280
+ {
281
+ type: "step-finish",
282
+ index: 0,
283
+ reason: "unknown",
284
+ usage: {
285
+ inputTokens: 10,
286
+ outputTokens: 5,
287
+ totalTokens: 15,
288
+ reasoningTokens: 1,
289
+ cacheReadInputTokens: 3,
290
+ cacheWriteInputTokens: 2,
291
+ },
292
+ providerMetadata: { openai: { step: true } },
293
+ },
294
+ {
295
+ type: "finish",
296
+ reason: "unknown",
297
+ usage: {
298
+ inputTokens: 11,
299
+ outputTokens: 6,
300
+ totalTokens: 17,
301
+ reasoningTokens: 2,
302
+ cacheReadInputTokens: 4,
303
+ },
304
+ },
305
+ ])
306
+ })
307
+
308
+ test("creates stable block ids when AI SDK omits them", async () => {
309
+ const events = await adapt([
310
+ uncheckedAdapterEvent({ type: "text-delta", text: "implicit text" }),
311
+ uncheckedAdapterEvent({ type: "text-end" }),
312
+ uncheckedAdapterEvent({ type: "reasoning-delta", text: "implicit reasoning" }),
313
+ uncheckedAdapterEvent({ type: "reasoning-end" }),
314
+ ])
315
+
316
+ expect(events).toMatchObject([
317
+ { type: "text-delta", id: "text-0", text: "implicit text" },
318
+ { type: "text-end", id: "text-0" },
319
+ { type: "reasoning-delta", id: "reasoning-0", text: "implicit reasoning" },
320
+ { type: "reasoning-end", id: "reasoning-0" },
321
+ ])
322
+ })
323
+
324
+ test("explicitly ignores non-session-visible AI SDK chunks", async () => {
325
+ expect(
326
+ await adapt([
327
+ uncheckedAdapterEvent({ type: "abort" }),
328
+ uncheckedAdapterEvent({ type: "source" }),
329
+ uncheckedAdapterEvent({ type: "file" }),
330
+ uncheckedAdapterEvent({ type: "raw" }),
331
+ uncheckedAdapterEvent({ type: "tool-output-denied" }),
332
+ uncheckedAdapterEvent({ type: "tool-approval-request" }),
333
+ ]),
334
+ ).toEqual([])
335
+ })
336
+
337
+ test("preserves tool-error cause", async () => {
338
+ const error = new PermissionV1.RejectedError()
339
+ const events = await Effect.runPromise(
340
+ LLMAISDK.toLLMEvents(LLMAISDK.adapterState(), {
341
+ type: "tool-error",
342
+ toolCallId: "call_123",
343
+ toolName: "bash",
344
+ input: {},
345
+ error,
346
+ }),
347
+ )
348
+
349
+ expect(events).toHaveLength(1)
350
+ expect(events[0]).toMatchObject({
351
+ type: "tool-error",
352
+ id: "call_123",
353
+ name: "bash",
354
+ message: error.message,
355
+ error,
356
+ })
357
+ })
358
+
359
+ test("emits undefined usage when every AI SDK usage field is missing", async () => {
360
+ // If every numeric field is undefined the translator should signal "no usage info"
361
+ // by emitting undefined, not by polluting the event with usage: {}. Downstream cost
362
+ // telemetry distinguishes "missing" from "zero," so emitting an empty object causes
363
+ // false positives ("usage was tracked, just empty") instead of correct nulls.
364
+ const events = await adapt([
365
+ {
366
+ type: "finish-step",
367
+ response: { id: "response-1", timestamp: new Date(0), modelId: "gpt-test" },
368
+ finishReason: "stop",
369
+ rawFinishReason: "stop",
370
+ providerMetadata: undefined,
371
+ usage: {
372
+ inputTokens: undefined,
373
+ outputTokens: undefined,
374
+ totalTokens: undefined,
375
+ reasoningTokens: undefined,
376
+ cachedInputTokens: undefined,
377
+ inputTokenDetails: { noCacheTokens: undefined, cacheReadTokens: undefined, cacheWriteTokens: undefined },
378
+ outputTokenDetails: { textTokens: undefined, reasoningTokens: undefined },
379
+ },
380
+ },
381
+ ])
382
+
383
+ expect(events).toHaveLength(1)
384
+ const stepFinish = events[0]
385
+ if (stepFinish.type !== "step-finish") throw new Error("expected step-finish")
386
+ expect(stepFinish.usage).toBeUndefined()
387
+ })
388
+
389
+ test("reuses adapter state cleanly across streams once finish has fired", async () => {
390
+ // adapterState() is meant to be per-stream, but the only thing finish currently clears
391
+ // is toolNames — step, text counters, and the current text/reasoning IDs all leak
392
+ // forward. A caller that reuses a state across two streams sees text-1/reasoning-1/
393
+ // step index 1 on the second stream's first events. The test pins the intended
394
+ // contract: after finish, the same state can be reused and starts fresh.
395
+ const state = LLMAISDK.adapterState()
396
+ const run = (events: ReadonlyArray<AISDKAdapterEvent>) =>
397
+ Effect.runPromise(
398
+ Effect.forEach(events, (event) => LLMAISDK.toLLMEvents(state, event)).pipe(Effect.map((items) => items.flat())),
399
+ )
400
+
401
+ await run([
402
+ { type: "start-step", request: {}, warnings: [] },
403
+ uncheckedAdapterEvent({ type: "text-delta", text: "first" }),
404
+ uncheckedAdapterEvent({ type: "text-end" }),
405
+ uncheckedAdapterEvent({ type: "reasoning-delta", text: "first reasoning" }),
406
+ uncheckedAdapterEvent({ type: "reasoning-end" }),
407
+ {
408
+ type: "finish-step",
409
+ response: { id: "r1", timestamp: new Date(0), modelId: "gpt-test" },
410
+ finishReason: "stop",
411
+ rawFinishReason: "stop",
412
+ providerMetadata: undefined,
413
+ usage: {
414
+ inputTokens: 1,
415
+ outputTokens: 1,
416
+ totalTokens: 2,
417
+ inputTokenDetails: { noCacheTokens: undefined, cacheReadTokens: undefined, cacheWriteTokens: undefined },
418
+ outputTokenDetails: { textTokens: undefined, reasoningTokens: undefined },
419
+ },
420
+ },
421
+ {
422
+ type: "finish",
423
+ finishReason: "stop",
424
+ rawFinishReason: "stop",
425
+ totalUsage: {
426
+ inputTokens: 1,
427
+ outputTokens: 1,
428
+ totalTokens: 2,
429
+ inputTokenDetails: { noCacheTokens: undefined, cacheReadTokens: undefined, cacheWriteTokens: undefined },
430
+ outputTokenDetails: { textTokens: undefined, reasoningTokens: undefined },
431
+ },
432
+ },
433
+ ])
434
+
435
+ const secondStream = await run([
436
+ { type: "start-step", request: {}, warnings: [] },
437
+ uncheckedAdapterEvent({ type: "text-delta", text: "second" }),
438
+ uncheckedAdapterEvent({ type: "text-end" }),
439
+ uncheckedAdapterEvent({ type: "reasoning-delta", text: "second reasoning" }),
440
+ uncheckedAdapterEvent({ type: "reasoning-end" }),
441
+ ])
442
+
443
+ expect(secondStream).toMatchObject([
444
+ { type: "step-start", index: 0 },
445
+ { type: "text-delta", id: "text-0", text: "second" },
446
+ { type: "text-end", id: "text-0" },
447
+ { type: "reasoning-delta", id: "reasoning-0", text: "second reasoning" },
448
+ { type: "reasoning-end", id: "reasoning-0" },
449
+ ])
450
+ })
451
+
452
+ // Anthropic emits cache write counts in providerMetadata.anthropic.cacheCreationInputTokens
453
+ // rather than usage.inputTokenDetails.cacheWriteTokens. Session.getUsage falls back to the
454
+ // metadata path — but only if the adapter preserves providerMetadata on step-finish.
455
+ test("preserves providerMetadata on step-finish so Anthropic cache writes survive getUsage", async () => {
456
+ const events = await adapt([
457
+ {
458
+ type: "finish-step",
459
+ response: { id: "msg_test", timestamp: new Date(0), modelId: "claude-3-5-sonnet" },
460
+ finishReason: "stop",
461
+ rawFinishReason: "stop",
462
+ // Anthropic's AI SDK shape: cacheWriteTokens is NOT in usage, it arrives via providerMetadata.
463
+ usage: {
464
+ inputTokens: 1000,
465
+ outputTokens: 500,
466
+ totalTokens: 1500,
467
+ inputTokenDetails: { noCacheTokens: 800, cacheReadTokens: 200, cacheWriteTokens: undefined },
468
+ outputTokenDetails: { textTokens: 500, reasoningTokens: undefined },
469
+ },
470
+ providerMetadata: { anthropic: { cacheCreationInputTokens: 300 } },
471
+ },
472
+ ])
473
+
474
+ expect(events).toHaveLength(1)
475
+ const stepFinish = events[0]
476
+ if (stepFinish.type !== "step-finish") throw new Error("expected step-finish")
477
+ expect(stepFinish.providerMetadata).toEqual({ anthropic: { cacheCreationInputTokens: 300 } })
478
+ expect(stepFinish.usage?.cacheWriteInputTokens).toBeUndefined()
479
+ expect(stepFinish.usage?.cacheReadInputTokens).toBe(200)
480
+
481
+ // End-to-end: with the metadata preserved, getUsage extracts cache.write from the fallback path.
482
+ const result = SessionNs.getUsage({
483
+ model: {
484
+ id: "claude-3-5-sonnet",
485
+ providerID: "anthropic",
486
+ name: "Claude",
487
+ limit: { context: 200_000, output: 8_000 },
488
+ cost: { input: 0, output: 0, cache: { read: 0, write: 0 } },
489
+ capabilities: {
490
+ toolcall: true,
491
+ attachment: false,
492
+ reasoning: false,
493
+ temperature: true,
494
+ input: { text: true, image: false, audio: false, video: false },
495
+ output: { text: true, image: false, audio: false, video: false },
496
+ },
497
+ api: { npm: "@ai-sdk/anthropic" },
498
+ options: {},
499
+ } as never,
500
+ usage: stepFinish.usage!,
501
+ metadata: stepFinish.providerMetadata,
502
+ })
503
+ expect(result.tokens.cache.write).toBe(300)
504
+ expect(result.tokens.cache.read).toBe(200)
505
+ })
506
+
507
+ test("captures Copilot billed usage from raw Anthropic message deltas per step", async () => {
508
+ const events = await adapt([
509
+ uncheckedAdapterEvent({
510
+ type: "raw",
511
+ rawValue: {
512
+ type: "message_delta",
513
+ copilot_usage: { total_nano_aiu: 4_473_525_000 },
514
+ },
515
+ }),
516
+ {
517
+ type: "finish-step",
518
+ response: { id: "msg_test", timestamp: new Date(0), modelId: "claude-sonnet-4.6" },
519
+ finishReason: "stop",
520
+ rawFinishReason: "end_turn",
521
+ usage: {
522
+ inputTokens: 11_774,
523
+ outputTokens: 39,
524
+ totalTokens: 11_813,
525
+ inputTokenDetails: { noCacheTokens: 3, cacheReadTokens: 0, cacheWriteTokens: 11_771 },
526
+ outputTokenDetails: { textTokens: 39, reasoningTokens: undefined },
527
+ },
528
+ providerMetadata: { anthropic: { cacheCreationInputTokens: 11_771 } },
529
+ },
530
+ {
531
+ type: "finish-step",
532
+ response: { id: "msg_follow_up", timestamp: new Date(0), modelId: "claude-sonnet-4.6" },
533
+ finishReason: "stop",
534
+ rawFinishReason: "end_turn",
535
+ usage: {
536
+ inputTokens: 1,
537
+ outputTokens: 1,
538
+ totalTokens: 2,
539
+ inputTokenDetails: { noCacheTokens: 1, cacheReadTokens: 0, cacheWriteTokens: 0 },
540
+ outputTokenDetails: { textTokens: 1, reasoningTokens: undefined },
541
+ },
542
+ providerMetadata: { anthropic: {} },
543
+ },
544
+ ])
545
+
546
+ expect(events[0]).toMatchObject({
547
+ type: "step-finish",
548
+ providerMetadata: {
549
+ anthropic: { cacheCreationInputTokens: 11_771 },
550
+ copilot: { totalNanoAiu: 4_473_525_000 },
551
+ },
552
+ })
553
+ expect(events[1]).toMatchObject({ type: "step-finish", providerMetadata: { anthropic: {} } })
554
+ if (events[1].type !== "step-finish") throw new Error("expected step-finish")
555
+ expect(events[1].providerMetadata?.copilot).toBeUndefined()
556
+ })
557
+ })
558
+
559
+ type Capture = {
560
+ url: URL
561
+ headers: Headers
562
+ body: Record<string, unknown>
563
+ }
564
+
565
+ const state = {
566
+ server: null as ReturnType<typeof Bun.serve> | null,
567
+ queue: [] as Array<{
568
+ path: string
569
+ response: Response | ((req: Request, capture: Capture) => Response)
570
+ resolve: (value: Capture) => void
571
+ }>,
572
+ }
573
+
574
+ function deferred<T>() {
575
+ const result = {} as { promise: Promise<T>; resolve: (value: T) => void }
576
+ result.promise = new Promise((resolve) => {
577
+ result.resolve = resolve
578
+ })
579
+ return result
580
+ }
581
+
582
+ function waitRequest(pathname: string, response: Response) {
583
+ const pending = deferred<Capture>()
584
+ state.queue.push({ path: pathname, response, resolve: pending.resolve })
585
+ return pending.promise
586
+ }
587
+
588
+ function timeout(ms: number) {
589
+ return new Promise<never>((_, reject) => {
590
+ setTimeout(() => reject(new Error(`timed out after ${ms}ms`)), ms)
591
+ })
592
+ }
593
+
594
+ function waitStreamingRequest(pathname: string) {
595
+ const request = deferred<Capture>()
596
+ const requestAborted = deferred<void>()
597
+ const responseCanceled = deferred<void>()
598
+ const encoder = new TextEncoder()
599
+
600
+ state.queue.push({
601
+ path: pathname,
602
+ resolve: request.resolve,
603
+ response(req: Request) {
604
+ req.signal.addEventListener("abort", () => requestAborted.resolve(), { once: true })
605
+
606
+ return new Response(
607
+ new ReadableStream<Uint8Array>({
608
+ start(controller) {
609
+ controller.enqueue(
610
+ encoder.encode(
611
+ [
612
+ `data: ${JSON.stringify({
613
+ id: "chatcmpl-abort",
614
+ object: "chat.completion.chunk",
615
+ choices: [{ delta: { role: "assistant" } }],
616
+ })}`,
617
+ ].join("\n\n") + "\n\n",
618
+ ),
619
+ )
620
+ },
621
+ cancel() {
622
+ responseCanceled.resolve()
623
+ },
624
+ }),
625
+ {
626
+ status: 200,
627
+ headers: { "Content-Type": "text/event-stream" },
628
+ },
629
+ )
630
+ },
631
+ })
632
+
633
+ return {
634
+ request: request.promise,
635
+ requestAborted: requestAborted.promise,
636
+ responseCanceled: responseCanceled.promise,
637
+ }
638
+ }
639
+
640
+ beforeAll(() => {
641
+ state.server = Bun.serve({
642
+ port: 0,
643
+ async fetch(req) {
644
+ const next = state.queue.shift()
645
+ if (!next) {
646
+ return new Response("unexpected request", { status: 500 })
647
+ }
648
+
649
+ const url = new URL(req.url)
650
+ const body = (await req.json()) as Record<string, unknown>
651
+ next.resolve({ url, headers: req.headers, body })
652
+
653
+ if (!url.pathname.endsWith(next.path)) {
654
+ return new Response("not found", { status: 404 })
655
+ }
656
+
657
+ return typeof next.response === "function"
658
+ ? next.response(req, { url, headers: req.headers, body })
659
+ : next.response
660
+ },
661
+ })
662
+ })
663
+
664
+ beforeEach(() => {
665
+ state.queue.length = 0
666
+ })
667
+
668
+ afterAll(() => {
669
+ void state.server?.stop()
670
+ })
671
+
672
+ function createChatStream(text: string) {
673
+ const payload =
674
+ [
675
+ `data: ${JSON.stringify({
676
+ id: "chatcmpl-1",
677
+ object: "chat.completion.chunk",
678
+ choices: [{ delta: { role: "assistant" } }],
679
+ })}`,
680
+ `data: ${JSON.stringify({
681
+ id: "chatcmpl-1",
682
+ object: "chat.completion.chunk",
683
+ choices: [{ delta: { content: text } }],
684
+ })}`,
685
+ `data: ${JSON.stringify({
686
+ id: "chatcmpl-1",
687
+ object: "chat.completion.chunk",
688
+ choices: [{ delta: {}, finish_reason: "stop" }],
689
+ })}`,
690
+ "data: [DONE]",
691
+ ].join("\n\n") + "\n\n"
692
+
693
+ const encoder = new TextEncoder()
694
+ return new ReadableStream<Uint8Array>({
695
+ start(controller) {
696
+ controller.enqueue(encoder.encode(payload))
697
+ controller.close()
698
+ },
699
+ })
700
+ }
701
+
702
+ const MODELS_FIXTURE = JSON.parse(
703
+ await Bun.file(path.join(import.meta.dir, "../tool/fixtures/models-api.json")).text(),
704
+ ) as Record<string, ModelsDev.Provider>
705
+
706
+ function loadFixture(providerID: string, modelID: string) {
707
+ const provider = MODELS_FIXTURE[providerID]
708
+ if (!provider) throw new Error(`Missing provider in fixture: ${providerID}`)
709
+ const model = provider.models[modelID]
710
+ if (!model) throw new Error(`Missing model in fixture: ${modelID}`)
711
+ return { provider, model }
712
+ }
713
+
714
+ function configModel(model: ModelsDev.Model) {
715
+ return {
716
+ id: model.id,
717
+ name: model.name,
718
+ family: model.family,
719
+ release_date: model.release_date,
720
+ attachment: model.attachment,
721
+ reasoning: model.reasoning,
722
+ temperature: model.temperature,
723
+ tool_call: model.tool_call,
724
+ interleaved: model.interleaved,
725
+ cost: model.cost ? { ...model.cost, tiers: undefined } : undefined,
726
+ limit: model.limit,
727
+ modalities: model.modalities,
728
+ status: model.status,
729
+ provider: model.provider,
730
+ }
731
+ }
732
+
733
+ function createEventStream(chunks: unknown[], includeDone = false) {
734
+ const lines = chunks.map((chunk) => `data: ${typeof chunk === "string" ? chunk : JSON.stringify(chunk)}`)
735
+ if (includeDone) {
736
+ lines.push("data: [DONE]")
737
+ }
738
+ const payload = lines.join("\n\n") + "\n\n"
739
+ const encoder = new TextEncoder()
740
+ return new ReadableStream<Uint8Array>({
741
+ start(controller) {
742
+ controller.enqueue(encoder.encode(payload))
743
+ controller.close()
744
+ },
745
+ })
746
+ }
747
+
748
+ function createEventResponse(chunks: unknown[], includeDone = false) {
749
+ return new Response(createEventStream(chunks, includeDone), {
750
+ status: 200,
751
+ headers: { "Content-Type": "text/event-stream" },
752
+ })
753
+ }
754
+
755
+ describe("session.llm.stream", () => {
756
+ const vivgridFixture = { providerID: "vivgrid", modelID: "gemini-3.1-pro-preview" }
757
+ it.instance(
758
+ "sends temperature, tokens, and reasoning options for openai-compatible models",
759
+ () =>
760
+ Effect.gen(function* () {
761
+ const fixture = loadFixture(vivgridFixture.providerID, vivgridFixture.modelID)
762
+ const request = waitRequest(
763
+ "/chat/completions",
764
+ new Response(createChatStream("Hello"), {
765
+ status: 200,
766
+ headers: { "Content-Type": "text/event-stream" },
767
+ }),
768
+ )
769
+
770
+ const resolved = yield* Provider.use.getModel(
771
+ ProviderV2.ID.make(vivgridFixture.providerID),
772
+ ModelV2.ID.make(fixture.model.id),
773
+ )
774
+ const sessionID = SessionID.make("session-test-1")
775
+ const agent = {
776
+ name: "test",
777
+ mode: "primary",
778
+ options: {},
779
+ permission: [{ permission: "*", pattern: "*", action: "allow" }],
780
+ temperature: 0.4,
781
+ topP: 0.8,
782
+ } satisfies Agent.Info
783
+
784
+ const user = {
785
+ id: MessageID.make("msg_user-1"),
786
+ sessionID,
787
+ role: "user",
788
+ time: { created: Date.now() },
789
+ agent: agent.name,
790
+ model: { providerID: ProviderV2.ID.make(vivgridFixture.providerID), modelID: resolved.id, variant: "high" },
791
+ } satisfies SessionV1.User
792
+
793
+ yield* drain({
794
+ user,
795
+ sessionID,
796
+ model: resolved,
797
+ agent,
798
+ system: ["You are a helpful assistant."],
799
+ messages: [{ role: "user", content: "Hello" }],
800
+ tools: {},
801
+ })
802
+
803
+ const capture = yield* Effect.promise(() => request)
804
+ const body = capture.body
805
+ const headers = capture.headers
806
+ const url = capture.url
807
+
808
+ expect(url.pathname.startsWith("/v1/")).toBe(true)
809
+ expect(url.pathname.endsWith("/chat/completions")).toBe(true)
810
+ expect(headers.get("Authorization")).toBe("Bearer test-key")
811
+
812
+ expect(body.model).toBe(resolved.api.id)
813
+ expect(body.temperature).toBe(0.4)
814
+ expect(body.top_p).toBe(0.8)
815
+ expect(body.stream).toBe(true)
816
+
817
+ const maxTokens = (body.max_tokens as number | undefined) ?? (body.max_output_tokens as number | undefined)
818
+ const expectedMaxTokens = ProviderTransform.maxOutputTokens(resolved)
819
+ expect(maxTokens).toBe(expectedMaxTokens)
820
+
821
+ const reasoning = (body.reasoningEffort as string | undefined) ?? (body.reasoning_effort as string | undefined)
822
+ expect(reasoning).toBe("high")
823
+ }),
824
+ {
825
+ config: () => ({
826
+ enabled_providers: [vivgridFixture.providerID],
827
+ provider: {
828
+ [vivgridFixture.providerID]: {
829
+ options: { apiKey: "test-key", baseURL: `${state.server!.url.origin}/v1` },
830
+ },
831
+ },
832
+ }),
833
+ },
834
+ )
835
+
836
+ it.instance(
837
+ "surfaces network_error finish reasons as retryable stream failures",
838
+ () =>
839
+ Effect.gen(function* () {
840
+ const fixture = loadFixture(vivgridFixture.providerID, vivgridFixture.modelID)
841
+ const request = waitRequest(
842
+ "/chat/completions",
843
+ createEventResponse(
844
+ [
845
+ {
846
+ id: "chatcmpl-network-error",
847
+ object: "chat.completion.chunk",
848
+ choices: [{ index: 0, delta: { role: "assistant", content: "" }, finish_reason: "network_error" }],
849
+ },
850
+ ],
851
+ true,
852
+ ),
853
+ )
854
+ const resolved = yield* Provider.use.getModel(
855
+ ProviderV2.ID.make(vivgridFixture.providerID),
856
+ ModelV2.ID.make(fixture.model.id),
857
+ )
858
+ const sessionID = SessionID.make("session-test-network-error")
859
+ const agent = {
860
+ name: "test",
861
+ mode: "primary",
862
+ options: {},
863
+ permission: [{ permission: "*", pattern: "*", action: "allow" }],
864
+ } satisfies Agent.Info
865
+ const user = {
866
+ id: MessageID.make("msg_user-network-error"),
867
+ sessionID,
868
+ role: "user",
869
+ time: { created: Date.now() },
870
+ agent: agent.name,
871
+ model: { providerID: ProviderV2.ID.make(vivgridFixture.providerID), modelID: resolved.id },
872
+ } satisfies SessionV1.User
873
+
874
+ const error = yield* drain({
875
+ user,
876
+ sessionID,
877
+ model: resolved,
878
+ agent,
879
+ system: ["You are a helpful assistant."],
880
+ messages: [{ role: "user", content: "Hello" }],
881
+ tools: {},
882
+ }).pipe(Effect.flip)
883
+ yield* Effect.promise(() => request)
884
+
885
+ if (!(error instanceof ProviderError.ResponseStreamError)) throw error
886
+ expect(error.message).toBe("Provider finish_reason: network_error")
887
+ }),
888
+ {
889
+ config: () => ({
890
+ enabled_providers: [vivgridFixture.providerID],
891
+ provider: {
892
+ [vivgridFixture.providerID]: {
893
+ options: { apiKey: "test-key", baseURL: `${state.server!.url.origin}/v1` },
894
+ },
895
+ },
896
+ }),
897
+ },
898
+ )
899
+
900
+ const cerebrasFixture = { providerID: "cerebras", modelID: "gpt-oss-120b" }
901
+ it.instance(
902
+ "replays Cerebras assistant reasoning using the provider-supported field",
903
+ () =>
904
+ Effect.gen(function* () {
905
+ const fixture = loadFixture(cerebrasFixture.providerID, cerebrasFixture.modelID)
906
+ const request = waitRequest(
907
+ "/chat/completions",
908
+ new Response(createChatStream("Hello"), {
909
+ status: 200,
910
+ headers: { "Content-Type": "text/event-stream" },
911
+ }),
912
+ )
913
+
914
+ const resolved = yield* Provider.use.getModel(
915
+ ProviderV2.ID.make(cerebrasFixture.providerID),
916
+ ModelV2.ID.make(fixture.model.id),
917
+ )
918
+ const sessionID = SessionID.make("session-test-cerebras-reasoning")
919
+ const agent = {
920
+ name: "test",
921
+ mode: "primary",
922
+ options: {},
923
+ permission: [{ permission: "*", pattern: "*", action: "allow" }],
924
+ } satisfies Agent.Info
925
+
926
+ const user = {
927
+ id: MessageID.make("msg_user-cerebras-reasoning"),
928
+ sessionID,
929
+ role: "user",
930
+ time: { created: Date.now() },
931
+ agent: agent.name,
932
+ model: { providerID: ProviderV2.ID.make(cerebrasFixture.providerID), modelID: resolved.id },
933
+ } satisfies SessionV1.User
934
+
935
+ yield* drain({
936
+ user,
937
+ sessionID,
938
+ model: resolved,
939
+ agent,
940
+ system: ["You are a helpful assistant."],
941
+ messages: [
942
+ { role: "user", content: "Hello" },
943
+ {
944
+ role: "assistant",
945
+ content: [
946
+ { type: "reasoning", text: "thinking" },
947
+ { type: "text", text: "Previous answer" },
948
+ ],
949
+ },
950
+ { role: "user", content: "Continue" },
951
+ ] satisfies ModelMessage[],
952
+ tools: {},
953
+ })
954
+
955
+ const capture = yield* Effect.promise(() => request)
956
+ const messages = capture.body.messages as Array<Record<string, unknown>>
957
+ const assistant = messages.find((msg) => msg.role === "assistant")
958
+
959
+ expect(assistant?.reasoning).toBe("thinking")
960
+ expect(assistant && "reasoning_content" in assistant).toBe(false)
961
+ }),
962
+ {
963
+ config: () => ({
964
+ enabled_providers: [cerebrasFixture.providerID],
965
+ provider: {
966
+ [cerebrasFixture.providerID]: {
967
+ options: { apiKey: "test-key", baseURL: `${state.server!.url.origin}/v1` },
968
+ },
969
+ },
970
+ }),
971
+ },
972
+ )
973
+
974
+ const mistralFixture = { providerID: "mistral", modelID: "mistral-small-latest" }
975
+ it.instance(
976
+ "replays native Mistral reasoning from chat history",
977
+ () =>
978
+ Effect.gen(function* () {
979
+ const fixture = loadFixture(mistralFixture.providerID, mistralFixture.modelID)
980
+ const request = waitRequest(
981
+ "/chat/completions",
982
+ createEventResponse(
983
+ [
984
+ {
985
+ id: "chatcmpl-mistral",
986
+ object: "chat.completion.chunk",
987
+ created: 0,
988
+ model: fixture.model.id,
989
+ choices: [{ index: 0, delta: { role: "assistant", content: "Hello" } }],
990
+ },
991
+ {
992
+ id: "chatcmpl-mistral",
993
+ object: "chat.completion.chunk",
994
+ created: 0,
995
+ model: fixture.model.id,
996
+ choices: [{ index: 0, delta: {}, finish_reason: "stop" }],
997
+ },
998
+ ],
999
+ true,
1000
+ ),
1001
+ )
1002
+
1003
+ const resolved = yield* Provider.use.getModel(
1004
+ ProviderV2.ID.make(mistralFixture.providerID),
1005
+ ModelV2.ID.make(fixture.model.id),
1006
+ )
1007
+ const sessionID = SessionID.make("session-test-mistral-reasoning")
1008
+ const agent = {
1009
+ name: "test",
1010
+ mode: "primary",
1011
+ options: {},
1012
+ permission: [{ permission: "*", pattern: "*", action: "allow" }],
1013
+ } satisfies Agent.Info
1014
+
1015
+ const user = {
1016
+ id: MessageID.make("msg_user-mistral-reasoning"),
1017
+ sessionID,
1018
+ role: "user",
1019
+ time: { created: Date.now() },
1020
+ agent: agent.name,
1021
+ model: { providerID: ProviderV2.ID.make(mistralFixture.providerID), modelID: resolved.id },
1022
+ } satisfies SessionV1.User
1023
+
1024
+ const thinking = {
1025
+ type: "thinking",
1026
+ thinking: [
1027
+ { type: "text", text: "thinking" },
1028
+ {
1029
+ type: "tool_reference",
1030
+ tool: "web_search",
1031
+ title: "Example result",
1032
+ url: "https://example.com/tool",
1033
+ favicon: "https://example.com/favicon.ico",
1034
+ description: "Example description",
1035
+ },
1036
+ { type: "reference", reference_ids: [1, "source-2"] },
1037
+ ],
1038
+ closed: true,
1039
+ signature: "sig-123",
1040
+ }
1041
+
1042
+ yield* drain({
1043
+ user,
1044
+ sessionID,
1045
+ model: resolved,
1046
+ agent,
1047
+ system: ["You are a helpful assistant."],
1048
+ messages: [
1049
+ { role: "user", content: "Hello" },
1050
+ {
1051
+ role: "assistant",
1052
+ content: [
1053
+ {
1054
+ type: "reasoning",
1055
+ text: "thinking",
1056
+ providerOptions: { mistral: { thinking } },
1057
+ },
1058
+ { type: "text", text: "Previous answer" },
1059
+ ],
1060
+ },
1061
+ { role: "user", content: "Continue" },
1062
+ ] satisfies ModelMessage[],
1063
+ tools: {},
1064
+ })
1065
+
1066
+ const capture = yield* Effect.promise(() => request)
1067
+ const messages = capture.body.messages as Array<Record<string, unknown>>
1068
+ expect(messages.find((message) => message.role === "assistant")).toEqual({
1069
+ role: "assistant",
1070
+ content: [thinking, { type: "text", text: "Previous answer" }],
1071
+ })
1072
+ }),
1073
+ {
1074
+ config: () => ({
1075
+ enabled_providers: [mistralFixture.providerID],
1076
+ provider: {
1077
+ [mistralFixture.providerID]: {
1078
+ options: { apiKey: "test-key", baseURL: `${state.server!.url.origin}/v1` },
1079
+ },
1080
+ },
1081
+ }),
1082
+ },
1083
+ )
1084
+
1085
+ const alibabaQwenFixture = { providerID: "alibaba", modelID: "qwen-plus" }
1086
+ it.instance(
1087
+ "service stream cancellation cancels provider response body promptly",
1088
+ () =>
1089
+ Effect.gen(function* () {
1090
+ const fixture = loadFixture(alibabaQwenFixture.providerID, alibabaQwenFixture.modelID)
1091
+ const pending = waitStreamingRequest("/chat/completions")
1092
+
1093
+ const resolved = yield* Provider.use.getModel(
1094
+ ProviderV2.ID.make(alibabaQwenFixture.providerID),
1095
+ ModelV2.ID.make(fixture.model.id),
1096
+ )
1097
+ const sessionID = SessionID.make("session-test-service-abort")
1098
+ const agent = {
1099
+ name: "test",
1100
+ mode: "primary",
1101
+ options: {},
1102
+ permission: [{ permission: "*", pattern: "*", action: "allow" }],
1103
+ } satisfies Agent.Info
1104
+ const user = {
1105
+ id: MessageID.make("msg_user-service-abort"),
1106
+ sessionID,
1107
+ role: "user",
1108
+ time: { created: Date.now() },
1109
+ agent: agent.name,
1110
+ model: { providerID: ProviderV2.ID.make(alibabaQwenFixture.providerID), modelID: resolved.id },
1111
+ } satisfies SessionV1.User
1112
+
1113
+ const fiber = yield* drain({
1114
+ user,
1115
+ sessionID,
1116
+ model: resolved,
1117
+ agent,
1118
+ system: ["You are a helpful assistant."],
1119
+ messages: [{ role: "user", content: "Hello" }],
1120
+ tools: {},
1121
+ }).pipe(Effect.exit, Effect.forkScoped)
1122
+
1123
+ yield* Effect.promise(() => pending.request)
1124
+ yield* Fiber.interrupt(fiber)
1125
+
1126
+ yield* Effect.promise(() => Promise.race([pending.responseCanceled, timeout(500)]))
1127
+ const exit = yield* Fiber.await(fiber)
1128
+ // Fiber.await returns an Exit<Exit<...>>. Unwrap once.
1129
+ const inner = Exit.isSuccess(exit) ? exit.value : exit
1130
+ expect(Exit.isFailure(inner)).toBe(true)
1131
+ if (Exit.isFailure(inner)) {
1132
+ expect(Cause.hasInterrupts(inner.cause)).toBe(true)
1133
+ }
1134
+ yield* Effect.promise(() => Promise.race([pending.requestAborted, timeout(500)]).catch(() => undefined))
1135
+ }),
1136
+ {
1137
+ config: () => ({
1138
+ enabled_providers: [alibabaQwenFixture.providerID],
1139
+ provider: {
1140
+ [alibabaQwenFixture.providerID]: {
1141
+ options: { apiKey: "test-key", baseURL: `${state.server!.url.origin}/v1` },
1142
+ },
1143
+ },
1144
+ }),
1145
+ },
1146
+ )
1147
+
1148
+ it.instance(
1149
+ "keeps tools enabled by prompt permissions",
1150
+ () =>
1151
+ Effect.gen(function* () {
1152
+ const fixture = loadFixture(alibabaQwenFixture.providerID, alibabaQwenFixture.modelID)
1153
+ const request = waitRequest(
1154
+ "/chat/completions",
1155
+ new Response(createChatStream("Hello"), {
1156
+ status: 200,
1157
+ headers: { "Content-Type": "text/event-stream" },
1158
+ }),
1159
+ )
1160
+
1161
+ const resolved = yield* Provider.use.getModel(
1162
+ ProviderV2.ID.make(alibabaQwenFixture.providerID),
1163
+ ModelV2.ID.make(fixture.model.id),
1164
+ )
1165
+ const sessionID = SessionID.make("session-test-tools")
1166
+ const agent = {
1167
+ name: "test",
1168
+ mode: "primary",
1169
+ options: {},
1170
+ permission: [{ permission: "question", pattern: "*", action: "deny" }],
1171
+ } satisfies Agent.Info
1172
+
1173
+ const user = {
1174
+ id: MessageID.make("msg_user-tools"),
1175
+ sessionID,
1176
+ role: "user",
1177
+ time: { created: Date.now() },
1178
+ agent: agent.name,
1179
+ model: { providerID: ProviderV2.ID.make(alibabaQwenFixture.providerID), modelID: resolved.id },
1180
+ tools: { question: true },
1181
+ } satisfies SessionV1.User
1182
+
1183
+ yield* drain({
1184
+ user,
1185
+ sessionID,
1186
+ model: resolved,
1187
+ agent,
1188
+ permission: [{ permission: "question", pattern: "*", action: "allow" }],
1189
+ system: ["You are a helpful assistant."],
1190
+ messages: [{ role: "user", content: "Hello" }],
1191
+ tools: {
1192
+ question: tool({
1193
+ description: "Ask a question",
1194
+ inputSchema: z.object({}),
1195
+ execute: async () => ({ output: "" }),
1196
+ }),
1197
+ },
1198
+ })
1199
+
1200
+ const capture = yield* Effect.promise(() => request)
1201
+ const tools = capture.body.tools as Array<{ function?: { name?: string } }> | undefined
1202
+ expect(tools?.some((item) => item.function?.name === "question")).toBe(true)
1203
+ }),
1204
+ {
1205
+ config: () => ({
1206
+ enabled_providers: [alibabaQwenFixture.providerID],
1207
+ provider: {
1208
+ [alibabaQwenFixture.providerID]: {
1209
+ options: { apiKey: "test-key", baseURL: `${state.server!.url.origin}/v1` },
1210
+ },
1211
+ },
1212
+ }),
1213
+ },
1214
+ )
1215
+
1216
+ it.instance(
1217
+ "sends responses API payload for OpenAI models",
1218
+ () =>
1219
+ Effect.gen(function* () {
1220
+ const model = loadFixture("openai", "gpt-5.2").model
1221
+
1222
+ const responseChunks = [
1223
+ {
1224
+ type: "response.created",
1225
+ response: {
1226
+ id: "resp-1",
1227
+ created_at: Math.floor(Date.now() / 1000),
1228
+ model: model.id,
1229
+ service_tier: null,
1230
+ },
1231
+ },
1232
+ {
1233
+ type: "response.output_item.added",
1234
+ output_index: 0,
1235
+ item: { type: "message", id: "item-1", status: "in_progress", role: "assistant", content: [] },
1236
+ },
1237
+ {
1238
+ type: "response.content_part.added",
1239
+ item_id: "item-1",
1240
+ output_index: 0,
1241
+ content_index: 0,
1242
+ part: { type: "output_text", text: "", annotations: [] },
1243
+ },
1244
+ {
1245
+ type: "response.output_text.delta",
1246
+ item_id: "item-1",
1247
+ delta: "Hello",
1248
+ logprobs: null,
1249
+ },
1250
+ {
1251
+ type: "response.completed",
1252
+ response: {
1253
+ incomplete_details: null,
1254
+ usage: {
1255
+ input_tokens: 1,
1256
+ input_tokens_details: null,
1257
+ output_tokens: 1,
1258
+ output_tokens_details: null,
1259
+ },
1260
+ service_tier: null,
1261
+ },
1262
+ },
1263
+ ]
1264
+ const request = waitRequest("/responses", createEventResponse(responseChunks, true))
1265
+
1266
+ const resolved = yield* Provider.use.getModel(ProviderV2.ID.openai, ModelV2.ID.make(model.id))
1267
+ const sessionID = SessionID.make("session-test-2")
1268
+ const agent = {
1269
+ name: "test",
1270
+ mode: "primary",
1271
+ options: { reasoningMode: "pro" },
1272
+ permission: [{ permission: "*", pattern: "*", action: "allow" }],
1273
+ temperature: 0.2,
1274
+ } satisfies Agent.Info
1275
+
1276
+ const user = {
1277
+ id: MessageID.make("msg_user-2"),
1278
+ sessionID,
1279
+ role: "user",
1280
+ time: { created: Date.now() },
1281
+ agent: agent.name,
1282
+ model: { providerID: ProviderV2.ID.make("openai"), modelID: resolved.id, variant: "high" },
1283
+ } satisfies SessionV1.User
1284
+
1285
+ yield* drain({
1286
+ user,
1287
+ sessionID,
1288
+ model: resolved,
1289
+ agent,
1290
+ system: ["You are a helpful assistant."],
1291
+ messages: [{ role: "user", content: "Hello" }],
1292
+ tools: {},
1293
+ })
1294
+
1295
+ const capture = yield* Effect.promise(() => request)
1296
+ const body = capture.body
1297
+
1298
+ expect(capture.url.pathname.endsWith("/responses")).toBe(true)
1299
+ expect(body.model).toBe(resolved.api.id)
1300
+ expect(body.stream).toBe(true)
1301
+ expect((body.reasoning as { effort?: string } | undefined)?.effort).toBe("high")
1302
+ expect((body.reasoning as { mode?: string } | undefined)?.mode).toBe("pro")
1303
+
1304
+ const maxTokens = body.max_output_tokens as number | undefined
1305
+ expect(maxTokens).toBe(undefined) // match codex cli behavior
1306
+ }),
1307
+ { config: () => openAIConfig(loadFixture("openai", "gpt-5.2").model, `${state.server!.url.origin}/v1`) },
1308
+ )
1309
+
1310
+ it.instance(
1311
+ "keeps supported OpenAI models on AI SDK path when native flag is off",
1312
+ () =>
1313
+ Effect.gen(function* () {
1314
+ const model = loadFixture("openai", "gpt-5.2").model
1315
+ const request = waitRequest(
1316
+ "/responses",
1317
+ createEventResponse(
1318
+ [
1319
+ {
1320
+ type: "response.created",
1321
+ response: {
1322
+ id: "resp-flag-off",
1323
+ created_at: Math.floor(Date.now() / 1000),
1324
+ model: model.id,
1325
+ service_tier: null,
1326
+ },
1327
+ },
1328
+ {
1329
+ type: "response.output_item.added",
1330
+ output_index: 0,
1331
+ item: { type: "message", id: "item-flag-off", status: "in_progress", role: "assistant", content: [] },
1332
+ },
1333
+ {
1334
+ type: "response.content_part.added",
1335
+ item_id: "item-flag-off",
1336
+ output_index: 0,
1337
+ content_index: 0,
1338
+ part: { type: "output_text", text: "", annotations: [] },
1339
+ },
1340
+ {
1341
+ type: "response.output_text.delta",
1342
+ item_id: "item-flag-off",
1343
+ delta: "Flag off",
1344
+ logprobs: null,
1345
+ },
1346
+ {
1347
+ type: "response.completed",
1348
+ response: {
1349
+ incomplete_details: null,
1350
+ usage: {
1351
+ input_tokens: 1,
1352
+ input_tokens_details: null,
1353
+ output_tokens: 1,
1354
+ output_tokens_details: null,
1355
+ },
1356
+ service_tier: null,
1357
+ },
1358
+ },
1359
+ ],
1360
+ true,
1361
+ ),
1362
+ )
1363
+ const failingNativeClient = Layer.succeed(
1364
+ LLMClient.Service,
1365
+ LLMClient.Service.of({
1366
+ prepare: () => Effect.die(new Error("native LLM client should not be used when the flag is off")),
1367
+ stream: () => Stream.die(new Error("native LLM client should not be used when the flag is off")),
1368
+ generate: () => Effect.die(new Error("native LLM client should not be used when the flag is off")),
1369
+ }),
1370
+ )
1371
+
1372
+ const resolved = yield* Provider.use.getModel(ProviderV2.ID.openai, ModelV2.ID.make(model.id))
1373
+ const sessionID = SessionID.make("session-test-native-flag-off")
1374
+ const agent = {
1375
+ name: "test",
1376
+ mode: "primary",
1377
+ options: {},
1378
+ permission: [{ permission: "*", pattern: "*", action: "allow" }],
1379
+ } satisfies Agent.Info
1380
+
1381
+ yield* drainWith(
1382
+ AppNodeBuilder.build(LLM.node, [
1383
+ [LayerNodePlatform.llmClient, failingNativeClient],
1384
+ [RuntimeFlags.node, RuntimeFlags.layer({ experimentalNativeLlm: false })],
1385
+ ]),
1386
+ {
1387
+ user: {
1388
+ id: MessageID.make("msg_user-native-flag-off"),
1389
+ sessionID,
1390
+ role: "user",
1391
+ time: { created: Date.now() },
1392
+ agent: agent.name,
1393
+ model: { providerID: ProviderV2.ID.make("openai"), modelID: resolved.id, variant: "high" },
1394
+ } satisfies SessionV1.User,
1395
+ sessionID,
1396
+ model: resolved,
1397
+ agent,
1398
+ system: ["You are a helpful assistant."],
1399
+ messages: [{ role: "user", content: "Hello" }],
1400
+ tools: {},
1401
+ },
1402
+ )
1403
+
1404
+ const capture = yield* Effect.promise(() => request)
1405
+ expect(capture.url.pathname.endsWith("/responses")).toBe(true)
1406
+ expect(capture.body.model).toBe(resolved.api.id)
1407
+ }),
1408
+ { config: () => openAIConfig(loadFixture("openai", "gpt-5.2").model, `${state.server!.url.origin}/v1`) },
1409
+ )
1410
+
1411
+ it.instance(
1412
+ "streams OpenAI through native runtime when opted in",
1413
+ () =>
1414
+ Effect.gen(function* () {
1415
+ const model = loadFixture("openai", "gpt-5.2").model
1416
+ const chunks = [
1417
+ { type: "response.created", response: { id: "resp-native" } },
1418
+ {
1419
+ type: "response.output_item.added",
1420
+ item: { type: "message", id: "item-native", status: "in_progress" },
1421
+ },
1422
+ { type: "response.output_text.delta", item_id: "item-native", delta: "Hello native" },
1423
+ {
1424
+ type: "response.completed",
1425
+ response: {
1426
+ incomplete_details: null,
1427
+ usage: {
1428
+ input_tokens: 1,
1429
+ input_tokens_details: null,
1430
+ output_tokens: 1,
1431
+ output_tokens_details: null,
1432
+ },
1433
+ },
1434
+ },
1435
+ ]
1436
+ const request = waitRequest("/responses", createEventResponse(chunks, true))
1437
+
1438
+ const resolved = yield* Provider.use.getModel(ProviderV2.ID.openai, ModelV2.ID.make(model.id))
1439
+ const sessionID = SessionID.make("session-test-native")
1440
+ const agent = {
1441
+ name: "test",
1442
+ mode: "primary",
1443
+ options: {},
1444
+ permission: [{ permission: "*", pattern: "*", action: "allow" }],
1445
+ temperature: 0.2,
1446
+ } satisfies Agent.Info
1447
+
1448
+ yield* drainWith(llmLayerWithExecutor({ flags: { experimentalNativeLlm: true } }), {
1449
+ user: {
1450
+ id: MessageID.make("msg_user-native"),
1451
+ sessionID,
1452
+ role: "user",
1453
+ time: { created: Date.now() },
1454
+ agent: agent.name,
1455
+ model: { providerID: ProviderV2.ID.make("openai"), modelID: resolved.id, variant: "high" },
1456
+ } satisfies SessionV1.User,
1457
+ sessionID,
1458
+ model: resolved,
1459
+ agent,
1460
+ system: ["You are a helpful assistant."],
1461
+ messages: [{ role: "user", content: "Hello" }],
1462
+ tools: {},
1463
+ })
1464
+
1465
+ const capture = yield* Effect.promise(() => request)
1466
+ expect(capture.url.pathname.endsWith("/responses")).toBe(true)
1467
+ expect(capture.headers.get("Authorization")).toBe("Bearer test-openai-key")
1468
+ expect(capture.body.model).toBe(model.id)
1469
+ expect(capture.body.stream).toBe(true)
1470
+ expect((capture.body.reasoning as { effort?: string } | undefined)?.effort).toBe("high")
1471
+ expect(capture.body.include).toEqual(["reasoning.encrypted_content"])
1472
+ expect(JSON.stringify(capture.body.input)).toContain("You are a helpful assistant.")
1473
+ expect(capture.body.input).toContainEqual({ role: "user", content: [{ type: "input_text", text: "Hello" }] })
1474
+ }),
1475
+ { config: () => openAIConfig(loadFixture("openai", "gpt-5.2").model, `${state.server!.url.origin}/v1`) },
1476
+ )
1477
+
1478
+ it.instance(
1479
+ "uses injected native request executor for tool calls",
1480
+ () =>
1481
+ Effect.gen(function* () {
1482
+ const model = loadFixture("openai", "gpt-5.2").model
1483
+ const chunks = [
1484
+ {
1485
+ type: "response.output_item.added",
1486
+ item: { type: "function_call", id: "item-injected-tool", call_id: "call-injected-tool", name: "lookup" },
1487
+ },
1488
+ {
1489
+ type: "response.function_call_arguments.delta",
1490
+ item_id: "item-injected-tool",
1491
+ delta: '{"query":"weather"}',
1492
+ },
1493
+ {
1494
+ type: "response.output_item.done",
1495
+ item: {
1496
+ type: "function_call",
1497
+ id: "item-injected-tool",
1498
+ call_id: "call-injected-tool",
1499
+ name: "lookup",
1500
+ arguments: '{"query":"weather"}',
1501
+ },
1502
+ },
1503
+ {
1504
+ type: "response.completed",
1505
+ response: { incomplete_details: null, usage: { input_tokens: 1, output_tokens: 1 } },
1506
+ },
1507
+ ]
1508
+ let captured: Record<string, unknown> | undefined
1509
+ let executed: unknown
1510
+ const executor = Layer.succeed(
1511
+ RequestExecutor.Service,
1512
+ RequestExecutor.Service.of({
1513
+ execute: (request) =>
1514
+ Effect.gen(function* () {
1515
+ const web = yield* HttpClientRequest.toWeb(request).pipe(Effect.orDie)
1516
+ captured = (yield* Effect.promise(() => web.json())) as Record<string, unknown>
1517
+ return HttpClientResponse.fromWeb(request, createEventResponse(chunks, true))
1518
+ }),
1519
+ }),
1520
+ )
1521
+
1522
+ const resolved = yield* Provider.use.getModel(ProviderV2.ID.openai, ModelV2.ID.make(model.id))
1523
+ const sessionID = SessionID.make("session-test-native-injected-tool")
1524
+ const agent = {
1525
+ name: "test",
1526
+ mode: "primary",
1527
+ options: {},
1528
+ permission: [{ permission: "*", pattern: "*", action: "allow" }],
1529
+ } satisfies Agent.Info
1530
+
1531
+ yield* drainWith(llmLayerWithExecutor({ executor, flags: { experimentalNativeLlm: true } }), {
1532
+ user: {
1533
+ id: MessageID.make("msg_user-native-injected-tool"),
1534
+ sessionID,
1535
+ role: "user",
1536
+ time: { created: Date.now() },
1537
+ agent: agent.name,
1538
+ model: { providerID: ProviderV2.ID.make("openai"), modelID: resolved.id },
1539
+ } satisfies SessionV1.User,
1540
+ sessionID,
1541
+ model: resolved,
1542
+ agent,
1543
+ system: [],
1544
+ messages: [{ role: "user", content: "Use lookup" }],
1545
+ tools: {
1546
+ lookup: tool({
1547
+ description: "Lookup data",
1548
+ inputSchema: z.object({ query: z.string() }),
1549
+ execute: async (args, options) => {
1550
+ executed = { args, toolCallId: options.toolCallId }
1551
+ return { output: "looked up" }
1552
+ },
1553
+ }),
1554
+ },
1555
+ })
1556
+
1557
+ expect(captured?.model).toBe(model.id)
1558
+ expect(captured?.tools).toEqual([
1559
+ {
1560
+ type: "function",
1561
+ name: "lookup",
1562
+ description: "Lookup data",
1563
+ strict: false,
1564
+ parameters: {
1565
+ type: "object",
1566
+ properties: { query: { type: "string" } },
1567
+ required: ["query"],
1568
+ additionalProperties: false,
1569
+ $schema: "http://json-schema.org/draft-07/schema#",
1570
+ },
1571
+ },
1572
+ ])
1573
+ expect(executed).toEqual({ args: { query: "weather" }, toolCallId: "call-injected-tool" })
1574
+ }),
1575
+ { config: () => openAIConfig(loadFixture("openai", "gpt-5.2").model, "https://injected-openai.test/v1") },
1576
+ )
1577
+
1578
+ it.instance(
1579
+ "executes OpenAI tool calls through native runtime",
1580
+ () =>
1581
+ Effect.gen(function* () {
1582
+ const model = loadFixture("openai", "gpt-5.2").model
1583
+ const chunks = [
1584
+ {
1585
+ type: "response.output_item.added",
1586
+ item: { type: "function_call", id: "item-native-tool", call_id: "call-native-tool", name: "lookup" },
1587
+ },
1588
+ {
1589
+ type: "response.function_call_arguments.delta",
1590
+ item_id: "item-native-tool",
1591
+ delta: '{"query":"weather"}',
1592
+ },
1593
+ {
1594
+ type: "response.output_item.done",
1595
+ item: {
1596
+ type: "function_call",
1597
+ id: "item-native-tool",
1598
+ call_id: "call-native-tool",
1599
+ name: "lookup",
1600
+ arguments: '{"query":"weather"}',
1601
+ },
1602
+ },
1603
+ {
1604
+ type: "response.completed",
1605
+ response: { incomplete_details: null, usage: { input_tokens: 1, output_tokens: 1 } },
1606
+ },
1607
+ ]
1608
+ const request = waitRequest("/responses", createEventResponse(chunks, true))
1609
+ let executed: unknown
1610
+
1611
+ const resolved = yield* Provider.use.getModel(ProviderV2.ID.openai, ModelV2.ID.make(model.id))
1612
+ const sessionID = SessionID.make("session-test-native-tool")
1613
+ const agent = {
1614
+ name: "test",
1615
+ mode: "primary",
1616
+ options: {},
1617
+ permission: [{ permission: "*", pattern: "*", action: "allow" }],
1618
+ } satisfies Agent.Info
1619
+
1620
+ yield* drainWith(llmLayerWithExecutor({ flags: { experimentalNativeLlm: true } }), {
1621
+ user: {
1622
+ id: MessageID.make("msg_user-native-tool"),
1623
+ sessionID,
1624
+ role: "user",
1625
+ time: { created: Date.now() },
1626
+ agent: agent.name,
1627
+ model: { providerID: ProviderV2.ID.make("openai"), modelID: resolved.id },
1628
+ } satisfies SessionV1.User,
1629
+ sessionID,
1630
+ model: resolved,
1631
+ agent,
1632
+ system: [],
1633
+ messages: [{ role: "user", content: "Use lookup" }],
1634
+ tools: {
1635
+ lookup: tool({
1636
+ description: "Lookup data",
1637
+ inputSchema: z.object({ query: z.string() }),
1638
+ execute: async (args, options) => {
1639
+ executed = { args, toolCallId: options.toolCallId }
1640
+ return { output: "looked up" }
1641
+ },
1642
+ }),
1643
+ },
1644
+ })
1645
+
1646
+ const capture = yield* Effect.promise(() => request)
1647
+ expect(capture.body.tools).toEqual([
1648
+ {
1649
+ type: "function",
1650
+ name: "lookup",
1651
+ description: "Lookup data",
1652
+ strict: false,
1653
+ parameters: {
1654
+ type: "object",
1655
+ properties: { query: { type: "string" } },
1656
+ required: ["query"],
1657
+ additionalProperties: false,
1658
+ $schema: "http://json-schema.org/draft-07/schema#",
1659
+ },
1660
+ },
1661
+ ])
1662
+ expect(executed).toEqual({ args: { query: "weather" }, toolCallId: "call-native-tool" })
1663
+ }),
1664
+ {
1665
+ config: () => {
1666
+ const model = loadFixture("openai", "gpt-5.2").model
1667
+ return {
1668
+ enabled_providers: ["openai"],
1669
+ provider: {
1670
+ openai: {
1671
+ name: "OpenAI",
1672
+ env: ["OPENAI_API_KEY"],
1673
+ npm: "@ai-sdk/openai",
1674
+ api: "https://api.openai.com/v1",
1675
+ models: { [model.id]: JSON.parse(JSON.stringify(model)) as ConfigModel },
1676
+ options: { apiKey: "test-openai-key", baseURL: `${state.server!.url.origin}/v1` },
1677
+ },
1678
+ },
1679
+ }
1680
+ },
1681
+ },
1682
+ )
1683
+
1684
+ it.instance(
1685
+ "accepts user image attachments as data URLs for OpenAI models",
1686
+ () =>
1687
+ Effect.gen(function* () {
1688
+ const model = loadFixture("openai", "gpt-5.2").model
1689
+ const chunks = [
1690
+ {
1691
+ type: "response.created",
1692
+ response: {
1693
+ id: "resp-data-url",
1694
+ created_at: Math.floor(Date.now() / 1000),
1695
+ model: model.id,
1696
+ service_tier: null,
1697
+ },
1698
+ },
1699
+ {
1700
+ type: "response.output_item.added",
1701
+ output_index: 0,
1702
+ item: { type: "message", id: "item-data-url", status: "in_progress", role: "assistant", content: [] },
1703
+ },
1704
+ {
1705
+ type: "response.content_part.added",
1706
+ item_id: "item-data-url",
1707
+ output_index: 0,
1708
+ content_index: 0,
1709
+ part: { type: "output_text", text: "", annotations: [] },
1710
+ },
1711
+ {
1712
+ type: "response.output_text.delta",
1713
+ item_id: "item-data-url",
1714
+ delta: "Looks good",
1715
+ logprobs: null,
1716
+ },
1717
+ {
1718
+ type: "response.completed",
1719
+ response: {
1720
+ incomplete_details: null,
1721
+ usage: {
1722
+ input_tokens: 1,
1723
+ input_tokens_details: null,
1724
+ output_tokens: 1,
1725
+ output_tokens_details: null,
1726
+ },
1727
+ service_tier: null,
1728
+ },
1729
+ },
1730
+ ]
1731
+ const request = waitRequest("/responses", createEventResponse(chunks, true))
1732
+ const image = `data:image/png;base64,${Buffer.from(
1733
+ yield* Effect.promise(() =>
1734
+ Bun.file(path.join(import.meta.dir, "../tool/fixtures/large-image.png")).arrayBuffer(),
1735
+ ),
1736
+ ).toString("base64")}`
1737
+
1738
+ const resolved = yield* Provider.use.getModel(ProviderV2.ID.openai, ModelV2.ID.make(model.id))
1739
+ const sessionID = SessionID.make("session-test-data-url")
1740
+ const agent = {
1741
+ name: "test",
1742
+ mode: "primary",
1743
+ options: {},
1744
+ permission: [{ permission: "*", pattern: "*", action: "allow" }],
1745
+ } satisfies Agent.Info
1746
+
1747
+ const user = {
1748
+ id: MessageID.make("msg_user-data-url"),
1749
+ sessionID,
1750
+ role: "user",
1751
+ time: { created: Date.now() },
1752
+ agent: agent.name,
1753
+ model: { providerID: ProviderV2.ID.make("openai"), modelID: resolved.id },
1754
+ } satisfies SessionV1.User
1755
+
1756
+ yield* drain({
1757
+ user,
1758
+ sessionID,
1759
+ model: resolved,
1760
+ agent,
1761
+ system: ["You are a helpful assistant."],
1762
+ messages: [
1763
+ {
1764
+ role: "user",
1765
+ content: [
1766
+ { type: "text", text: "Describe this image" },
1767
+ { type: "file", mediaType: "image/png", filename: "large-image.png", data: image },
1768
+ ],
1769
+ },
1770
+ ] as ModelMessage[],
1771
+ tools: {},
1772
+ })
1773
+
1774
+ const capture = yield* Effect.promise(() => request)
1775
+ expect(capture.url.pathname.endsWith("/responses")).toBe(true)
1776
+ }),
1777
+ { config: () => openAIConfig(loadFixture("openai", "gpt-5.2").model, `${state.server!.url.origin}/v1`) },
1778
+ )
1779
+
1780
+ const minimaxFixture = { providerID: "minimax", modelID: "MiniMax-M2.5" }
1781
+ it.instance(
1782
+ "sends messages API payload for Anthropic Compatible models",
1783
+ () =>
1784
+ Effect.gen(function* () {
1785
+ const model = loadFixture(minimaxFixture.providerID, minimaxFixture.modelID).model
1786
+
1787
+ const chunks = [
1788
+ {
1789
+ type: "message_start",
1790
+ message: {
1791
+ id: "msg-1",
1792
+ model: model.id,
1793
+ usage: {
1794
+ input_tokens: 3,
1795
+ cache_creation_input_tokens: null,
1796
+ cache_read_input_tokens: null,
1797
+ },
1798
+ },
1799
+ },
1800
+ {
1801
+ type: "content_block_start",
1802
+ index: 0,
1803
+ content_block: { type: "text", text: "" },
1804
+ },
1805
+ {
1806
+ type: "content_block_delta",
1807
+ index: 0,
1808
+ delta: { type: "text_delta", text: "Hello" },
1809
+ },
1810
+ { type: "content_block_stop", index: 0 },
1811
+ {
1812
+ type: "message_delta",
1813
+ delta: { stop_reason: "end_turn", stop_sequence: null, container: null },
1814
+ usage: {
1815
+ input_tokens: 3,
1816
+ output_tokens: 2,
1817
+ cache_creation_input_tokens: null,
1818
+ cache_read_input_tokens: null,
1819
+ },
1820
+ },
1821
+ { type: "message_stop" },
1822
+ ]
1823
+ const request = waitRequest("/messages", createEventResponse(chunks))
1824
+
1825
+ const resolved = yield* Provider.use.getModel(
1826
+ ProviderV2.ID.make(minimaxFixture.providerID),
1827
+ ModelV2.ID.make(model.id),
1828
+ )
1829
+ const sessionID = SessionID.make("session-test-3")
1830
+ const agent = {
1831
+ name: "test",
1832
+ mode: "primary",
1833
+ options: {},
1834
+ permission: [{ permission: "*", pattern: "*", action: "allow" }],
1835
+ temperature: 0.4,
1836
+ topP: 0.9,
1837
+ } satisfies Agent.Info
1838
+
1839
+ const user = {
1840
+ id: MessageID.make("msg_user-3"),
1841
+ sessionID,
1842
+ role: "user",
1843
+ time: { created: Date.now() },
1844
+ agent: agent.name,
1845
+ model: { providerID: ProviderV2.ID.make("minimax"), modelID: ModelV2.ID.make("MiniMax-M2.5") },
1846
+ } satisfies SessionV1.User
1847
+
1848
+ yield* drain({
1849
+ user,
1850
+ sessionID,
1851
+ model: resolved,
1852
+ agent,
1853
+ system: ["You are a helpful assistant."],
1854
+ messages: [{ role: "user", content: "Hello" }],
1855
+ tools: {},
1856
+ })
1857
+
1858
+ const capture = yield* Effect.promise(() => request)
1859
+ const body = capture.body
1860
+
1861
+ expect(capture.url.pathname.endsWith("/messages")).toBe(true)
1862
+ expect(body.model).toBe(resolved.api.id)
1863
+ expect(body.max_tokens).toBe(ProviderTransform.maxOutputTokens(resolved))
1864
+ expect(body.temperature).toBe(0.4)
1865
+ expect(body.top_p).toBe(0.9)
1866
+ }),
1867
+ {
1868
+ config: () => ({
1869
+ enabled_providers: [minimaxFixture.providerID],
1870
+ provider: {
1871
+ [minimaxFixture.providerID]: {
1872
+ options: { apiKey: "test-anthropic-key", baseURL: `${state.server!.url.origin}/v1` },
1873
+ },
1874
+ },
1875
+ }),
1876
+ },
1877
+ )
1878
+
1879
+ it.instance(
1880
+ "sends anthropic tool_use blocks with tool_result immediately after them",
1881
+ () =>
1882
+ Effect.gen(function* () {
1883
+ const model = loadFixture("anthropic", "claude-opus-4-6").model
1884
+ const chunks = [
1885
+ {
1886
+ type: "message_start",
1887
+ message: {
1888
+ id: "msg-tool-order",
1889
+ model: model.id,
1890
+ usage: {
1891
+ input_tokens: 3,
1892
+ cache_creation_input_tokens: null,
1893
+ cache_read_input_tokens: null,
1894
+ },
1895
+ },
1896
+ },
1897
+ {
1898
+ type: "content_block_start",
1899
+ index: 0,
1900
+ content_block: { type: "text", text: "" },
1901
+ },
1902
+ {
1903
+ type: "content_block_delta",
1904
+ index: 0,
1905
+ delta: { type: "text_delta", text: "ok" },
1906
+ },
1907
+ { type: "content_block_stop", index: 0 },
1908
+ {
1909
+ type: "message_delta",
1910
+ delta: { stop_reason: "end_turn", stop_sequence: null, container: null },
1911
+ usage: {
1912
+ input_tokens: 3,
1913
+ output_tokens: 2,
1914
+ cache_creation_input_tokens: null,
1915
+ cache_read_input_tokens: null,
1916
+ },
1917
+ },
1918
+ { type: "message_stop" },
1919
+ ]
1920
+ const request = waitRequest("/messages", createEventResponse(chunks))
1921
+
1922
+ const resolved = yield* Provider.use.getModel(ProviderV2.ID.make("anthropic"), ModelV2.ID.make(model.id))
1923
+ const sessionID = SessionID.make("session-test-anthropic-tools")
1924
+ const agent = {
1925
+ name: "test",
1926
+ mode: "primary",
1927
+ options: {},
1928
+ permission: [{ permission: "*", pattern: "*", action: "allow" }],
1929
+ } satisfies Agent.Info
1930
+ const user = {
1931
+ id: MessageID.make("msg_user-anthropic-tools"),
1932
+ sessionID,
1933
+ role: "user",
1934
+ time: { created: Date.now() },
1935
+ agent: agent.name,
1936
+ model: { providerID: ProviderV2.ID.make("anthropic"), modelID: resolved.id, variant: "max" },
1937
+ } satisfies SessionV1.User
1938
+
1939
+ const input = [
1940
+ {
1941
+ info: {
1942
+ id: "msg_user",
1943
+ sessionID,
1944
+ role: "user",
1945
+ time: { created: 1 },
1946
+ agent: "gentleman",
1947
+ model: { providerID: "anthropic", modelID: "claude-opus-4-6", variant: "max" },
1948
+ },
1949
+ parts: [
1950
+ {
1951
+ id: "p_user",
1952
+ sessionID,
1953
+ messageID: "msg_user",
1954
+ type: "text",
1955
+ text: "Can you check whether there are any PDF files in my home directory?",
1956
+ },
1957
+ ],
1958
+ },
1959
+ {
1960
+ info: {
1961
+ id: "msg_call",
1962
+ sessionID,
1963
+ parentID: "msg_user",
1964
+ role: "assistant",
1965
+ mode: "gentleman",
1966
+ agent: "gentleman",
1967
+ variant: "max",
1968
+ path: { cwd: "/root", root: "/" },
1969
+ cost: 0,
1970
+ tokens: { input: 0, output: 0, reasoning: 0, cache: { read: 0, write: 0 } },
1971
+ modelID: "claude-opus-4-6",
1972
+ providerID: "anthropic",
1973
+ time: { created: 2, completed: 3 },
1974
+ finish: "tool-calls",
1975
+ },
1976
+ parts: [
1977
+ {
1978
+ id: "p_step",
1979
+ sessionID,
1980
+ messageID: "msg_call",
1981
+ type: "step-start",
1982
+ },
1983
+ {
1984
+ id: "p_read",
1985
+ sessionID,
1986
+ messageID: "msg_call",
1987
+ type: "tool",
1988
+ tool: "read",
1989
+ callID: "toolu_01N8mDEzG8DSTs7UPHFtmgCT",
1990
+ state: {
1991
+ status: "completed",
1992
+ input: { filePath: "/root" },
1993
+ output: "<path>/root</path>",
1994
+ metadata: {},
1995
+ title: "root",
1996
+ time: { start: 10, end: 11 },
1997
+ },
1998
+ },
1999
+ {
2000
+ id: "p_glob",
2001
+ sessionID,
2002
+ messageID: "msg_call",
2003
+ type: "tool",
2004
+ tool: "glob",
2005
+ callID: "toolu_01APxrADs7VozN8uWzw9WwHr",
2006
+ state: {
2007
+ status: "completed",
2008
+ input: { pattern: "**/*.pdf", path: "/root" },
2009
+ output: "No files found",
2010
+ metadata: {},
2011
+ title: "root",
2012
+ time: { start: 12, end: 13 },
2013
+ },
2014
+ },
2015
+ {
2016
+ id: "p_text",
2017
+ sessionID,
2018
+ messageID: "msg_call",
2019
+ type: "text",
2020
+ text: "I checked your home directory and looked for PDF files.",
2021
+ time: { start: 14, end: 15 },
2022
+ },
2023
+ ],
2024
+ },
2025
+ ] as any[]
2026
+
2027
+ const modelMessages = yield* Effect.promise(() => MessageV2.toModelMessages(input as any, resolved))
2028
+ yield* drain({
2029
+ user,
2030
+ sessionID,
2031
+ model: resolved,
2032
+ agent,
2033
+ system: [],
2034
+ messages: modelMessages,
2035
+ tools: {
2036
+ read: tool({
2037
+ description: "Stub read tool",
2038
+ inputSchema: z.object({ filePath: z.string() }),
2039
+ execute: async () => ({ output: "stub" }),
2040
+ }),
2041
+ glob: tool({
2042
+ description: "Stub glob tool",
2043
+ inputSchema: z.object({ pattern: z.string(), path: z.string().optional() }),
2044
+ execute: async () => ({ output: "stub" }),
2045
+ }),
2046
+ },
2047
+ })
2048
+
2049
+ const capture = yield* Effect.promise(() => request)
2050
+ const body = capture.body
2051
+
2052
+ expect(capture.url.pathname.endsWith("/messages")).toBe(true)
2053
+ const messages = body.messages as Array<{ role: string; content: Array<Record<string, unknown>> }>
2054
+ expect(messages[0]?.role).toBe("user")
2055
+ expect(messages[0]?.content[0]).toMatchObject({
2056
+ type: "text",
2057
+ text: "Can you check whether there are any PDF files in my home directory?",
2058
+ })
2059
+ expect(messages.some((message) => message.content.some((part) => "cache_control" in part))).toBe(true)
2060
+ const toolUseIndex = messages.findIndex((message) => message.content.some((part) => part.type === "tool_use"))
2061
+ expect(toolUseIndex).toBeGreaterThan(0)
2062
+ expect(messages[toolUseIndex].role).toBe("assistant")
2063
+ expect(messages[toolUseIndex].content.filter((part) => part.type === "tool_use")).toMatchObject([
2064
+ {
2065
+ type: "tool_use",
2066
+ id: "toolu_01N8mDEzG8DSTs7UPHFtmgCT",
2067
+ name: "read",
2068
+ input: { filePath: "/root" },
2069
+ },
2070
+ {
2071
+ type: "tool_use",
2072
+ id: "toolu_01APxrADs7VozN8uWzw9WwHr",
2073
+ name: "glob",
2074
+ input: { pattern: "**/*.pdf", path: "/root" },
2075
+ },
2076
+ ])
2077
+ expect(messages[toolUseIndex + 1]).toMatchObject({
2078
+ role: "user",
2079
+ content: [
2080
+ { type: "tool_result", tool_use_id: "toolu_01N8mDEzG8DSTs7UPHFtmgCT", content: "<path>/root</path>" },
2081
+ { type: "tool_result", tool_use_id: "toolu_01APxrADs7VozN8uWzw9WwHr", content: "No files found" },
2082
+ ],
2083
+ })
2084
+ }),
2085
+ {
2086
+ config: () => {
2087
+ const model = loadFixture("anthropic", "claude-opus-4-6").model
2088
+ return {
2089
+ enabled_providers: ["anthropic"],
2090
+ provider: {
2091
+ anthropic: {
2092
+ name: "Anthropic",
2093
+ env: ["ANTHROPIC_API_KEY"],
2094
+ npm: "@ai-sdk/anthropic",
2095
+ api: "https://api.anthropic.com/v1",
2096
+ models: { [model.id]: configModel(model) as ConfigModel },
2097
+ options: { apiKey: "test-anthropic-key", baseURL: `${state.server!.url.origin}/v1` },
2098
+ },
2099
+ },
2100
+ }
2101
+ },
2102
+ },
2103
+ )
2104
+
2105
+ const geminiFixture = { providerID: "google", modelID: "gemini-2.5-flash" }
2106
+ it.instance(
2107
+ "sends Google API payload for Gemini models",
2108
+ () =>
2109
+ Effect.gen(function* () {
2110
+ const model = loadFixture(geminiFixture.providerID, geminiFixture.modelID).model
2111
+ const pathSuffix = `/v1beta/models/${model.id}:streamGenerateContent`
2112
+
2113
+ const chunks = [
2114
+ {
2115
+ candidates: [{ content: { parts: [{ text: "Hello" }] }, finishReason: "STOP" }],
2116
+ usageMetadata: { promptTokenCount: 1, candidatesTokenCount: 1, totalTokenCount: 2 },
2117
+ },
2118
+ ]
2119
+ const request = waitRequest(pathSuffix, createEventResponse(chunks))
2120
+
2121
+ const resolved = yield* Provider.use.getModel(
2122
+ ProviderV2.ID.make(geminiFixture.providerID),
2123
+ ModelV2.ID.make(model.id),
2124
+ )
2125
+ const sessionID = SessionID.make("session-test-4")
2126
+ const agent = {
2127
+ name: "test",
2128
+ mode: "primary",
2129
+ options: {},
2130
+ permission: [{ permission: "*", pattern: "*", action: "allow" }],
2131
+ temperature: 0.3,
2132
+ topP: 0.8,
2133
+ } satisfies Agent.Info
2134
+
2135
+ const user = {
2136
+ id: MessageID.make("msg_user-4"),
2137
+ sessionID,
2138
+ role: "user",
2139
+ time: { created: Date.now() },
2140
+ agent: agent.name,
2141
+ model: { providerID: ProviderV2.ID.make(geminiFixture.providerID), modelID: resolved.id },
2142
+ } satisfies SessionV1.User
2143
+
2144
+ yield* drain({
2145
+ user,
2146
+ sessionID,
2147
+ model: resolved,
2148
+ agent,
2149
+ system: ["You are a helpful assistant."],
2150
+ messages: [
2151
+ { role: "user", content: "Hello" },
2152
+ { role: "assistant", content: [{ type: "reasoning", text: "" }] },
2153
+ ],
2154
+ tools: {},
2155
+ })
2156
+
2157
+ const capture = yield* Effect.promise(() => request)
2158
+ const body = capture.body
2159
+ const config = body.generationConfig as
2160
+ | { temperature?: number; topP?: number; maxOutputTokens?: number }
2161
+ | undefined
2162
+
2163
+ expect(capture.url.pathname).toBe(pathSuffix)
2164
+ expect(body.contents).toEqual([{ role: "user", parts: [{ text: "Hello" }] }])
2165
+ expect(config?.temperature).toBe(0.3)
2166
+ expect(config?.topP).toBe(0.8)
2167
+ expect(config?.maxOutputTokens).toBe(ProviderTransform.maxOutputTokens(resolved))
2168
+ }),
2169
+ {
2170
+ config: () => ({
2171
+ enabled_providers: [geminiFixture.providerID],
2172
+ provider: {
2173
+ [geminiFixture.providerID]: {
2174
+ options: { apiKey: "test-google-key", baseURL: `${state.server!.url.origin}/v1beta` },
2175
+ },
2176
+ },
2177
+ }),
2178
+ },
2179
+ )
2180
+ })