mlx-bun 0.5.0 → 1.0.0

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 (640) hide show
  1. package/README.md +1449 -109
  2. package/THIRD_PARTY_NOTICES.md +882 -0
  3. package/bin/mlx-bun.convert.mjs +4 -0
  4. package/bin/mlx-bun.fuse.mjs +4 -0
  5. package/bin/mlx-bun.generate.mjs +4 -0
  6. package/bin/mlx-bun.lora.mjs +4 -0
  7. package/bin/mlx-bun.mjs +13 -21
  8. package/bin/mlx-bun.server.mjs +4 -0
  9. package/bin/mlx-bun.upload.mjs +4 -0
  10. package/dist/web/app.js +9584 -0
  11. package/package.json +50 -53
  12. package/src/cli/ane-install.ts +76 -0
  13. package/src/cli/args.ts +179 -0
  14. package/src/cli/engine-entry.ts +68 -0
  15. package/src/cli/inference.ts +214 -0
  16. package/src/cli/job-entry.ts +97 -0
  17. package/src/cli/library-host.ts +117 -0
  18. package/src/cli/main.ts +57 -0
  19. package/src/cli/memory-engine.ts +163 -0
  20. package/src/cli/memory.ts +36 -0
  21. package/src/cli/mlx-lm-aliases.ts +309 -0
  22. package/src/cli/model-choice.ts +33 -0
  23. package/src/cli/model-selection.ts +84 -0
  24. package/src/cli/module-host.ts +11 -0
  25. package/src/cli/module-verbs.ts +50 -0
  26. package/src/cli/numerical-policy.ts +12 -0
  27. package/src/cli/perplexity.ts +89 -0
  28. package/src/cli/serve-host.ts +423 -0
  29. package/src/cli/serve-isolated.ts +217 -0
  30. package/src/cli/serve-options.ts +106 -0
  31. package/src/cli/serve-state.ts +205 -0
  32. package/src/cli/serve.ts +500 -0
  33. package/src/cli/served-model-host.ts +102 -0
  34. package/src/cli/server-entry.ts +148 -0
  35. package/src/cli/serving-unit.ts +177 -0
  36. package/src/{tui.ts → cli/terminal.ts} +13 -8
  37. package/src/cli/worker-adapters.ts +34 -0
  38. package/src/cli/worker-entry.ts +272 -0
  39. package/src/cli/worker-unit.ts +179 -0
  40. package/src/engine/adapter-operation.ts +26 -0
  41. package/src/engine/cache-services.ts +114 -0
  42. package/src/engine/completion.ts +39 -0
  43. package/src/engine/generation-gateway.ts +358 -0
  44. package/src/engine/index.ts +59 -0
  45. package/src/engine/kv-budget.ts +120 -0
  46. package/src/engine/media-preparation.ts +232 -0
  47. package/src/engine/model-binding.ts +24 -0
  48. package/src/engine/model-host.ts +513 -0
  49. package/src/engine/model-serving.ts +89 -0
  50. package/src/{serve → engine}/preparation.ts +7 -5
  51. package/src/{serve → engine}/session-completion-engine.ts +6 -6
  52. package/src/engine/telemetry.ts +109 -0
  53. package/src/hub/downloads.ts +116 -0
  54. package/src/jobs/db.ts +16 -7
  55. package/src/jobs/events.ts +29 -0
  56. package/src/jobs/executable.ts +5 -0
  57. package/src/jobs/host.ts +51 -0
  58. package/src/jobs/in-process.ts +55 -0
  59. package/src/{contracts/jobs.ts → jobs/protocol.ts} +7 -9
  60. package/src/jobs/runner.ts +150 -231
  61. package/src/jobs/service.ts +75 -0
  62. package/src/jobs/sse.ts +59 -23
  63. package/src/jobs/worker-memory.ts +46 -0
  64. package/src/jobs/worker-process.ts +163 -0
  65. package/src/jobs/worker-supervisor.ts +258 -0
  66. package/src/modules.ts +58 -0
  67. package/src/publishing/catalog-hub.ts +18 -0
  68. package/src/publishing/credentials.ts +31 -0
  69. package/src/publishing/upload.ts +37 -0
  70. package/src/residency/leased-adapters.ts +21 -0
  71. package/src/residency/local-records.ts +16 -0
  72. package/src/residency/model-residency.ts +405 -0
  73. package/src/residency/record-index.ts +28 -0
  74. package/src/residency/resident-estimate.ts +50 -0
  75. package/src/{anthropic.ts → server/anthropic.ts} +3 -81
  76. package/src/server/cache-routes.ts +36 -0
  77. package/src/{serve → server}/chat-request.ts +10 -10
  78. package/src/{serve → server}/chat-stage.ts +14 -14
  79. package/src/{client.ts → server/client.ts} +28 -2
  80. package/src/{serve → server}/completion-executor.ts +33 -41
  81. package/src/server/decision-body.ts +55 -0
  82. package/src/server/decision-routes.ts +134 -0
  83. package/src/server/decision-wire.ts +228 -0
  84. package/src/{serve → server}/discovery-routes.ts +25 -82
  85. package/src/{serve → server}/http.ts +25 -13
  86. package/src/{serve → server}/inference-request.ts +5 -5
  87. package/src/server/job-routes.ts +20 -0
  88. package/src/server/media-prompt.ts +63 -0
  89. package/src/server/memory-completion-client.ts +46 -0
  90. package/src/server/model-routes.ts +192 -0
  91. package/src/{serve → server}/openai-wire.ts +3 -2
  92. package/src/{serve → server}/pipeline.ts +1 -1
  93. package/src/server/prompt-builder.ts +13 -0
  94. package/src/server/prompt-contracts.ts +17 -0
  95. package/src/server/proxy-routes.ts +187 -0
  96. package/src/server/publishing-routes.ts +41 -0
  97. package/src/{serve → server}/request-plan.ts +6 -8
  98. package/src/{serve → server}/request-prep.ts +47 -50
  99. package/src/{serve → server}/responses-client.ts +30 -13
  100. package/src/{responses.ts → server/responses.ts} +12 -93
  101. package/src/server/routes.ts +205 -0
  102. package/src/server/start.ts +72 -0
  103. package/src/server/status-routes.ts +108 -0
  104. package/src/{serve → server}/text-completion-stage.ts +25 -4
  105. package/src/server/text-prompt.ts +12 -0
  106. package/src/{serve → server}/token-streams.ts +36 -38
  107. package/src/server/worker-routes.ts +311 -0
  108. package/src/storage/paths.ts +61 -0
  109. package/src/web/assets.ts +37 -0
  110. package/src/web/{src → browser}/api.ts +5 -23
  111. package/src/web/browser/chat-host.ts +17 -0
  112. package/src/web/browser/installed-panels.ts +6 -0
  113. package/src/web/browser/main.ts +55 -0
  114. package/src/web/browser/markdown.ts +279 -0
  115. package/src/web/browser/memory-panel.ts +9 -0
  116. package/src/web/browser/palette.ts +64 -0
  117. package/src/web/browser/protocol.ts +13 -0
  118. package/src/web/browser/shell.ts +467 -0
  119. package/src/web/{src → browser}/status.ts +32 -36
  120. package/src/web/build.ts +51 -0
  121. package/src/web/public/app.html +1054 -0
  122. package/src/web/{sw.js → public/sw.js} +4 -3
  123. package/src/web/{vendor → public/vendor}/hljs-theme.css +2 -2
  124. package/src/web/{vendor → public/vendor}/hljs.js +2 -2
  125. package/THIRD_PARTY_LICENSES.md +0 -51
  126. package/docs/README.md +0 -107
  127. package/docs/archive/hlg-sampling.md +0 -350
  128. package/docs/archive/investigations/audio-input-plan.md +0 -356
  129. package/docs/archive/investigations/benchmark-ledger-through-v0.4.0.md +0 -3617
  130. package/docs/archive/investigations/chunk-finetune-experiment.md +0 -84
  131. package/docs/archive/investigations/colibri-glm52-port.md +0 -1486
  132. package/docs/archive/investigations/colibri-oracle-pin.md +0 -1284
  133. package/docs/archive/investigations/curve-bisector-routing.md +0 -123
  134. package/docs/archive/investigations/decode-roofline-lookagain.md +0 -348
  135. package/docs/archive/investigations/diffusion-gemma-port.md +0 -602
  136. package/docs/archive/investigations/dspark-handoff.md +0 -224
  137. package/docs/archive/investigations/expert-offload-single-user-moe.md +0 -292
  138. package/docs/archive/investigations/faithful-l1-consolidation.md +0 -310
  139. package/docs/archive/investigations/hlg-sampling-investigation.md +0 -970
  140. package/docs/archive/investigations/indexshare-performance-spike.md +0 -357
  141. package/docs/archive/investigations/kernel-perf-review-2026-07.md +0 -171
  142. package/docs/archive/investigations/lab-build-journal.md +0 -157
  143. package/docs/archive/investigations/minicpm5-decode-megakernel.md +0 -489
  144. package/docs/archive/investigations/openjarvis-fit.md +0 -198
  145. package/docs/archive/investigations/openwiki-evaluation.md +0 -249
  146. package/docs/archive/investigations/optimization_plan.md +0 -346
  147. package/docs/archive/investigations/orpo-base-uf-experiment-and-directions.md +0 -120
  148. package/docs/archive/investigations/orpo-flash-cce-pin-leak.md +0 -167
  149. package/docs/archive/investigations/orpo-uf-testing-handoff.md +0 -126
  150. package/docs/archive/investigations/pareto-specialized-runtime-findings.md +0 -92
  151. package/docs/archive/investigations/pi-builtin-investigation.md +0 -308
  152. package/docs/archive/investigations/pre-colibri-stabilization.md +0 -329
  153. package/docs/archive/investigations/qwen38-27b-campaign-log.md +0 -1092
  154. package/docs/archive/investigations/segmented-backward-handoff.md +0 -148
  155. package/docs/archive/investigations/starter-model-port-handoff.md +0 -342
  156. package/docs/archive/investigations/steel-flash-cce-handoff.md +0 -489
  157. package/docs/archive/investigations/the-dreaming-handoff.md +0 -141
  158. package/docs/archive/investigations/trainer-validation-experiment.md +0 -173
  159. package/docs/archive/planning/chat-ui-vision-memory-first.md +0 -199
  160. package/docs/archive/planning/chat-ui-vision.md +0 -50
  161. package/docs/archive/planning/chunk-task-roadmap.md +0 -77
  162. package/docs/archive/planning/curve-sampler-research-plan.md +0 -331
  163. package/docs/archive/planning/journal.md +0 -316
  164. package/docs/archive/planning/memory-docs-and-dag-plan.md +0 -194
  165. package/docs/archive/planning/pi-integration-review.md +0 -178
  166. package/docs/archive/planning/release-notes-v0.0.10.md +0 -57
  167. package/docs/archive/planning/release-notes-v0.0.11.md +0 -239
  168. package/docs/archive/planning/release-notes-v0.0.12.md +0 -80
  169. package/docs/archive/planning/release-notes-v0.0.13.md +0 -77
  170. package/docs/archive/planning/release-notes-v0.0.9.md +0 -122
  171. package/docs/archive/planning/release-notes-v0.2.0.md +0 -49
  172. package/docs/archive/planning/release-notes-v0.3.0.md +0 -35
  173. package/docs/archive/planning/release-notes-v0.4.0.md +0 -45
  174. package/docs/archive/planning/release-notes-v0.5.0.md +0 -37
  175. package/docs/archive/planning/web-ui-pass-plan.md +0 -213
  176. package/docs/archive/planning/website-readme-pass-plan.md +0 -224
  177. package/docs/design/batching.md +0 -2054
  178. package/docs/design/decode-speed-program.md +0 -2404
  179. package/docs/design/dreaming-nightly-pipeline.md +0 -684
  180. package/docs/design/generic-model-support.md +0 -821
  181. package/docs/design/kv-cache.md +0 -993
  182. package/docs/design/orpo-training.md +0 -680
  183. package/docs/design/speculative-decoding.md +0 -2129
  184. package/docs/design/turboquant.md +0 -2225
  185. package/docs/design/unified-engine-frontier-plan.md +0 -1987
  186. package/docs/design/web-chat-redesign.md +0 -695
  187. package/docs/planning/PRODUCT_ROADMAP.md +0 -593
  188. package/docs/planning/ResearchTopics.md +0 -119
  189. package/docs/planning/public-readiness-handoff.md +0 -244
  190. package/docs/planning/public-readiness.md +0 -149
  191. package/docs/planning/video-series.md +0 -421
  192. package/docs/reference/benchmarks.md +0 -3587
  193. package/docs/reference/cli.md +0 -646
  194. package/docs/reference/distribution.md +0 -380
  195. package/docs/reference/environment.md +0 -494
  196. package/docs/reference/glossary.md +0 -39
  197. package/docs/reference/library-api.md +0 -688
  198. package/docs/reference/memory.md +0 -260
  199. package/docs/reference/models.md +0 -346
  200. package/docs/reference/server-api.md +0 -1945
  201. package/docs/reference/server-config.md +0 -973
  202. package/docs/reference/training.md +0 -462
  203. package/docs/reference/troubleshooting.md +0 -158
  204. package/src/admission-plan.ts +0 -18
  205. package/src/audio/audiotoolbox.ts +0 -112
  206. package/src/audio/conformer.ts +0 -655
  207. package/src/audio/decode.ts +0 -133
  208. package/src/audio/features.ts +0 -202
  209. package/src/audio/mic-capture.ts +0 -98
  210. package/src/audio/silero-vad.ts +0 -379
  211. package/src/audio/transcode.ts +0 -67
  212. package/src/audio/whisper-decode.ts +0 -774
  213. package/src/audio/whisper-fast.ts +0 -732
  214. package/src/audio/whisper-format.ts +0 -73
  215. package/src/audio/whisper-mel.ts +0 -185
  216. package/src/audio/whisper-timing.ts +0 -274
  217. package/src/audio/whisper-tokenizer.ts +0 -239
  218. package/src/audio/whisper-transcribe.ts +0 -410
  219. package/src/backends/mlx/assistant-target.ts +0 -46
  220. package/src/backends/mlx/autoregressive-draft-rows.ts +0 -120
  221. package/src/backends/mlx/autoregressive.ts +0 -129
  222. package/src/backends/mlx/batch-group.ts +0 -1874
  223. package/src/backends/mlx/cache-attachment.ts +0 -67
  224. package/src/backends/mlx/cache-layout.ts +0 -31
  225. package/src/backends/mlx/checkpoint-state.ts +0 -53
  226. package/src/backends/mlx/continuation-persistence.ts +0 -107
  227. package/src/backends/mlx/continuation-request.ts +0 -43
  228. package/src/backends/mlx/continuation.ts +0 -77
  229. package/src/backends/mlx/diffusion.ts +0 -39
  230. package/src/backends/mlx/draft-target.ts +0 -32
  231. package/src/backends/mlx/encoder-cache.ts +0 -25
  232. package/src/backends/mlx/evaluation.ts +0 -105
  233. package/src/backends/mlx/fill-append.ts +0 -58
  234. package/src/backends/mlx/fill-group.ts +0 -497
  235. package/src/backends/mlx/full-kv-row-append.ts +0 -32
  236. package/src/backends/mlx/full-kv-row-donor.ts +0 -54
  237. package/src/backends/mlx/full-transitioning-kv-rows.ts +0 -65
  238. package/src/backends/mlx/gateway-binding.ts +0 -174
  239. package/src/backends/mlx/graph.ts +0 -38
  240. package/src/backends/mlx/kv-codec.ts +0 -9
  241. package/src/backends/mlx/kv-maintenance.ts +0 -161
  242. package/src/backends/mlx/kv-tensor-rows.ts +0 -212
  243. package/src/backends/mlx/memory-client.ts +0 -189
  244. package/src/backends/mlx/methods.ts +0 -90
  245. package/src/backends/mlx/mixed-iteration.ts +0 -62
  246. package/src/backends/mlx/model-prompt.ts +0 -178
  247. package/src/backends/mlx/model-serving.ts +0 -80
  248. package/src/backends/mlx/paged-cache-codec.ts +0 -27
  249. package/src/backends/mlx/prefill-cohort.ts +0 -100
  250. package/src/backends/mlx/prefill-policy.ts +0 -26
  251. package/src/backends/mlx/prefill-rows.ts +0 -244
  252. package/src/backends/mlx/prefill.ts +0 -33
  253. package/src/backends/mlx/projected-context-rows.ts +0 -72
  254. package/src/backends/mlx/prompt-input.ts +0 -29
  255. package/src/backends/mlx/qwen-prompt-input.ts +0 -40
  256. package/src/backends/mlx/request-policy.ts +0 -20
  257. package/src/backends/mlx/request-state-policy.ts +0 -54
  258. package/src/backends/mlx/rollback.ts +0 -92
  259. package/src/backends/mlx/row-sampling.ts +0 -49
  260. package/src/backends/mlx/serial-executor.ts +0 -251
  261. package/src/backends/mlx/speculative-group.ts +0 -438
  262. package/src/backends/mlx/speculative-round.ts +0 -161
  263. package/src/backends/mlx/speculative.ts +0 -131
  264. package/src/backends/mlx/state-rows.ts +0 -104
  265. package/src/backends/mlx/state-views.ts +0 -46
  266. package/src/backends/mlx/target-layout-capability.ts +0 -35
  267. package/src/backends/mlx/text-engine.ts +0 -73
  268. package/src/backends/mlx/transitioning-kv-rows.ts +0 -173
  269. package/src/chat-template.ts +0 -382
  270. package/src/cli.ts +0 -3490
  271. package/src/config.ts +0 -399
  272. package/src/contracts/completion.ts +0 -16
  273. package/src/contracts/execution.ts +0 -70
  274. package/src/contracts/generation.ts +0 -103
  275. package/src/contracts/host.ts +0 -7
  276. package/src/contracts/object-cache.ts +0 -8
  277. package/src/contracts/pi-provider.ts +0 -30
  278. package/src/contracts/pi.ts +0 -261
  279. package/src/contracts/prefix-cache.ts +0 -33
  280. package/src/contracts/resources.ts +0 -25
  281. package/src/contracts/sampling.ts +0 -23
  282. package/src/contracts/scheduling.ts +0 -54
  283. package/src/contracts/tsconfig.json +0 -14
  284. package/src/dataset/generators.ts +0 -1087
  285. package/src/dataset/index.ts +0 -12
  286. package/src/dataset/job.ts +0 -81
  287. package/src/dataset/llm.ts +0 -71
  288. package/src/dataset/registry.ts +0 -528
  289. package/src/diffusion/diffusion-generate.ts +0 -507
  290. package/src/download.ts +0 -505
  291. package/src/embed.ts +0 -77
  292. package/src/engine/admission.ts +0 -59
  293. package/src/engine/cancellation.ts +0 -38
  294. package/src/engine/engine.ts +0 -56
  295. package/src/engine/execution-coordinator.ts +0 -70
  296. package/src/engine/execution-plan.ts +0 -64
  297. package/src/engine/execution-tasks.ts +0 -44
  298. package/src/engine/namespaced-cache.ts +0 -13
  299. package/src/engine/resources.ts +0 -53
  300. package/src/engine/scheduler.ts +0 -92
  301. package/src/engine/session.ts +0 -247
  302. package/src/eval/README.md +0 -130
  303. package/src/eval/ifeval.ts +0 -81
  304. package/src/eval/kl-prompts.ts +0 -29
  305. package/src/eval/kl.ts +0 -590
  306. package/src/eval/perplexity.ts +0 -164
  307. package/src/eval/quality-db.ts +0 -92
  308. package/src/eval/reference-logits.ts +0 -207
  309. package/src/eval/runner.ts +0 -124
  310. package/src/eval/score.ts +0 -49
  311. package/src/eval/tasks/bfcl.ts +0 -535
  312. package/src/eval/tasks/gsm8k.ts +0 -98
  313. package/src/eval/tasks/hashhop.ts +0 -139
  314. package/src/eval/tasks/humaneval.ts +0 -275
  315. package/src/eval/tasks/ifeval.ts +0 -525
  316. package/src/eval/tasks/mmlu.ts +0 -157
  317. package/src/eval/ultrafeedback.ts +0 -60
  318. package/src/evaldb.ts +0 -199
  319. package/src/expert-io.ts +0 -393
  320. package/src/expert-offload-build.ts +0 -77
  321. package/src/expert-offload.ts +0 -76
  322. package/src/expert-residency.ts +0 -1134
  323. package/src/expert-usage.ts +0 -415
  324. package/src/fill/echo-index.ts +0 -188
  325. package/src/fill/fill-session.ts +0 -463
  326. package/src/fill/proposal.ts +0 -68
  327. package/src/fill/schema-rows.ts +0 -506
  328. package/src/fill/tool-boundary.ts +0 -103
  329. package/src/fit.ts +0 -249
  330. package/src/generate.ts +0 -1352
  331. package/src/grammar.ts +0 -604
  332. package/src/harness-pi.ts +0 -184
  333. package/src/hf-push.ts +0 -628
  334. package/src/hub-rest.ts +0 -242
  335. package/src/index.ts +0 -63
  336. package/src/inference/denoising.ts +0 -17
  337. package/src/inference/draft-acceptance.ts +0 -41
  338. package/src/inference/draft-output.ts +0 -49
  339. package/src/inference/graph.ts +0 -33
  340. package/src/inference/prefill.ts +0 -45
  341. package/src/inference/rollback.ts +0 -7
  342. package/src/jobs/index.ts +0 -31
  343. package/src/jobs/job-entry.ts +0 -107
  344. package/src/jobs/task-client.ts +0 -19
  345. package/src/jobs/types.ts +0 -2
  346. package/src/kv-scheme.ts +0 -279
  347. package/src/kv-store.ts +0 -1135
  348. package/src/lab/README.md +0 -22
  349. package/src/lab/curve/curve-designer.html +0 -303
  350. package/src/lab/curve/curve-sampler.ts +0 -122
  351. package/src/lab/expert-trace/expert-trace.ts +0 -69
  352. package/src/lab/paged-kv/paged-attention.ts +0 -117
  353. package/src/lab/paged-kv/paged-kv-rows.ts +0 -111
  354. package/src/lab/paged-kv/paged-kv.ts +0 -336
  355. package/src/library.ts +0 -40
  356. package/src/lora.ts +0 -408
  357. package/src/md-modules.d.ts +0 -54
  358. package/src/media-fetch.ts +0 -442
  359. package/src/memory/article.ts +0 -558
  360. package/src/memory/chunk-validation.ts +0 -150
  361. package/src/memory/chunk.ts +0 -318
  362. package/src/memory/cluster.ts +0 -351
  363. package/src/memory/coverage.ts +0 -134
  364. package/src/memory/crosslink.ts +0 -0
  365. package/src/memory/db.ts +0 -253
  366. package/src/memory/entity.ts +0 -203
  367. package/src/memory/gate.ts +0 -187
  368. package/src/memory/history.ts +0 -152
  369. package/src/memory/ingest.ts +0 -335
  370. package/src/memory/model.ts +0 -183
  371. package/src/memory/normalize.ts +0 -671
  372. package/src/memory/parse.ts +0 -22
  373. package/src/memory/pipeline.ts +0 -275
  374. package/src/memory/prompts.ts +0 -65
  375. package/src/memory/query.ts +0 -334
  376. package/src/memory/reconcile.ts +0 -734
  377. package/src/memory/redirect.ts +0 -108
  378. package/src/memory/reindex.ts +0 -325
  379. package/src/memory/resolve.ts +0 -307
  380. package/src/memory/rest.ts +0 -245
  381. package/src/memory/route.ts +0 -272
  382. package/src/memory/schedule.ts +0 -165
  383. package/src/memory/stages.ts +0 -777
  384. package/src/memory/synthesize.ts +0 -1638
  385. package/src/memory/tools.ts +0 -749
  386. package/src/memory/vault.ts +0 -1194
  387. package/src/memory/wikify.ts +0 -1064
  388. package/src/mlx/array.ts +0 -425
  389. package/src/mlx/autograd.ts +0 -304
  390. package/src/mlx/checkpoint.ts +0 -149
  391. package/src/mlx/compile.ts +0 -168
  392. package/src/mlx/custom-vjp.ts +0 -188
  393. package/src/mlx/ffi.ts +0 -531
  394. package/src/mlx/materialize.ts +0 -23
  395. package/src/mlx/metal-kernel.ts +0 -159
  396. package/src/mlx/normalized-argmax.ts +0 -116
  397. package/src/mlx/ops.ts +0 -1108
  398. package/src/mlx/quantized-rows.ts +0 -36
  399. package/src/mlx/token-bitmask.ts +0 -52
  400. package/src/mlx/turboquant-kv-decode.ts +0 -82
  401. package/src/mlx/turboquant-kv-inverse.ts +0 -122
  402. package/src/mlx/turboquant-ops.ts +0 -861
  403. package/src/mlx/turboquant-tables.ts +0 -237
  404. package/src/mmap.ts +0 -90
  405. package/src/model/artifact-identity.ts +0 -73
  406. package/src/model/batched-kv.ts +0 -235
  407. package/src/model/batched-mask.ts +0 -279
  408. package/src/model/batched-quant.ts +0 -212
  409. package/src/model/batched-quantized-kv.ts +0 -144
  410. package/src/model/batched-rotating-quant.ts +0 -353
  411. package/src/model/batched-rotating-state.ts +0 -149
  412. package/src/model/batched-rotating.ts +0 -338
  413. package/src/model/batched-row-storage.ts +0 -179
  414. package/src/model/batched-ssm.ts +0 -52
  415. package/src/model/batched-turboquant-kv.ts +0 -94
  416. package/src/model/compiled-decode.ts +0 -712
  417. package/src/model/decoded-kv-donor.ts +0 -12
  418. package/src/model/delayed-quantized-kv.ts +0 -72
  419. package/src/model/delayed-rotating-quantized-kv.ts +0 -93
  420. package/src/model/delayed-turboquant-kv.ts +0 -57
  421. package/src/model/diffusion-gemma.ts +0 -1026
  422. package/src/model/factory.ts +0 -215
  423. package/src/model/fingerprint.ts +0 -45
  424. package/src/model/flash-attention.ts +0 -694
  425. package/src/model/full-prefill-padding.ts +0 -56
  426. package/src/model/full-prefill-row.ts +0 -105
  427. package/src/model/gemma4-base.ts +0 -2370
  428. package/src/model/gemma4.ts +0 -1395
  429. package/src/model/generated/gemma4-12b.ts +0 -429
  430. package/src/model/generated/gemma4-26b.ts +0 -448
  431. package/src/model/generated/gemma4-e4b.ts +0 -717
  432. package/src/model/generated/index.ts +0 -18
  433. package/src/model/glm52-atlas.ts +0 -511
  434. package/src/model/glm52-cache.ts +0 -792
  435. package/src/model/glm52-config.ts +0 -262
  436. package/src/model/glm52-container.ts +0 -305
  437. package/src/model/glm52-coupling.ts +0 -388
  438. package/src/model/glm52-dsa.ts +0 -606
  439. package/src/model/glm52-expert-layout.ts +0 -253
  440. package/src/model/glm52-layout.ts +0 -152
  441. package/src/model/glm52-memory.ts +0 -500
  442. package/src/model/glm52-mla.ts +0 -792
  443. package/src/model/glm52-moe.ts +0 -223
  444. package/src/model/glm52-pilot.ts +0 -537
  445. package/src/model/glm52-reference.ts +0 -445
  446. package/src/model/glm52-residency.ts +0 -557
  447. package/src/model/glm52-resident-weights.ts +0 -135
  448. package/src/model/glm52-streamed-experts.ts +0 -453
  449. package/src/model/glm52-streamed-metal.ts +0 -610
  450. package/src/model/glm52-weights.ts +0 -503
  451. package/src/model/glm52.ts +0 -1132
  452. package/src/model/implementation.ts +0 -69
  453. package/src/model/kv-attention-view.ts +0 -70
  454. package/src/model/minicpm5.ts +0 -376
  455. package/src/model/profile.ts +0 -468
  456. package/src/model/quantized-append-attention.ts +0 -31
  457. package/src/model/qwen-append.ts +0 -12
  458. package/src/model/qwen-conv.ts +0 -52
  459. package/src/model/qwen3-delta-state.ts +0 -74
  460. package/src/model/qwen3-delta.ts +0 -451
  461. package/src/model/qwen3-moe.ts +0 -404
  462. package/src/model/qwen3-mrope.ts +0 -250
  463. package/src/model/qwen3.ts +0 -277
  464. package/src/model/qwen3_5-checkpoint.ts +0 -164
  465. package/src/model/qwen3_5.ts +0 -881
  466. package/src/model/rotating-kv-donor.ts +0 -35
  467. package/src/model/rotating-kv-layout.ts +0 -262
  468. package/src/model/rotating-row-transaction.ts +0 -35
  469. package/src/model/speculative-rotating-kv.ts +0 -101
  470. package/src/model/ssm-prefill-padding.ts +0 -39
  471. package/src/model/support.ts +0 -110
  472. package/src/model/token-groups.ts +0 -66
  473. package/src/model/trellis-balanced-scatter.ts +0 -51
  474. package/src/model/trellis-linear.ts +0 -672
  475. package/src/model/trellis-shared-m.ts +0 -151
  476. package/src/model/trellis-shared-scatter.ts +0 -144
  477. package/src/model/trellis-splitk-prefill.ts +0 -158
  478. package/src/model/trellis-tiled-prefill.ts +0 -136
  479. package/src/model/trellis-vector-expand.ts +0 -62
  480. package/src/model/trellis-wide-prefill.ts +0 -94
  481. package/src/model/turboquant-codec.ts +0 -89
  482. package/src/model/universal/archs.ts +0 -332
  483. package/src/model/universal/dense.ts +0 -612
  484. package/src/model/universal/modules.ts +0 -170
  485. package/src/model/universal/rope.ts +0 -381
  486. package/src/model/whisper.ts +0 -435
  487. package/src/native/expert_io.c +0 -523
  488. package/src/native/frame_extract.swift +0 -140
  489. package/src/native/mic_capture.swift +0 -92
  490. package/src/native-pack.ts +0 -163
  491. package/src/pi-provider.ts +0 -53
  492. package/src/pi-session.ts +0 -81
  493. package/src/pi-terminal.ts +0 -274
  494. package/src/pi-web.ts +0 -1940
  495. package/src/preflight.ts +0 -158
  496. package/src/prompt-cache.ts +0 -682
  497. package/src/quantize/allocator.ts +0 -627
  498. package/src/quantize/atomic-output.ts +0 -38
  499. package/src/quantize/calibration.ts +0 -181
  500. package/src/quantize/config-writer.ts +0 -145
  501. package/src/quantize/index.ts +0 -84
  502. package/src/quantize/job.ts +0 -162
  503. package/src/quantize/quantizer.ts +0 -666
  504. package/src/quantize/rotate.ts +0 -603
  505. package/src/quantize/safetensors-writer.ts +0 -249
  506. package/src/quantize/sensitivity.ts +0 -216
  507. package/src/quantize/trellis.ts +0 -586
  508. package/src/quantize/weight-transform.ts +0 -292
  509. package/src/registry.ts +0 -602
  510. package/src/runtime-config.ts +0 -77
  511. package/src/safetensors.ts +0 -165
  512. package/src/sampler.ts +0 -1012
  513. package/src/serve/admin-routes.ts +0 -184
  514. package/src/serve/assets/curve-terrain.html +0 -105
  515. package/src/serve/assets/training-inference-map.html +0 -575
  516. package/src/serve/audio-routes.ts +0 -323
  517. package/src/serve/aux-routes.ts +0 -132
  518. package/src/serve/batch-scheduler.ts +0 -8
  519. package/src/serve/checkpoint-identity.ts +0 -64
  520. package/src/serve/generation-gateway.ts +0 -537
  521. package/src/serve/isolate.ts +0 -609
  522. package/src/serve/kv-budget.ts +0 -14
  523. package/src/serve/lab-routes.ts +0 -341
  524. package/src/serve/lane-registry.ts +0 -45
  525. package/src/serve/model-admin-routes.ts +0 -155
  526. package/src/serve/model-binding.ts +0 -49
  527. package/src/serve/model-host.ts +0 -511
  528. package/src/serve/parent-application.ts +0 -56
  529. package/src/serve/prompt-response-trace.ts +0 -161
  530. package/src/serve/response-lease.ts +0 -24
  531. package/src/serve/session-search.ts +0 -326
  532. package/src/serve/static-routes.ts +0 -101
  533. package/src/serve/transcription-server.ts +0 -64
  534. package/src/serve/transcription-service.ts +0 -502
  535. package/src/server.ts +0 -1282
  536. package/src/spec/assistant-rows.ts +0 -93
  537. package/src/spec/assistant-source.ts +0 -126
  538. package/src/spec/cache-identity.ts +0 -22
  539. package/src/spec/deepspec-rows.ts +0 -29
  540. package/src/spec/deepspec-source.ts +0 -145
  541. package/src/spec/dflash-source.ts +0 -149
  542. package/src/spec/drafter.ts +0 -304
  543. package/src/spec/dspark/ab-stats.ts +0 -136
  544. package/src/spec/dspark/calibration.ts +0 -112
  545. package/src/spec/dspark/data-dflash.ts +0 -156
  546. package/src/spec/dspark/data.ts +0 -201
  547. package/src/spec/dspark/deepspec-module.ts +0 -776
  548. package/src/spec/dspark/generate-dflash.ts +0 -245
  549. package/src/spec/dspark/generate.ts +0 -290
  550. package/src/spec/dspark/loader.ts +0 -23
  551. package/src/spec/dspark/loss.ts +0 -213
  552. package/src/spec/dspark/module-dflash.ts +0 -560
  553. package/src/spec/dspark/module.ts +0 -540
  554. package/src/spec/dspark/quantize-drafter.ts +0 -91
  555. package/src/spec/dspark/sample.ts +0 -112
  556. package/src/spec/generate.ts +0 -276
  557. package/src/spec/glm52-mtp-graph.ts +0 -29
  558. package/src/spec/glm52-mtp-rows.ts +0 -132
  559. package/src/spec/glm52-mtp-source.ts +0 -318
  560. package/src/spec/glm52-mtp-state.ts +0 -33
  561. package/src/spec/ngram-source.ts +0 -226
  562. package/src/spec/projected-draft-rows.ts +0 -73
  563. package/src/spec/qwen-mtp-module.ts +0 -171
  564. package/src/spec/qwen-mtp-rows.ts +0 -230
  565. package/src/spec/qwen-mtp-source.ts +0 -386
  566. package/src/spec/qwen-mtp-state.ts +0 -30
  567. package/src/spec/serve-loop.ts +0 -587
  568. package/src/spec/source.ts +0 -228
  569. package/src/spec/two-model-rows.ts +0 -62
  570. package/src/spec/two-model.ts +0 -113
  571. package/src/ssd-cache.ts +0 -431
  572. package/src/ssd-durability.ts +0 -209
  573. package/src/storage/host-buffer.ts +0 -24
  574. package/src/storage/kv-writer.ts +0 -83
  575. package/src/storage/kv-writer.worker.js +0 -264
  576. package/src/storage/retention-policy.ts +0 -37
  577. package/src/tiered-prompt-cache.ts +0 -45
  578. package/src/tokenizer.ts +0 -335
  579. package/src/tool-approvals.ts +0 -131
  580. package/src/tool-call.ts +0 -695
  581. package/src/train/dataset.ts +0 -484
  582. package/src/train/diffusion-lora.ts +0 -216
  583. package/src/train/export.ts +0 -32
  584. package/src/train/flash-cce.ts +0 -1234
  585. package/src/train/forward.ts +0 -266
  586. package/src/train/fuse.ts +0 -243
  587. package/src/train/index.ts +0 -46
  588. package/src/train/job.ts +0 -205
  589. package/src/train/lora-params.ts +0 -254
  590. package/src/train/loss.ts +0 -1591
  591. package/src/train/merge.ts +0 -195
  592. package/src/train/optimizer.ts +0 -183
  593. package/src/train/prefix-shared.ts +0 -490
  594. package/src/train/rank.ts +0 -137
  595. package/src/train/segmented.ts +0 -1828
  596. package/src/train/steel-qmm-header.ts +0 -1461
  597. package/src/train/trainer.ts +0 -1221
  598. package/src/train/watch.ts +0 -542
  599. package/src/vision/diffusion-vision.ts +0 -620
  600. package/src/vision/embedder.ts +0 -176
  601. package/src/vision/preprocess.ts +0 -247
  602. package/src/vision/prompt.ts +0 -409
  603. package/src/vision/qwen3vl-preprocess.ts +0 -409
  604. package/src/vision/qwen3vl-prompt.ts +0 -233
  605. package/src/vision/qwen3vl-tower.ts +0 -566
  606. package/src/vision/siglip.ts +0 -607
  607. package/src/vision/video-frames.ts +0 -148
  608. package/src/web/app.html +0 -2155
  609. package/src/web/app.js +0 -5863
  610. package/src/web/skills/memory/SKILL.md +0 -120
  611. package/src/web/skills/web-research/SKILL.md +0 -46
  612. package/src/web/skills.ts +0 -97
  613. package/src/web/src/adapters-panel.ts +0 -288
  614. package/src/web/src/assistant.ts +0 -451
  615. package/src/web/src/chat.ts +0 -1060
  616. package/src/web/src/composer.ts +0 -962
  617. package/src/web/src/dataset.ts +0 -132
  618. package/src/web/src/finetune.ts +0 -267
  619. package/src/web/src/hub.ts +0 -428
  620. package/src/web/src/main.ts +0 -98
  621. package/src/web/src/markdown.ts +0 -0
  622. package/src/web/src/memory-panel.ts +0 -572
  623. package/src/web/src/model-picker.ts +0 -159
  624. package/src/web/src/palette.ts +0 -346
  625. package/src/web/src/protocol.ts +0 -26
  626. package/src/web/src/quantize.ts +0 -167
  627. package/src/web/src/rag.ts +0 -274
  628. package/src/web/src/sessions.ts +0 -469
  629. package/src/web/src/shell.ts +0 -804
  630. package/src/web/src/ui-catalog.ts +0 -78
  631. package/src/web/src/voice.ts +0 -141
  632. package/src/web/vendor/README.md +0 -61
  633. package/src/web-assets.ts +0 -47
  634. package/src/web-tools.ts +0 -478
  635. package/src/weights.ts +0 -255
  636. /package/src/{serve → server}/completion-sink.ts +0 -0
  637. /package/src/{serve → server}/generated-token-history.ts +0 -0
  638. /package/src/web/{icon.svg → public/icon.svg} +0 -0
  639. /package/src/web/{manifest.webmanifest → public/manifest.webmanifest} +0 -0
  640. /package/src/web/{vendor → public/vendor}/hljs-LICENSE +0 -0
@@ -0,0 +1,50 @@
1
+ // What a model holds once it is resident, for residency decisions: its weights
2
+ // plus the KV cache and prefill working set of a typical serving context, from
3
+ // the same fit model `/fit` reports. Loading replaces the estimate with what the
4
+ // loaded model reports (its real weights bytes and the state it carries).
5
+ import { totalmem } from "node:os";
6
+ import type { ModelConfig } from "@mlx-bun/inference/artifacts/config";
7
+ import { fit, WIRED_FRACTION, type MachineSpec } from "@mlx-bun/inference/execution/fit";
8
+ import type { KvSchemeOptions } from "@mlx-bun/inference/state/kv-scheme";
9
+ import type { ModelRecord } from "@mlx-bun/hub/registry";
10
+
11
+ /** The context the estimate reserves KV for; longer contexts grow into the budget's headroom or evict others. */
12
+ export const SERVING_CONTEXT_TOKENS = 8192;
13
+
14
+ /** The share of the device working set all resident models may use together by default; the rest is headroom for
15
+ * KV growth, prefill transients and the system. */
16
+ export const DEFAULT_BUDGET_FRACTION = 0.7;
17
+
18
+ /** The default residency budget, one rule for both compositions: a share of the device working set (Metal's recommended
19
+ * maximum, which `cache-services` also plans against), or of the RAM the GPU may wire (`fit`'s rule) when the device does
20
+ * not say. The in-process host reads the working set itself; an isolated host is told it by its first worker. */
21
+ export function defaultBudgetBytes(workingSetBytes?: number): number {
22
+ const workingSet = workingSetBytes !== undefined && workingSetBytes > 0 ? workingSetBytes : totalmem() * WIRED_FRACTION;
23
+ return Math.floor(workingSet * DEFAULT_BUDGET_FRACTION);
24
+ }
25
+
26
+ /** Bytes a model's KV and prefill working set take at the serving context. */
27
+ export function servingReserveBytes(config: ModelConfig, weightsBytes: number,
28
+ options: { expertsBytes?: number; kvScheme?: KvSchemeOptions; machine?: MachineSpec } = {}): number {
29
+ const context = Math.max(1, Math.min(SERVING_CONTEXT_TOKENS, config.text.maxPositionEmbeddings));
30
+ const report = fit(config, weightsBytes, context, options.machine, undefined, options.expertsBytes ?? 0, undefined, options.kvScheme);
31
+ const reserve = report.kvBytes + report.transientBytes;
32
+ // A config the fit model cannot size adds nothing: residency then counts the weights alone.
33
+ return Number.isFinite(reserve) && reserve > 0 ? reserve : 0;
34
+ }
35
+
36
+ /** What loading `record` would take: its weights and a typical context's KV and working set. A runtime that plans its
37
+ * own memory is served alone, so it needs the whole budget. Reads the model's config only; loads nothing. */
38
+ export async function estimateRecordBytes(record: ModelRecord, input: {
39
+ budgetBytes: number; machine?: MachineSpec;
40
+ cache: { kvQuant?: "off" | "config" | number; turboQuant?: import("@mlx-bun/inference/artifacts/config").TurboQuantScheme; quantizedKvStart?: number };
41
+ }): Promise<number> {
42
+ const [{ loadModelConfig }, { resolveModelProfile }, { plansMemory }, { resolveKvScheme }] = await Promise.all([
43
+ import("@mlx-bun/inference/artifacts/config"), import("@mlx-bun/inference/models/profile"), import("@mlx-bun/inference/models/memory-plan"),
44
+ import("@mlx-bun/inference/state/kv-scheme")]);
45
+ const config = await loadModelConfig(record.path);
46
+ if (plansMemory(resolveModelProfile(config))) return input.budgetBytes;
47
+ const kvScheme = resolveKvScheme({ override: input.cache.kvQuant, turboQuant: input.cache.turboQuant,
48
+ quantizedKvStart: input.cache.quantizedKvStart, config: config.kvQuant }).fitOptions;
49
+ return record.sizeBytes + servingReserveBytes(config, record.sizeBytes, { expertsBytes: record.expertsBytes, kvScheme, ...(input.machine ? { machine: input.machine } : {}) });
50
+ }
@@ -8,7 +8,7 @@
8
8
  //
9
9
  // Documented upgrades over the oracle (possible because our
10
10
  // chat-completions layer is natively capable where mlx-lm's gemma path
11
- // is not — same divergence class as the Phase 4 bidirectional-mask fix):
11
+ // is not — same divergence class as the bidirectional-mask fix):
12
12
  // - tools: Anthropic `tools`/`tool_use`/`tool_result` map onto our REAL
13
13
  // OpenAI tools surface (token-level gemma tool calling) instead of
14
14
  // the oracle's Qwen-style <tool_call> inline-text hack ("out of
@@ -22,11 +22,11 @@
22
22
  // - prior-turn `thinking` blocks are dropped on re-ingest (Anthropic
23
23
  // semantics) instead of the oracle's json.dumps fallback.
24
24
 
25
- import type { ChatMessage, ToolDefinition } from "./chat-template";
25
+ import type { ChatMessage, ToolDefinition } from "@mlx-bun/inference/input";
26
26
  import type {
27
27
  CompletionEvent,
28
28
  CompletionStreamProtocol,
29
- } from "./serve/completion-sink";
29
+ } from "./completion-sink";
30
30
 
31
31
  // ---------------------------------------------------------------------
32
32
  // Request translation: Anthropic → OpenAI (our ChatRequest shape)
@@ -545,81 +545,3 @@ export function createAnthropicStreamProtocol(model: string): CompletionStreamPr
545
545
  },
546
546
  };
547
547
  }
548
-
549
- /** Wrap our OpenAI SSE byte stream as an Anthropic SSE byte stream.
550
- * Frames translate as they arrive (per-chunk flush preserved — the
551
- * upstream already rate-limits socket flushes). */
552
- export function translateOpenAiSse(
553
- upstream: ReadableStream<Uint8Array>,
554
- model: string,
555
- ): ReadableStream<Uint8Array> {
556
- const translator = new AnthropicStreamTranslator(model);
557
- const enc = new TextEncoder();
558
- const dec = new TextDecoder();
559
- let buf = "";
560
- let finalized = false;
561
- let reader: ReadableStreamDefaultReader<Uint8Array> | null = null;
562
- let cancelled = false;
563
- return new ReadableStream<Uint8Array>({
564
- start(controller) {
565
- const emit = (frames: string[]) => {
566
- for (const f of frames) controller.enqueue(enc.encode(f));
567
- };
568
- reader = upstream.getReader();
569
- void (async () => {
570
- try {
571
- while (!cancelled) {
572
- const { done, value } = await reader!.read();
573
- if (done) break;
574
- buf += dec.decode(value, { stream: true });
575
- let sep: number;
576
- while ((sep = buf.indexOf("\n\n")) !== -1) {
577
- const frame = buf.slice(0, sep);
578
- buf = buf.slice(sep + 2);
579
- const data = frame
580
- .split("\n")
581
- .find((l) => l.startsWith("data: "))
582
- ?.slice(6);
583
- if (data == null) continue;
584
- if (data === "[DONE]" || data === '"[DONE]"') {
585
- if (!finalized) {
586
- finalized = true;
587
- emit(translator.finalize());
588
- }
589
- continue;
590
- }
591
- const parsed = JSON.parse(data);
592
- if (parsed?.error) {
593
- emit([
594
- sse("error", {
595
- type: "error",
596
- error: { type: "api_error", message: parsed.error.message ?? "generation failed" },
597
- }),
598
- ]);
599
- continue;
600
- }
601
- emit(translator.addChunk(parsed));
602
- }
603
- }
604
- if (!cancelled && !finalized) emit(translator.finalize());
605
- } catch (e) {
606
- if (!cancelled)
607
- emit([
608
- sse("error", {
609
- type: "error",
610
- error: { type: "api_error", message: (e as Error).message },
611
- }),
612
- ]);
613
- } finally {
614
- reader?.releaseLock();
615
- reader = null;
616
- if (!cancelled) controller.close();
617
- }
618
- })();
619
- },
620
- async cancel(reason) {
621
- cancelled = true;
622
- await reader?.cancel(reason);
623
- },
624
- });
625
- }
@@ -0,0 +1,36 @@
1
+ import type { KvCacheService } from "@mlx-bun/inference/state";
2
+ import { errorResponse } from "./http";
3
+
4
+ export type CacheRouteServices = Pick<KvCacheService, "closeSession" | "flush" | "report">;
5
+
6
+ /** Administrative HTTP policy over the loaded model's cache service, preserving
7
+ * main's `/admin/cache/*` wire contract. Both routes borrow the service: no
8
+ * execution lease is taken, a session close never waits for a flush in
9
+ * progress, and a close neither deletes checkpoints nor rejects an unknown
10
+ * session. The flush answer adds the service's own report of saved state. */
11
+ export function createCacheRoutes(cache: CacheRouteServices) {
12
+ return { async handle(request: Request): Promise<Response | null> {
13
+ const route = `${request.method} ${new URL(request.url).pathname}`;
14
+ if (route !== "POST /admin/cache/session/close" && route !== "POST /admin/cache/flush") return null;
15
+ try {
16
+ if (route === "POST /admin/cache/session/close") {
17
+ let body: unknown;
18
+ try {
19
+ const text = await request.text();
20
+ body = text.trim() ? JSON.parse(text) : {};
21
+ } catch (error) {
22
+ if (request.signal.aborted) return errorResponse(error, route, undefined, request.signal);
23
+ return Response.json({ error: { message: "invalid JSON body" } }, { status: 400 });
24
+ }
25
+ const id = body && typeof body === "object" && !Array.isArray(body) && "session_id" in body ? body.session_id : undefined;
26
+ if (typeof id === "string") cache.closeSession(id);
27
+ return Response.json({ closed: typeof id === "string" });
28
+ }
29
+ const result = await cache.flush();
30
+ const saved = cache.report().saved;
31
+ return Response.json({ ...result, entries: saved?.entries ?? 0,
32
+ longest_durable_prefix_tokens: saved?.longestDurablePrefixTokens ?? 0 },
33
+ { status: result.durable ? 200 : 503 });
34
+ } catch (error) { return errorResponse(error, route, undefined, request.signal); }
35
+ } };
36
+ }
@@ -1,9 +1,9 @@
1
1
  // Wire-level chat/completion request shapes and the pure per-request
2
2
  // helpers: message normalization, sampling-field coercion, grammar-degrade
3
3
  // prompt injection, and the mlx_lm.server-parity validators. No model, no
4
- // server state. Extracted from src/server.ts (repo-taming Phase 4).
5
- import type { ChatMessage, ToolDefinition } from "../chat-template";
6
- import type { HlgConfig } from "../sampler";
4
+ // server state.
5
+ import type { ChatMessage, ToolDefinition } from "@mlx-bun/inference/input";
6
+ import type { HlgConfig } from "@mlx-bun/inference/sampling";
7
7
  import { RequestError } from "./pipeline";
8
8
 
9
9
  export interface ChatRequestParams {
@@ -56,7 +56,9 @@ export interface ChatRequestParams {
56
56
  * adds preserve_thinking (keep think blocks from history; template default
57
57
  * true — better prompt-cache reuse in agent loops). */
58
58
  chat_template_kwargs?: {
59
- enable_thinking?: boolean;
59
+ /** Absent inherits server/model defaults; a boolean overrides them.
60
+ * null leaves thinking to the chat template itself, bypassing both defaults. */
61
+ enable_thinking?: boolean | null;
60
62
  preserve_thinking?: boolean;
61
63
  [key: string]: unknown;
62
64
  };
@@ -70,7 +72,7 @@ export interface ChatRequestParams {
70
72
  /** Mounted LoRA adapter selection: "id", "a+b" (stacked), or "none". */
71
73
  adapter?: string;
72
74
  /** HLG tone-curve sampling override (per request). Snake_case wire fields,
73
- * merged over the server's --hlg-sampling config. docs/archive/hlg-sampling.md. */
75
+ * merged over the server's --hlg-sampling config. `02d723a:docs/archive/hlg-sampling.md`. */
74
76
  hlg?: {
75
77
  enabled?: boolean;
76
78
  width?: number;
@@ -80,7 +82,7 @@ export interface ChatRequestParams {
80
82
  };
81
83
  /** OpenAI structured output: {type:"json_object"} | {type:"json_schema",
82
84
  * json_schema:{name,schema,strict?}} | {type:"text"}. Enforced at the
83
- * sampler via xgrammar token-bitmasks (src/grammar.ts). L2-class (oMLX
85
+ * sampler via xgrammar token-bitmasks (packages/inference/src/sampling/grammar.ts). L2-class (oMLX
84
86
  * oracle). On compile failure, degrades to a system-prompt injection +
85
87
  * Warning header (oMLX parity), never 500. */
86
88
  response_format?: unknown;
@@ -232,10 +234,8 @@ export function parseLogitBias(
232
234
  }
233
235
 
234
236
  /** Default seed for a request that didn't pin one. `Date.now()` alone is NOT
235
- * request-unique: under `--batch N` the batch lane serves ONLY default-seed
236
- * requests (explicit-seed requests use the serial mechanism, see
237
- * GenerationGateway.place),
238
- * so two identical prompts arriving in the same millisecond would share a seed
237
+ * request-unique: concurrent requests share the continuous batch, so two
238
+ * identical prompts arriving in the same millisecond would share a seed
239
239
  * and — with per-row RNG keyed as stepKey(seed, generatedCount) — produce
240
240
  * byte-identical completions, silently collapsing best-of-N diversity. Mix a
241
241
  * per-process Weyl counter (golden-ratio increment, period 2^32) into the
@@ -4,16 +4,16 @@
4
4
  // controller, and the token→text pipeline the model needs. Every chat-shaped
5
5
  // surface (OpenAI chat, Anthropic Messages, Responses) runs this stage; the
6
6
  // wire format is not its business.
7
- import type { PromptCache } from "../prompt-cache";
7
+ import type { KvCacheService } from "@mlx-bun/inference/state";
8
8
  import {
9
9
  applyGrammarDegrade,
10
10
  type ChatRequest,
11
11
  type ChatRequestParams,
12
12
  } from "./chat-request";
13
13
  import type { InferenceRequest } from "./inference-request";
14
- import type { ServingContext } from "./model-host";
15
- import type { BuiltPrompt, ModelPromptBuilder, PromptNativeWork } from "./model-binding";
16
- import { modelPromptBuilder } from "../backends/mlx/model-serving";
14
+ import type { LoadedModelContext as ModelContext } from "../engine/model-host";
15
+ import type { BuiltPrompt, ModelPromptBuilder, PromptNativeWork } from "./prompt-contracts";
16
+ import { modelPromptBuilder } from "./prompt-builder";
17
17
  import { RequestError } from "./pipeline";
18
18
  import { RequestOwnership } from "./request-plan";
19
19
  import type { RequestPrep } from "./request-prep";
@@ -21,15 +21,15 @@ import { StopMatcher, ThinkingTagSplitter } from "./token-streams";
21
21
 
22
22
  export class ChatStage {
23
23
  constructor(
24
- private readonly ctx: ServingContext,
24
+ private readonly ctx: ModelContext,
25
25
  private readonly prep: RequestPrep,
26
- private readonly promptCache: Pick<PromptCache, "peekPrefixLen"> & Partial<Pick<PromptCache, "objects">>,
27
- /** Admission ceiling (fit() / the GLM memory plan), resolved once. */
26
+ private readonly promptCache: Pick<KvCacheService, "peekPrefixLen"> & Partial<Pick<KvCacheService, "objects">>,
27
+ /** Admission ceiling (fit() / a runtime memory plan), resolved once. */
28
28
  private readonly contextLimit: number | null,
29
29
  /** `serve --adapter <dir>` startup default; a request's explicit
30
30
  * `adapter` (incl. "none") wins. */
31
31
  private readonly defaultAdapter?: string,
32
- private readonly preparation?: import("./preparation").PreparationExecutor,
32
+ private readonly preparation?: import("../engine/preparation").PreparationExecutor,
33
33
  private readonly buildPrompt: ModelPromptBuilder = modelPromptBuilder(ctx),
34
34
  ) {}
35
35
 
@@ -78,12 +78,12 @@ export class ChatStage {
78
78
  throw new RequestError(status, message);
79
79
  };
80
80
 
81
- // Grammar-constrained decoding (src/grammar.ts). Compile BEFORE prompt
81
+ // Grammar-constrained decoding (packages/inference/src/sampling/grammar.ts). Compile BEFORE prompt
82
82
  // rendering: on the degrade path (compile failed but a constraint was
83
83
  // requested) inject a system message instructing valid JSON so the
84
84
  // model still best-efforts schema-conformant output (oMLX parity —
85
85
  // _compile_grammar_for_request returning None + build_json_system_prompt).
86
- let grammarCtrl: import("../grammar").GrammarController | null = null;
86
+ let grammarCtrl: import("@mlx-bun/inference/sampling/grammar").GrammarController | null = null;
87
87
  const grammarReq =
88
88
  body.response_format != null || !!body.guided_grammar ||
89
89
  !!body.guided_regex || !!body.guided_choice?.length ||
@@ -137,10 +137,10 @@ export class ChatStage {
137
137
  // Token fast-forwarding (K3, MLX_BUN_FILL=strict): the determined-span
138
138
  // table for this request's tools. Refused for the shapes only this stage
139
139
  // can see — a compiled grammar (forced tokens are its job), media prompts
140
- // (embeddings prefill / mRoPE). Provider and cache-format support
141
- // are declared by the engine's method binding. The
140
+ // (embeddings prefill / mRoPE). Provider support
141
+ // is declared by the engine's method binding. The
142
142
  // body-level refusals live in prep.fillPlanFor.
143
- let fillSession: import("../fill/fill-session").FillSession | null = null;
143
+ let fillSession: import("@mlx-bun/inference/generation/fill").FillSession | null = null;
144
144
  if (!grammarCtrl && !vision && !diffusionPixels) {
145
145
  fillSession = prep.fillPlanFor(body, tools, promptIds);
146
146
  if (fillSession) options.fill = fillSession;
@@ -188,7 +188,7 @@ export class ChatStage {
188
188
  router: prep.toolRouter(tools, fillSession ? () => fillSession!.noteParseFailure() : undefined),
189
189
  stopper: new StopMatcher(options.stopSequences),
190
190
  thinking: new ThinkingTagSplitter(
191
- ctx.template.thinkingFormat === "think-tag",
191
+ ctx.template?.thinkingFormat === "think-tag",
192
192
  startInThinking,
193
193
  ),
194
194
  collectToolCalls: true,
@@ -1,5 +1,31 @@
1
- import type { CompletionClient } from "./contracts/completion";
2
- import type { EngineHost } from "./contracts/host";
1
+ // The public `mlx-bun/client` entry (main's src/client.ts with its host and
2
+ // completion contracts): an HTTP completion client over a borrowed transport,
3
+ // and an adapter that gives an in-process request handler the same host shape.
4
+ // It imports nothing at runtime, so a process without the MLX library can use it.
5
+ import type { Cancellation } from "@mlx-bun/inference/contracts/portable";
6
+
7
+ /** Transport-specific request/response values stay outside portable policy. */
8
+ export interface EngineHost<Request, Response> {
9
+ readonly ready: Promise<void>;
10
+ forward(request: Request): Promise<Response>;
11
+ /** Stop accepting work and wait until the owned worker has exited. */
12
+ close(): Promise<void>;
13
+ }
14
+
15
+ /** Application-facing calls. The implementation owns tokenization, numerical
16
+ * method and transport; callers provide request data and consume results. */
17
+ export interface CompletionClient<Request, Result> {
18
+ complete(request: Request): Promise<Result>;
19
+ }
20
+
21
+ export interface BatchCompletionClient<Request, Result> extends CompletionClient<Request, Result> {
22
+ /** Results preserve input order; the implementation chooses execution groups. */
23
+ completeBatch(requests: readonly Request[]): Promise<Result[]>;
24
+ }
25
+
26
+ export interface TaskClient<Request, Progress, Result> {
27
+ run(request: Request, report: (progress: Progress) => void, cancellation?: Cancellation): Promise<Result>;
28
+ }
3
29
 
4
30
  export interface CompletionCall {
5
31
  readonly body: Readonly<Record<string, unknown>>;
@@ -2,14 +2,13 @@ import type {
2
2
  GenerateOptions,
3
3
  GenerateStats,
4
4
  TokenLogprobs,
5
- } from "../generate";
5
+ } from "@mlx-bun/inference/generation";
6
6
  import type {
7
- GenerationMechanism,
8
7
  GenerationPlacement,
9
8
  OnToken,
10
9
  RequestShape,
11
10
  Vision,
12
- } from "./generation-gateway";
11
+ } from "../engine/completion";
13
12
  import {
14
13
  type CompletionEvent,
15
14
  type CompletionFinishReason,
@@ -19,29 +18,19 @@ import {
19
18
  CompletionSink,
20
19
  type CompletionToolCall,
21
20
  } from "./completion-sink";
22
- import { recordLane, type Lane } from "./lane-registry";
21
+ type Lane = "batched";
22
+ type GenerationMechanism = "continuous";
23
23
  import {
24
24
  planRequest,
25
25
  type RequestPlan,
26
26
  type RequestPlanInput,
27
27
  type RequestRejection,
28
28
  } from "./request-plan";
29
- import type { PromptResponseTrace } from "./prompt-response-trace";
30
- import { cleanupFailure } from "../engine/resources";
29
+ import type { PromptResponseTrace } from "@mlx-bun/inference/runtime/trace";
30
+ import { cleanupFailure } from "@mlx-bun/inference/runtime/resources";
31
31
 
32
- export interface CompletionEngine {
33
- place(shape: RequestShape, options?: GenerateOptions): GenerationPlacement;
34
- run(
35
- promptIds: number[],
36
- options: GenerateOptions & { stopSequences?: string[] },
37
- onToken: OnToken,
38
- vision: Vision | undefined,
39
- shape: RequestShape,
40
- placement: GenerationPlacement,
41
- signal?: AbortSignal,
42
- trace?: PromptResponseTrace,
43
- ): Promise<GenerateStats>;
44
- }
32
+ import type { CompletionEngine } from "../engine/completion";
33
+ export type { CompletionEngine } from "../engine/completion";
45
34
 
46
35
  export interface CompletionPreparation {
47
36
  requestId: string;
@@ -119,7 +108,7 @@ export interface CompletionUsage {
119
108
  completionTokens: number;
120
109
  totalTokens: number;
121
110
  speculation?: GenerateStats["spec"];
122
- /** Token fast-forwarding telemetry (serial lane, MLX_BUN_FILL=strict). */
111
+ /** Token fast-forwarding telemetry. */
123
112
  fill?: GenerateStats["fill"];
124
113
  }
125
114
 
@@ -191,20 +180,6 @@ class LogprobsCollector {
191
180
  }
192
181
  }
193
182
 
194
- function initialLane(mechanism: GenerationMechanism, shape: RequestShape): Lane {
195
- if (mechanism === "continuous") return "batched";
196
- return shape.hasDraft ? "serial+spec" : "serial";
197
- }
198
-
199
- function finalLane(
200
- mechanism: GenerationMechanism,
201
- lane: Lane,
202
- stats: GenerateStats,
203
- ): Lane {
204
- if (mechanism === "continuous") return "batched";
205
- return stats.spec ? "serial+spec" : "serial";
206
- }
207
-
208
183
  function finishReason(
209
184
  result: CompletionResult,
210
185
  stats: GenerateStats,
@@ -230,8 +205,28 @@ function combineControl(
230
205
  /** Owns one completion attempt from admission through terminal accounting.
231
206
  * Model math and scheduling policy remain in the supplied CompletionEngine. */
232
207
  export class CompletionExecutor {
208
+ readonly #placements = new WeakMap<PreparedCompletion, GenerationPlacement>();
233
209
  constructor(private readonly engine: CompletionEngine) {}
234
210
 
211
+ /** Resolve capability gaps before a transport commits streaming headers.
212
+ * A rejected placement releases the prepared request's owned resources. */
213
+ place(prepared: PreparedCompletion): GenerationPlacement {
214
+ const input = preparedData.get(prepared);
215
+ if (!input) throw new Error("prepared completion has already been executed");
216
+ const existing = this.#placements.get(prepared);
217
+ if (existing) return existing;
218
+ try {
219
+ const placement = this.engine.place(input.plan.shape, input.plan.options);
220
+ if (placement.shape !== input.plan.shape)
221
+ throw new Error("generation placement does not belong to this request shape");
222
+ this.#placements.set(prepared, placement);
223
+ return placement;
224
+ } catch (error) {
225
+ preparedData.delete(prepared);
226
+ return cleanupFailure(error, () => input.plan.dispose());
227
+ }
228
+ }
229
+
235
230
  async execute(
236
231
  prepared: PreparedCompletion,
237
232
  control: CompletionControl = {},
@@ -254,15 +249,13 @@ export class CompletionExecutor {
254
249
  try {
255
250
  if (usageProgress) control.onUsageProgress!(usageProgress);
256
251
  const closePlacement = control.trace?.begin("completion.placement");
257
- const enginePlacement = this.engine.place(planned.shape, planned.options);
252
+ const enginePlacement = this.#placements.get(prepared) ?? this.engine.place(planned.shape, planned.options);
253
+ this.#placements.delete(prepared);
258
254
  closePlacement?.();
259
255
  if (enginePlacement.shape !== planned.shape)
260
256
  throw new Error("generation placement does not belong to this request shape");
261
257
  const mechanism = enginePlacement.mechanism;
262
- const lane = enginePlacement.execution
263
- ? (mechanism === "continuous" ? "batched" : enginePlacement.execution.method === "speculative" ? "serial+spec" : "serial")
264
- : initialLane(mechanism, planned.shape);
265
- recordLane(input.requestId, lane);
258
+ const lane: Lane = "batched";
266
259
  const placement = { mechanism, lane, shape: planned.shape };
267
260
  input.onPlacement?.(placement);
268
261
  const flowControl = input.createFlowControl?.(placement);
@@ -318,8 +311,7 @@ export class CompletionExecutor {
318
311
  const result = sink.finish();
319
312
  if (result.events.length) await control.onEvents?.(result.events);
320
313
 
321
- const resolvedLane = finalLane(mechanism, lane, stats);
322
- if (resolvedLane !== lane) recordLane(input.requestId, resolvedLane);
314
+ const resolvedLane = lane;
323
315
  if (usageProgress) {
324
316
  usageProgress.promptTokens = stats.promptTokens;
325
317
  usageProgress.cachedTokens = stats.cachedTokens;
@@ -0,0 +1,55 @@
1
+ import { gunzipSync, zstdDecompressSync } from "node:zlib";
2
+ import { RequestError } from "./pipeline";
3
+
4
+ const MAX_BYTES = 32 * 1024 * 1024;
5
+ /** Preserve compressed and multipart bodies through model routing. The same
6
+ * parser finds the routing id and reads the final decision handler's input. */
7
+ export async function readDecisionBody(request: Request): Promise<unknown> {
8
+ const contentType = request.headers.get("content-type") ?? "application/json";
9
+ const encoding = request.headers.get("content-encoding")?.trim().toLowerCase();
10
+ const reader = request.body?.getReader(), chunks: Uint8Array[] = [];
11
+ let length = 0;
12
+ if (reader) try {
13
+ while (true) {
14
+ const { done, value } = await reader.read();
15
+ if (done) break;
16
+ length += value.length;
17
+ if (length > MAX_BYTES) { await reader.cancel(); throw new RequestError(413, "decision body exceeds 32 MiB"); }
18
+ chunks.push(value);
19
+ }
20
+ } finally { reader.releaseLock(); }
21
+ let bytes: Uint8Array = Buffer.concat(chunks, length);
22
+ try {
23
+ if (encoding === "gzip" || encoding === "x-gzip") bytes = gunzipSync(bytes, { maxOutputLength: MAX_BYTES });
24
+ else if (encoding === "zstd") bytes = zstdDecompressSync(bytes, { maxOutputLength: MAX_BYTES });
25
+ else if (encoding && encoding !== "identity") throw new RequestError(415, `unsupported Content-Encoding: ${encoding}`);
26
+ } catch (error) {
27
+ if (error instanceof RequestError) throw error;
28
+ if ((error as { code?: string }).code === "ERR_BUFFER_TOO_LARGE") throw new RequestError(413, "decompressed decision body exceeds 32 MiB");
29
+ throw new RequestError(400, "malformed compressed decision body");
30
+ }
31
+ try {
32
+ if (contentType.toLowerCase().startsWith("multipart/form-data")) {
33
+ const form = await new Response(new Uint8Array(bytes), { headers: { "content-type": contentType } }).formData();
34
+ const payload = form.get("request");
35
+ if (!payload) throw new Error("multipart body needs a request part");
36
+ const body = JSON.parse(typeof payload === "string" ? payload : await payload.text());
37
+ const used = new Set<string>(["request"]);
38
+ for (const asset of Object.values(body.assets ?? {}) as Record<string, unknown>[]) {
39
+ if (typeof asset?.part !== "string") continue;
40
+ if (used.has(asset.part)) throw new Error("duplicate image part");
41
+ used.add(asset.part);
42
+ const image = form.get(asset.part);
43
+ if (!(image instanceof File)) throw new Error("missing image part");
44
+ asset.format = asset.format ?? (image.type === "image/png" ? "png" : "jpg");
45
+ asset.data_base64 = Buffer.from(await image.arrayBuffer()).toString("base64");
46
+ delete asset.part;
47
+ }
48
+ if ([...form.keys()].some(key => !used.has(key)) || [...form.keys()].length !== used.size) throw new Error("unused or duplicate multipart part");
49
+ return body;
50
+ }
51
+ const json = new TextDecoder("utf-8", { fatal: true }).decode(bytes);
52
+ if (bytes[0] === 0xef && bytes[1] === 0xbb && bytes[2] === 0xbf) throw new Error("JSON BOM is unsupported");
53
+ return JSON.parse(json);
54
+ } catch (error) { throw new RequestError(400, error instanceof Error ? error.message : "invalid decision JSON body"); }
55
+ }