@qvac/sdk 0.15.0 → 0.17.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/NOTICE +87 -84
- package/README.md +65 -0
- package/dist/client/api/audio-gen-result.d.ts +4 -0
- package/dist/client/api/audio-gen-result.js +132 -0
- package/dist/client/api/audio-gen.d.ts +29 -0
- package/dist/client/api/audio-gen.js +31 -0
- package/dist/client/api/batch-completion.d.ts +2 -2
- package/dist/client/api/batch-completion.js +2 -2
- package/dist/client/api/bci-transcribe.d.ts +1 -1
- package/dist/client/api/bci-transcribe.js +3 -3
- package/dist/client/api/cancel.d.ts +1 -1
- package/dist/client/api/cancel.js +1 -1
- package/dist/client/api/classify.d.ts +1 -1
- package/dist/client/api/classify.js +1 -1
- package/dist/client/api/completion-stream.d.ts +3 -3
- package/dist/client/api/completion-stream.js +3 -3
- package/dist/client/api/delete-cache.js +1 -1
- package/dist/client/api/diffusion.d.ts +1 -1
- package/dist/client/api/diffusion.js +1 -1
- package/dist/client/api/download-asset.d.ts +1 -1
- package/dist/client/api/download-asset.js +3 -3
- package/dist/client/api/embed.d.ts +1 -1
- package/dist/client/api/embed.js +2 -2
- package/dist/client/api/finetune.d.ts +1 -1
- package/dist/client/api/finetune.js +2 -2
- package/dist/client/api/get-loaded-model-info.d.ts +2 -2
- package/dist/client/api/get-loaded-model-info.js +1 -1
- package/dist/client/api/get-model-info.d.ts +2 -2
- package/dist/client/api/get-model-info.js +1 -1
- package/dist/client/api/get-system-resources.d.ts +3 -0
- package/dist/client/api/get-system-resources.js +16 -0
- package/dist/client/api/heartbeat.d.ts +2 -2
- package/dist/client/api/heartbeat.js +1 -1
- package/dist/client/api/index.d.ts +35 -33
- package/dist/client/api/index.js +2 -0
- package/dist/client/api/invoke-plugin.d.ts +1 -1
- package/dist/client/api/invoke-plugin.js +1 -1
- package/dist/client/api/load-model.d.ts +2 -2
- package/dist/client/api/load-model.js +4 -4
- package/dist/client/api/logging-stream.d.ts +1 -1
- package/dist/client/api/logging-stream.js +1 -1
- package/dist/client/api/ocr.d.ts +10 -1
- package/dist/client/api/ocr.js +10 -1
- package/dist/client/api/provide.d.ts +1 -1
- package/dist/client/api/provide.js +1 -1
- package/dist/client/api/rag.d.ts +1 -1
- package/dist/client/api/rag.js +2 -2
- package/dist/client/api/registry.d.ts +1 -1
- package/dist/client/api/registry.js +1 -1
- package/dist/client/api/resume.js +1 -1
- package/dist/client/api/state.d.ts +1 -1
- package/dist/client/api/state.js +1 -1
- package/dist/client/api/stop-provider.js +1 -1
- package/dist/client/api/subscribe-logs.d.ts +1 -1
- package/dist/client/api/suspend.js +1 -1
- package/dist/client/api/text-to-speech.d.ts +3 -2
- package/dist/client/api/text-to-speech.js +31 -9
- package/dist/client/api/transcribe.d.ts +1 -1
- package/dist/client/api/transcribe.js +2 -2
- package/dist/client/api/translate.d.ts +1 -1
- package/dist/client/api/translate.js +7 -18
- package/dist/client/api/unload-model.d.ts +1 -1
- package/dist/client/api/unload-model.js +2 -2
- package/dist/client/api/upscale.d.ts +1 -1
- package/dist/client/api/upscale.js +1 -1
- package/dist/client/api/video.d.ts +8 -5
- package/dist/client/api/video.js +9 -6
- package/dist/client/api/vla.d.ts +15 -11
- package/dist/client/api/vla.js +14 -10
- package/dist/client/config-loader/config-utils.d.ts +1 -1
- package/dist/client/config-loader/resolve-config.bare.d.ts +1 -1
- package/dist/client/config-loader/resolve-config.expo.d.ts +1 -1
- package/dist/client/config-loader/resolve-config.node.d.ts +1 -1
- package/dist/client/index.d.ts +1 -1
- package/dist/client/init-hooks.d.ts +1 -1
- package/dist/client/logging-stream-registry.d.ts +1 -1
- package/dist/client/plugins-factory.d.ts +2 -2
- package/dist/client/plugins-factory.js +1 -1
- package/dist/client/rpc/bare-client.d.ts +1 -1
- package/dist/client/rpc/bare-client.js +3 -3
- package/dist/client/rpc/expo-rpc-client.js +3 -3
- package/dist/client/rpc/node-rpc-client.js +2 -2
- package/dist/client/rpc/profiling/index.d.ts +1 -1
- package/dist/client/rpc/profiling/profiler.d.ts +2 -2
- package/dist/client/rpc/profiling/profiler.js +4 -0
- package/dist/client/rpc/rpc-client.d.ts +1 -1
- package/dist/client/rpc/rpc-client.js +7 -4
- package/dist/client/rpc/rpc-error.d.ts +1 -1
- package/dist/client/rpc/rpc-error.js +7 -1
- package/dist/commands/bundle/bare-pack.d.ts +1 -1
- package/dist/commands/bundle/constants.js +1 -0
- package/dist/commands/bundle/entry-gen.js +1 -1
- package/dist/commands/bundle/index.js +6 -6
- package/dist/commands/bundle/manifest.d.ts +1 -1
- package/dist/commands/bundle/plugins.d.ts +2 -2
- package/dist/commands/bundle/plugins.js +1 -1
- package/dist/commands/index.d.ts +4 -4
- package/dist/commands/index.js +2 -2
- package/dist/commands/verify/abi.d.ts +1 -1
- package/dist/commands/verify/abi.js +1 -1
- package/dist/commands/verify/bundle-source.d.ts +1 -1
- package/dist/commands/verify/bundle-source.js +2 -2
- package/dist/commands/verify/index.d.ts +3 -3
- package/dist/commands/verify/index.js +5 -5
- package/dist/commands/verify/node-modules-source.d.ts +1 -1
- package/dist/commands/verify/node-modules-source.js +1 -1
- package/dist/commands/verify/prebuilds.d.ts +2 -1
- package/dist/commands/verify/prebuilds.js +8 -11
- package/dist/commands/verify/resource-collectors.d.ts +88 -0
- package/dist/commands/verify/resource-collectors.js +261 -0
- package/dist/constants/audio.js +1 -1
- package/dist/constants/index.d.ts +2 -2
- package/dist/examples/{transcription → asr}/parakeet-ctc-filesystem.js +2 -3
- package/dist/examples/{transcription → asr}/parakeet-microphone-record.js +2 -2
- package/dist/examples/{transcription → asr}/parakeet-microphone-stream.js +2 -2
- package/dist/examples/{transcription → asr}/parakeet-sortformer-streaming.js +2 -2
- package/dist/examples/{transcription → asr}/parakeet-sortformer.js +1 -1
- package/dist/examples/asr/parakeet-tdt-filesystem-chunked.d.ts +2 -0
- package/dist/examples/asr/parakeet-tdt-filesystem-chunked.js +203 -0
- package/dist/examples/{transcription → asr}/parakeet-tdt-filesystem.js +2 -2
- package/dist/examples/{transcription → asr}/whispercpp-filesystem-streaming.js +1 -1
- package/dist/examples/{transcription → asr}/whispercpp-filesystem.js +1 -1
- package/dist/examples/{transcription → asr}/whispercpp-microphone-conversation.js +1 -1
- package/dist/examples/{transcription → asr}/whispercpp-microphone-record.js +2 -2
- package/dist/examples/{transcription → asr}/whispercpp-prompt.js +1 -1
- package/dist/examples/audiogen/generate-music.d.ts +2 -0
- package/dist/examples/audiogen/generate-music.js +112 -0
- package/dist/examples/cache-management.js +1 -1
- package/dist/examples/config-reload.js +1 -1
- package/dist/examples/diffusion-ideogram.d.ts +2 -0
- package/dist/examples/diffusion-ideogram.js +108 -0
- package/dist/examples/diffusion-img2vid-ltx.d.ts +2 -0
- package/dist/examples/diffusion-img2vid-ltx.js +98 -0
- package/dist/examples/diffusion-txt2vid-ltx.d.ts +2 -0
- package/dist/examples/diffusion-txt2vid-ltx.js +92 -0
- package/dist/examples/diffusion-txt2vid-wan22.d.ts +2 -0
- package/dist/examples/diffusion-txt2vid-wan22.js +96 -0
- package/dist/examples/logging-streaming.js +5 -1
- package/dist/examples/quickstart.bare.js +1 -1
- package/dist/examples/rag/rag-mongodb.d.ts +2 -0
- package/dist/examples/rag/rag-mongodb.js +180 -0
- package/dist/examples/tools/llamacpp-tools-auto-dialect.d.ts +2 -0
- package/dist/examples/tools/llamacpp-tools-auto-dialect.js +77 -0
- package/dist/examples/tts/parler.d.ts +2 -0
- package/dist/examples/tts/parler.js +51 -0
- package/dist/examples/vla-groot.d.ts +2 -0
- package/dist/examples/vla-groot.js +120 -0
- package/dist/examples/voice-assistant/voice-assistant.js +2 -2
- package/dist/expo/plugins/index.d.ts +11 -11
- package/dist/expo/plugins/withDeviceInfo.js +2 -2
- package/dist/expo/plugins/withMobileBundle.js +3 -3
- package/dist/index.d.ts +16 -16
- package/dist/index.js +4 -4
- package/dist/logging/addon.d.ts +1 -1
- package/dist/logging/base-logger.d.ts +1 -1
- package/dist/logging/base-logger.js +1 -1
- package/dist/logging/client-logger.d.ts +1 -1
- package/dist/logging/index.d.ts +9 -9
- package/dist/logging/logger.d.ts +1 -1
- package/dist/logging/namespaces.d.ts +1 -1
- package/dist/logging/registry.d.ts +1 -1
- package/dist/logging/server-logger.d.ts +1 -1
- package/dist/logging/stream-logger.d.ts +1 -1
- package/dist/logging/transport.d.ts +1 -1
- package/dist/logging/utils.d.ts +1 -1
- package/dist/logging/utils.js +1 -1
- package/dist/models/registry/index.d.ts +1 -1
- package/dist/models/registry/models.d.ts +6443 -9832
- package/dist/models/registry/models.js +4076 -7658
- package/dist/models/update-models/codegen.d.ts +1 -1
- package/dist/models/update-models/companions.d.ts +1 -1
- package/dist/models/update-models/history.d.ts +1 -1
- package/dist/models/update-models/history.js +4 -4
- package/dist/models/update-models/naming.d.ts +1 -1
- package/dist/models/update-models/naming.js +12 -2
- package/dist/models/update-models/processing.d.ts +1 -1
- package/dist/models/update-models/processing.js +1 -1
- package/dist/models/update-models/registry.d.ts +2 -2
- package/dist/models/update-models/shards.d.ts +1 -1
- package/dist/models/update-models/types.d.ts +1 -1
- package/dist/pear/pre.js +2 -0
- package/dist/profiling/aggregator.d.ts +1 -1
- package/dist/profiling/backend-diagnostics.d.ts +8 -0
- package/dist/profiling/backend-diagnostics.js +21 -0
- package/dist/profiling/controller.d.ts +4 -2
- package/dist/profiling/controller.js +8 -0
- package/dist/profiling/envelope.d.ts +3 -2
- package/dist/profiling/envelope.js +13 -2
- package/dist/profiling/events.d.ts +2 -2
- package/dist/profiling/exporters.d.ts +1 -1
- package/dist/profiling/index.d.ts +9 -8
- package/dist/profiling/index.js +3 -2
- package/dist/profiling/types.d.ts +6 -0
- package/dist/schemas/audio-gen.d.ts +253 -0
- package/dist/schemas/audio-gen.js +69 -0
- package/dist/schemas/batch-completion-stream.d.ts +11 -6
- package/dist/schemas/bci-config.d.ts +2 -0
- package/dist/schemas/bci.d.ts +1 -1
- package/dist/schemas/cancel.d.ts +5 -0
- package/dist/schemas/cancel.js +1 -0
- package/dist/schemas/common.d.ts +1938 -58
- package/dist/schemas/common.js +9 -1
- package/dist/schemas/completion-event.d.ts +5 -2
- package/dist/schemas/completion-event.js +3 -0
- package/dist/schemas/completion-stream.d.ts +195 -3
- package/dist/schemas/completion-stream.js +72 -2
- package/dist/schemas/constants-registry.d.ts +69 -0
- package/dist/schemas/constants-registry.js +52 -0
- package/dist/schemas/delete-cache.js +7 -3
- package/dist/schemas/download-asset.d.ts +8 -4
- package/dist/schemas/engine-addon-map.d.ts +3 -1
- package/dist/schemas/engine-addon-map.js +6 -2
- package/dist/schemas/finetune.js +6 -3
- package/dist/schemas/get-loaded-model-info.d.ts +18 -0
- package/dist/schemas/get-loaded-model-info.js +14 -6
- package/dist/schemas/get-model-info.d.ts +2 -0
- package/dist/schemas/get-model-info.js +1 -0
- package/dist/schemas/index.d.ts +53 -51
- package/dist/schemas/index.js +4 -2
- package/dist/schemas/llamacpp-config.d.ts +30 -6
- package/dist/schemas/llamacpp-config.js +1 -6
- package/dist/schemas/load-model.d.ts +7654 -2919
- package/dist/schemas/load-model.js +45 -6
- package/dist/schemas/mcp-adapter.d.ts +1 -1
- package/dist/schemas/model-src-utils.d.ts +10 -2
- package/dist/schemas/model-types.d.ts +19 -1
- package/dist/schemas/model-types.js +11 -0
- package/dist/schemas/ocr.d.ts +2 -0
- package/dist/schemas/plugin.d.ts +17 -8
- package/dist/schemas/plugin.js +16 -6
- package/dist/schemas/profiling.d.ts +587 -0
- package/dist/schemas/profiling.js +22 -0
- package/dist/schemas/registry.d.ts +18 -0
- package/dist/schemas/registry.js +2 -0
- package/dist/schemas/reload-config.d.ts +10 -2
- package/dist/schemas/reload-config.js +3 -1
- package/dist/schemas/sdcpp-config.d.ts +296 -0
- package/dist/schemas/sdcpp-config.js +165 -17
- package/dist/schemas/sdk-config.d.ts +958 -40
- package/dist/schemas/sdk-errors-server.js +3 -1
- package/dist/schemas/system-resources.d.ts +3967 -0
- package/dist/schemas/system-resources.js +149 -0
- package/dist/schemas/text-to-speech.d.ts +340 -6
- package/dist/schemas/text-to-speech.js +136 -17
- package/dist/schemas/tools.d.ts +1 -1
- package/dist/schemas/tools.js +2 -1
- package/dist/schemas/transcription-config.d.ts +2 -0
- package/dist/schemas/translate.d.ts +2 -2
- package/dist/schemas/translate.js +12 -6
- package/dist/schemas/translation-config.d.ts +30 -10
- package/dist/schemas/vla.d.ts +10 -0
- package/dist/schemas/vla.js +31 -11
- package/dist/scripts/check-resource-collectors-packaging.d.ts +33 -0
- package/dist/scripts/check-resource-collectors-packaging.js +136 -0
- package/dist/scripts/contract/build-contract.d.ts +38 -0
- package/dist/scripts/contract/build-contract.js +620 -0
- package/dist/scripts/contract/build-error-codes.d.ts +17 -0
- package/dist/scripts/contract/build-error-codes.js +17 -0
- package/dist/scripts/contract/build-model-type-maps.d.ts +22 -0
- package/dist/scripts/contract/build-model-type-maps.js +16 -0
- package/dist/scripts/contract/build-models-registry.d.ts +17 -0
- package/dist/scripts/contract/build-models-registry.js +31 -0
- package/dist/scripts/contract/plugin-contract.d.ts +23 -0
- package/dist/scripts/contract/plugin-contract.js +28 -0
- package/dist/scripts/export-contract.d.ts +2 -0
- package/dist/scripts/export-contract.js +27 -0
- package/dist/scripts/export-plugin-contract.d.ts +2 -0
- package/dist/scripts/export-plugin-contract.js +18 -0
- package/dist/server/bare/delegate-rpc-client.d.ts +1 -1
- package/dist/server/bare/delegate-rpc-client.js +3 -3
- package/dist/server/bare/hyperswarm.d.ts +1 -1
- package/dist/server/bare/hyperswarm.js +2 -2
- package/dist/server/bare/ops/bci-transcribe.d.ts +1 -1
- package/dist/server/bare/ops/bci-transcribe.js +4 -4
- package/dist/server/bare/ops/cancel.d.ts +2 -2
- package/dist/server/bare/ops/cancel.js +2 -2
- package/dist/server/bare/ops/embed.d.ts +1 -1
- package/dist/server/bare/ops/embed.js +3 -3
- package/dist/server/bare/ops/kv-cache-retention.d.ts +12 -0
- package/dist/server/bare/ops/kv-cache-retention.js +121 -0
- package/dist/server/bare/ops/kv-cache-utils.d.ts +2 -1
- package/dist/server/bare/ops/kv-cache-utils.js +26 -5
- package/dist/server/bare/ops/load-model.d.ts +2 -2
- package/dist/server/bare/ops/load-model.js +7 -5
- package/dist/server/bare/ops/transcribe.d.ts +2 -2
- package/dist/server/bare/ops/transcribe.js +20 -40
- package/dist/server/bare/ops/translate.d.ts +1 -1
- package/dist/server/bare/ops/translate.js +21 -12
- package/dist/server/bare/ops/unload-model.d.ts +1 -1
- package/dist/server/bare/ops/unload-model.js +5 -5
- package/dist/server/bare/plugins/asr-ggml/config.d.ts +32 -0
- package/dist/server/bare/plugins/asr-ggml/config.js +46 -0
- package/dist/server/bare/plugins/asr-ggml/logging.d.ts +11 -0
- package/dist/server/bare/plugins/asr-ggml/logging.js +12 -0
- package/dist/server/bare/plugins/audiogen-ggml/config.d.ts +18 -0
- package/dist/server/bare/plugins/audiogen-ggml/config.js +23 -0
- package/dist/server/bare/plugins/audiogen-ggml/ops/audio-gen-stream.d.ts +3 -0
- package/dist/server/bare/plugins/audiogen-ggml/ops/audio-gen-stream.js +162 -0
- package/dist/server/bare/plugins/audiogen-ggml/plugin.d.ts +213 -0
- package/dist/server/bare/plugins/audiogen-ggml/plugin.js +58 -0
- package/dist/server/bare/plugins/bci-whispercpp-transcription/plugin.d.ts +8 -5
- package/dist/server/bare/plugins/bci-whispercpp-transcription/plugin.js +4 -4
- package/dist/server/bare/plugins/bci-whispercpp-transcription/resolve-config.d.ts +1 -1
- package/dist/server/bare/plugins/ggml-classification/ops/classify.d.ts +1 -1
- package/dist/server/bare/plugins/ggml-classification/ops/classify.js +1 -1
- package/dist/server/bare/plugins/ggml-classification/plugin.d.ts +2 -2
- package/dist/server/bare/plugins/ggml-classification/plugin.js +1 -1
- package/dist/server/bare/plugins/ggml-ocr/ops/ocr-stream.d.ts +1 -1
- package/dist/server/bare/plugins/ggml-ocr/ops/ocr-stream.js +2 -2
- package/dist/server/bare/plugins/ggml-ocr/plugin.d.ts +4 -2
- package/dist/server/bare/plugins/ggml-ocr/plugin.js +4 -25
- package/dist/server/bare/plugins/ggml-ocr/resolve-config.d.ts +6 -0
- package/dist/server/bare/plugins/ggml-ocr/resolve-config.js +73 -0
- package/dist/server/bare/plugins/ggml-vla/ops/vla-hparams.d.ts +1 -1
- package/dist/server/bare/plugins/ggml-vla/ops/vla-hparams.js +1 -1
- package/dist/server/bare/plugins/ggml-vla/ops/vla-run.d.ts +1 -1
- package/dist/server/bare/plugins/ggml-vla/ops/vla-run.js +1 -1
- package/dist/server/bare/plugins/ggml-vla/plugin.d.ts +8 -3
- package/dist/server/bare/plugins/ggml-vla/plugin.js +1 -1
- package/dist/server/bare/plugins/index.d.ts +12 -11
- package/dist/server/bare/plugins/index.js +1 -0
- package/dist/server/bare/plugins/llamacpp-completion/ops/batch-completion-stream.d.ts +3 -3
- package/dist/server/bare/plugins/llamacpp-completion/ops/batch-completion-stream.js +5 -5
- package/dist/server/bare/plugins/llamacpp-completion/ops/completion-stats.d.ts +11 -1
- package/dist/server/bare/plugins/llamacpp-completion/ops/completion-stats.js +16 -0
- package/dist/server/bare/plugins/llamacpp-completion/ops/completion-stream.d.ts +3 -3
- package/dist/server/bare/plugins/llamacpp-completion/ops/completion-stream.js +26 -35
- package/dist/server/bare/plugins/llamacpp-completion/ops/finetune.d.ts +1 -1
- package/dist/server/bare/plugins/llamacpp-completion/ops/finetune.js +3 -3
- package/dist/server/bare/plugins/llamacpp-completion/ops/kv-cache-session.d.ts +20 -10
- package/dist/server/bare/plugins/llamacpp-completion/ops/kv-cache-session.js +217 -80
- package/dist/server/bare/plugins/llamacpp-completion/ops/kv-cache-state.d.ts +8 -0
- package/dist/server/bare/plugins/llamacpp-completion/ops/kv-cache-state.js +8 -0
- package/dist/server/bare/plugins/llamacpp-completion/plugin.d.ts +20 -8
- package/dist/server/bare/plugins/llamacpp-completion/plugin.js +16 -16
- package/dist/server/bare/plugins/llamacpp-completion/transform.d.ts +1 -1
- package/dist/server/bare/plugins/llamacpp-embedding/plugin.d.ts +8 -3
- package/dist/server/bare/plugins/llamacpp-embedding/plugin.js +5 -5
- package/dist/server/bare/plugins/nmtcpp-translation/plugin.d.ts +12 -2
- package/dist/server/bare/plugins/nmtcpp-translation/plugin.js +3 -3
- package/dist/server/bare/plugins/nmtcpp-translation/resolve-vocab.d.ts +1 -1
- package/dist/server/bare/plugins/parakeet-transcription/plugin.d.ts +6 -6
- package/dist/server/bare/plugins/parakeet-transcription/plugin.js +13 -20
- package/dist/server/bare/plugins/sdcpp-generation/ops/diffusion.d.ts +1 -1
- package/dist/server/bare/plugins/sdcpp-generation/ops/diffusion.js +1 -1
- package/dist/server/bare/plugins/sdcpp-generation/ops/upscale.d.ts +1 -1
- package/dist/server/bare/plugins/sdcpp-generation/ops/upscale.js +1 -1
- package/dist/server/bare/plugins/sdcpp-generation/ops/video.d.ts +4 -1
- package/dist/server/bare/plugins/sdcpp-generation/ops/video.js +40 -5
- package/dist/server/bare/plugins/sdcpp-generation/plugin.d.ts +155 -5
- package/dist/server/bare/plugins/sdcpp-generation/plugin.js +109 -23
- package/dist/server/bare/plugins/tts-ggml/ops/parler-options.d.ts +7 -0
- package/dist/server/bare/plugins/tts-ggml/ops/parler-options.js +30 -0
- package/dist/server/bare/plugins/tts-ggml/ops/text-to-speech-stream.d.ts +2 -2
- package/dist/server/bare/plugins/tts-ggml/ops/text-to-speech-stream.js +8 -5
- package/dist/server/bare/plugins/tts-ggml/ops/text-to-speech.d.ts +2 -2
- package/dist/server/bare/plugins/tts-ggml/ops/text-to-speech.js +16 -22
- package/dist/server/bare/plugins/tts-ggml/plugin.d.ts +120 -5
- package/dist/server/bare/plugins/tts-ggml/plugin.js +63 -5
- package/dist/server/bare/plugins/whispercpp-transcription/plugin.d.ts +8 -6
- package/dist/server/bare/plugins/whispercpp-transcription/plugin.js +17 -25
- package/dist/server/bare/rag-hyperdb/handlers/chunk.d.ts +1 -1
- package/dist/server/bare/rag-hyperdb/handlers/close-workspace.d.ts +1 -1
- package/dist/server/bare/rag-hyperdb/handlers/close-workspace.js +1 -1
- package/dist/server/bare/rag-hyperdb/handlers/delete-embeddings.d.ts +1 -1
- package/dist/server/bare/rag-hyperdb/handlers/delete-embeddings.js +1 -1
- package/dist/server/bare/rag-hyperdb/handlers/delete-workspace.d.ts +1 -1
- package/dist/server/bare/rag-hyperdb/handlers/delete-workspace.js +1 -1
- package/dist/server/bare/rag-hyperdb/handlers/ingest.d.ts +1 -1
- package/dist/server/bare/rag-hyperdb/handlers/ingest.js +2 -2
- package/dist/server/bare/rag-hyperdb/handlers/list-workspaces.d.ts +1 -1
- package/dist/server/bare/rag-hyperdb/handlers/list-workspaces.js +1 -1
- package/dist/server/bare/rag-hyperdb/handlers/reindex.d.ts +1 -1
- package/dist/server/bare/rag-hyperdb/handlers/reindex.js +1 -1
- package/dist/server/bare/rag-hyperdb/handlers/save-embeddings.d.ts +1 -1
- package/dist/server/bare/rag-hyperdb/handlers/save-embeddings.js +1 -1
- package/dist/server/bare/rag-hyperdb/handlers/search.d.ts +1 -1
- package/dist/server/bare/rag-hyperdb/handlers/search.js +2 -2
- package/dist/server/bare/rag-hyperdb/index.d.ts +11 -11
- package/dist/server/bare/rag-hyperdb/index.js +11 -11
- package/dist/server/bare/rag-hyperdb/rag-operation-manager.js +1 -1
- package/dist/server/bare/rag-hyperdb/rag-workspace-manager.js +4 -4
- package/dist/server/bare/registry/config-registry.d.ts +1 -1
- package/dist/server/bare/registry/config-registry.js +1 -1
- package/dist/server/bare/registry/model-config-registry.d.ts +2 -2
- package/dist/server/bare/registry/model-config-registry.js +4 -4
- package/dist/server/bare/registry/model-config-utils.d.ts +1 -1
- package/dist/server/bare/registry/model-config-utils.js +3 -0
- package/dist/server/bare/registry/model-registry.d.ts +15 -4
- package/dist/server/bare/registry/registry-client.js +2 -2
- package/dist/server/bare/registry/runtime-context-registry.d.ts +1 -1
- package/dist/server/bare/resources/collector.d.ts +644 -0
- package/dist/server/bare/resources/collector.js +226 -0
- package/dist/server/bare/resources/native.d.ts +3 -0
- package/dist/server/bare/resources/native.js +35 -0
- package/dist/server/bare/resources/normalize.d.ts +13 -0
- package/dist/server/bare/resources/normalize.js +73 -0
- package/dist/server/bare/resources/types.d.ts +66 -0
- package/dist/server/bare/resources/types.js +1 -0
- package/dist/server/bare/resources/worker-collector.d.ts +5 -0
- package/dist/server/bare/resources/worker-collector.js +15 -0
- package/dist/server/bare/runtime/index.d.ts +7 -7
- package/dist/server/bare/runtime/index.js +3 -3
- package/dist/server/bare/runtime/request-context.d.ts +2 -2
- package/dist/server/bare/runtime/request-registry-singleton.d.ts +1 -1
- package/dist/server/bare/runtime/request-registry-singleton.js +11 -1
- package/dist/server/bare/runtime/request-registry.d.ts +2 -2
- package/dist/server/bare/runtime/request-registry.js +1 -1
- package/dist/server/bare/runtime/with-request-context.d.ts +1 -1
- package/dist/server/bare/runtime-lifecycle.d.ts +1 -1
- package/dist/server/bare/types/addon-responses.d.ts +8 -34
- package/dist/server/bare/utils/asr-events.d.ts +15 -0
- package/dist/server/bare/utils/asr-events.js +30 -0
- package/dist/server/bare/utils/audio-input.d.ts +1 -1
- package/dist/server/bare/utils/audio-input.js +1 -1
- package/dist/server/bare/utils/transcribe-metadata.d.ts +1 -1
- package/dist/server/bare/utils/tts-stats.d.ts +9 -5
- package/dist/server/bare/utils/tts-stats.js +6 -0
- package/dist/server/error-handlers.d.ts +1 -1
- package/dist/server/plugins/index.d.ts +1 -1
- package/dist/server/plugins/registry.d.ts +1 -1
- package/dist/server/plugins/registry.js +35 -8
- package/dist/server/rpc/create-server.js +23 -1
- package/dist/server/rpc/delegate-transport.d.ts +2 -2
- package/dist/server/rpc/delegate-transport.js +5 -10
- package/dist/server/rpc/handle-request.js +3 -3
- package/dist/server/rpc/handler-registry.d.ts +1 -1
- package/dist/server/rpc/handler-registry.js +45 -36
- package/dist/server/rpc/handler-selection.d.ts +2 -2
- package/dist/server/rpc/handler-utils.d.ts +2 -2
- package/dist/server/rpc/handler-utils.js +5 -5
- package/dist/server/rpc/handlers/audio-gen-stream.d.ts +3 -0
- package/dist/server/rpc/handlers/audio-gen-stream.js +4 -0
- package/dist/server/rpc/handlers/batch-completion-stream.d.ts +3 -1
- package/dist/server/rpc/handlers/batch-completion-stream.js +1 -1
- package/dist/server/rpc/handlers/bci-transcribe-stream.d.ts +1 -1
- package/dist/server/rpc/handlers/bci-transcribe-stream.js +1 -1
- package/dist/server/rpc/handlers/bci-transcribe.d.ts +1 -1
- package/dist/server/rpc/handlers/bci-transcribe.js +1 -1
- package/dist/server/rpc/handlers/cancel-delegated.d.ts +2 -2
- package/dist/server/rpc/handlers/cancel-delegated.js +3 -3
- package/dist/server/rpc/handlers/cancelHandler.d.ts +1 -1
- package/dist/server/rpc/handlers/cancelHandler.js +3 -3
- package/dist/server/rpc/handlers/classify.d.ts +1 -1
- package/dist/server/rpc/handlers/classify.js +1 -1
- package/dist/server/rpc/handlers/completion-orchestrate.d.ts +42 -0
- package/dist/server/rpc/handlers/completion-orchestrate.js +190 -0
- package/dist/server/rpc/handlers/completion-stream-delegated.d.ts +2 -2
- package/dist/server/rpc/handlers/completion-stream-delegated.js +3 -3
- package/dist/server/rpc/handlers/completion-stream.d.ts +1 -1
- package/dist/server/rpc/handlers/completion-stream.js +1 -1
- package/dist/server/rpc/handlers/delete-cache.d.ts +1 -1
- package/dist/server/rpc/handlers/delete-cache.js +1 -1
- package/dist/server/rpc/handlers/diffusion-stream.d.ts +1 -1
- package/dist/server/rpc/handlers/diffusion-stream.js +1 -1
- package/dist/server/rpc/handlers/download-asset.d.ts +1 -1
- package/dist/server/rpc/handlers/download-asset.js +4 -4
- package/dist/server/rpc/handlers/embed.d.ts +1 -1
- package/dist/server/rpc/handlers/embed.js +1 -1
- package/dist/server/rpc/handlers/finetune.d.ts +1 -1
- package/dist/server/rpc/handlers/finetune.js +2 -2
- package/dist/server/rpc/handlers/get-loaded-model-info.d.ts +1 -1
- package/dist/server/rpc/handlers/get-loaded-model-info.js +10 -4
- package/dist/server/rpc/handlers/get-model-info.d.ts +1 -1
- package/dist/server/rpc/handlers/get-model-info.js +4 -4
- package/dist/server/rpc/handlers/get-system-resources.d.ts +3 -0
- package/dist/server/rpc/handlers/get-system-resources.js +53 -0
- package/dist/server/rpc/handlers/heartbeat-delegated.d.ts +2 -2
- package/dist/server/rpc/handlers/heartbeat-delegated.js +2 -2
- package/dist/server/rpc/handlers/heartbeat.d.ts +1 -1
- package/dist/server/rpc/handlers/index.d.ts +38 -32
- package/dist/server/rpc/handlers/index.js +6 -0
- package/dist/server/rpc/handlers/load-model/download-manager.d.ts +3 -3
- package/dist/server/rpc/handlers/load-model/download-manager.js +1 -1
- package/dist/server/rpc/handlers/load-model/download-stats.d.ts +3 -3
- package/dist/server/rpc/handlers/load-model/handler.d.ts +1 -1
- package/dist/server/rpc/handlers/load-model/handler.js +12 -8
- package/dist/server/rpc/handlers/load-model/http.d.ts +2 -2
- package/dist/server/rpc/handlers/load-model/http.js +4 -4
- package/dist/server/rpc/handlers/load-model/hyperdrive.d.ts +2 -2
- package/dist/server/rpc/handlers/load-model/hyperdrive.js +5 -5
- package/dist/server/rpc/handlers/load-model/index.d.ts +2 -2
- package/dist/server/rpc/handlers/load-model/registry-client-options.d.ts +1 -1
- package/dist/server/rpc/handlers/load-model/registry-companion-set.d.ts +3 -3
- package/dist/server/rpc/handlers/load-model/registry-companion-set.js +3 -3
- package/dist/server/rpc/handlers/load-model/registry-download-utils.d.ts +4 -4
- package/dist/server/rpc/handlers/load-model/registry-download-utils.js +3 -3
- package/dist/server/rpc/handlers/load-model/registry-metadata.d.ts +1 -1
- package/dist/server/rpc/handlers/load-model/registry.d.ts +3 -3
- package/dist/server/rpc/handlers/load-model/registry.js +6 -6
- package/dist/server/rpc/handlers/load-model/resolve-session.d.ts +2 -2
- package/dist/server/rpc/handlers/load-model/resolve-session.js +3 -3
- package/dist/server/rpc/handlers/load-model/resolve.d.ts +2 -2
- package/dist/server/rpc/handlers/load-model/resolve.js +2 -2
- package/dist/server/rpc/handlers/load-model/types.d.ts +2 -2
- package/dist/server/rpc/handlers/load-model-delegated.d.ts +2 -2
- package/dist/server/rpc/handlers/load-model-delegated.js +3 -3
- package/dist/server/rpc/handlers/logging-stream.d.ts +1 -1
- package/dist/server/rpc/handlers/logging-stream.js +1 -1
- package/dist/server/rpc/handlers/ocr-stream.d.ts +1 -1
- package/dist/server/rpc/handlers/ocr-stream.js +1 -1
- package/dist/server/rpc/handlers/plugin-dispatch.js +3 -3
- package/dist/server/rpc/handlers/plugin-invoke.d.ts +1 -1
- package/dist/server/rpc/handlers/plugin-invoke.js +3 -3
- package/dist/server/rpc/handlers/provideHandler/connection.js +1 -1
- package/dist/server/rpc/handlers/provideHandler/handler.d.ts +1 -1
- package/dist/server/rpc/handlers/provideHandler/handler.js +1 -1
- package/dist/server/rpc/handlers/provideHandler/proxy.js +1 -1
- package/dist/server/rpc/handlers/provideHandler.d.ts +1 -1
- package/dist/server/rpc/handlers/rag.d.ts +1 -1
- package/dist/server/rpc/handlers/rag.js +4 -4
- package/dist/server/rpc/handlers/registry.d.ts +1 -1
- package/dist/server/rpc/handlers/registry.js +1 -1
- package/dist/server/rpc/handlers/resume.d.ts +1 -1
- package/dist/server/rpc/handlers/resume.js +1 -1
- package/dist/server/rpc/handlers/state.d.ts +1 -1
- package/dist/server/rpc/handlers/state.js +1 -1
- package/dist/server/rpc/handlers/stopProvideHandler.d.ts +1 -1
- package/dist/server/rpc/handlers/stopProvideHandler.js +1 -1
- package/dist/server/rpc/handlers/suspend.d.ts +1 -1
- package/dist/server/rpc/handlers/suspend.js +1 -1
- package/dist/server/rpc/handlers/text-to-speech-stream.d.ts +1 -1
- package/dist/server/rpc/handlers/text-to-speech-stream.js +1 -1
- package/dist/server/rpc/handlers/text-to-speech.d.ts +1 -1
- package/dist/server/rpc/handlers/text-to-speech.js +1 -1
- package/dist/server/rpc/handlers/transcribe-stream.d.ts +1 -1
- package/dist/server/rpc/handlers/transcribe-stream.js +1 -1
- package/dist/server/rpc/handlers/transcribe.d.ts +1 -1
- package/dist/server/rpc/handlers/transcribe.js +1 -1
- package/dist/server/rpc/handlers/translate.d.ts +1 -1
- package/dist/server/rpc/handlers/translate.js +1 -1
- package/dist/server/rpc/handlers/unload-model-delegated.d.ts +1 -1
- package/dist/server/rpc/handlers/unload-model-delegated.js +5 -5
- package/dist/server/rpc/handlers/unload-model.d.ts +1 -1
- package/dist/server/rpc/handlers/unload-model.js +3 -3
- package/dist/server/rpc/handlers/upscale-stream.d.ts +1 -1
- package/dist/server/rpc/handlers/upscale-stream.js +1 -1
- package/dist/server/rpc/handlers/video-stream.d.ts +1 -1
- package/dist/server/rpc/handlers/video-stream.js +1 -1
- package/dist/server/rpc/index.d.ts +2 -2
- package/dist/server/rpc/method-shapes.d.ts +53 -0
- package/dist/server/rpc/method-shapes.js +50 -0
- package/dist/server/rpc/profiling/backend-diagnostics.d.ts +3 -0
- package/dist/server/rpc/profiling/backend-diagnostics.js +16 -0
- package/dist/server/rpc/profiling/context.d.ts +1 -1
- package/dist/server/rpc/profiling/delegation-profiler.d.ts +2 -2
- package/dist/server/rpc/profiling/index.d.ts +4 -4
- package/dist/server/rpc/profiling/operation-metrics.d.ts +2 -2
- package/dist/server/rpc/profiling/operation-metrics.js +29 -1
- package/dist/server/rpc/profiling/operation-wrappers.d.ts +1 -1
- package/dist/server/rpc/profiling/operation-wrappers.js +51 -13
- package/dist/server/rpc/profiling/profiler.d.ts +2 -2
- package/dist/server/rpc/transport-selector.d.ts +1 -1
- package/dist/server/utils/archive.js +1 -1
- package/dist/server/utils/audio/decoder.d.ts +1 -1
- package/dist/server/utils/audio/decoder.js +1 -1
- package/dist/server/utils/cache/auto-history.d.ts +1 -1
- package/dist/server/utils/cache/auto-history.js +13 -9
- package/dist/server/utils/cache/paths.d.ts +1 -1
- package/dist/server/utils/cache/paths.js +6 -6
- package/dist/server/utils/checksum.d.ts +1 -1
- package/dist/server/utils/completion-normalizer.d.ts +2 -1
- package/dist/server/utils/completion-normalizer.js +31 -1
- package/dist/server/utils/config-transform.d.ts +1 -1
- package/dist/server/utils/config-transform.js +5 -7
- package/dist/server/utils/index.d.ts +11 -11
- package/dist/server/utils/path-security.d.ts +1 -1
- package/dist/server/utils/qvac-paths.js +1 -1
- package/dist/server/utils/response-format.d.ts +1 -1
- package/dist/server/utils/shard-utils.d.ts +1 -1
- package/dist/server/utils/tool-integration.d.ts +1 -1
- package/dist/server/utils/tool-integration.js +2 -2
- package/dist/server/utils/tools/dialect.d.ts +1 -1
- package/dist/server/utils/tools/dialect.js +2 -0
- package/dist/server/utils/tools/index.d.ts +2 -2
- package/dist/server/utils/tools/parser.d.ts +1 -1
- package/dist/server/utils/tools/parser.js +16 -8
- package/dist/server/utils/tools/parsers/dsml.d.ts +4 -0
- package/dist/server/utils/tools/parsers/dsml.js +176 -0
- package/dist/server/utils/tools/parsers/gemma4native.d.ts +2 -2
- package/dist/server/utils/tools/parsers/gemma4native.js +1 -1
- package/dist/server/utils/tools/parsers/harmony.d.ts +2 -2
- package/dist/server/utils/tools/parsers/harmony.js +1 -1
- package/dist/server/utils/tools/parsers/hermes.d.ts +2 -2
- package/dist/server/utils/tools/parsers/hermes.js +1 -1
- package/dist/server/utils/tools/parsers/json.d.ts +2 -2
- package/dist/server/utils/tools/parsers/json.js +1 -1
- package/dist/server/utils/tools/parsers/pythonic.d.ts +2 -2
- package/dist/server/utils/tools/parsers/pythonic.js +1 -1
- package/dist/server/utils/tools/parsers/qwen35.d.ts +2 -2
- package/dist/server/utils/tools/parsers/qwen35.js +1 -1
- package/dist/server/utils/tools/shared.d.ts +1 -1
- package/dist/server/utils/worker-lock.js +1 -1
- package/dist/server/worker-core.js +14 -10
- package/dist/server/worker.js +4 -3
- package/dist/utils/aggregate-events.d.ts +2 -2
- package/dist/utils/aggregate-events.js +2 -2
- package/dist/utils/errors-client.d.ts +0 -3
- package/dist/utils/errors-client.js +0 -5
- package/dist/utils/errors-server.d.ts +1 -1
- package/dist/utils/mcp-adapter.d.ts +4 -4
- package/dist/utils/mcp-adapter.js +1 -1
- package/dist/utils/tool-helpers.d.ts +1 -1
- package/dist/utils/tool-helpers.js +1 -1
- package/docs/system-resources-support-matrix.md +103 -0
- package/package.json +37 -19
- /package/dist/examples/{transcription → asr}/parakeet-ctc-filesystem.d.ts +0 -0
- /package/dist/examples/{transcription → asr}/parakeet-microphone-record.d.ts +0 -0
- /package/dist/examples/{transcription → asr}/parakeet-microphone-stream.d.ts +0 -0
- /package/dist/examples/{transcription → asr}/parakeet-sortformer-streaming.d.ts +0 -0
- /package/dist/examples/{transcription → asr}/parakeet-sortformer.d.ts +0 -0
- /package/dist/examples/{transcription → asr}/parakeet-tdt-filesystem.d.ts +0 -0
- /package/dist/examples/{transcription → asr}/whispercpp-filesystem-streaming.d.ts +0 -0
- /package/dist/examples/{transcription → asr}/whispercpp-filesystem.d.ts +0 -0
- /package/dist/examples/{transcription → asr}/whispercpp-microphone-conversation.d.ts +0 -0
- /package/dist/examples/{transcription → asr}/whispercpp-microphone-record.d.ts +0 -0
- /package/dist/examples/{transcription → asr}/whispercpp-prompt.d.ts +0 -0
|
@@ -0,0 +1,190 @@
|
|
|
1
|
+
import { toolCallbackResultSchema } from '../../../schemas/completion-stream.js';
|
|
2
|
+
import { aggregateEvents } from '../../../utils/aggregate-events.js';
|
|
3
|
+
import { CompletionFailedError } from '../../../utils/errors-server.js';
|
|
4
|
+
import { dispatchPluginStream } from '../../../server/rpc/handlers/plugin-dispatch.js';
|
|
5
|
+
const DEFAULT_MAX_TOOL_TURNS = 8;
|
|
6
|
+
// Generous ceiling on one client-side tool execution: long enough for a slow
|
|
7
|
+
// MCP round trip, short enough that a client that stopped reading can't pin a
|
|
8
|
+
// model slot forever.
|
|
9
|
+
const TOOL_RESULT_TIMEOUT_MS = 5 * 60 * 1000;
|
|
10
|
+
// A single tool result is small JSON; anything past this without a newline is a
|
|
11
|
+
// malformed/hostile client trying to grow the buffer without bound. Fail loudly
|
|
12
|
+
// instead of accumulating forever.
|
|
13
|
+
const MAX_TOOL_RESULT_LINE_BYTES = 16 * 1024 * 1024;
|
|
14
|
+
/**
|
|
15
|
+
* Reads newline-delimited JSON tool results off the duplex request stream
|
|
16
|
+
* and resolves per-callId waiters. Chunk boundaries don't align with lines,
|
|
17
|
+
* so a partial tail is buffered until its newline arrives.
|
|
18
|
+
*
|
|
19
|
+
* The pump is torn down by `close()` (called from a `finally` when
|
|
20
|
+
* orchestration ends, however it ends) so the upstream iterator is returned
|
|
21
|
+
* and the reader stops holding the request stream open.
|
|
22
|
+
*/
|
|
23
|
+
export class ToolResultReader {
|
|
24
|
+
buffer = '';
|
|
25
|
+
results = new Map();
|
|
26
|
+
waiters = new Map();
|
|
27
|
+
failed;
|
|
28
|
+
closed = false;
|
|
29
|
+
iterator;
|
|
30
|
+
constructor(inputStream) {
|
|
31
|
+
this.iterator = inputStream[Symbol.asyncIterator]();
|
|
32
|
+
void this.pump();
|
|
33
|
+
}
|
|
34
|
+
async pump() {
|
|
35
|
+
try {
|
|
36
|
+
while (!this.closed) {
|
|
37
|
+
const next = await this.iterator.next();
|
|
38
|
+
if (next.done)
|
|
39
|
+
break;
|
|
40
|
+
this.buffer += next.value.toString();
|
|
41
|
+
let nl;
|
|
42
|
+
while ((nl = this.buffer.indexOf('\n')) !== -1) {
|
|
43
|
+
const line = this.buffer.slice(0, nl);
|
|
44
|
+
this.buffer = this.buffer.slice(nl + 1);
|
|
45
|
+
if (line.trim())
|
|
46
|
+
this.accept(line);
|
|
47
|
+
}
|
|
48
|
+
if (this.buffer.length > MAX_TOOL_RESULT_LINE_BYTES) {
|
|
49
|
+
throw new CompletionFailedError('tool result frame exceeds the maximum size');
|
|
50
|
+
}
|
|
51
|
+
}
|
|
52
|
+
if (!this.closed && this.buffer.trim())
|
|
53
|
+
this.accept(this.buffer);
|
|
54
|
+
}
|
|
55
|
+
catch (error) {
|
|
56
|
+
this.failWaiters(error instanceof Error ? error : new Error(String(error)));
|
|
57
|
+
}
|
|
58
|
+
}
|
|
59
|
+
failWaiters(error) {
|
|
60
|
+
this.failed = error;
|
|
61
|
+
// Wake every waiter; each rethrows via waitFor's failure check.
|
|
62
|
+
for (const [callId, resolve] of this.waiters) {
|
|
63
|
+
this.waiters.delete(callId);
|
|
64
|
+
resolve({ callId, error: error.message });
|
|
65
|
+
}
|
|
66
|
+
}
|
|
67
|
+
accept(line) {
|
|
68
|
+
const parsed = toolCallbackResultSchema.parse(JSON.parse(line));
|
|
69
|
+
const waiter = this.waiters.get(parsed.callId);
|
|
70
|
+
if (waiter) {
|
|
71
|
+
this.waiters.delete(parsed.callId);
|
|
72
|
+
waiter(parsed);
|
|
73
|
+
}
|
|
74
|
+
else {
|
|
75
|
+
this.results.set(parsed.callId, parsed);
|
|
76
|
+
}
|
|
77
|
+
}
|
|
78
|
+
async waitFor(callId, timeoutMs = TOOL_RESULT_TIMEOUT_MS) {
|
|
79
|
+
const buffered = this.results.get(callId);
|
|
80
|
+
if (buffered) {
|
|
81
|
+
this.results.delete(callId);
|
|
82
|
+
return buffered;
|
|
83
|
+
}
|
|
84
|
+
if (this.failed)
|
|
85
|
+
throw this.failed;
|
|
86
|
+
return await new Promise((resolve, reject) => {
|
|
87
|
+
const timer = setTimeout(() => {
|
|
88
|
+
this.waiters.delete(callId);
|
|
89
|
+
reject(new CompletionFailedError(`tool callback ${callId} timed out`));
|
|
90
|
+
}, timeoutMs);
|
|
91
|
+
this.waiters.set(callId, (result) => {
|
|
92
|
+
clearTimeout(timer);
|
|
93
|
+
resolve(result);
|
|
94
|
+
});
|
|
95
|
+
});
|
|
96
|
+
}
|
|
97
|
+
/**
|
|
98
|
+
* Stop the pump and release the upstream iterator. Idempotent. Pending
|
|
99
|
+
* waiters are failed so a caller blocked in `waitFor` can't hang after the
|
|
100
|
+
* stream is torn down.
|
|
101
|
+
*/
|
|
102
|
+
close() {
|
|
103
|
+
if (this.closed)
|
|
104
|
+
return;
|
|
105
|
+
this.closed = true;
|
|
106
|
+
this.buffer = '';
|
|
107
|
+
this.results.clear();
|
|
108
|
+
if (!this.failed)
|
|
109
|
+
this.failWaiters(new CompletionFailedError('tool result stream closed'));
|
|
110
|
+
void this.iterator.return?.();
|
|
111
|
+
}
|
|
112
|
+
}
|
|
113
|
+
/**
|
|
114
|
+
* The orchestration core, pure over its collaborators so it unit-tests
|
|
115
|
+
* without the plugin registry: runs completion turns, forwards their events
|
|
116
|
+
* downstream, and when a turn requests tool calls, emits `toolCallback`
|
|
117
|
+
* frames and blocks on the client's results before starting the next turn
|
|
118
|
+
* with the extended history.
|
|
119
|
+
*/
|
|
120
|
+
export async function* orchestrateCompletion(request, runTurn, reader) {
|
|
121
|
+
// Pull the loop bound out of the per-turn base; everything else (including
|
|
122
|
+
// `requestId`) flows into each inner completionStream turn. Threading the
|
|
123
|
+
// orchestrate `requestId` down is what makes `cancel({ requestId })` reach
|
|
124
|
+
// the turn that's currently generating — the expensive part a Stop button
|
|
125
|
+
// targets. Turns run strictly sequentially: turn N's request-registry entry
|
|
126
|
+
// is disposed (on the `for await` break below) before turn N+1 calls
|
|
127
|
+
// `begin(...)`, so reusing the id can't collide in `entries`. A cancel that
|
|
128
|
+
// lands between turns (during a tool-result wait) is caught by the
|
|
129
|
+
// registry's cancel-before-begin tripwire and aborts the next turn's
|
|
130
|
+
// `begin(...)`. Each turn's `history` and `type` are set below.
|
|
131
|
+
const { maxToolTurns, ...turnBase } = request;
|
|
132
|
+
const maxTurns = maxToolTurns ?? DEFAULT_MAX_TOOL_TURNS;
|
|
133
|
+
let history = [...request.history];
|
|
134
|
+
for (let turn = 0; turn < maxTurns; turn++) {
|
|
135
|
+
const turnEvents = [];
|
|
136
|
+
const innerRequest = {
|
|
137
|
+
...turnBase,
|
|
138
|
+
history,
|
|
139
|
+
type: 'completionStream'
|
|
140
|
+
};
|
|
141
|
+
for await (const response of runTurn(innerRequest)) {
|
|
142
|
+
turnEvents.push(...response.events);
|
|
143
|
+
yield { type: 'completionOrchestrate', turn, events: response.events };
|
|
144
|
+
if (response.done)
|
|
145
|
+
break;
|
|
146
|
+
}
|
|
147
|
+
const { toolCalls, rawFullText, contentText, error, cancelled } = aggregateEvents(turnEvents);
|
|
148
|
+
// Terminal turn: the model answered (or failed/was cancelled) without
|
|
149
|
+
// requesting tools — the client folds the forwarded events itself.
|
|
150
|
+
if (error || cancelled || toolCalls.length === 0) {
|
|
151
|
+
yield { type: 'completionOrchestrate', done: true };
|
|
152
|
+
return;
|
|
153
|
+
}
|
|
154
|
+
// The assistant's tool-call turn goes back verbatim so the model sees
|
|
155
|
+
// its own call syntax; each result follows as a `tool` message.
|
|
156
|
+
history = [...history, { role: 'assistant', content: rawFullText ?? contentText }];
|
|
157
|
+
for (const call of toolCalls) {
|
|
158
|
+
yield {
|
|
159
|
+
type: 'completionOrchestrate',
|
|
160
|
+
turn,
|
|
161
|
+
toolCallback: { callId: call.id, name: call.name, arguments: call.arguments }
|
|
162
|
+
};
|
|
163
|
+
const result = await reader.waitFor(call.id);
|
|
164
|
+
const content = result.error !== undefined
|
|
165
|
+
? JSON.stringify({ error: result.error })
|
|
166
|
+
: JSON.stringify(result.result ?? null);
|
|
167
|
+
history = [...history, { role: 'tool', content }];
|
|
168
|
+
}
|
|
169
|
+
}
|
|
170
|
+
// Turn cap reached without a natural end. Emit a terminal done frame with a
|
|
171
|
+
// stopReason so the client can tell this was truncated (the model still
|
|
172
|
+
// wanted a tool) rather than a natural finish; the last turn's events carry
|
|
173
|
+
// whatever the model produced.
|
|
174
|
+
yield { type: 'completionOrchestrate', done: true, stopReason: 'maxToolTurns' };
|
|
175
|
+
}
|
|
176
|
+
export async function* handleCompletionOrchestrate(request, inputStream) {
|
|
177
|
+
const reader = new ToolResultReader(inputStream);
|
|
178
|
+
const runTurn = async function* (innerRequest) {
|
|
179
|
+
yield* dispatchPluginStream(innerRequest.modelId, 'completionStream', innerRequest);
|
|
180
|
+
};
|
|
181
|
+
try {
|
|
182
|
+
yield* orchestrateCompletion(request, runTurn, reader);
|
|
183
|
+
}
|
|
184
|
+
finally {
|
|
185
|
+
// Tear the reader down however orchestration ends -- natural finish, turn
|
|
186
|
+
// cap, error, or the client abandoning the stream -- so the pump stops and
|
|
187
|
+
// the request stream isn't held open.
|
|
188
|
+
reader.close();
|
|
189
|
+
}
|
|
190
|
+
}
|
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import type { CompletionStreamRequest, CompletionStreamResponse } from '../../../schemas/index';
|
|
2
|
-
import type { DelegatedHandlerOptions } from '
|
|
1
|
+
import type { CompletionStreamRequest, CompletionStreamResponse } from '../../../schemas/index.js';
|
|
2
|
+
import type { DelegatedHandlerOptions } from '../../../server/rpc/profiling/index.js';
|
|
3
3
|
export type HandleCompletionStreamDelegatedOptions = DelegatedHandlerOptions;
|
|
4
4
|
export declare function handleCompletionStreamDelegated(request: CompletionStreamRequest, options?: HandleCompletionStreamDelegatedOptions): AsyncGenerator<CompletionStreamResponse>;
|
|
5
5
|
//# sourceMappingURL=completion-stream-delegated.d.ts.map
|
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
import { getModelEntry } from '
|
|
2
|
-
import { getRPC } from '
|
|
3
|
-
import { stream } from '
|
|
1
|
+
import { getModelEntry } from '../../../server/bare/registry/model-registry.js';
|
|
2
|
+
import { getRPC } from '../../../server/bare/delegate-rpc-client.js';
|
|
3
|
+
import { stream } from '../../../server/rpc/delegate-transport.js';
|
|
4
4
|
import { ModelIsDelegatedError } from '../../../utils/errors-server.js';
|
|
5
5
|
import { getServerLogger } from '../../../logging/index.js';
|
|
6
6
|
const logger = getServerLogger();
|
|
@@ -1,3 +1,3 @@
|
|
|
1
|
-
import type { CompletionStreamRequest, CompletionStreamResponse } from '../../../schemas/index';
|
|
1
|
+
import type { CompletionStreamRequest, CompletionStreamResponse } from '../../../schemas/index.js';
|
|
2
2
|
export declare function handleCompletionStream(request: CompletionStreamRequest): AsyncGenerator<CompletionStreamResponse>;
|
|
3
3
|
//# sourceMappingURL=completion-stream.d.ts.map
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { dispatchPluginStream } from '
|
|
1
|
+
import { dispatchPluginStream } from '../../../server/rpc/handlers/plugin-dispatch.js';
|
|
2
2
|
export async function* handleCompletionStream(request) {
|
|
3
3
|
yield* dispatchPluginStream(request.modelId, 'completionStream', request);
|
|
4
4
|
}
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import type { DeleteCacheRequest, DeleteCacheResponse } from '../../../schemas/index';
|
|
1
|
+
import type { DeleteCacheRequest, DeleteCacheResponse } from '../../../schemas/index.js';
|
|
2
2
|
/**
|
|
3
3
|
* RPC handler for `deleteCache(...)`. The three KV-cache bookkeeping
|
|
4
4
|
* layers (on-disk `.bin`, `initializedCaches`, `cachedMessageCounts`)
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { deleteKvCacheState } from '
|
|
1
|
+
import { deleteKvCacheState } from '../../../server/bare/plugins/llamacpp-completion/ops/kv-cache-session.js';
|
|
2
2
|
import { getServerLogger } from '../../../logging/index.js';
|
|
3
3
|
const logger = getServerLogger();
|
|
4
4
|
/**
|
|
@@ -1,3 +1,3 @@
|
|
|
1
|
-
import type { DiffusionStreamRequest, DiffusionStreamResponse } from '../../../schemas/sdcpp-config';
|
|
1
|
+
import type { DiffusionStreamRequest, DiffusionStreamResponse } from '../../../schemas/sdcpp-config.js';
|
|
2
2
|
export declare function handleDiffusionStream(request: DiffusionStreamRequest): AsyncGenerator<DiffusionStreamResponse>;
|
|
3
3
|
//# sourceMappingURL=diffusion-stream.d.ts.map
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import { dispatchPluginStream } from '
|
|
1
|
+
import { dispatchPluginStream } from '../../../server/rpc/handlers/plugin-dispatch.js';
|
|
2
2
|
export async function* handleDiffusionStream(request) {
|
|
3
3
|
yield* dispatchPluginStream(request.modelId, 'diffusionStream', request);
|
|
4
4
|
}
|
|
@@ -1,3 +1,3 @@
|
|
|
1
|
-
import type { DownloadAssetRequest, DownloadAssetResponse, ModelProgressUpdate } from '../../../schemas/index';
|
|
1
|
+
import type { DownloadAssetRequest, DownloadAssetResponse, ModelProgressUpdate } from '../../../schemas/index.js';
|
|
2
2
|
export declare function handleDownloadAsset(request: DownloadAssetRequest, progressCallback?: (update: ModelProgressUpdate) => void): Promise<DownloadAssetResponse>;
|
|
3
3
|
//# sourceMappingURL=download-asset.d.ts.map
|
|
@@ -51,12 +51,12 @@ var __disposeResources = (this && this.__disposeResources) || (function (Suppres
|
|
|
51
51
|
return e.name = "SuppressedError", e.error = error, e.suppressed = suppressed, e;
|
|
52
52
|
});
|
|
53
53
|
import { PROFILING_KEY, OPERATION_EVENT_KEY } from '../../../schemas/index.js';
|
|
54
|
-
import { resolveModelPath, resolveModelPathWithStats } from '
|
|
55
|
-
import { buildDownloadProfilingFields } from '
|
|
54
|
+
import { resolveModelPath, resolveModelPathWithStats } from '../../../server/rpc/handlers/load-model/resolve.js';
|
|
55
|
+
import { buildDownloadProfilingFields } from '../../../server/rpc/handlers/load-model/types.js';
|
|
56
56
|
import { nowMs, generateProfileId } from '../../../profiling/clock.js';
|
|
57
57
|
import { getServerLogger } from '../../../logging/index.js';
|
|
58
|
-
import { getRequestRegistry, withRequestContext } from '
|
|
59
|
-
import { generateServerRequestId } from '
|
|
58
|
+
import { getRequestRegistry, withRequestContext } from '../../../server/bare/runtime/index.js';
|
|
59
|
+
import { generateServerRequestId } from '../../../server/bare/runtime/request-id.js';
|
|
60
60
|
import { InferenceCancelledError } from '../../../utils/errors-server.js';
|
|
61
61
|
const logger = getServerLogger();
|
|
62
62
|
export async function handleDownloadAsset(request, progressCallback) {
|
|
@@ -1,3 +1,3 @@
|
|
|
1
|
-
import type { EmbedRequest, EmbedResponse } from '../../../schemas/index';
|
|
1
|
+
import type { EmbedRequest, EmbedResponse } from '../../../schemas/index.js';
|
|
2
2
|
export declare function handleEmbed(request: EmbedRequest): Promise<EmbedResponse>;
|
|
3
3
|
//# sourceMappingURL=embed.d.ts.map
|
|
@@ -1,3 +1,3 @@
|
|
|
1
|
-
import type { FinetuneProgressResponse, FinetuneRequest, FinetuneResult } from '../../../schemas/index';
|
|
1
|
+
import type { FinetuneProgressResponse, FinetuneRequest, FinetuneResult } from '../../../schemas/index.js';
|
|
2
2
|
export declare function handleFinetune(request: FinetuneRequest, progressCallback?: (update: FinetuneProgressResponse) => void): Promise<FinetuneResult>;
|
|
3
3
|
//# sourceMappingURL=finetune.d.ts.map
|
|
@@ -1,5 +1,5 @@
|
|
|
1
|
-
import { dispatchPluginReply } from '
|
|
2
|
-
import { startFinetune } from '
|
|
1
|
+
import { dispatchPluginReply } from '../../../server/rpc/handlers/plugin-dispatch.js';
|
|
2
|
+
import { startFinetune } from '../../../server/bare/plugins/llamacpp-completion/ops/finetune.js';
|
|
3
3
|
export async function handleFinetune(request, progressCallback) {
|
|
4
4
|
if (progressCallback &&
|
|
5
5
|
(request.operation === 'start' ||
|
|
@@ -1,3 +1,3 @@
|
|
|
1
|
-
import type { GetLoadedModelInfoRequest, GetLoadedModelInfoResponse } from '../../../schemas/index';
|
|
1
|
+
import type { GetLoadedModelInfoRequest, GetLoadedModelInfoResponse } from '../../../schemas/index.js';
|
|
2
2
|
export declare function handleGetLoadedModelInfo(request: GetLoadedModelInfoRequest): GetLoadedModelInfoResponse;
|
|
3
3
|
//# sourceMappingURL=get-loaded-model-info.d.ts.map
|
|
@@ -1,5 +1,6 @@
|
|
|
1
|
-
import { getModelEntry } from '
|
|
2
|
-
import { getPlugin } from '
|
|
1
|
+
import { getModelEntry } from '../../../server/bare/registry/model-registry.js';
|
|
2
|
+
import { getPlugin } from '../../../server/plugins/registry.js';
|
|
3
|
+
import { detectToolDialectFromName } from '../../../server/utils/tools/index.js';
|
|
3
4
|
import { ModelNotFoundError } from '../../../utils/errors-server.js';
|
|
4
5
|
import { getServerLogger } from '../../../logging/index.js';
|
|
5
6
|
const logger = getServerLogger();
|
|
@@ -24,16 +25,21 @@ export function handleGetLoadedModelInfo(request) {
|
|
|
24
25
|
if (!plugin) {
|
|
25
26
|
logger.warn(`getLoadedModelInfo: no plugin registered for modelType "${entry.local.modelType}" on loaded model "${modelId}"`);
|
|
26
27
|
}
|
|
28
|
+
const handlers = plugin ? Object.keys(plugin.handlers) : [];
|
|
27
29
|
const info = {
|
|
28
30
|
modelId: entry.id,
|
|
29
31
|
isDelegated: false,
|
|
30
32
|
modelType: entry.local.modelType,
|
|
31
|
-
handlers
|
|
33
|
+
handlers,
|
|
32
34
|
loadedAt: entry.local.loadedAt,
|
|
33
35
|
...(plugin && { displayName: plugin.displayName }),
|
|
34
36
|
...(plugin && { addonPackage: plugin.addonPackage }),
|
|
35
37
|
...(entry.local.name && { name: entry.local.name }),
|
|
36
|
-
...(entry.local.path && { path: entry.local.path })
|
|
38
|
+
...(entry.local.path && { path: entry.local.path }),
|
|
39
|
+
// Same detection the completion normalizer uses, so callers see the dialect it parses.
|
|
40
|
+
...(handlers.includes('completionStream') && {
|
|
41
|
+
toolDialect: detectToolDialectFromName(entry.local.name, entry.local.path)
|
|
42
|
+
})
|
|
37
43
|
};
|
|
38
44
|
return { type: 'getLoadedModelInfo', info };
|
|
39
45
|
}
|
|
@@ -1,3 +1,3 @@
|
|
|
1
|
-
import type { GetModelInfoRequest, GetModelInfoResponse } from '../../../schemas/index';
|
|
1
|
+
import type { GetModelInfoRequest, GetModelInfoResponse } from '../../../schemas/index.js';
|
|
2
2
|
export declare function handleGetModelInfo(request: GetModelInfoRequest): Promise<GetModelInfoResponse>;
|
|
3
3
|
//# sourceMappingURL=get-model-info.d.ts.map
|
|
@@ -1,9 +1,9 @@
|
|
|
1
1
|
import { models } from '../../../models/registry/models.js';
|
|
2
|
-
import { getAllModelIds, getModelEntry } from '
|
|
3
|
-
import { generateShortHash } from '
|
|
2
|
+
import { getAllModelIds, getModelEntry } from '../../../server/bare/registry/model-registry.js';
|
|
3
|
+
import { generateShortHash } from '../../../server/utils/index.js';
|
|
4
4
|
import { promises as fsPromises } from 'bare-fs';
|
|
5
|
-
import { getShardPath, getModelsCacheDir, getSingleFileCachePath } from '
|
|
6
|
-
import { validateAndJoinPath } from '
|
|
5
|
+
import { getShardPath, getModelsCacheDir, getSingleFileCachePath } from '../../../server/utils/cache/paths.js';
|
|
6
|
+
import { validateAndJoinPath } from '../../../server/utils/path-security.js';
|
|
7
7
|
import { ModelNotFoundError } from '../../../utils/errors-server.js';
|
|
8
8
|
export async function handleGetModelInfo(request) {
|
|
9
9
|
const { name } = request;
|
|
@@ -0,0 +1,3 @@
|
|
|
1
|
+
import type { GetSystemResourcesRequest, GetSystemResourcesResponse } from '../../../schemas/system-resources.js';
|
|
2
|
+
export declare function handleGetSystemResources(request: GetSystemResourcesRequest): GetSystemResourcesResponse;
|
|
3
|
+
//# sourceMappingURL=get-system-resources.d.ts.map
|
|
@@ -0,0 +1,53 @@
|
|
|
1
|
+
import { getWorkerResourceCollector } from '../../../server/bare/resources/worker-collector.js';
|
|
2
|
+
export function handleGetSystemResources(request) {
|
|
3
|
+
const collector = getWorkerResourceCollector();
|
|
4
|
+
if (!collector) {
|
|
5
|
+
return {
|
|
6
|
+
type: 'getSystemResources',
|
|
7
|
+
capabilities: {
|
|
8
|
+
cpu: {
|
|
9
|
+
status: 'failed',
|
|
10
|
+
reason: 'CPU resource collector is not initialized'
|
|
11
|
+
},
|
|
12
|
+
memory: {
|
|
13
|
+
totalBytes: {
|
|
14
|
+
status: 'failed',
|
|
15
|
+
reason: 'CPU resource collector is not initialized'
|
|
16
|
+
}
|
|
17
|
+
},
|
|
18
|
+
gpus: {
|
|
19
|
+
status: 'failed',
|
|
20
|
+
reason: 'GPU resource collector is not initialized'
|
|
21
|
+
}
|
|
22
|
+
},
|
|
23
|
+
...(request.sample && {
|
|
24
|
+
sample: {
|
|
25
|
+
sampledAt: Date.now(),
|
|
26
|
+
cpu: {
|
|
27
|
+
status: 'failed',
|
|
28
|
+
reason: 'CPU resource collector is not initialized'
|
|
29
|
+
},
|
|
30
|
+
memory: {
|
|
31
|
+
usedBytes: {
|
|
32
|
+
status: 'failed',
|
|
33
|
+
reason: 'CPU resource collector is not initialized'
|
|
34
|
+
},
|
|
35
|
+
totalBytes: {
|
|
36
|
+
status: 'failed',
|
|
37
|
+
reason: 'CPU resource collector is not initialized'
|
|
38
|
+
}
|
|
39
|
+
},
|
|
40
|
+
gpus: {
|
|
41
|
+
status: 'failed',
|
|
42
|
+
reason: 'GPU resource collector is not initialized'
|
|
43
|
+
}
|
|
44
|
+
}
|
|
45
|
+
})
|
|
46
|
+
};
|
|
47
|
+
}
|
|
48
|
+
return {
|
|
49
|
+
type: 'getSystemResources',
|
|
50
|
+
capabilities: collector.getCapabilities(),
|
|
51
|
+
...(request.sample && { sample: collector.sample() })
|
|
52
|
+
};
|
|
53
|
+
}
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import type { HeartbeatRequest, HeartbeatResponse } from '../../../schemas/index';
|
|
2
|
-
import type { DelegatedHandlerOptions } from '
|
|
1
|
+
import type { HeartbeatRequest, HeartbeatResponse } from '../../../schemas/index.js';
|
|
2
|
+
import type { DelegatedHandlerOptions } from '../../../server/rpc/profiling/index.js';
|
|
3
3
|
export declare function handleHeartbeatDelegated(request: HeartbeatRequest, options?: DelegatedHandlerOptions): Promise<HeartbeatResponse>;
|
|
4
4
|
//# sourceMappingURL=heartbeat-delegated.d.ts.map
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
import { getServerLogger } from '../../../logging/index.js';
|
|
2
|
-
import { getRPC } from '
|
|
3
|
-
import { send } from '
|
|
2
|
+
import { getRPC } from '../../../server/bare/delegate-rpc-client.js';
|
|
3
|
+
import { send } from '../../../server/rpc/delegate-transport.js';
|
|
4
4
|
import { DelegateConnectionFailedError } from '../../../utils/errors-server.js';
|
|
5
5
|
const logger = getServerLogger();
|
|
6
6
|
export async function handleHeartbeatDelegated(request, options) {
|
|
@@ -1,39 +1,44 @@
|
|
|
1
|
-
import { handleBatchCompletionStream } from './batch-completion-stream';
|
|
2
|
-
import {
|
|
3
|
-
import {
|
|
4
|
-
import {
|
|
5
|
-
import {
|
|
6
|
-
import {
|
|
7
|
-
import {
|
|
8
|
-
import {
|
|
9
|
-
import {
|
|
10
|
-
import {
|
|
11
|
-
import {
|
|
12
|
-
import {
|
|
13
|
-
import {
|
|
14
|
-
import {
|
|
15
|
-
import {
|
|
16
|
-
import {
|
|
17
|
-
import {
|
|
18
|
-
import {
|
|
19
|
-
import {
|
|
20
|
-
import {
|
|
21
|
-
import {
|
|
22
|
-
import {
|
|
23
|
-
import {
|
|
24
|
-
import {
|
|
25
|
-
import {
|
|
26
|
-
import {
|
|
27
|
-
import {
|
|
28
|
-
import {
|
|
29
|
-
import {
|
|
30
|
-
import {
|
|
31
|
-
import {
|
|
32
|
-
import {
|
|
1
|
+
import { handleBatchCompletionStream } from './batch-completion-stream.js';
|
|
2
|
+
import { handleCompletionOrchestrate } from './completion-orchestrate.js';
|
|
3
|
+
import { handleCompletionStream } from './completion-stream.js';
|
|
4
|
+
import { handleDownloadAsset } from './download-asset.js';
|
|
5
|
+
import { handleLoadModel } from './load-model/index.js';
|
|
6
|
+
import { handleUnloadModel } from './unload-model.js';
|
|
7
|
+
import { handleEmbed } from './embed.js';
|
|
8
|
+
import { handleTranscribe } from './transcribe.js';
|
|
9
|
+
import { handleTranscribeStream } from './transcribe-stream.js';
|
|
10
|
+
import { handleBciTranscribe } from './bci-transcribe.js';
|
|
11
|
+
import { handleBciTranscribeStream } from './bci-transcribe-stream.js';
|
|
12
|
+
import { provideHandler } from './provideHandler.js';
|
|
13
|
+
import { stopProvideHandler } from './stopProvideHandler.js';
|
|
14
|
+
import { handleTranslate } from './translate.js';
|
|
15
|
+
import { handleLoggingStream } from './logging-stream.js';
|
|
16
|
+
import { handleRag } from './rag.js';
|
|
17
|
+
import { cancelHandler } from './cancelHandler.js';
|
|
18
|
+
import { handleDeleteCache } from './delete-cache.js';
|
|
19
|
+
import { handleTextToSpeech } from './text-to-speech.js';
|
|
20
|
+
import { handleTextToSpeechStream } from './text-to-speech-stream.js';
|
|
21
|
+
import { handleGetModelInfo } from './get-model-info.js';
|
|
22
|
+
import { handleGetLoadedModelInfo } from './get-loaded-model-info.js';
|
|
23
|
+
import { handleGetSystemResources } from './get-system-resources.js';
|
|
24
|
+
import { handleFinetune } from './finetune.js';
|
|
25
|
+
import { handleOCRStream } from './ocr-stream.js';
|
|
26
|
+
import { handleHeartbeat } from './heartbeat.js';
|
|
27
|
+
import { handleDiffusionStream } from './diffusion-stream.js';
|
|
28
|
+
import { handleVideoStream } from './video-stream.js';
|
|
29
|
+
import { handleUpscaleStream } from './upscale-stream.js';
|
|
30
|
+
import { handlePluginInvoke, handlePluginInvokeStream } from './plugin-invoke.js';
|
|
31
|
+
import { handleModelRegistryList, handleModelRegistrySearch, handleModelRegistryGetModel } from './registry.js';
|
|
32
|
+
import { handleSuspend } from './suspend.js';
|
|
33
|
+
import { handleResume } from './resume.js';
|
|
34
|
+
import { handleState } from './state.js';
|
|
35
|
+
import { handleAudioGenStream } from '../../../server/rpc/handlers/audio-gen-stream.js';
|
|
33
36
|
export declare const handlers: {
|
|
37
|
+
audioGenStream: typeof handleAudioGenStream;
|
|
34
38
|
heartbeat: typeof handleHeartbeat;
|
|
35
39
|
batchCompletionStream: typeof handleBatchCompletionStream;
|
|
36
40
|
completionStream: typeof handleCompletionStream;
|
|
41
|
+
completionOrchestrate: typeof handleCompletionOrchestrate;
|
|
37
42
|
downloadAsset: typeof handleDownloadAsset;
|
|
38
43
|
deleteCache: typeof handleDeleteCache;
|
|
39
44
|
loadModel: typeof handleLoadModel;
|
|
@@ -53,6 +58,7 @@ export declare const handlers: {
|
|
|
53
58
|
textToSpeechStream: typeof handleTextToSpeechStream;
|
|
54
59
|
getModelInfo: typeof handleGetModelInfo;
|
|
55
60
|
getLoadedModelInfo: typeof handleGetLoadedModelInfo;
|
|
61
|
+
getSystemResources: typeof handleGetSystemResources;
|
|
56
62
|
finetune: typeof handleFinetune;
|
|
57
63
|
ocrStream: typeof handleOCRStream;
|
|
58
64
|
diffusionStream: typeof handleDiffusionStream;
|
|
@@ -1,4 +1,5 @@
|
|
|
1
1
|
import { handleBatchCompletionStream } from './batch-completion-stream.js';
|
|
2
|
+
import { handleCompletionOrchestrate } from './completion-orchestrate.js';
|
|
2
3
|
import { handleCompletionStream } from './completion-stream.js';
|
|
3
4
|
import { handleDownloadAsset } from './download-asset.js';
|
|
4
5
|
import { handleLoadModel } from './load-model/index.js';
|
|
@@ -19,6 +20,7 @@ import { handleTextToSpeech } from './text-to-speech.js';
|
|
|
19
20
|
import { handleTextToSpeechStream } from './text-to-speech-stream.js';
|
|
20
21
|
import { handleGetModelInfo } from './get-model-info.js';
|
|
21
22
|
import { handleGetLoadedModelInfo } from './get-loaded-model-info.js';
|
|
23
|
+
import { handleGetSystemResources } from './get-system-resources.js';
|
|
22
24
|
import { handleFinetune } from './finetune.js';
|
|
23
25
|
import { handleOCRStream } from './ocr-stream.js';
|
|
24
26
|
import { handleHeartbeat } from './heartbeat.js';
|
|
@@ -30,10 +32,13 @@ import { handleModelRegistryList, handleModelRegistrySearch, handleModelRegistry
|
|
|
30
32
|
import { handleSuspend } from './suspend.js';
|
|
31
33
|
import { handleResume } from './resume.js';
|
|
32
34
|
import { handleState } from './state.js';
|
|
35
|
+
import { handleAudioGenStream } from '../../../server/rpc/handlers/audio-gen-stream.js';
|
|
33
36
|
export const handlers = {
|
|
37
|
+
audioGenStream: handleAudioGenStream,
|
|
34
38
|
heartbeat: handleHeartbeat,
|
|
35
39
|
batchCompletionStream: handleBatchCompletionStream,
|
|
36
40
|
completionStream: handleCompletionStream,
|
|
41
|
+
completionOrchestrate: handleCompletionOrchestrate,
|
|
37
42
|
downloadAsset: handleDownloadAsset,
|
|
38
43
|
deleteCache: handleDeleteCache,
|
|
39
44
|
loadModel: handleLoadModel,
|
|
@@ -53,6 +58,7 @@ export const handlers = {
|
|
|
53
58
|
textToSpeechStream: handleTextToSpeechStream,
|
|
54
59
|
getModelInfo: handleGetModelInfo,
|
|
55
60
|
getLoadedModelInfo: handleGetLoadedModelInfo,
|
|
61
|
+
getSystemResources: handleGetSystemResources,
|
|
56
62
|
finetune: handleFinetune,
|
|
57
63
|
ocrStream: handleOCRStream,
|
|
58
64
|
diffusionStream: handleDiffusionStream,
|
|
@@ -1,7 +1,7 @@
|
|
|
1
|
-
import type { ModelProgressUpdate } from '../../../../schemas/index';
|
|
1
|
+
import type { ModelProgressUpdate } from '../../../../schemas/index.js';
|
|
2
2
|
import { AbortController, type AbortSignal } from 'bare-abort-controller';
|
|
3
|
-
import type { DisposableScope } from '
|
|
4
|
-
import type { DownloadHooks } from '
|
|
3
|
+
import type { DisposableScope } from '../../../../server/bare/runtime/disposable-scope.js';
|
|
4
|
+
import type { DownloadHooks } from '../../../../server/rpc/handlers/load-model/types.js';
|
|
5
5
|
/**
|
|
6
6
|
* Per-subscriber binding to a registry-tracked request.
|
|
7
7
|
* `startOrJoinDownload` is request-aware: each caller (the
|
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
import { AbortController } from 'bare-abort-controller';
|
|
2
2
|
import { DownloadCancelledError, InferenceCancelledError } from '../../../../utils/errors-server.js';
|
|
3
3
|
import { getServerLogger } from '../../../../logging/index.js';
|
|
4
|
-
import { getRequestRegistry } from '
|
|
4
|
+
import { getRequestRegistry } from '../../../../server/bare/runtime/index.js';
|
|
5
5
|
const logger = getServerLogger();
|
|
6
6
|
const activeTransfers = new Map();
|
|
7
7
|
let nextSubscriberId = 0;
|
|
@@ -1,6 +1,6 @@
|
|
|
1
|
-
import type { ModelProgressUpdate } from '../../../../schemas/index';
|
|
2
|
-
import type { DownloadStats, DownloadResult, DownloadHooks, ResolveResult } from './types';
|
|
3
|
-
import type { ExplicitRegistryMetadata } from './registry-metadata';
|
|
1
|
+
import type { ModelProgressUpdate } from '../../../../schemas/index.js';
|
|
2
|
+
import type { DownloadStats, DownloadResult, DownloadHooks, ResolveResult } from './types.js';
|
|
3
|
+
import type { ExplicitRegistryMetadata } from './registry-metadata.js';
|
|
4
4
|
export declare function downloadModelFromHttpWithStats(url: string, progressCallback?: (progress: ModelProgressUpdate) => void, downloadHooks?: DownloadHooks): Promise<DownloadResult>;
|
|
5
5
|
export declare function downloadModelFromRegistryWithStats(registryPath: string, registrySource: string, progressCallback?: (progress: ModelProgressUpdate) => void, expectedChecksum?: string, downloadHooks?: DownloadHooks, explicitMetadata?: ExplicitRegistryMetadata): Promise<DownloadResult>;
|
|
6
6
|
export declare function downloadModelFromHyperdriveWithStats(hyperdriveKey: string, modelFileName: string, seed?: boolean, progressCallback?: (progress: ModelProgressUpdate) => void, downloadHooks?: DownloadHooks): Promise<DownloadResult>;
|
|
@@ -1,3 +1,3 @@
|
|
|
1
|
-
import type { LoadModelRequest, LoadModelResponse, ModelProgressUpdate } from '../../../../schemas/index';
|
|
1
|
+
import type { LoadModelRequest, LoadModelResponse, ModelProgressUpdate } from '../../../../schemas/index.js';
|
|
2
2
|
export declare function handleLoadModel(request: LoadModelRequest, progressCallback?: (update: ModelProgressUpdate) => void): Promise<LoadModelResponse>;
|
|
3
3
|
//# sourceMappingURL=handler.d.ts.map
|