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.
- package/README.md +1449 -109
- package/THIRD_PARTY_NOTICES.md +882 -0
- package/bin/mlx-bun.convert.mjs +4 -0
- package/bin/mlx-bun.fuse.mjs +4 -0
- package/bin/mlx-bun.generate.mjs +4 -0
- package/bin/mlx-bun.lora.mjs +4 -0
- package/bin/mlx-bun.mjs +13 -21
- package/bin/mlx-bun.server.mjs +4 -0
- package/bin/mlx-bun.upload.mjs +4 -0
- package/dist/web/app.js +9584 -0
- package/package.json +50 -53
- package/src/cli/ane-install.ts +76 -0
- package/src/cli/args.ts +179 -0
- package/src/cli/engine-entry.ts +68 -0
- package/src/cli/inference.ts +214 -0
- package/src/cli/job-entry.ts +97 -0
- package/src/cli/library-host.ts +117 -0
- package/src/cli/main.ts +57 -0
- package/src/cli/memory-engine.ts +163 -0
- package/src/cli/memory.ts +36 -0
- package/src/cli/mlx-lm-aliases.ts +309 -0
- package/src/cli/model-choice.ts +33 -0
- package/src/cli/model-selection.ts +84 -0
- package/src/cli/module-host.ts +11 -0
- package/src/cli/module-verbs.ts +50 -0
- package/src/cli/numerical-policy.ts +12 -0
- package/src/cli/perplexity.ts +89 -0
- package/src/cli/serve-host.ts +423 -0
- package/src/cli/serve-isolated.ts +217 -0
- package/src/cli/serve-options.ts +106 -0
- package/src/cli/serve-state.ts +205 -0
- package/src/cli/serve.ts +500 -0
- package/src/cli/served-model-host.ts +102 -0
- package/src/cli/server-entry.ts +148 -0
- package/src/cli/serving-unit.ts +177 -0
- package/src/{tui.ts → cli/terminal.ts} +13 -8
- package/src/cli/worker-adapters.ts +34 -0
- package/src/cli/worker-entry.ts +272 -0
- package/src/cli/worker-unit.ts +179 -0
- package/src/engine/adapter-operation.ts +26 -0
- package/src/engine/cache-services.ts +114 -0
- package/src/engine/completion.ts +39 -0
- package/src/engine/generation-gateway.ts +358 -0
- package/src/engine/index.ts +59 -0
- package/src/engine/kv-budget.ts +120 -0
- package/src/engine/media-preparation.ts +232 -0
- package/src/engine/model-binding.ts +24 -0
- package/src/engine/model-host.ts +513 -0
- package/src/engine/model-serving.ts +89 -0
- package/src/{serve → engine}/preparation.ts +7 -5
- package/src/{serve → engine}/session-completion-engine.ts +6 -6
- package/src/engine/telemetry.ts +109 -0
- package/src/hub/downloads.ts +116 -0
- package/src/jobs/db.ts +16 -7
- package/src/jobs/events.ts +29 -0
- package/src/jobs/executable.ts +5 -0
- package/src/jobs/host.ts +51 -0
- package/src/jobs/in-process.ts +55 -0
- package/src/{contracts/jobs.ts → jobs/protocol.ts} +7 -9
- package/src/jobs/runner.ts +150 -231
- package/src/jobs/service.ts +75 -0
- package/src/jobs/sse.ts +59 -23
- package/src/jobs/worker-memory.ts +46 -0
- package/src/jobs/worker-process.ts +163 -0
- package/src/jobs/worker-supervisor.ts +258 -0
- package/src/modules.ts +58 -0
- package/src/publishing/catalog-hub.ts +18 -0
- package/src/publishing/credentials.ts +31 -0
- package/src/publishing/upload.ts +37 -0
- package/src/residency/leased-adapters.ts +21 -0
- package/src/residency/local-records.ts +16 -0
- package/src/residency/model-residency.ts +405 -0
- package/src/residency/record-index.ts +28 -0
- package/src/residency/resident-estimate.ts +50 -0
- package/src/{anthropic.ts → server/anthropic.ts} +3 -81
- package/src/server/cache-routes.ts +36 -0
- package/src/{serve → server}/chat-request.ts +10 -10
- package/src/{serve → server}/chat-stage.ts +14 -14
- package/src/{client.ts → server/client.ts} +28 -2
- package/src/{serve → server}/completion-executor.ts +33 -41
- package/src/server/decision-body.ts +55 -0
- package/src/server/decision-routes.ts +134 -0
- package/src/server/decision-wire.ts +228 -0
- package/src/{serve → server}/discovery-routes.ts +25 -82
- package/src/{serve → server}/http.ts +25 -13
- package/src/{serve → server}/inference-request.ts +5 -5
- package/src/server/job-routes.ts +20 -0
- package/src/server/media-prompt.ts +63 -0
- package/src/server/memory-completion-client.ts +46 -0
- package/src/server/model-routes.ts +192 -0
- package/src/{serve → server}/openai-wire.ts +3 -2
- package/src/{serve → server}/pipeline.ts +1 -1
- package/src/server/prompt-builder.ts +13 -0
- package/src/server/prompt-contracts.ts +17 -0
- package/src/server/proxy-routes.ts +187 -0
- package/src/server/publishing-routes.ts +41 -0
- package/src/{serve → server}/request-plan.ts +6 -8
- package/src/{serve → server}/request-prep.ts +47 -50
- package/src/{serve → server}/responses-client.ts +30 -13
- package/src/{responses.ts → server/responses.ts} +12 -93
- package/src/server/routes.ts +205 -0
- package/src/server/start.ts +72 -0
- package/src/server/status-routes.ts +108 -0
- package/src/{serve → server}/text-completion-stage.ts +25 -4
- package/src/server/text-prompt.ts +12 -0
- package/src/{serve → server}/token-streams.ts +36 -38
- package/src/server/worker-routes.ts +311 -0
- package/src/storage/paths.ts +61 -0
- package/src/web/assets.ts +37 -0
- package/src/web/{src → browser}/api.ts +5 -23
- package/src/web/browser/chat-host.ts +17 -0
- package/src/web/browser/installed-panels.ts +6 -0
- package/src/web/browser/main.ts +55 -0
- package/src/web/browser/markdown.ts +279 -0
- package/src/web/browser/memory-panel.ts +9 -0
- package/src/web/browser/palette.ts +64 -0
- package/src/web/browser/protocol.ts +13 -0
- package/src/web/browser/shell.ts +467 -0
- package/src/web/{src → browser}/status.ts +32 -36
- package/src/web/build.ts +51 -0
- package/src/web/public/app.html +1054 -0
- package/src/web/{sw.js → public/sw.js} +4 -3
- package/src/web/{vendor → public/vendor}/hljs-theme.css +2 -2
- package/src/web/{vendor → public/vendor}/hljs.js +2 -2
- package/THIRD_PARTY_LICENSES.md +0 -51
- package/docs/README.md +0 -107
- package/docs/archive/hlg-sampling.md +0 -350
- package/docs/archive/investigations/audio-input-plan.md +0 -356
- package/docs/archive/investigations/benchmark-ledger-through-v0.4.0.md +0 -3617
- package/docs/archive/investigations/chunk-finetune-experiment.md +0 -84
- package/docs/archive/investigations/colibri-glm52-port.md +0 -1486
- package/docs/archive/investigations/colibri-oracle-pin.md +0 -1284
- package/docs/archive/investigations/curve-bisector-routing.md +0 -123
- package/docs/archive/investigations/decode-roofline-lookagain.md +0 -348
- package/docs/archive/investigations/diffusion-gemma-port.md +0 -602
- package/docs/archive/investigations/dspark-handoff.md +0 -224
- package/docs/archive/investigations/expert-offload-single-user-moe.md +0 -292
- package/docs/archive/investigations/faithful-l1-consolidation.md +0 -310
- package/docs/archive/investigations/hlg-sampling-investigation.md +0 -970
- package/docs/archive/investigations/indexshare-performance-spike.md +0 -357
- package/docs/archive/investigations/kernel-perf-review-2026-07.md +0 -171
- package/docs/archive/investigations/lab-build-journal.md +0 -157
- package/docs/archive/investigations/minicpm5-decode-megakernel.md +0 -489
- package/docs/archive/investigations/openjarvis-fit.md +0 -198
- package/docs/archive/investigations/openwiki-evaluation.md +0 -249
- package/docs/archive/investigations/optimization_plan.md +0 -346
- package/docs/archive/investigations/orpo-base-uf-experiment-and-directions.md +0 -120
- package/docs/archive/investigations/orpo-flash-cce-pin-leak.md +0 -167
- package/docs/archive/investigations/orpo-uf-testing-handoff.md +0 -126
- package/docs/archive/investigations/pareto-specialized-runtime-findings.md +0 -92
- package/docs/archive/investigations/pi-builtin-investigation.md +0 -308
- package/docs/archive/investigations/pre-colibri-stabilization.md +0 -329
- package/docs/archive/investigations/qwen38-27b-campaign-log.md +0 -1092
- package/docs/archive/investigations/segmented-backward-handoff.md +0 -148
- package/docs/archive/investigations/starter-model-port-handoff.md +0 -342
- package/docs/archive/investigations/steel-flash-cce-handoff.md +0 -489
- package/docs/archive/investigations/the-dreaming-handoff.md +0 -141
- package/docs/archive/investigations/trainer-validation-experiment.md +0 -173
- package/docs/archive/planning/chat-ui-vision-memory-first.md +0 -199
- package/docs/archive/planning/chat-ui-vision.md +0 -50
- package/docs/archive/planning/chunk-task-roadmap.md +0 -77
- package/docs/archive/planning/curve-sampler-research-plan.md +0 -331
- package/docs/archive/planning/journal.md +0 -316
- package/docs/archive/planning/memory-docs-and-dag-plan.md +0 -194
- package/docs/archive/planning/pi-integration-review.md +0 -178
- package/docs/archive/planning/release-notes-v0.0.10.md +0 -57
- package/docs/archive/planning/release-notes-v0.0.11.md +0 -239
- package/docs/archive/planning/release-notes-v0.0.12.md +0 -80
- package/docs/archive/planning/release-notes-v0.0.13.md +0 -77
- package/docs/archive/planning/release-notes-v0.0.9.md +0 -122
- package/docs/archive/planning/release-notes-v0.2.0.md +0 -49
- package/docs/archive/planning/release-notes-v0.3.0.md +0 -35
- package/docs/archive/planning/release-notes-v0.4.0.md +0 -45
- package/docs/archive/planning/release-notes-v0.5.0.md +0 -37
- package/docs/archive/planning/web-ui-pass-plan.md +0 -213
- package/docs/archive/planning/website-readme-pass-plan.md +0 -224
- package/docs/design/batching.md +0 -2054
- package/docs/design/decode-speed-program.md +0 -2404
- package/docs/design/dreaming-nightly-pipeline.md +0 -684
- package/docs/design/generic-model-support.md +0 -821
- package/docs/design/kv-cache.md +0 -993
- package/docs/design/orpo-training.md +0 -680
- package/docs/design/speculative-decoding.md +0 -2129
- package/docs/design/turboquant.md +0 -2225
- package/docs/design/unified-engine-frontier-plan.md +0 -1987
- package/docs/design/web-chat-redesign.md +0 -695
- package/docs/planning/PRODUCT_ROADMAP.md +0 -593
- package/docs/planning/ResearchTopics.md +0 -119
- package/docs/planning/public-readiness-handoff.md +0 -244
- package/docs/planning/public-readiness.md +0 -149
- package/docs/planning/video-series.md +0 -421
- package/docs/reference/benchmarks.md +0 -3587
- package/docs/reference/cli.md +0 -646
- package/docs/reference/distribution.md +0 -380
- package/docs/reference/environment.md +0 -494
- package/docs/reference/glossary.md +0 -39
- package/docs/reference/library-api.md +0 -688
- package/docs/reference/memory.md +0 -260
- package/docs/reference/models.md +0 -346
- package/docs/reference/server-api.md +0 -1945
- package/docs/reference/server-config.md +0 -973
- package/docs/reference/training.md +0 -462
- package/docs/reference/troubleshooting.md +0 -158
- package/src/admission-plan.ts +0 -18
- package/src/audio/audiotoolbox.ts +0 -112
- package/src/audio/conformer.ts +0 -655
- package/src/audio/decode.ts +0 -133
- package/src/audio/features.ts +0 -202
- package/src/audio/mic-capture.ts +0 -98
- package/src/audio/silero-vad.ts +0 -379
- package/src/audio/transcode.ts +0 -67
- package/src/audio/whisper-decode.ts +0 -774
- package/src/audio/whisper-fast.ts +0 -732
- package/src/audio/whisper-format.ts +0 -73
- package/src/audio/whisper-mel.ts +0 -185
- package/src/audio/whisper-timing.ts +0 -274
- package/src/audio/whisper-tokenizer.ts +0 -239
- package/src/audio/whisper-transcribe.ts +0 -410
- package/src/backends/mlx/assistant-target.ts +0 -46
- package/src/backends/mlx/autoregressive-draft-rows.ts +0 -120
- package/src/backends/mlx/autoregressive.ts +0 -129
- package/src/backends/mlx/batch-group.ts +0 -1874
- package/src/backends/mlx/cache-attachment.ts +0 -67
- package/src/backends/mlx/cache-layout.ts +0 -31
- package/src/backends/mlx/checkpoint-state.ts +0 -53
- package/src/backends/mlx/continuation-persistence.ts +0 -107
- package/src/backends/mlx/continuation-request.ts +0 -43
- package/src/backends/mlx/continuation.ts +0 -77
- package/src/backends/mlx/diffusion.ts +0 -39
- package/src/backends/mlx/draft-target.ts +0 -32
- package/src/backends/mlx/encoder-cache.ts +0 -25
- package/src/backends/mlx/evaluation.ts +0 -105
- package/src/backends/mlx/fill-append.ts +0 -58
- package/src/backends/mlx/fill-group.ts +0 -497
- package/src/backends/mlx/full-kv-row-append.ts +0 -32
- package/src/backends/mlx/full-kv-row-donor.ts +0 -54
- package/src/backends/mlx/full-transitioning-kv-rows.ts +0 -65
- package/src/backends/mlx/gateway-binding.ts +0 -174
- package/src/backends/mlx/graph.ts +0 -38
- package/src/backends/mlx/kv-codec.ts +0 -9
- package/src/backends/mlx/kv-maintenance.ts +0 -161
- package/src/backends/mlx/kv-tensor-rows.ts +0 -212
- package/src/backends/mlx/memory-client.ts +0 -189
- package/src/backends/mlx/methods.ts +0 -90
- package/src/backends/mlx/mixed-iteration.ts +0 -62
- package/src/backends/mlx/model-prompt.ts +0 -178
- package/src/backends/mlx/model-serving.ts +0 -80
- package/src/backends/mlx/paged-cache-codec.ts +0 -27
- package/src/backends/mlx/prefill-cohort.ts +0 -100
- package/src/backends/mlx/prefill-policy.ts +0 -26
- package/src/backends/mlx/prefill-rows.ts +0 -244
- package/src/backends/mlx/prefill.ts +0 -33
- package/src/backends/mlx/projected-context-rows.ts +0 -72
- package/src/backends/mlx/prompt-input.ts +0 -29
- package/src/backends/mlx/qwen-prompt-input.ts +0 -40
- package/src/backends/mlx/request-policy.ts +0 -20
- package/src/backends/mlx/request-state-policy.ts +0 -54
- package/src/backends/mlx/rollback.ts +0 -92
- package/src/backends/mlx/row-sampling.ts +0 -49
- package/src/backends/mlx/serial-executor.ts +0 -251
- package/src/backends/mlx/speculative-group.ts +0 -438
- package/src/backends/mlx/speculative-round.ts +0 -161
- package/src/backends/mlx/speculative.ts +0 -131
- package/src/backends/mlx/state-rows.ts +0 -104
- package/src/backends/mlx/state-views.ts +0 -46
- package/src/backends/mlx/target-layout-capability.ts +0 -35
- package/src/backends/mlx/text-engine.ts +0 -73
- package/src/backends/mlx/transitioning-kv-rows.ts +0 -173
- package/src/chat-template.ts +0 -382
- package/src/cli.ts +0 -3490
- package/src/config.ts +0 -399
- package/src/contracts/completion.ts +0 -16
- package/src/contracts/execution.ts +0 -70
- package/src/contracts/generation.ts +0 -103
- package/src/contracts/host.ts +0 -7
- package/src/contracts/object-cache.ts +0 -8
- package/src/contracts/pi-provider.ts +0 -30
- package/src/contracts/pi.ts +0 -261
- package/src/contracts/prefix-cache.ts +0 -33
- package/src/contracts/resources.ts +0 -25
- package/src/contracts/sampling.ts +0 -23
- package/src/contracts/scheduling.ts +0 -54
- package/src/contracts/tsconfig.json +0 -14
- package/src/dataset/generators.ts +0 -1087
- package/src/dataset/index.ts +0 -12
- package/src/dataset/job.ts +0 -81
- package/src/dataset/llm.ts +0 -71
- package/src/dataset/registry.ts +0 -528
- package/src/diffusion/diffusion-generate.ts +0 -507
- package/src/download.ts +0 -505
- package/src/embed.ts +0 -77
- package/src/engine/admission.ts +0 -59
- package/src/engine/cancellation.ts +0 -38
- package/src/engine/engine.ts +0 -56
- package/src/engine/execution-coordinator.ts +0 -70
- package/src/engine/execution-plan.ts +0 -64
- package/src/engine/execution-tasks.ts +0 -44
- package/src/engine/namespaced-cache.ts +0 -13
- package/src/engine/resources.ts +0 -53
- package/src/engine/scheduler.ts +0 -92
- package/src/engine/session.ts +0 -247
- package/src/eval/README.md +0 -130
- package/src/eval/ifeval.ts +0 -81
- package/src/eval/kl-prompts.ts +0 -29
- package/src/eval/kl.ts +0 -590
- package/src/eval/perplexity.ts +0 -164
- package/src/eval/quality-db.ts +0 -92
- package/src/eval/reference-logits.ts +0 -207
- package/src/eval/runner.ts +0 -124
- package/src/eval/score.ts +0 -49
- package/src/eval/tasks/bfcl.ts +0 -535
- package/src/eval/tasks/gsm8k.ts +0 -98
- package/src/eval/tasks/hashhop.ts +0 -139
- package/src/eval/tasks/humaneval.ts +0 -275
- package/src/eval/tasks/ifeval.ts +0 -525
- package/src/eval/tasks/mmlu.ts +0 -157
- package/src/eval/ultrafeedback.ts +0 -60
- package/src/evaldb.ts +0 -199
- package/src/expert-io.ts +0 -393
- package/src/expert-offload-build.ts +0 -77
- package/src/expert-offload.ts +0 -76
- package/src/expert-residency.ts +0 -1134
- package/src/expert-usage.ts +0 -415
- package/src/fill/echo-index.ts +0 -188
- package/src/fill/fill-session.ts +0 -463
- package/src/fill/proposal.ts +0 -68
- package/src/fill/schema-rows.ts +0 -506
- package/src/fill/tool-boundary.ts +0 -103
- package/src/fit.ts +0 -249
- package/src/generate.ts +0 -1352
- package/src/grammar.ts +0 -604
- package/src/harness-pi.ts +0 -184
- package/src/hf-push.ts +0 -628
- package/src/hub-rest.ts +0 -242
- package/src/index.ts +0 -63
- package/src/inference/denoising.ts +0 -17
- package/src/inference/draft-acceptance.ts +0 -41
- package/src/inference/draft-output.ts +0 -49
- package/src/inference/graph.ts +0 -33
- package/src/inference/prefill.ts +0 -45
- package/src/inference/rollback.ts +0 -7
- package/src/jobs/index.ts +0 -31
- package/src/jobs/job-entry.ts +0 -107
- package/src/jobs/task-client.ts +0 -19
- package/src/jobs/types.ts +0 -2
- package/src/kv-scheme.ts +0 -279
- package/src/kv-store.ts +0 -1135
- package/src/lab/README.md +0 -22
- package/src/lab/curve/curve-designer.html +0 -303
- package/src/lab/curve/curve-sampler.ts +0 -122
- package/src/lab/expert-trace/expert-trace.ts +0 -69
- package/src/lab/paged-kv/paged-attention.ts +0 -117
- package/src/lab/paged-kv/paged-kv-rows.ts +0 -111
- package/src/lab/paged-kv/paged-kv.ts +0 -336
- package/src/library.ts +0 -40
- package/src/lora.ts +0 -408
- package/src/md-modules.d.ts +0 -54
- package/src/media-fetch.ts +0 -442
- package/src/memory/article.ts +0 -558
- package/src/memory/chunk-validation.ts +0 -150
- package/src/memory/chunk.ts +0 -318
- package/src/memory/cluster.ts +0 -351
- package/src/memory/coverage.ts +0 -134
- package/src/memory/crosslink.ts +0 -0
- package/src/memory/db.ts +0 -253
- package/src/memory/entity.ts +0 -203
- package/src/memory/gate.ts +0 -187
- package/src/memory/history.ts +0 -152
- package/src/memory/ingest.ts +0 -335
- package/src/memory/model.ts +0 -183
- package/src/memory/normalize.ts +0 -671
- package/src/memory/parse.ts +0 -22
- package/src/memory/pipeline.ts +0 -275
- package/src/memory/prompts.ts +0 -65
- package/src/memory/query.ts +0 -334
- package/src/memory/reconcile.ts +0 -734
- package/src/memory/redirect.ts +0 -108
- package/src/memory/reindex.ts +0 -325
- package/src/memory/resolve.ts +0 -307
- package/src/memory/rest.ts +0 -245
- package/src/memory/route.ts +0 -272
- package/src/memory/schedule.ts +0 -165
- package/src/memory/stages.ts +0 -777
- package/src/memory/synthesize.ts +0 -1638
- package/src/memory/tools.ts +0 -749
- package/src/memory/vault.ts +0 -1194
- package/src/memory/wikify.ts +0 -1064
- package/src/mlx/array.ts +0 -425
- package/src/mlx/autograd.ts +0 -304
- package/src/mlx/checkpoint.ts +0 -149
- package/src/mlx/compile.ts +0 -168
- package/src/mlx/custom-vjp.ts +0 -188
- package/src/mlx/ffi.ts +0 -531
- package/src/mlx/materialize.ts +0 -23
- package/src/mlx/metal-kernel.ts +0 -159
- package/src/mlx/normalized-argmax.ts +0 -116
- package/src/mlx/ops.ts +0 -1108
- package/src/mlx/quantized-rows.ts +0 -36
- package/src/mlx/token-bitmask.ts +0 -52
- package/src/mlx/turboquant-kv-decode.ts +0 -82
- package/src/mlx/turboquant-kv-inverse.ts +0 -122
- package/src/mlx/turboquant-ops.ts +0 -861
- package/src/mlx/turboquant-tables.ts +0 -237
- package/src/mmap.ts +0 -90
- package/src/model/artifact-identity.ts +0 -73
- package/src/model/batched-kv.ts +0 -235
- package/src/model/batched-mask.ts +0 -279
- package/src/model/batched-quant.ts +0 -212
- package/src/model/batched-quantized-kv.ts +0 -144
- package/src/model/batched-rotating-quant.ts +0 -353
- package/src/model/batched-rotating-state.ts +0 -149
- package/src/model/batched-rotating.ts +0 -338
- package/src/model/batched-row-storage.ts +0 -179
- package/src/model/batched-ssm.ts +0 -52
- package/src/model/batched-turboquant-kv.ts +0 -94
- package/src/model/compiled-decode.ts +0 -712
- package/src/model/decoded-kv-donor.ts +0 -12
- package/src/model/delayed-quantized-kv.ts +0 -72
- package/src/model/delayed-rotating-quantized-kv.ts +0 -93
- package/src/model/delayed-turboquant-kv.ts +0 -57
- package/src/model/diffusion-gemma.ts +0 -1026
- package/src/model/factory.ts +0 -215
- package/src/model/fingerprint.ts +0 -45
- package/src/model/flash-attention.ts +0 -694
- package/src/model/full-prefill-padding.ts +0 -56
- package/src/model/full-prefill-row.ts +0 -105
- package/src/model/gemma4-base.ts +0 -2370
- package/src/model/gemma4.ts +0 -1395
- package/src/model/generated/gemma4-12b.ts +0 -429
- package/src/model/generated/gemma4-26b.ts +0 -448
- package/src/model/generated/gemma4-e4b.ts +0 -717
- package/src/model/generated/index.ts +0 -18
- package/src/model/glm52-atlas.ts +0 -511
- package/src/model/glm52-cache.ts +0 -792
- package/src/model/glm52-config.ts +0 -262
- package/src/model/glm52-container.ts +0 -305
- package/src/model/glm52-coupling.ts +0 -388
- package/src/model/glm52-dsa.ts +0 -606
- package/src/model/glm52-expert-layout.ts +0 -253
- package/src/model/glm52-layout.ts +0 -152
- package/src/model/glm52-memory.ts +0 -500
- package/src/model/glm52-mla.ts +0 -792
- package/src/model/glm52-moe.ts +0 -223
- package/src/model/glm52-pilot.ts +0 -537
- package/src/model/glm52-reference.ts +0 -445
- package/src/model/glm52-residency.ts +0 -557
- package/src/model/glm52-resident-weights.ts +0 -135
- package/src/model/glm52-streamed-experts.ts +0 -453
- package/src/model/glm52-streamed-metal.ts +0 -610
- package/src/model/glm52-weights.ts +0 -503
- package/src/model/glm52.ts +0 -1132
- package/src/model/implementation.ts +0 -69
- package/src/model/kv-attention-view.ts +0 -70
- package/src/model/minicpm5.ts +0 -376
- package/src/model/profile.ts +0 -468
- package/src/model/quantized-append-attention.ts +0 -31
- package/src/model/qwen-append.ts +0 -12
- package/src/model/qwen-conv.ts +0 -52
- package/src/model/qwen3-delta-state.ts +0 -74
- package/src/model/qwen3-delta.ts +0 -451
- package/src/model/qwen3-moe.ts +0 -404
- package/src/model/qwen3-mrope.ts +0 -250
- package/src/model/qwen3.ts +0 -277
- package/src/model/qwen3_5-checkpoint.ts +0 -164
- package/src/model/qwen3_5.ts +0 -881
- package/src/model/rotating-kv-donor.ts +0 -35
- package/src/model/rotating-kv-layout.ts +0 -262
- package/src/model/rotating-row-transaction.ts +0 -35
- package/src/model/speculative-rotating-kv.ts +0 -101
- package/src/model/ssm-prefill-padding.ts +0 -39
- package/src/model/support.ts +0 -110
- package/src/model/token-groups.ts +0 -66
- package/src/model/trellis-balanced-scatter.ts +0 -51
- package/src/model/trellis-linear.ts +0 -672
- package/src/model/trellis-shared-m.ts +0 -151
- package/src/model/trellis-shared-scatter.ts +0 -144
- package/src/model/trellis-splitk-prefill.ts +0 -158
- package/src/model/trellis-tiled-prefill.ts +0 -136
- package/src/model/trellis-vector-expand.ts +0 -62
- package/src/model/trellis-wide-prefill.ts +0 -94
- package/src/model/turboquant-codec.ts +0 -89
- package/src/model/universal/archs.ts +0 -332
- package/src/model/universal/dense.ts +0 -612
- package/src/model/universal/modules.ts +0 -170
- package/src/model/universal/rope.ts +0 -381
- package/src/model/whisper.ts +0 -435
- package/src/native/expert_io.c +0 -523
- package/src/native/frame_extract.swift +0 -140
- package/src/native/mic_capture.swift +0 -92
- package/src/native-pack.ts +0 -163
- package/src/pi-provider.ts +0 -53
- package/src/pi-session.ts +0 -81
- package/src/pi-terminal.ts +0 -274
- package/src/pi-web.ts +0 -1940
- package/src/preflight.ts +0 -158
- package/src/prompt-cache.ts +0 -682
- package/src/quantize/allocator.ts +0 -627
- package/src/quantize/atomic-output.ts +0 -38
- package/src/quantize/calibration.ts +0 -181
- package/src/quantize/config-writer.ts +0 -145
- package/src/quantize/index.ts +0 -84
- package/src/quantize/job.ts +0 -162
- package/src/quantize/quantizer.ts +0 -666
- package/src/quantize/rotate.ts +0 -603
- package/src/quantize/safetensors-writer.ts +0 -249
- package/src/quantize/sensitivity.ts +0 -216
- package/src/quantize/trellis.ts +0 -586
- package/src/quantize/weight-transform.ts +0 -292
- package/src/registry.ts +0 -602
- package/src/runtime-config.ts +0 -77
- package/src/safetensors.ts +0 -165
- package/src/sampler.ts +0 -1012
- package/src/serve/admin-routes.ts +0 -184
- package/src/serve/assets/curve-terrain.html +0 -105
- package/src/serve/assets/training-inference-map.html +0 -575
- package/src/serve/audio-routes.ts +0 -323
- package/src/serve/aux-routes.ts +0 -132
- package/src/serve/batch-scheduler.ts +0 -8
- package/src/serve/checkpoint-identity.ts +0 -64
- package/src/serve/generation-gateway.ts +0 -537
- package/src/serve/isolate.ts +0 -609
- package/src/serve/kv-budget.ts +0 -14
- package/src/serve/lab-routes.ts +0 -341
- package/src/serve/lane-registry.ts +0 -45
- package/src/serve/model-admin-routes.ts +0 -155
- package/src/serve/model-binding.ts +0 -49
- package/src/serve/model-host.ts +0 -511
- package/src/serve/parent-application.ts +0 -56
- package/src/serve/prompt-response-trace.ts +0 -161
- package/src/serve/response-lease.ts +0 -24
- package/src/serve/session-search.ts +0 -326
- package/src/serve/static-routes.ts +0 -101
- package/src/serve/transcription-server.ts +0 -64
- package/src/serve/transcription-service.ts +0 -502
- package/src/server.ts +0 -1282
- package/src/spec/assistant-rows.ts +0 -93
- package/src/spec/assistant-source.ts +0 -126
- package/src/spec/cache-identity.ts +0 -22
- package/src/spec/deepspec-rows.ts +0 -29
- package/src/spec/deepspec-source.ts +0 -145
- package/src/spec/dflash-source.ts +0 -149
- package/src/spec/drafter.ts +0 -304
- package/src/spec/dspark/ab-stats.ts +0 -136
- package/src/spec/dspark/calibration.ts +0 -112
- package/src/spec/dspark/data-dflash.ts +0 -156
- package/src/spec/dspark/data.ts +0 -201
- package/src/spec/dspark/deepspec-module.ts +0 -776
- package/src/spec/dspark/generate-dflash.ts +0 -245
- package/src/spec/dspark/generate.ts +0 -290
- package/src/spec/dspark/loader.ts +0 -23
- package/src/spec/dspark/loss.ts +0 -213
- package/src/spec/dspark/module-dflash.ts +0 -560
- package/src/spec/dspark/module.ts +0 -540
- package/src/spec/dspark/quantize-drafter.ts +0 -91
- package/src/spec/dspark/sample.ts +0 -112
- package/src/spec/generate.ts +0 -276
- package/src/spec/glm52-mtp-graph.ts +0 -29
- package/src/spec/glm52-mtp-rows.ts +0 -132
- package/src/spec/glm52-mtp-source.ts +0 -318
- package/src/spec/glm52-mtp-state.ts +0 -33
- package/src/spec/ngram-source.ts +0 -226
- package/src/spec/projected-draft-rows.ts +0 -73
- package/src/spec/qwen-mtp-module.ts +0 -171
- package/src/spec/qwen-mtp-rows.ts +0 -230
- package/src/spec/qwen-mtp-source.ts +0 -386
- package/src/spec/qwen-mtp-state.ts +0 -30
- package/src/spec/serve-loop.ts +0 -587
- package/src/spec/source.ts +0 -228
- package/src/spec/two-model-rows.ts +0 -62
- package/src/spec/two-model.ts +0 -113
- package/src/ssd-cache.ts +0 -431
- package/src/ssd-durability.ts +0 -209
- package/src/storage/host-buffer.ts +0 -24
- package/src/storage/kv-writer.ts +0 -83
- package/src/storage/kv-writer.worker.js +0 -264
- package/src/storage/retention-policy.ts +0 -37
- package/src/tiered-prompt-cache.ts +0 -45
- package/src/tokenizer.ts +0 -335
- package/src/tool-approvals.ts +0 -131
- package/src/tool-call.ts +0 -695
- package/src/train/dataset.ts +0 -484
- package/src/train/diffusion-lora.ts +0 -216
- package/src/train/export.ts +0 -32
- package/src/train/flash-cce.ts +0 -1234
- package/src/train/forward.ts +0 -266
- package/src/train/fuse.ts +0 -243
- package/src/train/index.ts +0 -46
- package/src/train/job.ts +0 -205
- package/src/train/lora-params.ts +0 -254
- package/src/train/loss.ts +0 -1591
- package/src/train/merge.ts +0 -195
- package/src/train/optimizer.ts +0 -183
- package/src/train/prefix-shared.ts +0 -490
- package/src/train/rank.ts +0 -137
- package/src/train/segmented.ts +0 -1828
- package/src/train/steel-qmm-header.ts +0 -1461
- package/src/train/trainer.ts +0 -1221
- package/src/train/watch.ts +0 -542
- package/src/vision/diffusion-vision.ts +0 -620
- package/src/vision/embedder.ts +0 -176
- package/src/vision/preprocess.ts +0 -247
- package/src/vision/prompt.ts +0 -409
- package/src/vision/qwen3vl-preprocess.ts +0 -409
- package/src/vision/qwen3vl-prompt.ts +0 -233
- package/src/vision/qwen3vl-tower.ts +0 -566
- package/src/vision/siglip.ts +0 -607
- package/src/vision/video-frames.ts +0 -148
- package/src/web/app.html +0 -2155
- package/src/web/app.js +0 -5863
- package/src/web/skills/memory/SKILL.md +0 -120
- package/src/web/skills/web-research/SKILL.md +0 -46
- package/src/web/skills.ts +0 -97
- package/src/web/src/adapters-panel.ts +0 -288
- package/src/web/src/assistant.ts +0 -451
- package/src/web/src/chat.ts +0 -1060
- package/src/web/src/composer.ts +0 -962
- package/src/web/src/dataset.ts +0 -132
- package/src/web/src/finetune.ts +0 -267
- package/src/web/src/hub.ts +0 -428
- package/src/web/src/main.ts +0 -98
- package/src/web/src/markdown.ts +0 -0
- package/src/web/src/memory-panel.ts +0 -572
- package/src/web/src/model-picker.ts +0 -159
- package/src/web/src/palette.ts +0 -346
- package/src/web/src/protocol.ts +0 -26
- package/src/web/src/quantize.ts +0 -167
- package/src/web/src/rag.ts +0 -274
- package/src/web/src/sessions.ts +0 -469
- package/src/web/src/shell.ts +0 -804
- package/src/web/src/ui-catalog.ts +0 -78
- package/src/web/src/voice.ts +0 -141
- package/src/web/vendor/README.md +0 -61
- package/src/web-assets.ts +0 -47
- package/src/web-tools.ts +0 -478
- package/src/weights.ts +0 -255
- /package/src/{serve → server}/completion-sink.ts +0 -0
- /package/src/{serve → server}/generated-token-history.ts +0 -0
- /package/src/web/{icon.svg → public/icon.svg} +0 -0
- /package/src/web/{manifest.webmanifest → public/manifest.webmanifest} +0 -0
- /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
|
|
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 "
|
|
25
|
+
import type { ChatMessage, ToolDefinition } from "@mlx-bun/inference/input";
|
|
26
26
|
import type {
|
|
27
27
|
CompletionEvent,
|
|
28
28
|
CompletionStreamProtocol,
|
|
29
|
-
} from "./
|
|
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.
|
|
5
|
-
import type { ChatMessage, ToolDefinition } from "
|
|
6
|
-
import type { HlgConfig } from "
|
|
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
|
-
|
|
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:
|
|
236
|
-
*
|
|
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 {
|
|
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 {
|
|
15
|
-
import type { BuiltPrompt, ModelPromptBuilder, PromptNativeWork } from "./
|
|
16
|
-
import { modelPromptBuilder } from "
|
|
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:
|
|
24
|
+
private readonly ctx: ModelContext,
|
|
25
25
|
private readonly prep: RequestPrep,
|
|
26
|
-
private readonly promptCache: Pick<
|
|
27
|
-
/** Admission ceiling (fit() /
|
|
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("
|
|
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("
|
|
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
|
|
141
|
-
//
|
|
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("
|
|
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
|
|
191
|
+
ctx.template?.thinkingFormat === "think-tag",
|
|
192
192
|
startInThinking,
|
|
193
193
|
),
|
|
194
194
|
collectToolCalls: true,
|
|
@@ -1,5 +1,31 @@
|
|
|
1
|
-
|
|
2
|
-
|
|
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 "
|
|
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 "
|
|
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
|
-
|
|
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 "
|
|
30
|
-
import { cleanupFailure } from "
|
|
29
|
+
import type { PromptResponseTrace } from "@mlx-bun/inference/runtime/trace";
|
|
30
|
+
import { cleanupFailure } from "@mlx-bun/inference/runtime/resources";
|
|
31
31
|
|
|
32
|
-
|
|
33
|
-
|
|
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
|
|
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 =
|
|
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 =
|
|
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
|
+
}
|