@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,122 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* `/api/voice/audio-frames` — WebView → agent transport for live on-device
|
|
3
|
+
* speaker diarization.
|
|
4
|
+
*
|
|
5
|
+
* The Android `audioFrame` PCM stream is captured in the Capacitor WebView but
|
|
6
|
+
* the bun:ffi voice libs run in the agent process. The WebView batches frames
|
|
7
|
+
* (~49 fps) and POSTs them here; this route feeds them to the single
|
|
8
|
+
* {@link LiveDiarizationSession}, which runs the real ggml VAD / encoder /
|
|
9
|
+
* diarizer / attribution pipeline and emits VOICE_TURN_OBSERVED.
|
|
10
|
+
*
|
|
11
|
+
* Routes:
|
|
12
|
+
* POST /api/voice/audio-frames body: { frames: AudioFrameEvent[],
|
|
13
|
+
* flush?: boolean }
|
|
14
|
+
* → { ok, framesReceived, turnsObserved }
|
|
15
|
+
* GET /api/voice/audio-frames/status → LiveDiarizationStatus (device evidence)
|
|
16
|
+
*
|
|
17
|
+
* Auth follows the compat pattern: trusted-loopback OR the compat API token.
|
|
18
|
+
* The WebView reaches this over 127.0.0.1 (trusted local), matching the rest of
|
|
19
|
+
* the on-device agent surface.
|
|
20
|
+
*/
|
|
21
|
+
|
|
22
|
+
import type http from "node:http";
|
|
23
|
+
import type { AudioFrameEvent } from "../services/voice/audio-frame-consumer.js";
|
|
24
|
+
import {
|
|
25
|
+
LiveDiarizationSession,
|
|
26
|
+
type RuntimeEventSink,
|
|
27
|
+
} from "../services/voice/live-diarization-session.js";
|
|
28
|
+
import {
|
|
29
|
+
type CompatRuntimeState,
|
|
30
|
+
ensureCompatApiAuthorized,
|
|
31
|
+
readCompatJsonBody,
|
|
32
|
+
sendJson,
|
|
33
|
+
sendJsonError,
|
|
34
|
+
} from "./compat-helpers.js";
|
|
35
|
+
|
|
36
|
+
let session: LiveDiarizationSession | null = null;
|
|
37
|
+
|
|
38
|
+
/** Lazily own one session per agent process, bound to the live runtime. */
|
|
39
|
+
function getSession(state: CompatRuntimeState): LiveDiarizationSession | null {
|
|
40
|
+
const runtime = state.current as unknown as RuntimeEventSink | null;
|
|
41
|
+
if (!runtime || typeof runtime.emitEvent !== "function") return null;
|
|
42
|
+
if (!session) session = new LiveDiarizationSession(runtime);
|
|
43
|
+
return session;
|
|
44
|
+
}
|
|
45
|
+
|
|
46
|
+
/** Reset the module-level session (tests + capture teardown). */
|
|
47
|
+
export async function resetLiveDiarizationSession(): Promise<void> {
|
|
48
|
+
const current = session;
|
|
49
|
+
session = null;
|
|
50
|
+
if (current) await current.close();
|
|
51
|
+
}
|
|
52
|
+
|
|
53
|
+
function isAudioFrameEvent(value: unknown): value is AudioFrameEvent {
|
|
54
|
+
if (!value || typeof value !== "object") return false;
|
|
55
|
+
const f = value as Partial<AudioFrameEvent>;
|
|
56
|
+
return (
|
|
57
|
+
typeof f.pcm16 === "string" &&
|
|
58
|
+
typeof f.sampleRate === "number" &&
|
|
59
|
+
typeof f.channels === "number" &&
|
|
60
|
+
typeof f.samples === "number" &&
|
|
61
|
+
typeof f.rms === "number" &&
|
|
62
|
+
typeof f.timestamp === "number" &&
|
|
63
|
+
typeof f.frameIndex === "number"
|
|
64
|
+
);
|
|
65
|
+
}
|
|
66
|
+
|
|
67
|
+
export async function handleLiveDiarizationRoute(
|
|
68
|
+
req: http.IncomingMessage,
|
|
69
|
+
res: http.ServerResponse,
|
|
70
|
+
state: CompatRuntimeState,
|
|
71
|
+
): Promise<boolean> {
|
|
72
|
+
const url = new URL(req.url ?? "/", "http://localhost");
|
|
73
|
+
const method = req.method ?? "GET";
|
|
74
|
+
|
|
75
|
+
if (url.pathname === "/api/voice/audio-frames/status" && method === "GET") {
|
|
76
|
+
if (!ensureCompatApiAuthorized(req, res)) return true;
|
|
77
|
+
const current = getSession(state);
|
|
78
|
+
if (!current) {
|
|
79
|
+
sendJsonError(res, 503, "Runtime not ready");
|
|
80
|
+
return true;
|
|
81
|
+
}
|
|
82
|
+
sendJson(res, 200, await current.status());
|
|
83
|
+
return true;
|
|
84
|
+
}
|
|
85
|
+
|
|
86
|
+
if (url.pathname === "/api/voice/audio-frames" && method === "POST") {
|
|
87
|
+
if (!ensureCompatApiAuthorized(req, res)) return true;
|
|
88
|
+
const current = getSession(state);
|
|
89
|
+
if (!current) {
|
|
90
|
+
sendJsonError(res, 503, "Runtime not ready");
|
|
91
|
+
return true;
|
|
92
|
+
}
|
|
93
|
+
const body = await readCompatJsonBody(req, res);
|
|
94
|
+
if (!body) return true;
|
|
95
|
+
const rawFrames = body.frames;
|
|
96
|
+
if (!Array.isArray(rawFrames)) {
|
|
97
|
+
sendJsonError(res, 400, "Expected { frames: AudioFrameEvent[] }");
|
|
98
|
+
return true;
|
|
99
|
+
}
|
|
100
|
+
const frames = rawFrames.filter(isAudioFrameEvent);
|
|
101
|
+
if (frames.length !== rawFrames.length) {
|
|
102
|
+
sendJsonError(
|
|
103
|
+
res,
|
|
104
|
+
400,
|
|
105
|
+
`Malformed frame(s): ${rawFrames.length - frames.length} of ${rawFrames.length} did not match AudioFrameEvent`,
|
|
106
|
+
);
|
|
107
|
+
return true;
|
|
108
|
+
}
|
|
109
|
+
await current.ingest(frames);
|
|
110
|
+
if (body.flush === true) await current.flush();
|
|
111
|
+
const status = await current.status();
|
|
112
|
+
sendJson(res, 200, {
|
|
113
|
+
ok: true,
|
|
114
|
+
framesReceived: status.framesReceived,
|
|
115
|
+
framesDropped: status.framesDropped,
|
|
116
|
+
turnsObserved: status.turnsObserved,
|
|
117
|
+
});
|
|
118
|
+
return true;
|
|
119
|
+
}
|
|
120
|
+
|
|
121
|
+
return false;
|
|
122
|
+
}
|
|
@@ -0,0 +1,4 @@
|
|
|
1
|
+
import type http from "node:http";
|
|
2
|
+
import { type CompatRuntimeState } from "./compat-helpers";
|
|
3
|
+
export declare function handleLocalInferenceAsrRoute(req: http.IncomingMessage, res: http.ServerResponse, state: CompatRuntimeState): Promise<boolean>;
|
|
4
|
+
//# sourceMappingURL=local-inference-asr-route.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"local-inference-asr-route.d.ts","sourceRoot":"","sources":["local-inference-asr-route.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,IAAI,MAAM,WAAW,CAAC;AAElC,OAAO,EACN,KAAK,kBAAkB,EAIvB,MAAM,kBAAkB,CAAC;AA4E1B,wBAAsB,4BAA4B,CACjD,GAAG,EAAE,IAAI,CAAC,eAAe,EACzB,GAAG,EAAE,IAAI,CAAC,cAAc,EACxB,KAAK,EAAE,kBAAkB,GACvB,OAAO,CAAC,OAAO,CAAC,CA2ElB"}
|
|
@@ -0,0 +1,205 @@
|
|
|
1
|
+
import * as http from "node:http";
|
|
2
|
+
import { Socket } from "node:net";
|
|
3
|
+
import { ModelType } from "@elizaos/core";
|
|
4
|
+
import { describe, expect, it, vi } from "vitest";
|
|
5
|
+
import type { CompatRuntimeState } from "./compat-helpers";
|
|
6
|
+
import { handleLocalInferenceAsrRoute } from "./local-inference-asr-route";
|
|
7
|
+
import { transcribeWavWithWords } from "./local-inference-asr-transcribe";
|
|
8
|
+
|
|
9
|
+
vi.mock("./local-inference-asr-transcribe", () => ({
|
|
10
|
+
transcribeWavWithWords: vi.fn(),
|
|
11
|
+
}));
|
|
12
|
+
|
|
13
|
+
const transcribeWavWithWordsMock = vi.mocked(transcribeWavWithWords);
|
|
14
|
+
|
|
15
|
+
function wavBytes(): Uint8Array {
|
|
16
|
+
const pcm = new Int16Array([0, 900, -900, 0]);
|
|
17
|
+
const buffer = new ArrayBuffer(44 + pcm.length * 2);
|
|
18
|
+
const view = new DataView(buffer);
|
|
19
|
+
const writeAscii = (offset: number, value: string) => {
|
|
20
|
+
for (let index = 0; index < value.length; index += 1) {
|
|
21
|
+
view.setUint8(offset + index, value.charCodeAt(index));
|
|
22
|
+
}
|
|
23
|
+
};
|
|
24
|
+
writeAscii(0, "RIFF");
|
|
25
|
+
view.setUint32(4, 36 + pcm.length * 2, true);
|
|
26
|
+
writeAscii(8, "WAVE");
|
|
27
|
+
writeAscii(12, "fmt ");
|
|
28
|
+
view.setUint32(16, 16, true);
|
|
29
|
+
view.setUint16(20, 1, true);
|
|
30
|
+
view.setUint16(22, 1, true);
|
|
31
|
+
view.setUint32(24, 16_000, true);
|
|
32
|
+
view.setUint32(28, 16_000 * 2, true);
|
|
33
|
+
view.setUint16(32, 2, true);
|
|
34
|
+
view.setUint16(34, 16, true);
|
|
35
|
+
writeAscii(36, "data");
|
|
36
|
+
view.setUint32(40, pcm.length * 2, true);
|
|
37
|
+
for (let index = 0; index < pcm.length; index += 1) {
|
|
38
|
+
view.setInt16(44 + index * 2, pcm[index] ?? 0, true);
|
|
39
|
+
}
|
|
40
|
+
return new Uint8Array(buffer);
|
|
41
|
+
}
|
|
42
|
+
|
|
43
|
+
function fakeReq(
|
|
44
|
+
body?: unknown,
|
|
45
|
+
opts?: { method?: string; url?: string },
|
|
46
|
+
): http.IncomingMessage {
|
|
47
|
+
const req = new http.IncomingMessage(new Socket());
|
|
48
|
+
req.method = opts?.method ?? "POST";
|
|
49
|
+
req.url = opts?.url ?? "/api/asr/local-inference";
|
|
50
|
+
req.headers = {
|
|
51
|
+
host: "localhost:2138",
|
|
52
|
+
"content-type": "audio/wav",
|
|
53
|
+
};
|
|
54
|
+
Object.defineProperty(req.socket, "remoteAddress", {
|
|
55
|
+
value: "127.0.0.1",
|
|
56
|
+
configurable: true,
|
|
57
|
+
});
|
|
58
|
+
if (body !== undefined) {
|
|
59
|
+
(req as { body?: unknown }).body = body;
|
|
60
|
+
}
|
|
61
|
+
return req;
|
|
62
|
+
}
|
|
63
|
+
|
|
64
|
+
function fakeRes(): {
|
|
65
|
+
res: http.ServerResponse;
|
|
66
|
+
bodyJson: () => Record<string, unknown>;
|
|
67
|
+
status: () => number;
|
|
68
|
+
} {
|
|
69
|
+
const req = new http.IncomingMessage(new Socket());
|
|
70
|
+
const res = new http.ServerResponse(req);
|
|
71
|
+
let body = Buffer.alloc(0);
|
|
72
|
+
let status = 200;
|
|
73
|
+
res.setHeader = (() => res) as typeof res.setHeader;
|
|
74
|
+
res.writeHead = ((code: number) => {
|
|
75
|
+
status = code;
|
|
76
|
+
res.statusCode = code;
|
|
77
|
+
return res;
|
|
78
|
+
}) as typeof res.writeHead;
|
|
79
|
+
res.end = ((chunk?: string | Uint8Array | Buffer) => {
|
|
80
|
+
if (typeof chunk === "string") {
|
|
81
|
+
body = Buffer.concat([body, Buffer.from(chunk)]);
|
|
82
|
+
} else if (chunk) {
|
|
83
|
+
body = Buffer.concat([body, Buffer.from(chunk)]);
|
|
84
|
+
}
|
|
85
|
+
return res;
|
|
86
|
+
}) as typeof res.end;
|
|
87
|
+
return {
|
|
88
|
+
res,
|
|
89
|
+
bodyJson: () => JSON.parse(body.toString("utf8")),
|
|
90
|
+
status: () => status,
|
|
91
|
+
};
|
|
92
|
+
}
|
|
93
|
+
|
|
94
|
+
describe("local inference ASR route", () => {
|
|
95
|
+
it("reports readiness off the registered fused TRANSCRIPTION handler", async () => {
|
|
96
|
+
const getModel = vi.fn(() => () => "transcript");
|
|
97
|
+
const useModel = vi.fn();
|
|
98
|
+
const state: CompatRuntimeState = {
|
|
99
|
+
current: {
|
|
100
|
+
getModel,
|
|
101
|
+
useModel,
|
|
102
|
+
} as unknown as CompatRuntimeState["current"],
|
|
103
|
+
};
|
|
104
|
+
const out = fakeRes();
|
|
105
|
+
|
|
106
|
+
const handled = await handleLocalInferenceAsrRoute(
|
|
107
|
+
fakeReq(undefined, {
|
|
108
|
+
method: "GET",
|
|
109
|
+
url: "/api/asr/local-inference/status",
|
|
110
|
+
}),
|
|
111
|
+
out.res,
|
|
112
|
+
state,
|
|
113
|
+
);
|
|
114
|
+
|
|
115
|
+
expect(handled).toBe(true);
|
|
116
|
+
expect(out.status()).toBe(200);
|
|
117
|
+
expect(out.bodyJson()).toEqual({
|
|
118
|
+
ready: true,
|
|
119
|
+
provider: "local-inference",
|
|
120
|
+
});
|
|
121
|
+
expect(getModel).toHaveBeenCalledWith(ModelType.TRANSCRIPTION);
|
|
122
|
+
expect(useModel).not.toHaveBeenCalled();
|
|
123
|
+
});
|
|
124
|
+
|
|
125
|
+
it("reports not-ready when no TRANSCRIPTION handler is registered", async () => {
|
|
126
|
+
const getModel = vi.fn(() => undefined);
|
|
127
|
+
const state: CompatRuntimeState = {
|
|
128
|
+
current: {
|
|
129
|
+
getModel,
|
|
130
|
+
} as unknown as CompatRuntimeState["current"],
|
|
131
|
+
};
|
|
132
|
+
const out = fakeRes();
|
|
133
|
+
|
|
134
|
+
const handled = await handleLocalInferenceAsrRoute(
|
|
135
|
+
fakeReq(undefined, {
|
|
136
|
+
method: "GET",
|
|
137
|
+
url: "/api/asr/local-inference/status",
|
|
138
|
+
}),
|
|
139
|
+
out.res,
|
|
140
|
+
state,
|
|
141
|
+
);
|
|
142
|
+
|
|
143
|
+
expect(handled).toBe(true);
|
|
144
|
+
expect(out.status()).toBe(200);
|
|
145
|
+
expect(out.bodyJson()).toEqual({ ready: false, provider: null });
|
|
146
|
+
});
|
|
147
|
+
|
|
148
|
+
it("transcribes raw WAV audio and returns text + per-word timings", async () => {
|
|
149
|
+
transcribeWavWithWordsMock.mockResolvedValue({
|
|
150
|
+
text: "hello local voice",
|
|
151
|
+
words: [
|
|
152
|
+
{ text: "hello", startMs: 0, endMs: 400 },
|
|
153
|
+
{ text: "local", startMs: 400, endMs: 700 },
|
|
154
|
+
{ text: "voice", startMs: 700, endMs: 1000 },
|
|
155
|
+
],
|
|
156
|
+
});
|
|
157
|
+
const state: CompatRuntimeState = {
|
|
158
|
+
current: {} as unknown as CompatRuntimeState["current"],
|
|
159
|
+
};
|
|
160
|
+
const out = fakeRes();
|
|
161
|
+
|
|
162
|
+
const handled = await handleLocalInferenceAsrRoute(
|
|
163
|
+
fakeReq(wavBytes()),
|
|
164
|
+
out.res,
|
|
165
|
+
state,
|
|
166
|
+
);
|
|
167
|
+
|
|
168
|
+
expect(handled).toBe(true);
|
|
169
|
+
// The raw WAV bytes are forwarded to the single FFI-pipe transcriber.
|
|
170
|
+
expect(
|
|
171
|
+
Array.from(transcribeWavWithWordsMock.mock.calls[0]?.[1] as Uint8Array),
|
|
172
|
+
).toEqual(Array.from(wavBytes()));
|
|
173
|
+
expect(out.status()).toBe(200);
|
|
174
|
+
expect(out.bodyJson()).toEqual({
|
|
175
|
+
text: "hello local voice",
|
|
176
|
+
words: [
|
|
177
|
+
{ text: "hello", startMs: 0, endMs: 400 },
|
|
178
|
+
{ text: "local", startMs: 400, endMs: 700 },
|
|
179
|
+
{ text: "voice", startMs: 700, endMs: 1000 },
|
|
180
|
+
],
|
|
181
|
+
});
|
|
182
|
+
});
|
|
183
|
+
|
|
184
|
+
it("accepts JSON base64 audio for route clients that cannot send raw WAV", async () => {
|
|
185
|
+
transcribeWavWithWordsMock.mockResolvedValue({
|
|
186
|
+
text: "hello from json",
|
|
187
|
+
words: [],
|
|
188
|
+
});
|
|
189
|
+
const state: CompatRuntimeState = {
|
|
190
|
+
current: {} as unknown as CompatRuntimeState["current"],
|
|
191
|
+
};
|
|
192
|
+
const req = fakeReq({
|
|
193
|
+
audioBase64: Buffer.from(wavBytes()).toString("base64"),
|
|
194
|
+
});
|
|
195
|
+
req.headers["content-type"] = "application/json";
|
|
196
|
+
const out = fakeRes();
|
|
197
|
+
|
|
198
|
+
await handleLocalInferenceAsrRoute(req, out.res, state);
|
|
199
|
+
|
|
200
|
+
expect(
|
|
201
|
+
Array.from(transcribeWavWithWordsMock.mock.calls[0]?.[1] as Uint8Array),
|
|
202
|
+
).toEqual(Array.from(wavBytes()));
|
|
203
|
+
expect(out.bodyJson()).toEqual({ text: "hello from json", words: [] });
|
|
204
|
+
});
|
|
205
|
+
});
|
|
@@ -0,0 +1,163 @@
|
|
|
1
|
+
import type http from "node:http";
|
|
2
|
+
import { ModelType } from "@elizaos/core";
|
|
3
|
+
import {
|
|
4
|
+
type CompatRuntimeState,
|
|
5
|
+
ensureRouteAuthorized,
|
|
6
|
+
readCompatJsonBody,
|
|
7
|
+
sendJson,
|
|
8
|
+
} from "./compat-helpers";
|
|
9
|
+
import { transcribeWavWithWords } from "./local-inference-asr-transcribe";
|
|
10
|
+
|
|
11
|
+
const MAX_LOCAL_ASR_AUDIO_BYTES = 16 * 1024 * 1024;
|
|
12
|
+
|
|
13
|
+
function toUint8Array(value: Uint8Array | ArrayBuffer): Uint8Array {
|
|
14
|
+
return value instanceof Uint8Array ? value : new Uint8Array(value);
|
|
15
|
+
}
|
|
16
|
+
|
|
17
|
+
function coercePreParsedAudio(value: unknown): Uint8Array | null {
|
|
18
|
+
if (value instanceof Uint8Array) return toUint8Array(value);
|
|
19
|
+
if (value instanceof ArrayBuffer) return toUint8Array(value);
|
|
20
|
+
if (ArrayBuffer.isView(value)) {
|
|
21
|
+
return new Uint8Array(value.buffer, value.byteOffset, value.byteLength);
|
|
22
|
+
}
|
|
23
|
+
return null;
|
|
24
|
+
}
|
|
25
|
+
|
|
26
|
+
async function readRawAudioBody(
|
|
27
|
+
req: http.IncomingMessage,
|
|
28
|
+
res: http.ServerResponse,
|
|
29
|
+
): Promise<Uint8Array | null> {
|
|
30
|
+
const preParsed = coercePreParsedAudio((req as { body?: unknown }).body);
|
|
31
|
+
if (preParsed) return preParsed;
|
|
32
|
+
|
|
33
|
+
const chunks: Buffer[] = [];
|
|
34
|
+
let totalBytes = 0;
|
|
35
|
+
try {
|
|
36
|
+
for await (const chunk of req) {
|
|
37
|
+
const buf = Buffer.isBuffer(chunk) ? chunk : Buffer.from(chunk);
|
|
38
|
+
totalBytes += buf.byteLength;
|
|
39
|
+
if (totalBytes > MAX_LOCAL_ASR_AUDIO_BYTES) {
|
|
40
|
+
req.destroy();
|
|
41
|
+
sendJson(res, 413, { error: "Audio body too large" });
|
|
42
|
+
return null;
|
|
43
|
+
}
|
|
44
|
+
chunks.push(buf);
|
|
45
|
+
}
|
|
46
|
+
} catch {
|
|
47
|
+
sendJson(res, 400, { error: "Invalid audio body" });
|
|
48
|
+
return null;
|
|
49
|
+
}
|
|
50
|
+
|
|
51
|
+
return new Uint8Array(Buffer.concat(chunks));
|
|
52
|
+
}
|
|
53
|
+
|
|
54
|
+
function firstHeaderValue(value: string | string[] | undefined): string {
|
|
55
|
+
return Array.isArray(value) ? (value[0] ?? "") : (value ?? "");
|
|
56
|
+
}
|
|
57
|
+
|
|
58
|
+
async function readLocalInferenceAsrAudio(
|
|
59
|
+
req: http.IncomingMessage,
|
|
60
|
+
res: http.ServerResponse,
|
|
61
|
+
): Promise<Uint8Array | null> {
|
|
62
|
+
const contentType = firstHeaderValue(req.headers["content-type"])
|
|
63
|
+
.toLowerCase()
|
|
64
|
+
.split(";", 1)[0]
|
|
65
|
+
.trim();
|
|
66
|
+
|
|
67
|
+
if (contentType === "application/json") {
|
|
68
|
+
const body = await readCompatJsonBody(req, res);
|
|
69
|
+
if (!body) return null;
|
|
70
|
+
if (typeof body.audioBase64 === "string") {
|
|
71
|
+
return new Uint8Array(Buffer.from(body.audioBase64, "base64"));
|
|
72
|
+
}
|
|
73
|
+
sendJson(res, 400, { error: "Missing audioBase64" });
|
|
74
|
+
return null;
|
|
75
|
+
}
|
|
76
|
+
|
|
77
|
+
return readRawAudioBody(req, res);
|
|
78
|
+
}
|
|
79
|
+
|
|
80
|
+
function isClosed(res: http.ServerResponse): boolean {
|
|
81
|
+
return res.destroyed || res.writableEnded;
|
|
82
|
+
}
|
|
83
|
+
|
|
84
|
+
export async function handleLocalInferenceAsrRoute(
|
|
85
|
+
req: http.IncomingMessage,
|
|
86
|
+
res: http.ServerResponse,
|
|
87
|
+
state: CompatRuntimeState,
|
|
88
|
+
): Promise<boolean> {
|
|
89
|
+
const method = req.method?.toUpperCase() ?? "GET";
|
|
90
|
+
const url = new URL(req.url ?? "/", "http://localhost");
|
|
91
|
+
if (method === "GET" && url.pathname === "/api/asr/local-inference/status") {
|
|
92
|
+
if (!(await ensureRouteAuthorized(req, res, state))) return true;
|
|
93
|
+
// Transcription runs through the registered TRANSCRIPTION model handler,
|
|
94
|
+
// which is backed by the fused libelizainference ASR runtime. Report ready
|
|
95
|
+
// when a TRANSCRIPTION handler is registered.
|
|
96
|
+
const getModel = state.current?.getModel;
|
|
97
|
+
const runtimeAsr =
|
|
98
|
+
typeof getModel === "function" &&
|
|
99
|
+
Boolean(getModel.call(state.current, ModelType.TRANSCRIPTION));
|
|
100
|
+
sendJson(res, 200, {
|
|
101
|
+
ready: runtimeAsr,
|
|
102
|
+
provider: runtimeAsr ? "local-inference" : null,
|
|
103
|
+
});
|
|
104
|
+
return true;
|
|
105
|
+
}
|
|
106
|
+
|
|
107
|
+
if (method !== "POST" || url.pathname !== "/api/asr/local-inference") {
|
|
108
|
+
return false;
|
|
109
|
+
}
|
|
110
|
+
|
|
111
|
+
if (!(await ensureRouteAuthorized(req, res, state))) return true;
|
|
112
|
+
|
|
113
|
+
const audio = await readLocalInferenceAsrAudio(req, res);
|
|
114
|
+
if (!audio) return true;
|
|
115
|
+
if (audio.byteLength === 0) {
|
|
116
|
+
sendJson(res, 400, { error: "Missing audio" });
|
|
117
|
+
return true;
|
|
118
|
+
}
|
|
119
|
+
|
|
120
|
+
const abortController = new AbortController();
|
|
121
|
+
let completed = false;
|
|
122
|
+
let clientClosed = false;
|
|
123
|
+
const abortOnClose = () => {
|
|
124
|
+
clientClosed = true;
|
|
125
|
+
if (!completed && !abortController.signal.aborted) {
|
|
126
|
+
abortController.abort();
|
|
127
|
+
}
|
|
128
|
+
};
|
|
129
|
+
req.on("close", abortOnClose);
|
|
130
|
+
res.on("close", abortOnClose);
|
|
131
|
+
|
|
132
|
+
try {
|
|
133
|
+
const runtime = state.current;
|
|
134
|
+
if (!runtime) {
|
|
135
|
+
completed = true;
|
|
136
|
+
sendJson(res, 503, {
|
|
137
|
+
error: "Local inference TRANSCRIPTION is not available",
|
|
138
|
+
});
|
|
139
|
+
return true;
|
|
140
|
+
}
|
|
141
|
+
const { text, words } = await transcribeWavWithWords(
|
|
142
|
+
runtime,
|
|
143
|
+
audio,
|
|
144
|
+
abortController.signal,
|
|
145
|
+
);
|
|
146
|
+
completed = true;
|
|
147
|
+
sendJson(res, 200, { text, words });
|
|
148
|
+
} catch (err) {
|
|
149
|
+
if (!clientClosed && !abortController.signal.aborted && !isClosed(res)) {
|
|
150
|
+
sendJson(res, 502, {
|
|
151
|
+
error: `Local inference ASR error: ${
|
|
152
|
+
err instanceof Error ? err.message : String(err)
|
|
153
|
+
}`,
|
|
154
|
+
});
|
|
155
|
+
}
|
|
156
|
+
} finally {
|
|
157
|
+
completed = true;
|
|
158
|
+
req.off("close", abortOnClose);
|
|
159
|
+
res.off("close", abortOnClose);
|
|
160
|
+
}
|
|
161
|
+
|
|
162
|
+
return true;
|
|
163
|
+
}
|
|
@@ -0,0 +1,20 @@
|
|
|
1
|
+
import { type AgentRuntime } from "@elizaos/core";
|
|
2
|
+
import { type AsrWordTiming } from "../services/voice";
|
|
3
|
+
/** A transcript plus optional per-word timings (ms from audio start). */
|
|
4
|
+
export interface LocalAsrTranscript {
|
|
5
|
+
text: string;
|
|
6
|
+
words: AsrWordTiming[];
|
|
7
|
+
}
|
|
8
|
+
/**
|
|
9
|
+
* Transcribe WAV audio to text + per-word timings.
|
|
10
|
+
*
|
|
11
|
+
* The primary path is the single fused libelizainference FFI pipe
|
|
12
|
+
* (`transcribePcmTimed`, ASR ABI v12) — the cross-platform on-device runtime
|
|
13
|
+
* that emits per-word `[startMs,endMs)` timings the transcript player
|
|
14
|
+
* highlights against. When the in-process engine is not the active local
|
|
15
|
+
* backend (a device-bridge / capacitor / AOSP topology owns transcription),
|
|
16
|
+
* fall back to the `useModel` provider chain for the transcript text; word
|
|
17
|
+
* timings are unavailable on that path (`words: []`).
|
|
18
|
+
*/
|
|
19
|
+
export declare function transcribeWavWithWords(runtime: AgentRuntime, audioWav: Uint8Array, signal?: AbortSignal): Promise<LocalAsrTranscript>;
|
|
20
|
+
//# sourceMappingURL=local-inference-asr-transcribe.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"local-inference-asr-transcribe.d.ts","sourceRoot":"","sources":["local-inference-asr-transcribe.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,YAAY,EAAa,MAAM,eAAe,CAAC;AAE7D,OAAO,EAAE,KAAK,aAAa,EAAsB,MAAM,mBAAmB,CAAC;AAE3E,yEAAyE;AACzE,MAAM,WAAW,kBAAkB;IAClC,IAAI,EAAE,MAAM,CAAC;IACb,KAAK,EAAE,aAAa,EAAE,CAAC;CACvB;AA+BD;;;;;;;;;;GAUG;AACH,wBAAsB,sBAAsB,CAC3C,OAAO,EAAE,YAAY,EACrB,QAAQ,EAAE,UAAU,EACpB,MAAM,CAAC,EAAE,WAAW,GAClB,OAAO,CAAC,kBAAkB,CAAC,CAc7B"}
|
|
@@ -0,0 +1,118 @@
|
|
|
1
|
+
import { type AgentRuntime, ModelType } from "@elizaos/core";
|
|
2
|
+
import { afterEach, describe, expect, it, vi } from "vitest";
|
|
3
|
+
import { localInferenceEngine } from "../services/engine";
|
|
4
|
+
import { transcribeWavWithWords } from "./local-inference-asr-transcribe";
|
|
5
|
+
|
|
6
|
+
vi.mock("../services/engine", () => ({
|
|
7
|
+
localInferenceEngine: {
|
|
8
|
+
available: vi.fn(),
|
|
9
|
+
ensureActiveBundleVoiceReady: vi.fn(),
|
|
10
|
+
transcribePcmTimed: vi.fn(),
|
|
11
|
+
},
|
|
12
|
+
}));
|
|
13
|
+
|
|
14
|
+
const engine = vi.mocked(localInferenceEngine, true) as unknown as {
|
|
15
|
+
available: ReturnType<typeof vi.fn>;
|
|
16
|
+
ensureActiveBundleVoiceReady: ReturnType<typeof vi.fn>;
|
|
17
|
+
transcribePcmTimed: ReturnType<typeof vi.fn>;
|
|
18
|
+
};
|
|
19
|
+
|
|
20
|
+
/** Minimal mono PCM16 16 kHz WAV with four samples. */
|
|
21
|
+
function wavBytes(): Uint8Array {
|
|
22
|
+
const pcm = new Int16Array([0, 900, -900, 0]);
|
|
23
|
+
const buffer = new ArrayBuffer(44 + pcm.length * 2);
|
|
24
|
+
const view = new DataView(buffer);
|
|
25
|
+
const writeAscii = (offset: number, value: string) => {
|
|
26
|
+
for (let i = 0; i < value.length; i += 1) {
|
|
27
|
+
view.setUint8(offset + i, value.charCodeAt(i));
|
|
28
|
+
}
|
|
29
|
+
};
|
|
30
|
+
writeAscii(0, "RIFF");
|
|
31
|
+
view.setUint32(4, 36 + pcm.length * 2, true);
|
|
32
|
+
writeAscii(8, "WAVE");
|
|
33
|
+
writeAscii(12, "fmt ");
|
|
34
|
+
view.setUint32(16, 16, true);
|
|
35
|
+
view.setUint16(20, 1, true);
|
|
36
|
+
view.setUint16(22, 1, true);
|
|
37
|
+
view.setUint32(24, 16_000, true);
|
|
38
|
+
view.setUint32(28, 16_000 * 2, true);
|
|
39
|
+
view.setUint16(32, 2, true);
|
|
40
|
+
view.setUint16(34, 16, true);
|
|
41
|
+
writeAscii(36, "data");
|
|
42
|
+
view.setUint32(40, pcm.length * 2, true);
|
|
43
|
+
for (let i = 0; i < pcm.length; i += 1) {
|
|
44
|
+
view.setInt16(44 + i * 2, pcm[i] ?? 0, true);
|
|
45
|
+
}
|
|
46
|
+
return new Uint8Array(buffer);
|
|
47
|
+
}
|
|
48
|
+
|
|
49
|
+
describe("transcribeWavWithWords", () => {
|
|
50
|
+
afterEach(() => {
|
|
51
|
+
vi.clearAllMocks();
|
|
52
|
+
});
|
|
53
|
+
|
|
54
|
+
it("runs the fused FFI timed pipe when the in-process engine is active", async () => {
|
|
55
|
+
engine.available.mockResolvedValue(true);
|
|
56
|
+
engine.ensureActiveBundleVoiceReady.mockResolvedValue(undefined);
|
|
57
|
+
engine.transcribePcmTimed.mockResolvedValue({
|
|
58
|
+
text: " hello world ",
|
|
59
|
+
words: [
|
|
60
|
+
{ text: "hello", startMs: 0, endMs: 500 },
|
|
61
|
+
{ text: "world", startMs: 500, endMs: 1000 },
|
|
62
|
+
],
|
|
63
|
+
});
|
|
64
|
+
const runtime = {
|
|
65
|
+
useModel: vi.fn(),
|
|
66
|
+
} as unknown as AgentRuntime;
|
|
67
|
+
|
|
68
|
+
const result = await transcribeWavWithWords(runtime, wavBytes());
|
|
69
|
+
|
|
70
|
+
// useModel is never touched — the single FFI pipe owns this path.
|
|
71
|
+
expect(runtime.useModel).not.toHaveBeenCalled();
|
|
72
|
+
expect(engine.ensureActiveBundleVoiceReady).toHaveBeenCalledTimes(1);
|
|
73
|
+
// The WAV decoded to a Float32Array @ 16 kHz before the timed call.
|
|
74
|
+
const [audioArg] = engine.transcribePcmTimed.mock.calls[0] ?? [];
|
|
75
|
+
expect((audioArg as { pcm: Float32Array }).pcm).toBeInstanceOf(
|
|
76
|
+
Float32Array,
|
|
77
|
+
);
|
|
78
|
+
expect((audioArg as { sampleRate: number }).sampleRate).toBe(16_000);
|
|
79
|
+
expect(result).toEqual({
|
|
80
|
+
text: "hello world",
|
|
81
|
+
words: [
|
|
82
|
+
{ text: "hello", startMs: 0, endMs: 500 },
|
|
83
|
+
{ text: "world", startMs: 500, endMs: 1000 },
|
|
84
|
+
],
|
|
85
|
+
});
|
|
86
|
+
});
|
|
87
|
+
|
|
88
|
+
it("falls back to the useModel provider chain when the engine is inactive", async () => {
|
|
89
|
+
engine.available.mockResolvedValue(false);
|
|
90
|
+
const useModel = vi
|
|
91
|
+
.fn()
|
|
92
|
+
.mockRejectedValueOnce(
|
|
93
|
+
new Error("No handler found for delegate type: TRANSCRIPTION"),
|
|
94
|
+
)
|
|
95
|
+
.mockResolvedValueOnce({ text: "hello local voice" });
|
|
96
|
+
const runtime = { useModel } as unknown as AgentRuntime;
|
|
97
|
+
|
|
98
|
+
const result = await transcribeWavWithWords(runtime, wavBytes());
|
|
99
|
+
|
|
100
|
+
expect(useModel).toHaveBeenCalledTimes(2);
|
|
101
|
+
expect(useModel.mock.calls[0]?.[0]).toBe(ModelType.TRANSCRIPTION);
|
|
102
|
+
expect(useModel.mock.calls[0]?.[2]).toBe("eliza-local-inference");
|
|
103
|
+
expect(useModel.mock.calls[1]?.[2]).toBe("capacitor-llama");
|
|
104
|
+
// Fallback path carries no word timings (TRANSCRIPTION ⇒ string).
|
|
105
|
+
expect(result).toEqual({ text: "hello local voice", words: [] });
|
|
106
|
+
expect(engine.transcribePcmTimed).not.toHaveBeenCalled();
|
|
107
|
+
});
|
|
108
|
+
|
|
109
|
+
it("rejects an invalid transcript shape from the provider chain", async () => {
|
|
110
|
+
engine.available.mockResolvedValue(false);
|
|
111
|
+
const useModel = vi.fn().mockResolvedValue(42);
|
|
112
|
+
const runtime = { useModel } as unknown as AgentRuntime;
|
|
113
|
+
|
|
114
|
+
await expect(transcribeWavWithWords(runtime, wavBytes())).rejects.toThrow(
|
|
115
|
+
/invalid transcript/,
|
|
116
|
+
);
|
|
117
|
+
});
|
|
118
|
+
});
|