@elizaos/plugin-local-inference 2.0.0-beta.1 → 2.0.3-beta.2
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/LICENSE +21 -0
- package/README.md +83 -0
- package/package.json +82 -15
- package/src/actions/generate-media.d.ts +59 -0
- package/src/actions/generate-media.d.ts.map +1 -0
- package/src/actions/generate-media.ts +647 -0
- package/src/actions/identify-speaker.d.ts +23 -0
- package/src/actions/identify-speaker.d.ts.map +1 -0
- package/src/actions/identify-speaker.ts +171 -0
- package/src/actions/transcription-control.d.ts +29 -0
- package/src/actions/transcription-control.d.ts.map +1 -0
- package/src/actions/transcription-control.test.ts +100 -0
- package/src/actions/transcription-control.ts +127 -0
- package/src/adapters/capacitor-llama/__tests__/compat-behavior.test.ts +218 -0
- package/src/adapters/capacitor-llama/__tests__/index.test.ts +68 -0
- package/src/adapters/capacitor-llama/__tests__/structured-output.test.ts +215 -0
- package/src/adapters/capacitor-llama/__tests__/text-streaming.test.ts +174 -0
- package/src/adapters/capacitor-llama/environment.ts +71 -0
- package/src/adapters/capacitor-llama/index.browser.ts +83 -0
- package/src/adapters/capacitor-llama/index.ts +807 -0
- package/src/adapters/capacitor-llama/loader.ts +109 -0
- package/src/adapters/capacitor-llama/structured-output.ts +165 -0
- package/src/adapters/capacitor-llama/text-streaming.ts +227 -0
- package/src/adapters/capacitor-llama/types.ts +374 -0
- package/src/backends/apple-foundation.ts +127 -0
- package/src/index.d.ts +8 -0
- package/src/index.d.ts.map +1 -0
- package/src/index.ts +62 -0
- package/src/local-inference-routes.d.ts +38 -0
- package/src/local-inference-routes.d.ts.map +1 -0
- package/src/local-inference-routes.test.ts +344 -0
- package/src/local-inference-routes.ts +1543 -0
- package/src/provider.d.ts +21 -0
- package/src/provider.d.ts.map +1 -0
- package/src/provider.ts +1082 -0
- package/src/routes/compat-helpers.d.ts +18 -0
- package/src/routes/compat-helpers.d.ts.map +1 -0
- package/src/routes/compat-helpers.ts +274 -0
- package/src/routes/family-member-route.d.ts +62 -0
- package/src/routes/family-member-route.d.ts.map +1 -0
- package/src/routes/family-member-route.ts +353 -0
- package/src/routes/index.d.ts +19 -0
- package/src/routes/index.d.ts.map +1 -0
- package/src/routes/index.ts +60 -0
- package/src/routes/live-diarization-route.d.ts +26 -0
- package/src/routes/live-diarization-route.d.ts.map +1 -0
- package/src/routes/live-diarization-route.test.ts +213 -0
- package/src/routes/live-diarization-route.ts +122 -0
- package/src/routes/local-inference-asr-route.d.ts +4 -0
- package/src/routes/local-inference-asr-route.d.ts.map +1 -0
- package/src/routes/local-inference-asr-route.test.ts +205 -0
- package/src/routes/local-inference-asr-route.ts +163 -0
- package/src/routes/local-inference-asr-transcribe.d.ts +20 -0
- package/src/routes/local-inference-asr-transcribe.d.ts.map +1 -0
- package/src/routes/local-inference-asr-transcribe.test.ts +118 -0
- package/src/routes/local-inference-asr-transcribe.ts +97 -0
- package/src/routes/local-inference-compat-routes.d.ts +16 -0
- package/src/routes/local-inference-compat-routes.d.ts.map +1 -0
- package/src/routes/local-inference-compat-routes.test.ts +485 -0
- package/src/routes/local-inference-compat-routes.ts +808 -0
- package/src/routes/local-inference-tts-route.d.ts +7 -0
- package/src/routes/local-inference-tts-route.d.ts.map +1 -0
- package/src/routes/local-inference-tts-route.test.ts +179 -0
- package/src/routes/local-inference-tts-route.ts +230 -0
- package/src/routes/transcript-audio-store.d.ts +15 -0
- package/src/routes/transcript-audio-store.d.ts.map +1 -0
- package/src/routes/transcript-audio-store.ts +27 -0
- package/src/routes/transcripts-routes.d.ts +36 -0
- package/src/routes/transcripts-routes.d.ts.map +1 -0
- package/src/routes/transcripts-routes.test.ts +144 -0
- package/src/routes/transcripts-routes.ts +159 -0
- package/src/routes/voice-first-run-routes.d.ts +62 -0
- package/src/routes/voice-first-run-routes.d.ts.map +1 -0
- package/src/routes/voice-first-run-routes.ts +524 -0
- package/src/routes/voice-models-routes.d.ts +62 -0
- package/src/routes/voice-models-routes.d.ts.map +1 -0
- package/src/routes/voice-models-routes.ts +554 -0
- package/src/routes/voice-profile-plugin-routes.d.ts +19 -0
- package/src/routes/voice-profile-plugin-routes.d.ts.map +1 -0
- package/src/routes/voice-profile-plugin-routes.ts +138 -0
- package/src/routes/voice-profiles-management-routes.d.ts +52 -0
- package/src/routes/voice-profiles-management-routes.d.ts.map +1 -0
- package/src/routes/voice-profiles-management-routes.ts +476 -0
- package/src/routes/voice-speaker-profile-routes.d.ts +57 -0
- package/src/routes/voice-speaker-profile-routes.d.ts.map +1 -0
- package/src/routes/voice-speaker-profile-routes.ts +199 -0
- package/src/runtime/aosp-llama-loader-selection.test.ts +80 -0
- package/src/runtime/capacitor-llama.d.ts +25 -0
- package/src/runtime/embedding-manager-support.d.ts +77 -0
- package/src/runtime/embedding-manager-support.d.ts.map +1 -0
- package/src/runtime/embedding-manager-support.ts +497 -0
- package/src/runtime/embedding-presets.d.ts +16 -0
- package/src/runtime/embedding-presets.d.ts.map +1 -0
- package/src/runtime/embedding-presets.ts +81 -0
- package/src/runtime/embedding-warmup-policy.d.ts +14 -0
- package/src/runtime/embedding-warmup-policy.d.ts.map +1 -0
- package/src/runtime/embedding-warmup-policy.test.ts +53 -0
- package/src/runtime/embedding-warmup-policy.ts +48 -0
- package/src/runtime/ensure-local-inference-handler.d.ts +62 -0
- package/src/runtime/ensure-local-inference-handler.d.ts.map +1 -0
- package/src/runtime/ensure-local-inference-handler.test.ts +528 -0
- package/src/runtime/ensure-local-inference-handler.ts +1448 -0
- package/src/runtime/index.d.ts +15 -0
- package/src/runtime/index.d.ts.map +1 -0
- package/src/runtime/index.ts +33 -0
- package/src/runtime/mobile-local-inference-gate.d.ts +31 -0
- package/src/runtime/mobile-local-inference-gate.d.ts.map +1 -0
- package/src/runtime/mobile-local-inference-gate.test.ts +69 -0
- package/src/runtime/mobile-local-inference-gate.ts +44 -0
- package/src/runtime/voice-entity-binding.d.ts +103 -0
- package/src/runtime/voice-entity-binding.d.ts.map +1 -0
- package/src/runtime/voice-entity-binding.transcript.test.ts +69 -0
- package/src/runtime/voice-entity-binding.ts +328 -0
- package/src/services/README.md +71 -0
- package/src/services/__tests__/backend-selector.test.ts +101 -0
- package/src/services/__tests__/checkpoint-manager.test.ts +376 -0
- package/src/services/__tests__/gpu-autotune.test.ts +400 -0
- package/src/services/__tests__/llm-streaming-binding.test.ts +85 -0
- package/src/services/__tests__/planner-grammar.test.ts +372 -0
- package/src/services/__tests__/runtime-target.test.ts +176 -0
- package/src/services/active-model-switch-rollback.test.ts +183 -0
- package/src/services/active-model.d.ts +282 -0
- package/src/services/active-model.d.ts.map +1 -0
- package/src/services/active-model.ts +1213 -0
- package/src/services/assignments.d.ts +71 -0
- package/src/services/assignments.d.ts.map +1 -0
- package/src/services/assignments.test.ts +80 -0
- package/src/services/assignments.ts +230 -0
- package/src/services/backend-selector.ts +95 -0
- package/src/services/backend.d.ts +346 -0
- package/src/services/backend.d.ts.map +1 -0
- package/src/services/backend.ts +612 -0
- package/src/services/bionic-host-loader.d.ts +46 -0
- package/src/services/bionic-host-loader.d.ts.map +1 -0
- package/src/services/bionic-host-loader.test.ts +133 -0
- package/src/services/bionic-host-loader.ts +180 -0
- package/src/services/bundled-models.d.ts +34 -0
- package/src/services/bundled-models.d.ts.map +1 -0
- package/src/services/bundled-models.ts +129 -0
- package/src/services/cache-bridge.d.ts +206 -0
- package/src/services/cache-bridge.d.ts.map +1 -0
- package/src/services/cache-bridge.test.ts +516 -0
- package/src/services/cache-bridge.ts +423 -0
- package/src/services/catalog.d.ts +10 -0
- package/src/services/catalog.d.ts.map +1 -0
- package/src/services/catalog.test.ts +238 -0
- package/src/services/catalog.ts +27 -0
- package/src/services/checkpoint-client.d.ts +109 -0
- package/src/services/checkpoint-client.d.ts.map +1 -0
- package/src/services/checkpoint-client.ts +258 -0
- package/src/services/checkpoint-manager.ts +474 -0
- package/src/services/cloud-fallback.d.ts +102 -0
- package/src/services/cloud-fallback.d.ts.map +1 -0
- package/src/services/cloud-fallback.ts +230 -0
- package/src/services/conversation-registry.d.ts +142 -0
- package/src/services/conversation-registry.d.ts.map +1 -0
- package/src/services/conversation-registry.test.ts +235 -0
- package/src/services/conversation-registry.ts +264 -0
- package/src/services/desktop-fused-ffi-backend-runtime.d.ts +95 -0
- package/src/services/desktop-fused-ffi-backend-runtime.d.ts.map +1 -0
- package/src/services/desktop-fused-ffi-backend-runtime.ts +339 -0
- package/src/services/device-bridge.d.ts +188 -0
- package/src/services/device-bridge.d.ts.map +1 -0
- package/src/services/device-bridge.ts +1237 -0
- package/src/services/device-resource-metrics.d.ts +149 -0
- package/src/services/device-resource-metrics.d.ts.map +1 -0
- package/src/services/device-resource-metrics.test.ts +98 -0
- package/src/services/device-resource-metrics.ts +346 -0
- package/src/services/device-tier.d.ts +115 -0
- package/src/services/device-tier.d.ts.map +1 -0
- package/src/services/device-tier.test.ts +371 -0
- package/src/services/device-tier.ts +410 -0
- package/src/services/downloader.d.ts +82 -0
- package/src/services/downloader.d.ts.map +1 -0
- package/src/services/downloader.test.ts +747 -0
- package/src/services/downloader.ts +925 -0
- package/src/services/engine-direct-bundle.test.ts +58 -0
- package/src/services/engine-streaming.test.ts +80 -0
- package/src/services/engine.d.ts +540 -0
- package/src/services/engine.d.ts.map +1 -0
- package/src/services/engine.ts +1909 -0
- package/src/services/ensure-local-artifacts.integration.test.ts +273 -0
- package/src/services/ensure-local-artifacts.test.ts +368 -0
- package/src/services/ensure-local-artifacts.ts +351 -0
- package/src/services/external-scanner.d.ts +17 -0
- package/src/services/external-scanner.d.ts.map +1 -0
- package/src/services/external-scanner.ts +312 -0
- package/src/services/ffi-llm-mock.ts +354 -0
- package/src/services/ffi-llm-streaming-abi.ts +442 -0
- package/src/services/ffi-streaming-backend.d.ts +180 -0
- package/src/services/ffi-streaming-backend.d.ts.map +1 -0
- package/src/services/ffi-streaming-backend.ts +382 -0
- package/src/services/ffi-streaming-runner.d.ts +122 -0
- package/src/services/ffi-streaming-runner.d.ts.map +1 -0
- package/src/services/ffi-streaming-runner.test.ts +60 -0
- package/src/services/ffi-streaming-runner.ts +354 -0
- package/src/services/ffi-unload-ordering.test.ts +162 -0
- package/src/services/gpu-autotune.ts +534 -0
- package/src/services/gpu-detect.d.ts +56 -0
- package/src/services/gpu-detect.d.ts.map +1 -0
- package/src/services/gpu-detect.ts +139 -0
- package/src/services/handler-registry.d.ts +72 -0
- package/src/services/handler-registry.d.ts.map +1 -0
- package/src/services/handler-registry.ts +240 -0
- package/src/services/hardware.d.ts +63 -0
- package/src/services/hardware.d.ts.map +1 -0
- package/src/services/hardware.test.ts +231 -0
- package/src/services/hardware.ts +410 -0
- package/src/services/hf-search.d.ts +26 -0
- package/src/services/hf-search.d.ts.map +1 -0
- package/src/services/hf-search.test.ts +69 -0
- package/src/services/hf-search.ts +420 -0
- package/src/services/image-description-runtime.d.ts +14 -0
- package/src/services/image-description-runtime.d.ts.map +1 -0
- package/src/services/image-description-runtime.test.ts +61 -0
- package/src/services/image-description-runtime.ts +118 -0
- package/src/services/imagegen/aosp-unavailable.d.ts +134 -0
- package/src/services/imagegen/aosp-unavailable.d.ts.map +1 -0
- package/src/services/imagegen/aosp-unavailable.ts +229 -0
- package/src/services/imagegen/backend-selector.d.ts +118 -0
- package/src/services/imagegen/backend-selector.d.ts.map +1 -0
- package/src/services/imagegen/backend-selector.ts +277 -0
- package/src/services/imagegen/coreml-unavailable.d.ts +105 -0
- package/src/services/imagegen/coreml-unavailable.d.ts.map +1 -0
- package/src/services/imagegen/coreml-unavailable.ts +237 -0
- package/src/services/imagegen/errors.d.ts +16 -0
- package/src/services/imagegen/errors.d.ts.map +1 -0
- package/src/services/imagegen/errors.ts +40 -0
- package/src/services/imagegen/index.d.ts +58 -0
- package/src/services/imagegen/index.d.ts.map +1 -0
- package/src/services/imagegen/index.ts +144 -0
- package/src/services/imagegen/mflux.d.ts +74 -0
- package/src/services/imagegen/mflux.d.ts.map +1 -0
- package/src/services/imagegen/mflux.ts +313 -0
- package/src/services/imagegen/sd-cpp.d.ts +180 -0
- package/src/services/imagegen/sd-cpp.d.ts.map +1 -0
- package/src/services/imagegen/sd-cpp.ts +718 -0
- package/src/services/imagegen/tensorrt-unavailable.d.ts +83 -0
- package/src/services/imagegen/tensorrt-unavailable.d.ts.map +1 -0
- package/src/services/imagegen/tensorrt-unavailable.ts +295 -0
- package/src/services/imagegen/types.d.ts +181 -0
- package/src/services/imagegen/types.d.ts.map +1 -0
- package/src/services/imagegen/types.ts +193 -0
- package/src/services/index.d.ts +29 -0
- package/src/services/index.d.ts.map +1 -0
- package/src/services/index.ts +211 -0
- package/src/services/inference-capabilities.d.ts +132 -0
- package/src/services/inference-capabilities.d.ts.map +1 -0
- package/src/services/inference-capabilities.test.ts +75 -0
- package/src/services/inference-capabilities.ts +204 -0
- package/src/services/inference-telemetry.d.ts +59 -0
- package/src/services/inference-telemetry.d.ts.map +1 -0
- package/src/services/inference-telemetry.ts +143 -0
- package/src/services/ios-llama-streaming.ts +248 -0
- package/src/services/kv-spill.d.ts +189 -0
- package/src/services/kv-spill.d.ts.map +1 -0
- package/src/services/kv-spill.test.ts +222 -0
- package/src/services/kv-spill.ts +356 -0
- package/src/services/latency-trace.d.ts +346 -0
- package/src/services/latency-trace.d.ts.map +1 -0
- package/src/services/latency-trace.test.ts +266 -0
- package/src/services/latency-trace.ts +844 -0
- package/src/services/llama-server-metrics.ts +304 -0
- package/src/services/llm-streaming-binding.d.ts +96 -0
- package/src/services/llm-streaming-binding.d.ts.map +1 -0
- package/src/services/llm-streaming-binding.ts +136 -0
- package/src/services/load-args.d.ts +82 -0
- package/src/services/load-args.d.ts.map +1 -0
- package/src/services/load-args.ts +81 -0
- package/src/services/manifest/eliza-1.manifest.v1.json +708 -0
- package/src/services/manifest/index.d.ts +4 -0
- package/src/services/manifest/index.d.ts.map +1 -0
- package/src/services/manifest/index.ts +66 -0
- package/src/services/manifest/manifest.test.ts +689 -0
- package/src/services/manifest/schema.d.ts +713 -0
- package/src/services/manifest/schema.d.ts.map +1 -0
- package/src/services/manifest/schema.ts +653 -0
- package/src/services/manifest/types.d.ts +30 -0
- package/src/services/manifest/types.d.ts.map +1 -0
- package/src/services/manifest/types.ts +55 -0
- package/src/services/manifest/validator.d.ts +66 -0
- package/src/services/manifest/validator.d.ts.map +1 -0
- package/src/services/manifest/validator.ts +567 -0
- package/src/services/memory-arbiter.d.ts +318 -0
- package/src/services/memory-arbiter.d.ts.map +1 -0
- package/src/services/memory-arbiter.test.ts +419 -0
- package/src/services/memory-arbiter.ts +925 -0
- package/src/services/memory-monitor.d.ts +122 -0
- package/src/services/memory-monitor.d.ts.map +1 -0
- package/src/services/memory-monitor.test.ts +208 -0
- package/src/services/memory-monitor.ts +297 -0
- package/src/services/memory-pressure.d.ts +130 -0
- package/src/services/memory-pressure.d.ts.map +1 -0
- package/src/services/memory-pressure.ts +414 -0
- package/src/services/mtp-doctor.d.ts +13 -0
- package/src/services/mtp-doctor.d.ts.map +1 -0
- package/src/services/mtp-doctor.ts +78 -0
- package/src/services/network-policy.d.ts +127 -0
- package/src/services/network-policy.d.ts.map +1 -0
- package/src/services/network-policy.ts +346 -0
- package/src/services/paths.d.ts +6 -0
- package/src/services/paths.d.ts.map +1 -0
- package/src/services/paths.ts +25 -0
- package/src/services/planner-skeleton.d.ts +124 -0
- package/src/services/planner-skeleton.d.ts.map +1 -0
- package/src/services/planner-skeleton.ts +175 -0
- package/src/services/providers.d.ts +38 -0
- package/src/services/providers.d.ts.map +1 -0
- package/src/services/providers.ts +507 -0
- package/src/services/ram-budget-cache.test.ts +163 -0
- package/src/services/ram-budget.d.ts +110 -0
- package/src/services/ram-budget.d.ts.map +1 -0
- package/src/services/ram-budget.ts +0 -0
- package/src/services/readiness.d.ts +9 -0
- package/src/services/readiness.d.ts.map +1 -0
- package/src/services/readiness.test.ts +87 -0
- package/src/services/readiness.ts +238 -0
- package/src/services/recommendation.d.ts +111 -0
- package/src/services/recommendation.d.ts.map +1 -0
- package/src/services/recommendation.ts +671 -0
- package/src/services/registry.d.ts +35 -0
- package/src/services/registry.d.ts.map +1 -0
- package/src/services/registry.ts +151 -0
- package/src/services/router-handler.d.ts +92 -0
- package/src/services/router-handler.d.ts.map +1 -0
- package/src/services/router-handler.test.ts +45 -0
- package/src/services/router-handler.ts +407 -0
- package/src/services/routing-policy.d.ts +69 -0
- package/src/services/routing-policy.d.ts.map +1 -0
- package/src/services/routing-policy.test.ts +164 -0
- package/src/services/routing-policy.ts +297 -0
- package/src/services/routing-preferences.d.ts +8 -0
- package/src/services/routing-preferences.d.ts.map +1 -0
- package/src/services/routing-preferences.ts +17 -0
- package/src/services/runtime-target.d.ts +98 -0
- package/src/services/runtime-target.d.ts.map +1 -0
- package/src/services/runtime-target.ts +154 -0
- package/src/services/service.d.ts +128 -0
- package/src/services/service.d.ts.map +1 -0
- package/src/services/service.test.ts +223 -0
- package/src/services/service.ts +735 -0
- package/src/services/session-pool.d.ts +72 -0
- package/src/services/session-pool.d.ts.map +1 -0
- package/src/services/session-pool.ts +153 -0
- package/src/services/structured-output/deterministic-repair.d.ts +23 -0
- package/src/services/structured-output/deterministic-repair.d.ts.map +1 -0
- package/src/services/structured-output/deterministic-repair.test.ts +169 -0
- package/src/services/structured-output/deterministic-repair.ts +443 -0
- package/src/services/structured-output/index.ts +4 -0
- package/src/services/structured-output.d.ts +311 -0
- package/src/services/structured-output.d.ts.map +1 -0
- package/src/services/structured-output.test.ts +483 -0
- package/src/services/structured-output.ts +712 -0
- package/src/services/system-memory.d.ts +33 -0
- package/src/services/system-memory.d.ts.map +1 -0
- package/src/services/system-memory.test.ts +47 -0
- package/src/services/system-memory.ts +67 -0
- package/src/services/transcription-priority.test.ts +211 -0
- package/src/services/types.d.ts +19 -0
- package/src/services/types.d.ts.map +1 -0
- package/src/services/types.ts +55 -0
- package/src/services/verify-on-device.d.ts +34 -0
- package/src/services/verify-on-device.d.ts.map +1 -0
- package/src/services/verify-on-device.test.ts +87 -0
- package/src/services/verify-on-device.ts +127 -0
- package/src/services/verify.d.ts +8 -0
- package/src/services/verify.d.ts.map +1 -0
- package/src/services/verify.ts +13 -0
- package/src/services/vision/aosp-unavailable.d.ts +115 -0
- package/src/services/vision/aosp-unavailable.d.ts.map +1 -0
- package/src/services/vision/aosp-unavailable.ts +163 -0
- package/src/services/vision/capacitor-llama.d.ts +99 -0
- package/src/services/vision/capacitor-llama.d.ts.map +1 -0
- package/src/services/vision/capacitor-llama.ts +255 -0
- package/src/services/vision/cloud-fallback.d.ts +47 -0
- package/src/services/vision/cloud-fallback.d.ts.map +1 -0
- package/src/services/vision/cloud-fallback.test.ts +243 -0
- package/src/services/vision/cloud-fallback.ts +268 -0
- package/src/services/vision/fallback-chain.test.ts +86 -0
- package/src/services/vision/hash.d.ts +71 -0
- package/src/services/vision/hash.d.ts.map +1 -0
- package/src/services/vision/hash.ts +157 -0
- package/src/services/vision/index.d.ts +95 -0
- package/src/services/vision/index.d.ts.map +1 -0
- package/src/services/vision/index.ts +251 -0
- package/src/services/vision/llama-server.d.ts +73 -0
- package/src/services/vision/llama-server.d.ts.map +1 -0
- package/src/services/vision/llama-server.ts +177 -0
- package/src/services/vision/types.d.ts +153 -0
- package/src/services/vision/types.d.ts.map +1 -0
- package/src/services/vision/types.ts +154 -0
- package/src/services/vision/vast-fallback.d.ts +18 -0
- package/src/services/vision/vast-fallback.d.ts.map +1 -0
- package/src/services/vision/vast-fallback.ts +127 -0
- package/src/services/vision-embedding-cache.d.ts +98 -0
- package/src/services/vision-embedding-cache.d.ts.map +1 -0
- package/src/services/vision-embedding-cache.ts +189 -0
- package/src/services/voice/VOICE_WORKBENCH.md +88 -0
- package/src/services/voice/__test-helpers__/fake-ffi.ts +94 -0
- package/src/services/voice/__test-helpers__/synthetic-speech.ts +124 -0
- package/src/services/voice/__tests__/checkpoint-manager.test.ts +241 -0
- package/src/services/voice/__tests__/checkpoint-policy.test.ts +270 -0
- package/src/services/voice/__tests__/eager-context-builder.test.ts +257 -0
- package/src/services/voice/__tests__/eliza1-eot-scorer.test.ts +288 -0
- package/src/services/voice/__tests__/eot-classifier.test.ts +431 -0
- package/src/services/voice/__tests__/optimistic-rollback.test.ts +312 -0
- package/src/services/voice/__tests__/prefill-client.test.ts +266 -0
- package/src/services/voice/__tests__/prefix-preserving-queue.test.ts +208 -0
- package/src/services/voice/__tests__/streaming-asr.test.ts +450 -0
- package/src/services/voice/__tests__/streaming-transcriber.test.ts +339 -0
- package/src/services/voice/__tests__/turn-detector-resolver.test.ts +195 -0
- package/src/services/voice/__tests__/voice-state-machine-prefill.test.ts +275 -0
- package/src/services/voice/__tests__/voice-state-machine.test.ts +354 -0
- package/src/services/voice/asr-timed.real.test.ts +141 -0
- package/src/services/voice/audio-frame-consumer.d.ts +212 -0
- package/src/services/voice/audio-frame-consumer.d.ts.map +1 -0
- package/src/services/voice/audio-frame-consumer.test.ts +343 -0
- package/src/services/voice/audio-frame-consumer.ts +491 -0
- package/src/services/voice/barge-in.d.ts +112 -0
- package/src/services/voice/barge-in.d.ts.map +1 -0
- package/src/services/voice/barge-in.test.ts +244 -0
- package/src/services/voice/barge-in.ts +336 -0
- package/src/services/voice/cancellation-coordinator.d.ts +127 -0
- package/src/services/voice/cancellation-coordinator.d.ts.map +1 -0
- package/src/services/voice/cancellation-coordinator.test.ts +196 -0
- package/src/services/voice/cancellation-coordinator.ts +269 -0
- package/src/services/voice/checkpoint-manager.d.ts +199 -0
- package/src/services/voice/checkpoint-manager.d.ts.map +1 -0
- package/src/services/voice/checkpoint-manager.ts +401 -0
- package/src/services/voice/checkpoint-policy.ts +336 -0
- package/src/services/voice/composite-eot-classifier.test.ts +59 -0
- package/src/services/voice/e2e-harness.test.ts +182 -0
- package/src/services/voice/e2e-harness.ts +743 -0
- package/src/services/voice/eager-context-builder.d.ts +170 -0
- package/src/services/voice/eager-context-builder.d.ts.map +1 -0
- package/src/services/voice/eager-context-builder.ts +262 -0
- package/src/services/voice/eliza1-eot-scorer.d.ts +124 -0
- package/src/services/voice/eliza1-eot-scorer.d.ts.map +1 -0
- package/src/services/voice/eliza1-eot-scorer.ts +242 -0
- package/src/services/voice/embedding-server.ts +200 -0
- package/src/services/voice/embedding.d.ts +133 -0
- package/src/services/voice/embedding.d.ts.map +1 -0
- package/src/services/voice/embedding.test.ts +131 -0
- package/src/services/voice/embedding.ts +243 -0
- package/src/services/voice/emotion-attribution.d.ts +68 -0
- package/src/services/voice/emotion-attribution.d.ts.map +1 -0
- package/src/services/voice/emotion-attribution.test.ts +129 -0
- package/src/services/voice/emotion-attribution.ts +361 -0
- package/src/services/voice/engine-bridge-cancellation.test.ts +422 -0
- package/src/services/voice/engine-bridge.d.ts +759 -0
- package/src/services/voice/engine-bridge.d.ts.map +1 -0
- package/src/services/voice/engine-bridge.test.ts +384 -0
- package/src/services/voice/engine-bridge.ts +2302 -0
- package/src/services/voice/eot-classifier-ggml.d.ts +179 -0
- package/src/services/voice/eot-classifier-ggml.d.ts.map +1 -0
- package/src/services/voice/eot-classifier-ggml.ts +566 -0
- package/src/services/voice/eot-classifier.d.ts +214 -0
- package/src/services/voice/eot-classifier.d.ts.map +1 -0
- package/src/services/voice/eot-classifier.ts +533 -0
- package/src/services/voice/errors.d.ts +20 -0
- package/src/services/voice/errors.d.ts.map +1 -0
- package/src/services/voice/errors.ts +32 -0
- package/src/services/voice/expressive-tags.d.ts +158 -0
- package/src/services/voice/expressive-tags.d.ts.map +1 -0
- package/src/services/voice/expressive-tags.ts +405 -0
- package/src/services/voice/ffi-bindings.d.ts +674 -0
- package/src/services/voice/ffi-bindings.d.ts.map +1 -0
- package/src/services/voice/ffi-bindings.test.ts +728 -0
- package/src/services/voice/ffi-bindings.ts +3225 -0
- package/src/services/voice/first-line-cache.d.ts +181 -0
- package/src/services/voice/first-line-cache.d.ts.map +1 -0
- package/src/services/voice/first-line-cache.ts +725 -0
- package/src/services/voice/fused-eot-scorer.d.ts +51 -0
- package/src/services/voice/fused-eot-scorer.d.ts.map +1 -0
- package/src/services/voice/fused-eot-scorer.ts +135 -0
- package/src/services/voice/index.d.ts +91 -0
- package/src/services/voice/index.d.ts.map +1 -0
- package/src/services/voice/index.ts +481 -0
- package/src/services/voice/kokoro/__tests__/kokoro-backend.test.ts +151 -0
- package/src/services/voice/kokoro/__tests__/kokoro-engine-bridge.real.test.ts +151 -0
- package/src/services/voice/kokoro/__tests__/kokoro-engine-bridge.test.ts +60 -0
- package/src/services/voice/kokoro/__tests__/kokoro-engine-discovery.test.ts +277 -0
- package/src/services/voice/kokoro/__tests__/kokoro-ffi-runtime.test.ts +235 -0
- package/src/services/voice/kokoro/__tests__/kokoro-runtime.test.ts +95 -0
- package/src/services/voice/kokoro/__tests__/phonemizer.test.ts +53 -0
- package/src/services/voice/kokoro/__tests__/runtime-selection.test.ts +231 -0
- package/src/services/voice/kokoro/__tests__/voices.test.ts +57 -0
- package/src/services/voice/kokoro/index.ts +79 -0
- package/src/services/voice/kokoro/kokoro-backend.d.ts +72 -0
- package/src/services/voice/kokoro/kokoro-backend.d.ts.map +1 -0
- package/src/services/voice/kokoro/kokoro-backend.ts +207 -0
- package/src/services/voice/kokoro/kokoro-engine-discovery.d.ts +58 -0
- package/src/services/voice/kokoro/kokoro-engine-discovery.d.ts.map +1 -0
- package/src/services/voice/kokoro/kokoro-engine-discovery.ts +177 -0
- package/src/services/voice/kokoro/kokoro-ffi-runtime.d.ts +75 -0
- package/src/services/voice/kokoro/kokoro-ffi-runtime.d.ts.map +1 -0
- package/src/services/voice/kokoro/kokoro-ffi-runtime.ts +233 -0
- package/src/services/voice/kokoro/kokoro-runtime.d.ts +100 -0
- package/src/services/voice/kokoro/kokoro-runtime.d.ts.map +1 -0
- package/src/services/voice/kokoro/kokoro-runtime.ts +170 -0
- package/src/services/voice/kokoro/phoneme-stream.ts +123 -0
- package/src/services/voice/kokoro/phonemizer.d.ts +50 -0
- package/src/services/voice/kokoro/phonemizer.d.ts.map +1 -0
- package/src/services/voice/kokoro/phonemizer.ts +344 -0
- package/src/services/voice/kokoro/pick-runtime.d.ts +61 -0
- package/src/services/voice/kokoro/pick-runtime.d.ts.map +1 -0
- package/src/services/voice/kokoro/pick-runtime.test.ts +91 -0
- package/src/services/voice/kokoro/pick-runtime.ts +130 -0
- package/src/services/voice/kokoro/runtime-selection.d.ts +92 -0
- package/src/services/voice/kokoro/runtime-selection.d.ts.map +1 -0
- package/src/services/voice/kokoro/runtime-selection.ts +237 -0
- package/src/services/voice/kokoro/types.d.ts +82 -0
- package/src/services/voice/kokoro/types.d.ts.map +1 -0
- package/src/services/voice/kokoro/types.ts +95 -0
- package/src/services/voice/kokoro/voice-presets.d.ts +23 -0
- package/src/services/voice/kokoro/voice-presets.d.ts.map +1 -0
- package/src/services/voice/kokoro/voice-presets.ts +129 -0
- package/src/services/voice/kokoro/voices.d.ts +30 -0
- package/src/services/voice/kokoro/voices.d.ts.map +1 -0
- package/src/services/voice/kokoro/voices.ts +64 -0
- package/src/services/voice/lifecycle.d.ts +135 -0
- package/src/services/voice/lifecycle.d.ts.map +1 -0
- package/src/services/voice/lifecycle.test.ts +315 -0
- package/src/services/voice/lifecycle.ts +301 -0
- package/src/services/voice/live-diarization-session.d.ts +96 -0
- package/src/services/voice/live-diarization-session.d.ts.map +1 -0
- package/src/services/voice/live-diarization-session.ts +289 -0
- package/src/services/voice/mic-source.d.ts +136 -0
- package/src/services/voice/mic-source.d.ts.map +1 -0
- package/src/services/voice/mic-source.test.ts +210 -0
- package/src/services/voice/mic-source.ts +503 -0
- package/src/services/voice/optimistic-policy.d.ts +109 -0
- package/src/services/voice/optimistic-policy.d.ts.map +1 -0
- package/src/services/voice/optimistic-policy.test.ts +101 -0
- package/src/services/voice/optimistic-policy.ts +192 -0
- package/src/services/voice/optimistic-rollback.ts +343 -0
- package/src/services/voice/partial-stabilizer.d.ts +73 -0
- package/src/services/voice/partial-stabilizer.d.ts.map +1 -0
- package/src/services/voice/partial-stabilizer.test.ts +68 -0
- package/src/services/voice/partial-stabilizer.ts +140 -0
- package/src/services/voice/phoneme-tokenizer.d.ts +49 -0
- package/src/services/voice/phoneme-tokenizer.d.ts.map +1 -0
- package/src/services/voice/phoneme-tokenizer.ts +158 -0
- package/src/services/voice/phrase-cache.d.ts +76 -0
- package/src/services/voice/phrase-cache.d.ts.map +1 -0
- package/src/services/voice/phrase-cache.test.ts +242 -0
- package/src/services/voice/phrase-cache.ts +186 -0
- package/src/services/voice/phrase-chunker.d.ts +62 -0
- package/src/services/voice/phrase-chunker.d.ts.map +1 -0
- package/src/services/voice/phrase-chunker.test.ts +239 -0
- package/src/services/voice/phrase-chunker.ts +281 -0
- package/src/services/voice/pipeline-impls.d.ts +151 -0
- package/src/services/voice/pipeline-impls.d.ts.map +1 -0
- package/src/services/voice/pipeline-impls.l6.test.ts +110 -0
- package/src/services/voice/pipeline-impls.test.ts +292 -0
- package/src/services/voice/pipeline-impls.ts +315 -0
- package/src/services/voice/pipeline.d.ts +216 -0
- package/src/services/voice/pipeline.d.ts.map +1 -0
- package/src/services/voice/pipeline.ts +505 -0
- package/src/services/voice/prefill-client.d.ts +123 -0
- package/src/services/voice/prefill-client.d.ts.map +1 -0
- package/src/services/voice/prefill-client.ts +316 -0
- package/src/services/voice/prefix-preserving-queue.d.ts +113 -0
- package/src/services/voice/prefix-preserving-queue.d.ts.map +1 -0
- package/src/services/voice/prefix-preserving-queue.ts +162 -0
- package/src/services/voice/profile-store.d.ts +248 -0
- package/src/services/voice/profile-store.d.ts.map +1 -0
- package/src/services/voice/profile-store.ts +887 -0
- package/src/services/voice/real-audio-decode.test.ts +148 -0
- package/src/services/voice/ring-buffer.d.ts +40 -0
- package/src/services/voice/ring-buffer.d.ts.map +1 -0
- package/src/services/voice/ring-buffer.test.ts +129 -0
- package/src/services/voice/ring-buffer.ts +123 -0
- package/src/services/voice/rollback-queue.d.ts +24 -0
- package/src/services/voice/rollback-queue.d.ts.map +1 -0
- package/src/services/voice/rollback-queue.ts +74 -0
- package/src/services/voice/samantha-preset-placeholder.d.ts +67 -0
- package/src/services/voice/samantha-preset-placeholder.d.ts.map +1 -0
- package/src/services/voice/samantha-preset-placeholder.test.ts +97 -0
- package/src/services/voice/samantha-preset-placeholder.ts +148 -0
- package/src/services/voice/samantha-preset-regenerator.d.ts +87 -0
- package/src/services/voice/samantha-preset-regenerator.d.ts.map +1 -0
- package/src/services/voice/samantha-preset-regenerator.ts +393 -0
- package/src/services/voice/scheduler.d.ts +146 -0
- package/src/services/voice/scheduler.d.ts.map +1 -0
- package/src/services/voice/scheduler.t2.test.ts +141 -0
- package/src/services/voice/scheduler.ts +927 -0
- package/src/services/voice/shared-resources.d.ts +190 -0
- package/src/services/voice/shared-resources.d.ts.map +1 -0
- package/src/services/voice/shared-resources.ts +320 -0
- package/src/services/voice/speaker/attribution-pipeline.d.ts +74 -0
- package/src/services/voice/speaker/attribution-pipeline.d.ts.map +1 -0
- package/src/services/voice/speaker/attribution-pipeline.ts +386 -0
- package/src/services/voice/speaker/diarizer-fused.d.ts +59 -0
- package/src/services/voice/speaker/diarizer-fused.d.ts.map +1 -0
- package/src/services/voice/speaker/diarizer-fused.real.test.ts +100 -0
- package/src/services/voice/speaker/diarizer-fused.ts +154 -0
- package/src/services/voice/speaker/diarizer.d.ts +75 -0
- package/src/services/voice/speaker/diarizer.d.ts.map +1 -0
- package/src/services/voice/speaker/diarizer.ts +218 -0
- package/src/services/voice/speaker/encoder-fused.d.ts +60 -0
- package/src/services/voice/speaker/encoder-fused.d.ts.map +1 -0
- package/src/services/voice/speaker/encoder-fused.real.test.ts +113 -0
- package/src/services/voice/speaker/encoder-fused.ts +138 -0
- package/src/services/voice/speaker/encoder-ggml.d.ts +33 -0
- package/src/services/voice/speaker/encoder-ggml.d.ts.map +1 -0
- package/src/services/voice/speaker/encoder-ggml.ts +79 -0
- package/src/services/voice/speaker/encoder.d.ts +37 -0
- package/src/services/voice/speaker/encoder.d.ts.map +1 -0
- package/src/services/voice/speaker/encoder.ts +105 -0
- package/src/services/voice/speaker-imprint.d.ts +83 -0
- package/src/services/voice/speaker-imprint.d.ts.map +1 -0
- package/src/services/voice/speaker-imprint.test.ts +185 -0
- package/src/services/voice/speaker-imprint.ts +312 -0
- package/src/services/voice/speaker-preset-cache.d.ts +77 -0
- package/src/services/voice/speaker-preset-cache.d.ts.map +1 -0
- package/src/services/voice/speaker-preset-cache.test.ts +154 -0
- package/src/services/voice/speaker-preset-cache.ts +195 -0
- package/src/services/voice/streaming-asr/streaming-pipeline-adapter.ts +292 -0
- package/src/services/voice/system-audio-sink.d.ts +73 -0
- package/src/services/voice/system-audio-sink.d.ts.map +1 -0
- package/src/services/voice/system-audio-sink.test.ts +29 -0
- package/src/services/voice/system-audio-sink.ts +366 -0
- package/src/services/voice/transcriber.d.ts +244 -0
- package/src/services/voice/transcriber.d.ts.map +1 -0
- package/src/services/voice/transcriber.test.ts +392 -0
- package/src/services/voice/transcriber.ts +704 -0
- package/src/services/voice/transcript-knowledge.d.ts +37 -0
- package/src/services/voice/transcript-knowledge.d.ts.map +1 -0
- package/src/services/voice/transcript-knowledge.test.ts +68 -0
- package/src/services/voice/transcript-knowledge.ts +75 -0
- package/src/services/voice/transcript-service.d.ts +41 -0
- package/src/services/voice/transcript-service.d.ts.map +1 -0
- package/src/services/voice/transcript-service.test.ts +137 -0
- package/src/services/voice/transcript-service.ts +141 -0
- package/src/services/voice/transcript-store.d.ts +53 -0
- package/src/services/voice/transcript-store.d.ts.map +1 -0
- package/src/services/voice/transcript-store.test.ts +153 -0
- package/src/services/voice/transcript-store.ts +132 -0
- package/src/services/voice/turn-controller.d.ts +183 -0
- package/src/services/voice/turn-controller.d.ts.map +1 -0
- package/src/services/voice/turn-controller.test.ts +575 -0
- package/src/services/voice/turn-controller.ts +596 -0
- package/src/services/voice/types.d.ts +643 -0
- package/src/services/voice/types.d.ts.map +1 -0
- package/src/services/voice/types.ts +699 -0
- package/src/services/voice/vad.d.ts +282 -0
- package/src/services/voice/vad.d.ts.map +1 -0
- package/src/services/voice/vad.test.ts +480 -0
- package/src/services/voice/vad.ts +827 -0
- package/src/services/voice/vad.v1-v4.test.ts +222 -0
- package/src/services/voice/voice-budget.d.ts +241 -0
- package/src/services/voice/voice-budget.d.ts.map +1 -0
- package/src/services/voice/voice-budget.test.ts +418 -0
- package/src/services/voice/voice-budget.ts +635 -0
- package/src/services/voice/voice-duet.test.ts +375 -0
- package/src/services/voice/voice-emotion-classifier.d.ts +95 -0
- package/src/services/voice/voice-emotion-classifier.d.ts.map +1 -0
- package/src/services/voice/voice-emotion-classifier.test.ts +210 -0
- package/src/services/voice/voice-emotion-classifier.ts +273 -0
- package/src/services/voice/voice-preset-format.d.ts +158 -0
- package/src/services/voice/voice-preset-format.d.ts.map +1 -0
- package/src/services/voice/voice-preset-format.ts +700 -0
- package/src/services/voice/voice-preset-generator.test.ts +89 -0
- package/src/services/voice/voice-profile-artifact.d.ts +116 -0
- package/src/services/voice/voice-profile-artifact.d.ts.map +1 -0
- package/src/services/voice/voice-profile-artifact.test.ts +138 -0
- package/src/services/voice/voice-profile-artifact.ts +518 -0
- package/src/services/voice/voice-profile-routes.d.ts +83 -0
- package/src/services/voice/voice-profile-routes.d.ts.map +1 -0
- package/src/services/voice/voice-profile-routes.test.ts +429 -0
- package/src/services/voice/voice-profile-routes.ts +425 -0
- package/src/services/voice/voice-scenario.ts +154 -0
- package/src/services/voice/voice-settings.d.ts +82 -0
- package/src/services/voice/voice-settings.d.ts.map +1 -0
- package/src/services/voice/voice-settings.ts +172 -0
- package/src/services/voice/voice-state-machine.d.ts +364 -0
- package/src/services/voice/voice-state-machine.d.ts.map +1 -0
- package/src/services/voice/voice-state-machine.ts +727 -0
- package/src/services/voice/voice-workbench-report.test.ts +168 -0
- package/src/services/voice/voice-workbench-report.ts +326 -0
- package/src/services/voice/voice-workbench.test.ts +158 -0
- package/src/services/voice/voice.test.ts +1070 -0
- package/src/services/voice/wake-word-ggml.d.ts +101 -0
- package/src/services/voice/wake-word-ggml.d.ts.map +1 -0
- package/src/services/voice/wake-word-ggml.ts +320 -0
- package/src/services/voice/wake-word.d.ts +255 -0
- package/src/services/voice/wake-word.d.ts.map +1 -0
- package/src/services/voice/wake-word.test.ts +298 -0
- package/src/services/voice/wake-word.ts +554 -0
- package/src/services/voice/wrap-with-first-line-cache.d.ts +70 -0
- package/src/services/voice/wrap-with-first-line-cache.d.ts.map +1 -0
- package/src/services/voice/wrap-with-first-line-cache.ts +267 -0
- package/src/services/voice-model-updater.d.ts +240 -0
- package/src/services/voice-model-updater.d.ts.map +1 -0
- package/src/services/voice-model-updater.ts +724 -0
- package/src/services/voice-prewarm.d.ts +3 -0
- package/src/services/voice-prewarm.d.ts.map +1 -0
- package/src/services/voice-prewarm.ts +51 -0
- package/dist/index.d.ts +0 -37
- package/dist/index.js +0 -1098
|
@@ -0,0 +1,69 @@
|
|
|
1
|
+
import { afterEach, describe, expect, it, vi } from "vitest";
|
|
2
|
+
import { searchHuggingFaceGguf } from "./hf-search";
|
|
3
|
+
|
|
4
|
+
function jsonResponse(body: unknown): Response {
|
|
5
|
+
return new Response(JSON.stringify(body), {
|
|
6
|
+
status: 200,
|
|
7
|
+
headers: { "content-type": "application/json" },
|
|
8
|
+
});
|
|
9
|
+
}
|
|
10
|
+
|
|
11
|
+
describe("searchHuggingFaceGguf", () => {
|
|
12
|
+
afterEach(() => {
|
|
13
|
+
vi.unstubAllGlobals();
|
|
14
|
+
});
|
|
15
|
+
|
|
16
|
+
it("classifies Qwen3.5 decimal tiers without legacy Qwen3 aliases", async () => {
|
|
17
|
+
const details = new Map<string, unknown>([
|
|
18
|
+
[
|
|
19
|
+
"Qwen/Qwen3.5-0.8B-GGUF",
|
|
20
|
+
{
|
|
21
|
+
id: "Qwen/Qwen3.5-0.8B-GGUF",
|
|
22
|
+
tags: ["gguf", "text-generation"],
|
|
23
|
+
siblings: [{ rfilename: "qwen3.5-0.8b-q4_k_m.gguf", size: 512 }],
|
|
24
|
+
pipeline_tag: "text-generation",
|
|
25
|
+
},
|
|
26
|
+
],
|
|
27
|
+
[
|
|
28
|
+
"Qwen/Qwen3.5-2B-GGUF",
|
|
29
|
+
{
|
|
30
|
+
id: "Qwen/Qwen3.5-2B-GGUF",
|
|
31
|
+
tags: ["gguf", "text-generation"],
|
|
32
|
+
siblings: [{ rfilename: "qwen3.5-2b-q4_k_m.gguf", size: 512 }],
|
|
33
|
+
pipeline_tag: "text-generation",
|
|
34
|
+
},
|
|
35
|
+
],
|
|
36
|
+
[
|
|
37
|
+
"org/tiny-0.8b-GGUF",
|
|
38
|
+
{
|
|
39
|
+
id: "org/tiny-0.8b-GGUF",
|
|
40
|
+
tags: ["gguf"],
|
|
41
|
+
siblings: [{ rfilename: "tiny-0.8b-q4_k_m.gguf", size: 512 }],
|
|
42
|
+
pipeline_tag: "text-generation",
|
|
43
|
+
},
|
|
44
|
+
],
|
|
45
|
+
]);
|
|
46
|
+
const fetchMock = vi.fn(async (input: RequestInfo | URL) => {
|
|
47
|
+
const url = String(input);
|
|
48
|
+
if (url.startsWith("https://huggingface.co/api/models?")) {
|
|
49
|
+
return jsonResponse([...details.keys()].map((id) => ({ id })));
|
|
50
|
+
}
|
|
51
|
+
const encodedId = url.replace("https://huggingface.co/api/models/", "");
|
|
52
|
+
const detail = details.get(decodeURIComponent(encodedId));
|
|
53
|
+
if (detail) return jsonResponse(detail);
|
|
54
|
+
return new Response("not found", { status: 404 });
|
|
55
|
+
});
|
|
56
|
+
vi.stubGlobal("fetch", fetchMock);
|
|
57
|
+
|
|
58
|
+
const results = await searchHuggingFaceGguf("qwen", 4);
|
|
59
|
+
|
|
60
|
+
expect(results.map((result) => [result.params, result.bucket])).toEqual([
|
|
61
|
+
["0.8B", "small"],
|
|
62
|
+
["2B", "small"],
|
|
63
|
+
["0.8B", "small"],
|
|
64
|
+
]);
|
|
65
|
+
expect(
|
|
66
|
+
results.map((result) => result.parameterLabel ?? result.params),
|
|
67
|
+
).toEqual(["0.8B", "2B", "0.8B"]);
|
|
68
|
+
});
|
|
69
|
+
});
|
|
@@ -0,0 +1,420 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* HuggingFace Hub search for GGUF models.
|
|
3
|
+
*
|
|
4
|
+
* Calls `https://huggingface.co/api/models` with `filter=gguf` to narrow
|
|
5
|
+
* results to repos that actually ship GGUF quantisations. Each matching
|
|
6
|
+
* repo is expanded with `/api/models/<repo>` to pick a representative
|
|
7
|
+
* quant file (preferring Q4_K_M when present). Results are shaped like
|
|
8
|
+
* `CatalogModel` so the existing ModelCard renders them directly.
|
|
9
|
+
*
|
|
10
|
+
* We deliberately do not persist these — they're dynamic, and a curated
|
|
11
|
+
* entry with the same hfRepo always takes precedence in the UI.
|
|
12
|
+
*/
|
|
13
|
+
|
|
14
|
+
import type { CatalogModel, ModelBucket } from "./types";
|
|
15
|
+
|
|
16
|
+
const HF_API = "https://huggingface.co/api";
|
|
17
|
+
const SEARCH_TIMEOUT_MS = 10_000;
|
|
18
|
+
|
|
19
|
+
interface HfSearchResultRaw {
|
|
20
|
+
id?: string;
|
|
21
|
+
modelId?: string;
|
|
22
|
+
author?: string;
|
|
23
|
+
likes?: number;
|
|
24
|
+
downloads?: number;
|
|
25
|
+
tags?: string[];
|
|
26
|
+
pipeline_tag?: string;
|
|
27
|
+
}
|
|
28
|
+
|
|
29
|
+
interface HfSiblingRaw {
|
|
30
|
+
rfilename?: string;
|
|
31
|
+
size?: number;
|
|
32
|
+
}
|
|
33
|
+
|
|
34
|
+
interface HfModelDetailRaw {
|
|
35
|
+
id?: string;
|
|
36
|
+
author?: string;
|
|
37
|
+
tags?: string[];
|
|
38
|
+
siblings?: HfSiblingRaw[];
|
|
39
|
+
pipeline_tag?: string;
|
|
40
|
+
likes?: number;
|
|
41
|
+
downloads?: number;
|
|
42
|
+
}
|
|
43
|
+
|
|
44
|
+
interface ModelScopeEnvelope<T> {
|
|
45
|
+
Code?: number;
|
|
46
|
+
Message?: string;
|
|
47
|
+
Data?: T;
|
|
48
|
+
}
|
|
49
|
+
|
|
50
|
+
interface ModelScopeModelRaw {
|
|
51
|
+
ModelId?: string;
|
|
52
|
+
model_id?: string;
|
|
53
|
+
Name?: string;
|
|
54
|
+
Path?: string;
|
|
55
|
+
Downloads?: number;
|
|
56
|
+
Tags?: string[];
|
|
57
|
+
Tasks?: string[];
|
|
58
|
+
Frameworks?: string[];
|
|
59
|
+
}
|
|
60
|
+
|
|
61
|
+
interface ModelScopeListRaw {
|
|
62
|
+
Models?: ModelScopeModelRaw[];
|
|
63
|
+
}
|
|
64
|
+
|
|
65
|
+
interface ModelScopeFileRaw {
|
|
66
|
+
Name?: string;
|
|
67
|
+
Path?: string;
|
|
68
|
+
Size?: number;
|
|
69
|
+
}
|
|
70
|
+
|
|
71
|
+
interface ModelScopeFilesRaw {
|
|
72
|
+
Files?: ModelScopeFileRaw[];
|
|
73
|
+
}
|
|
74
|
+
|
|
75
|
+
const QUANT_PREFERENCE = [
|
|
76
|
+
"Q4_K_M",
|
|
77
|
+
"Q5_K_M",
|
|
78
|
+
"Q4_0",
|
|
79
|
+
"Q5_0",
|
|
80
|
+
"Q3_K_M",
|
|
81
|
+
"Q8_0",
|
|
82
|
+
"Q2_K",
|
|
83
|
+
];
|
|
84
|
+
|
|
85
|
+
function pickQuantFile(siblings: HfSiblingRaw[]): HfSiblingRaw | null {
|
|
86
|
+
const ggufs = siblings.filter((s) =>
|
|
87
|
+
s.rfilename?.toLowerCase().endsWith(".gguf"),
|
|
88
|
+
);
|
|
89
|
+
if (ggufs.length === 0) return null;
|
|
90
|
+
|
|
91
|
+
for (const quant of QUANT_PREFERENCE) {
|
|
92
|
+
const match = ggufs.find((s) => s.rfilename?.toUpperCase().includes(quant));
|
|
93
|
+
if (match) return match;
|
|
94
|
+
}
|
|
95
|
+
// No preferred quant — pick the smallest GGUF so we don't surprise the
|
|
96
|
+
// user with a 40 GB file.
|
|
97
|
+
return [...ggufs].sort((a, b) => (a.size ?? 0) - (b.size ?? 0))[0] ?? null;
|
|
98
|
+
}
|
|
99
|
+
|
|
100
|
+
function extractQuantLabel(filename: string): string {
|
|
101
|
+
for (const quant of QUANT_PREFERENCE) {
|
|
102
|
+
if (filename.toUpperCase().includes(quant)) return quant;
|
|
103
|
+
}
|
|
104
|
+
return "GGUF";
|
|
105
|
+
}
|
|
106
|
+
|
|
107
|
+
/**
|
|
108
|
+
* Very rough parameter-count inference from model name / tags. We use this
|
|
109
|
+
* only to pick a bucket label — not for any hard memory check.
|
|
110
|
+
*/
|
|
111
|
+
function inferParams(
|
|
112
|
+
name: string,
|
|
113
|
+
tags: string[],
|
|
114
|
+
): {
|
|
115
|
+
params: CatalogModel["params"];
|
|
116
|
+
bucket: ModelBucket;
|
|
117
|
+
parameterLabel?: string;
|
|
118
|
+
} {
|
|
119
|
+
const lower = `${name} ${tags.join(" ")}`.toLowerCase();
|
|
120
|
+
const sizes: Array<[RegExp, CatalogModel["params"], ModelBucket, string?]> = [
|
|
121
|
+
[/\b32b\b/, "32B", "xl"],
|
|
122
|
+
[/\b27b\b/, "27B", "large"],
|
|
123
|
+
[/\b24b\b/, "24B", "large"],
|
|
124
|
+
[/\b22b\b/, "22B", "large"],
|
|
125
|
+
[/\b16b\b/, "16B", "large"],
|
|
126
|
+
[/\b14b\b/, "14B", "large"],
|
|
127
|
+
[/\b13b\b/, "14B", "large"],
|
|
128
|
+
[/\b9b\b/, "9B", "mid"],
|
|
129
|
+
[/(?:^|[^0-9])0\.8b\b/, "0.8B", "small"],
|
|
130
|
+
[/\b8b\b/, "8B", "mid"],
|
|
131
|
+
[/\b7b\b/, "7B", "mid"],
|
|
132
|
+
[/\b3b\b/, "3B", "small"],
|
|
133
|
+
[/\b2b\b/, "2B", "small"],
|
|
134
|
+
[/\b1b\b/, "1B", "small"],
|
|
135
|
+
];
|
|
136
|
+
for (const [re, params, bucket, parameterLabel] of sizes) {
|
|
137
|
+
if (re.test(lower)) {
|
|
138
|
+
return {
|
|
139
|
+
params,
|
|
140
|
+
bucket,
|
|
141
|
+
...(parameterLabel ? { parameterLabel } : {}),
|
|
142
|
+
};
|
|
143
|
+
}
|
|
144
|
+
}
|
|
145
|
+
return { params: "7B", bucket: "mid" };
|
|
146
|
+
}
|
|
147
|
+
|
|
148
|
+
function inferCategory(
|
|
149
|
+
tags: string[],
|
|
150
|
+
pipelineTag: string | undefined,
|
|
151
|
+
): CatalogModel["category"] {
|
|
152
|
+
const lowerTags = tags.map((t) => t.toLowerCase());
|
|
153
|
+
if (lowerTags.some((t) => t.includes("code") || t.includes("coder"))) {
|
|
154
|
+
return "code";
|
|
155
|
+
}
|
|
156
|
+
if (
|
|
157
|
+
lowerTags.some(
|
|
158
|
+
(t) => t.includes("reasoning") || t === "math" || t.includes("r1"),
|
|
159
|
+
)
|
|
160
|
+
) {
|
|
161
|
+
return "reasoning";
|
|
162
|
+
}
|
|
163
|
+
if (
|
|
164
|
+
lowerTags.some(
|
|
165
|
+
(t) =>
|
|
166
|
+
t.includes("function") || t.includes("tool") || t.includes("hermes"),
|
|
167
|
+
)
|
|
168
|
+
) {
|
|
169
|
+
return "tools";
|
|
170
|
+
}
|
|
171
|
+
if (pipelineTag === "text-generation") {
|
|
172
|
+
return "chat";
|
|
173
|
+
}
|
|
174
|
+
return "chat";
|
|
175
|
+
}
|
|
176
|
+
|
|
177
|
+
async function fetchWithTimeout(
|
|
178
|
+
url: string,
|
|
179
|
+
init?: RequestInit,
|
|
180
|
+
): Promise<Response> {
|
|
181
|
+
const controller = new AbortController();
|
|
182
|
+
const timer = setTimeout(() => controller.abort(), SEARCH_TIMEOUT_MS);
|
|
183
|
+
try {
|
|
184
|
+
return await fetch(url, { ...init, signal: controller.signal });
|
|
185
|
+
} finally {
|
|
186
|
+
clearTimeout(timer);
|
|
187
|
+
}
|
|
188
|
+
}
|
|
189
|
+
|
|
190
|
+
/**
|
|
191
|
+
* Search HuggingFace for GGUF repos matching `query`, returning
|
|
192
|
+
* catalog-shaped entries ready for the Model Hub UI.
|
|
193
|
+
*/
|
|
194
|
+
export async function searchHuggingFaceGguf(
|
|
195
|
+
query: string,
|
|
196
|
+
limit = 12,
|
|
197
|
+
): Promise<CatalogModel[]> {
|
|
198
|
+
const trimmed = query.trim();
|
|
199
|
+
if (trimmed.length === 0) return [];
|
|
200
|
+
|
|
201
|
+
const searchUrl = new URL(`${HF_API}/models`);
|
|
202
|
+
searchUrl.searchParams.set("search", trimmed);
|
|
203
|
+
searchUrl.searchParams.set("filter", "gguf");
|
|
204
|
+
searchUrl.searchParams.set(
|
|
205
|
+
"limit",
|
|
206
|
+
String(Math.min(50, Math.max(1, limit * 2))),
|
|
207
|
+
);
|
|
208
|
+
searchUrl.searchParams.set("sort", "downloads");
|
|
209
|
+
searchUrl.searchParams.set("direction", "-1");
|
|
210
|
+
|
|
211
|
+
const searchRes = await fetchWithTimeout(searchUrl.toString(), {
|
|
212
|
+
headers: { accept: "application/json" },
|
|
213
|
+
});
|
|
214
|
+
if (!searchRes.ok) {
|
|
215
|
+
throw new Error(`HuggingFace search failed: HTTP ${searchRes.status}`);
|
|
216
|
+
}
|
|
217
|
+
const rawList = (await searchRes.json()) as HfSearchResultRaw[];
|
|
218
|
+
const candidates = rawList
|
|
219
|
+
.map((r) => r.id ?? r.modelId)
|
|
220
|
+
.filter((id): id is string => typeof id === "string" && id.length > 0)
|
|
221
|
+
.slice(0, limit);
|
|
222
|
+
|
|
223
|
+
// Parallel detail fetches; failures drop quietly so one bad repo doesn't
|
|
224
|
+
// take down the whole search.
|
|
225
|
+
const details = await Promise.all(
|
|
226
|
+
candidates.map(async (id) => {
|
|
227
|
+
try {
|
|
228
|
+
const res = await fetchWithTimeout(
|
|
229
|
+
`${HF_API}/models/${encodeURIComponent(id)}`,
|
|
230
|
+
{ headers: { accept: "application/json" } },
|
|
231
|
+
);
|
|
232
|
+
if (!res.ok) return null;
|
|
233
|
+
return (await res.json()) as HfModelDetailRaw;
|
|
234
|
+
} catch {
|
|
235
|
+
return null;
|
|
236
|
+
}
|
|
237
|
+
}),
|
|
238
|
+
);
|
|
239
|
+
|
|
240
|
+
const results: CatalogModel[] = [];
|
|
241
|
+
for (let i = 0; i < candidates.length; i++) {
|
|
242
|
+
const id = candidates[i];
|
|
243
|
+
const detail = details[i];
|
|
244
|
+
if (!id || !detail?.siblings) continue;
|
|
245
|
+
const sibling = pickQuantFile(detail.siblings);
|
|
246
|
+
if (!sibling?.rfilename) continue;
|
|
247
|
+
|
|
248
|
+
const sizeBytes = sibling.size ?? 0;
|
|
249
|
+
const sizeGb = sizeBytes > 0 ? sizeBytes / 1024 ** 3 : 4;
|
|
250
|
+
const { params, bucket, parameterLabel } = inferParams(
|
|
251
|
+
id,
|
|
252
|
+
detail.tags ?? [],
|
|
253
|
+
);
|
|
254
|
+
const quant = extractQuantLabel(sibling.rfilename);
|
|
255
|
+
const category = inferCategory(detail.tags ?? [], detail.pipeline_tag);
|
|
256
|
+
const displayName = id.split("/").pop() ?? id;
|
|
257
|
+
|
|
258
|
+
// minRam heuristic: 2x the file size, minimum 4 GB. Close enough for
|
|
259
|
+
// the fit badge; the ModelCard's fit computation still does the work.
|
|
260
|
+
const minRamGb = Math.max(4, Math.round(sizeGb * 2));
|
|
261
|
+
|
|
262
|
+
results.push({
|
|
263
|
+
id: `hf:${id}::${sibling.rfilename}`,
|
|
264
|
+
displayName,
|
|
265
|
+
hfRepo: id,
|
|
266
|
+
ggufFile: sibling.rfilename,
|
|
267
|
+
params,
|
|
268
|
+
...(parameterLabel ? { parameterLabel } : {}),
|
|
269
|
+
quant,
|
|
270
|
+
sizeGb: Math.round(sizeGb * 10) / 10,
|
|
271
|
+
minRamGb,
|
|
272
|
+
category,
|
|
273
|
+
bucket,
|
|
274
|
+
blurb:
|
|
275
|
+
(detail.tags ?? []).slice(0, 4).join(" · ") ||
|
|
276
|
+
`${detail.downloads ?? 0} downloads · ${detail.likes ?? 0} likes`,
|
|
277
|
+
});
|
|
278
|
+
}
|
|
279
|
+
return results;
|
|
280
|
+
}
|
|
281
|
+
|
|
282
|
+
function normalizeModelScopeRepoId(query: string): string | null {
|
|
283
|
+
const trimmed = query.trim().replace(/^modelscope:\/\//i, "");
|
|
284
|
+
const match = /^[A-Za-z0-9_.-]+\/[A-Za-z0-9_.-]+$/.exec(trimmed);
|
|
285
|
+
return match ? trimmed : null;
|
|
286
|
+
}
|
|
287
|
+
|
|
288
|
+
function modelScopeOwnerQuery(query: string): string | null {
|
|
289
|
+
const trimmed = query.trim().replace(/^modelscope:\/\//i, "");
|
|
290
|
+
const first = trimmed.split(/\s+/)[0] ?? "";
|
|
291
|
+
if (/^[A-Za-z0-9_.-]+$/.test(first)) return first;
|
|
292
|
+
return null;
|
|
293
|
+
}
|
|
294
|
+
|
|
295
|
+
function pickModelScopeGguf(files: ModelScopeFileRaw[]): HfSiblingRaw | null {
|
|
296
|
+
return pickQuantFile(
|
|
297
|
+
files.map((file) => ({
|
|
298
|
+
rfilename: file.Path ?? file.Name,
|
|
299
|
+
size: file.Size,
|
|
300
|
+
})),
|
|
301
|
+
);
|
|
302
|
+
}
|
|
303
|
+
|
|
304
|
+
async function fetchModelScopeJson<T>(
|
|
305
|
+
url: string,
|
|
306
|
+
init?: RequestInit,
|
|
307
|
+
): Promise<T> {
|
|
308
|
+
const headers = new Headers(init?.headers);
|
|
309
|
+
headers.set("accept", "application/json");
|
|
310
|
+
const res = await fetchWithTimeout(url, {
|
|
311
|
+
...init,
|
|
312
|
+
headers,
|
|
313
|
+
redirect: "follow",
|
|
314
|
+
});
|
|
315
|
+
if (!res.ok) {
|
|
316
|
+
throw new Error(`ModelScope request failed: HTTP ${res.status}`);
|
|
317
|
+
}
|
|
318
|
+
const envelope = (await res.json()) as ModelScopeEnvelope<T>;
|
|
319
|
+
if (envelope.Code !== 200 || envelope.Data === undefined) {
|
|
320
|
+
throw new Error(envelope.Message ?? "ModelScope request failed");
|
|
321
|
+
}
|
|
322
|
+
return envelope.Data;
|
|
323
|
+
}
|
|
324
|
+
|
|
325
|
+
async function listModelScopeOwnerModels(
|
|
326
|
+
owner: string,
|
|
327
|
+
limit: number,
|
|
328
|
+
): Promise<string[]> {
|
|
329
|
+
const data = await fetchModelScopeJson<ModelScopeListRaw>(
|
|
330
|
+
"https://www.modelscope.cn/api/v1/models",
|
|
331
|
+
{
|
|
332
|
+
method: "PUT",
|
|
333
|
+
headers: { "content-type": "application/json" },
|
|
334
|
+
body: JSON.stringify({
|
|
335
|
+
Path: owner,
|
|
336
|
+
PageNumber: 1,
|
|
337
|
+
PageSize: Math.min(50, Math.max(1, limit * 3)),
|
|
338
|
+
}),
|
|
339
|
+
},
|
|
340
|
+
);
|
|
341
|
+
return (data.Models ?? [])
|
|
342
|
+
.map((m) => m.ModelId ?? m.model_id)
|
|
343
|
+
.filter((id): id is string => typeof id === "string" && id.length > 0);
|
|
344
|
+
}
|
|
345
|
+
|
|
346
|
+
async function modelScopeResultForRepo(
|
|
347
|
+
repoId: string,
|
|
348
|
+
): Promise<CatalogModel | null> {
|
|
349
|
+
const filesData = await fetchModelScopeJson<ModelScopeFilesRaw>(
|
|
350
|
+
`https://www.modelscope.cn/api/v1/models/${repoId}/repo/files?Revision=master&Recursive=true`,
|
|
351
|
+
);
|
|
352
|
+
const sibling = pickModelScopeGguf(filesData.Files ?? []);
|
|
353
|
+
if (!sibling?.rfilename) return null;
|
|
354
|
+
|
|
355
|
+
const sizeBytes = sibling.size ?? 0;
|
|
356
|
+
const sizeGb = sizeBytes > 0 ? sizeBytes / 1024 ** 3 : 4;
|
|
357
|
+
const { params, bucket } = inferParams(repoId, []);
|
|
358
|
+
const quant = extractQuantLabel(sibling.rfilename);
|
|
359
|
+
const displayName = repoId.split("/").pop() ?? repoId;
|
|
360
|
+
|
|
361
|
+
return {
|
|
362
|
+
id: `modelscope:${repoId}::${sibling.rfilename}`,
|
|
363
|
+
displayName,
|
|
364
|
+
hub: "modelscope",
|
|
365
|
+
hfRepo: repoId,
|
|
366
|
+
ggufFile: sibling.rfilename,
|
|
367
|
+
params,
|
|
368
|
+
quant,
|
|
369
|
+
sizeGb: Math.round(sizeGb * 10) / 10,
|
|
370
|
+
minRamGb: Math.max(4, Math.round(sizeGb * 2)),
|
|
371
|
+
category: "chat",
|
|
372
|
+
bucket,
|
|
373
|
+
blurb: "ModelScope GGUF result · explicit opt-in only",
|
|
374
|
+
};
|
|
375
|
+
}
|
|
376
|
+
|
|
377
|
+
/**
|
|
378
|
+
* Search ModelScope for GGUF repos. ModelScope exposes reliable owner-list
|
|
379
|
+
* and repo-file APIs, so this accepts either `owner/model` or an owner name
|
|
380
|
+
* and filters the returned repos for GGUF files.
|
|
381
|
+
*/
|
|
382
|
+
export async function searchModelScopeGguf(
|
|
383
|
+
query: string,
|
|
384
|
+
limit = 12,
|
|
385
|
+
): Promise<CatalogModel[]> {
|
|
386
|
+
const trimmed = query.trim();
|
|
387
|
+
if (trimmed.length === 0) return [];
|
|
388
|
+
|
|
389
|
+
const direct = normalizeModelScopeRepoId(trimmed);
|
|
390
|
+
const repoIds = direct
|
|
391
|
+
? [direct]
|
|
392
|
+
: await listModelScopeOwnerModels(
|
|
393
|
+
modelScopeOwnerQuery(trimmed) ?? trimmed,
|
|
394
|
+
limit,
|
|
395
|
+
);
|
|
396
|
+
|
|
397
|
+
const results = await Promise.all(
|
|
398
|
+
repoIds.slice(0, Math.min(50, limit * 3)).map(async (id) => {
|
|
399
|
+
try {
|
|
400
|
+
return await modelScopeResultForRepo(id);
|
|
401
|
+
} catch {
|
|
402
|
+
return null;
|
|
403
|
+
}
|
|
404
|
+
}),
|
|
405
|
+
);
|
|
406
|
+
|
|
407
|
+
return results
|
|
408
|
+
.filter((model): model is CatalogModel => Boolean(model))
|
|
409
|
+
.slice(0, limit);
|
|
410
|
+
}
|
|
411
|
+
|
|
412
|
+
export async function searchModelHubGguf(
|
|
413
|
+
query: string,
|
|
414
|
+
hub: "huggingface" | "modelscope" = "huggingface",
|
|
415
|
+
limit = 12,
|
|
416
|
+
): Promise<CatalogModel[]> {
|
|
417
|
+
return hub === "modelscope"
|
|
418
|
+
? searchModelScopeGguf(query, limit)
|
|
419
|
+
: searchHuggingFaceGguf(query, limit);
|
|
420
|
+
}
|
|
@@ -0,0 +1,14 @@
|
|
|
1
|
+
export interface ImageDescriptionRuntimeOptions {
|
|
2
|
+
tier: string;
|
|
3
|
+
modelPath: string;
|
|
4
|
+
}
|
|
5
|
+
export interface ImageDescriptionRuntime {
|
|
6
|
+
describe(args: {
|
|
7
|
+
imagePath: string;
|
|
8
|
+
prompt: string;
|
|
9
|
+
maxTokens?: number;
|
|
10
|
+
}): Promise<string>;
|
|
11
|
+
cleanup?(): Promise<void>;
|
|
12
|
+
}
|
|
13
|
+
export declare function createImageDescriptionRuntime(args: ImageDescriptionRuntimeOptions): Promise<ImageDescriptionRuntime>;
|
|
14
|
+
//# sourceMappingURL=image-description-runtime.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"image-description-runtime.d.ts","sourceRoot":"","sources":["image-description-runtime.ts"],"names":[],"mappings":"AAkBA,MAAM,WAAW,8BAA8B;IAC9C,IAAI,EAAE,MAAM,CAAC;IACb,SAAS,EAAE,MAAM,CAAC;CAClB;AAED,MAAM,WAAW,uBAAuB;IACvC,QAAQ,CAAC,IAAI,EAAE;QACd,SAAS,EAAE,MAAM,CAAC;QAClB,MAAM,EAAE,MAAM,CAAC;QACf,SAAS,CAAC,EAAE,MAAM,CAAC;KACnB,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;IACpB,OAAO,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CAC1B;AAED,wBAAsB,6BAA6B,CAClD,IAAI,EAAE,8BAA8B,GAClC,OAAO,CAAC,uBAAuB,CAAC,CA8BlC"}
|
|
@@ -0,0 +1,61 @@
|
|
|
1
|
+
import { mkdirSync, mkdtempSync, writeFileSync } from "node:fs";
|
|
2
|
+
import { tmpdir } from "node:os";
|
|
3
|
+
import path from "node:path";
|
|
4
|
+
import { afterEach, describe, expect, it, vi } from "vitest";
|
|
5
|
+
|
|
6
|
+
const mocks = vi.hoisted(() => ({
|
|
7
|
+
load: vi.fn(async () => undefined),
|
|
8
|
+
describeImage: vi.fn(async () => ({ text: "ok" })),
|
|
9
|
+
unload: vi.fn(async () => undefined),
|
|
10
|
+
}));
|
|
11
|
+
|
|
12
|
+
vi.mock("./engine", () => ({
|
|
13
|
+
localInferenceEngine: {
|
|
14
|
+
load: mocks.load,
|
|
15
|
+
describeImage: mocks.describeImage,
|
|
16
|
+
unload: mocks.unload,
|
|
17
|
+
},
|
|
18
|
+
}));
|
|
19
|
+
|
|
20
|
+
import { createImageDescriptionRuntime } from "./image-description-runtime";
|
|
21
|
+
|
|
22
|
+
afterEach(() => {
|
|
23
|
+
vi.clearAllMocks();
|
|
24
|
+
});
|
|
25
|
+
|
|
26
|
+
describe("createImageDescriptionRuntime", () => {
|
|
27
|
+
it("passes the manifest catalog id for direct bundle loads", async () => {
|
|
28
|
+
const bundleRoot = mkdtempSync(path.join(tmpdir(), "eliza-vision-"));
|
|
29
|
+
mkdirSync(path.join(bundleRoot, "text"), { recursive: true });
|
|
30
|
+
mkdirSync(path.join(bundleRoot, "vision"), { recursive: true });
|
|
31
|
+
const textPath = path.join(bundleRoot, "text", "eliza-1-9b-128k.gguf");
|
|
32
|
+
const mmprojPath = path.join(bundleRoot, "vision", "mmproj-9b.gguf");
|
|
33
|
+
writeFileSync(textPath, "GGUF", "utf8");
|
|
34
|
+
writeFileSync(mmprojPath, "GGUF", "utf8");
|
|
35
|
+
writeFileSync(
|
|
36
|
+
path.join(bundleRoot, "eliza-1.manifest.json"),
|
|
37
|
+
JSON.stringify({
|
|
38
|
+
id: "eliza-1-9b",
|
|
39
|
+
files: {
|
|
40
|
+
text: [{ path: "text/eliza-1-9b-128k.gguf", ctx: 131072 }],
|
|
41
|
+
vision: [{ path: "vision/mmproj-9b.gguf" }],
|
|
42
|
+
},
|
|
43
|
+
}),
|
|
44
|
+
"utf8",
|
|
45
|
+
);
|
|
46
|
+
|
|
47
|
+
const runtime = await createImageDescriptionRuntime({
|
|
48
|
+
tier: "eliza-1-9b",
|
|
49
|
+
modelPath: bundleRoot,
|
|
50
|
+
});
|
|
51
|
+
|
|
52
|
+
expect(mocks.load).toHaveBeenCalledWith(textPath, {
|
|
53
|
+
modelPath: textPath,
|
|
54
|
+
modelId: "eliza-1-9b",
|
|
55
|
+
mmprojPath,
|
|
56
|
+
contextSize: 4096,
|
|
57
|
+
});
|
|
58
|
+
await runtime.cleanup?.();
|
|
59
|
+
expect(mocks.unload).toHaveBeenCalled();
|
|
60
|
+
});
|
|
61
|
+
});
|
|
@@ -0,0 +1,118 @@
|
|
|
1
|
+
import { readFile } from "node:fs/promises";
|
|
2
|
+
import path from "node:path";
|
|
3
|
+
import { localInferenceEngine } from "./engine";
|
|
4
|
+
import type { LocalInferenceLoadArgs } from "./load-args";
|
|
5
|
+
|
|
6
|
+
interface BundleManifestFile {
|
|
7
|
+
path: string;
|
|
8
|
+
ctx?: number;
|
|
9
|
+
}
|
|
10
|
+
|
|
11
|
+
interface BundleManifest {
|
|
12
|
+
id?: string;
|
|
13
|
+
files?: {
|
|
14
|
+
text?: BundleManifestFile[];
|
|
15
|
+
vision?: BundleManifestFile[];
|
|
16
|
+
};
|
|
17
|
+
}
|
|
18
|
+
|
|
19
|
+
export interface ImageDescriptionRuntimeOptions {
|
|
20
|
+
tier: string;
|
|
21
|
+
modelPath: string;
|
|
22
|
+
}
|
|
23
|
+
|
|
24
|
+
export interface ImageDescriptionRuntime {
|
|
25
|
+
describe(args: {
|
|
26
|
+
imagePath: string;
|
|
27
|
+
prompt: string;
|
|
28
|
+
maxTokens?: number;
|
|
29
|
+
}): Promise<string>;
|
|
30
|
+
cleanup?(): Promise<void>;
|
|
31
|
+
}
|
|
32
|
+
|
|
33
|
+
export async function createImageDescriptionRuntime(
|
|
34
|
+
args: ImageDescriptionRuntimeOptions,
|
|
35
|
+
): Promise<ImageDescriptionRuntime> {
|
|
36
|
+
const bundleRoot = path.resolve(args.modelPath);
|
|
37
|
+
const manifest = await readBundleManifest(bundleRoot);
|
|
38
|
+
const textModelPath = resolveTextModelPath(bundleRoot, manifest, args.tier);
|
|
39
|
+
const mmprojPath = resolveVisionModelPath(bundleRoot, manifest, args.tier);
|
|
40
|
+
const loadArgs: LocalInferenceLoadArgs = {
|
|
41
|
+
modelPath: textModelPath,
|
|
42
|
+
modelId: manifest.id || args.tier,
|
|
43
|
+
mmprojPath,
|
|
44
|
+
contextSize: 4096,
|
|
45
|
+
};
|
|
46
|
+
|
|
47
|
+
await localInferenceEngine.load(textModelPath, loadArgs);
|
|
48
|
+
|
|
49
|
+
return {
|
|
50
|
+
async describe({ imagePath, prompt, maxTokens }) {
|
|
51
|
+
const bytes = await readImageInput(imagePath);
|
|
52
|
+
const result = await localInferenceEngine.describeImage({
|
|
53
|
+
bytes,
|
|
54
|
+
mimeType: mimeTypeFor(imagePath),
|
|
55
|
+
prompt,
|
|
56
|
+
maxTokens,
|
|
57
|
+
temperature: 0,
|
|
58
|
+
});
|
|
59
|
+
return result.text;
|
|
60
|
+
},
|
|
61
|
+
async cleanup() {
|
|
62
|
+
await localInferenceEngine.unload();
|
|
63
|
+
},
|
|
64
|
+
};
|
|
65
|
+
}
|
|
66
|
+
|
|
67
|
+
async function readBundleManifest(bundleRoot: string): Promise<BundleManifest> {
|
|
68
|
+
const manifestPath = path.join(bundleRoot, "eliza-1.manifest.json");
|
|
69
|
+
const raw = await readFile(manifestPath, "utf8");
|
|
70
|
+
return JSON.parse(raw) as BundleManifest;
|
|
71
|
+
}
|
|
72
|
+
|
|
73
|
+
function resolveTextModelPath(
|
|
74
|
+
bundleRoot: string,
|
|
75
|
+
manifest: BundleManifest,
|
|
76
|
+
tier: string,
|
|
77
|
+
): string {
|
|
78
|
+
const textFiles = manifest.files?.text ?? [];
|
|
79
|
+
const selected = [...textFiles].sort(
|
|
80
|
+
(a, b) => (a.ctx ?? 0) - (b.ctx ?? 0),
|
|
81
|
+
)[0];
|
|
82
|
+
if (selected?.path) return path.join(bundleRoot, selected.path);
|
|
83
|
+
const slug = tier.replace(/^eliza-1-/, "");
|
|
84
|
+
return path.join(bundleRoot, "text", `eliza-1-${slug}-64k.gguf`);
|
|
85
|
+
}
|
|
86
|
+
|
|
87
|
+
function resolveVisionModelPath(
|
|
88
|
+
bundleRoot: string,
|
|
89
|
+
manifest: BundleManifest,
|
|
90
|
+
tier: string,
|
|
91
|
+
): string {
|
|
92
|
+
const selected = manifest.files?.vision?.[0];
|
|
93
|
+
if (selected?.path) return path.join(bundleRoot, selected.path);
|
|
94
|
+
const slug = tier.replace(/^eliza-1-/, "");
|
|
95
|
+
return path.join(bundleRoot, "vision", `mmproj-${slug}.gguf`);
|
|
96
|
+
}
|
|
97
|
+
|
|
98
|
+
async function readImageInput(imagePath: string): Promise<Uint8Array> {
|
|
99
|
+
if (imagePath.startsWith("data:")) {
|
|
100
|
+
const comma = imagePath.indexOf(",");
|
|
101
|
+
if (comma < 0) throw new Error("invalid data URL image input");
|
|
102
|
+
const encoded = imagePath.slice(comma + 1);
|
|
103
|
+
return Uint8Array.from(Buffer.from(encoded, "base64"));
|
|
104
|
+
}
|
|
105
|
+
return Uint8Array.from(await readFile(imagePath));
|
|
106
|
+
}
|
|
107
|
+
|
|
108
|
+
function mimeTypeFor(imagePath: string): string | undefined {
|
|
109
|
+
if (imagePath.startsWith("data:")) {
|
|
110
|
+
const match = /^data:([^;,]+)/.exec(imagePath);
|
|
111
|
+
return match?.[1];
|
|
112
|
+
}
|
|
113
|
+
const ext = path.extname(imagePath).toLowerCase();
|
|
114
|
+
if (ext === ".png") return "image/png";
|
|
115
|
+
if (ext === ".jpg" || ext === ".jpeg") return "image/jpeg";
|
|
116
|
+
if (ext === ".webp") return "image/webp";
|
|
117
|
+
return undefined;
|
|
118
|
+
}
|