@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,79 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Speaker-embedding encoder — shared constants, error class, and the
|
|
3
|
+
* embedding-distance helper.
|
|
4
|
+
*
|
|
5
|
+
* The speaker encoder runs EXCLUSIVELY through the fused `libelizainference`
|
|
6
|
+
* `eliza_inference_speaker_*` ABI (`FusedSpeakerEncoder` in `encoder-fused.ts`).
|
|
7
|
+
* The standalone `libvoice_classifier` binding that previously lived here has
|
|
8
|
+
* been removed — there is one on-device voice runtime.
|
|
9
|
+
*
|
|
10
|
+
* This module retains the pieces the fused path shares:
|
|
11
|
+
* - the canonical dims (`SPEAKER_GGML_*`), pinned at 256 to match the C-side
|
|
12
|
+
* `VOICE_SPEAKER_EMBEDDING_DIM` and the WeSpeaker ResNet34-LM head,
|
|
13
|
+
* - the structured `SpeakerEncoderGgmlUnavailableError` the fused encoder
|
|
14
|
+
* throws (no synthetic embedding fallback),
|
|
15
|
+
* - the pure `voiceSpeakerDistance` cosine-distance helper.
|
|
16
|
+
*/
|
|
17
|
+
|
|
18
|
+
/** Output embedding dim. Matches `VOICE_SPEAKER_EMBEDDING_DIM`. */
|
|
19
|
+
export const SPEAKER_GGML_EMBEDDING_DIM = 256;
|
|
20
|
+
|
|
21
|
+
/** Required input sample rate. */
|
|
22
|
+
export const SPEAKER_GGML_SAMPLE_RATE = 16_000;
|
|
23
|
+
|
|
24
|
+
/** Minimum useful audio window (~1.0 s). */
|
|
25
|
+
export const SPEAKER_GGML_MIN_SAMPLES = 16_000;
|
|
26
|
+
|
|
27
|
+
export class SpeakerEncoderGgmlUnavailableError extends Error {
|
|
28
|
+
readonly code:
|
|
29
|
+
| "native-missing"
|
|
30
|
+
| "library-missing"
|
|
31
|
+
| "model-missing"
|
|
32
|
+
| "model-load-failed"
|
|
33
|
+
| "model-shape-mismatch"
|
|
34
|
+
| "forward-not-implemented"
|
|
35
|
+
| "invalid-input";
|
|
36
|
+
constructor(
|
|
37
|
+
code: SpeakerEncoderGgmlUnavailableError["code"],
|
|
38
|
+
message: string,
|
|
39
|
+
) {
|
|
40
|
+
super(message);
|
|
41
|
+
this.name = "SpeakerEncoderGgmlUnavailableError";
|
|
42
|
+
this.code = code;
|
|
43
|
+
}
|
|
44
|
+
}
|
|
45
|
+
|
|
46
|
+
/**
|
|
47
|
+
* Cosine distance between two 256-dim speaker embeddings. Defined as
|
|
48
|
+
* `1 - cos_similarity(a, b)`, range [0, 2]. Mirrors the C-side
|
|
49
|
+
* `voice_speaker_distance` helper exactly.
|
|
50
|
+
*/
|
|
51
|
+
export function voiceSpeakerDistance(a: Float32Array, b: Float32Array): number {
|
|
52
|
+
if (a.length !== SPEAKER_GGML_EMBEDDING_DIM) {
|
|
53
|
+
throw new SpeakerEncoderGgmlUnavailableError(
|
|
54
|
+
"invalid-input",
|
|
55
|
+
`[speaker] left embedding has dim ${a.length}, expected ${SPEAKER_GGML_EMBEDDING_DIM}`,
|
|
56
|
+
);
|
|
57
|
+
}
|
|
58
|
+
if (b.length !== SPEAKER_GGML_EMBEDDING_DIM) {
|
|
59
|
+
throw new SpeakerEncoderGgmlUnavailableError(
|
|
60
|
+
"invalid-input",
|
|
61
|
+
`[speaker] right embedding has dim ${b.length}, expected ${SPEAKER_GGML_EMBEDDING_DIM}`,
|
|
62
|
+
);
|
|
63
|
+
}
|
|
64
|
+
let dot = 0;
|
|
65
|
+
let normA = 0;
|
|
66
|
+
let normB = 0;
|
|
67
|
+
for (let i = 0; i < SPEAKER_GGML_EMBEDDING_DIM; i += 1) {
|
|
68
|
+
const av = a[i];
|
|
69
|
+
const bv = b[i];
|
|
70
|
+
dot += av * bv;
|
|
71
|
+
normA += av * av;
|
|
72
|
+
normB += bv * bv;
|
|
73
|
+
}
|
|
74
|
+
if (normA <= 0 || normB <= 0) return 1;
|
|
75
|
+
let cosine = dot / (Math.sqrt(normA) * Math.sqrt(normB));
|
|
76
|
+
if (cosine > 1) cosine = 1;
|
|
77
|
+
if (cosine < -1) cosine = -1;
|
|
78
|
+
return 1 - cosine;
|
|
79
|
+
}
|
|
@@ -0,0 +1,37 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Speaker-embedding encoder — shared contract, model-id constants, and the
|
|
3
|
+
* `averageEmbeddings` centroid helper.
|
|
4
|
+
*
|
|
5
|
+
* The speaker encoder runs EXCLUSIVELY through the fused `libelizainference`
|
|
6
|
+
* `eliza_inference_speaker_*` ABI (`FusedSpeakerEncoder` in `encoder-fused.ts`).
|
|
7
|
+
* The standalone `WespeakerEncoder` (`libvoice_classifier`) binding has been
|
|
8
|
+
* removed — there is one on-device voice runtime.
|
|
9
|
+
*
|
|
10
|
+
* This module retains the cross-cutting pieces every encoder caller shares:
|
|
11
|
+
* - the `SpeakerEncoder` interface the fused encoder implements,
|
|
12
|
+
* - the stored model-id strings (kept stable so existing voice profiles in
|
|
13
|
+
* the database stay valid),
|
|
14
|
+
* - the canonical dims (re-exported from `encoder-ggml`),
|
|
15
|
+
* - the `SpeakerEncoderUnavailableError` the enrollment routes raise,
|
|
16
|
+
* - the pure `averageEmbeddings` centroid helper.
|
|
17
|
+
*/
|
|
18
|
+
export declare const WESPEAKER_RESNET34_LM_INT8_MODEL_ID: "wespeaker-resnet34-lm-int8";
|
|
19
|
+
export declare const WESPEAKER_RESNET34_LM_FP32_MODEL_ID: "wespeaker-resnet34-lm-fp32";
|
|
20
|
+
export type WespeakerModelId = typeof WESPEAKER_RESNET34_LM_INT8_MODEL_ID | typeof WESPEAKER_RESNET34_LM_FP32_MODEL_ID;
|
|
21
|
+
export declare const WESPEAKER_EMBEDDING_DIM = 256;
|
|
22
|
+
export declare const WESPEAKER_SAMPLE_RATE = 16000;
|
|
23
|
+
export declare const WESPEAKER_MIN_SAMPLES = 16000;
|
|
24
|
+
export declare class SpeakerEncoderUnavailableError extends Error {
|
|
25
|
+
readonly code: "native-missing" | "library-missing" | "model-missing" | "model-load-failed" | "model-shape-mismatch" | "forward-not-implemented" | "invalid-input";
|
|
26
|
+
constructor(code: SpeakerEncoderUnavailableError["code"], message: string);
|
|
27
|
+
}
|
|
28
|
+
/** The minimal contract every speaker encoder honors. */
|
|
29
|
+
export interface SpeakerEncoder {
|
|
30
|
+
readonly embeddingDim: number;
|
|
31
|
+
readonly sampleRate: number;
|
|
32
|
+
readonly modelId?: string;
|
|
33
|
+
encode(pcm: Float32Array): Promise<Float32Array>;
|
|
34
|
+
dispose(): Promise<void>;
|
|
35
|
+
}
|
|
36
|
+
export declare function averageEmbeddings(embeddings: readonly Float32Array[]): Float32Array;
|
|
37
|
+
//# sourceMappingURL=encoder.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"encoder.d.ts","sourceRoot":"","sources":["encoder.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;;GAgBG;AAeH,eAAO,MAAM,mCAAmC,EAC/C,4BAAqC,CAAC;AACvC,eAAO,MAAM,mCAAmC,EAC/C,4BAAqC,CAAC;AACvC,MAAM,MAAM,gBAAgB,GACzB,OAAO,mCAAmC,GAC1C,OAAO,mCAAmC,CAAC;AAE9C,eAAO,MAAM,uBAAuB,MAA6B,CAAC;AAClE,eAAO,MAAM,qBAAqB,QAA2B,CAAC;AAC9D,eAAO,MAAM,qBAAqB,QAA2B,CAAC;AAM9D,qBAAa,8BAA+B,SAAQ,KAAK;IACxD,QAAQ,CAAC,IAAI,EACV,gBAAgB,GAChB,iBAAiB,GACjB,eAAe,GACf,mBAAmB,GACnB,sBAAsB,GACtB,yBAAyB,GACzB,eAAe,CAAC;gBACP,IAAI,EAAE,8BAA8B,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,MAAM;CAKzE;AAMD,yDAAyD;AACzD,MAAM,WAAW,cAAc;IAC9B,QAAQ,CAAC,YAAY,EAAE,MAAM,CAAC;IAC9B,QAAQ,CAAC,UAAU,EAAE,MAAM,CAAC;IAC5B,QAAQ,CAAC,OAAO,CAAC,EAAE,MAAM,CAAC;IAC1B,MAAM,CAAC,GAAG,EAAE,YAAY,GAAG,OAAO,CAAC,YAAY,CAAC,CAAC;IACjD,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CACzB;AAMD,wBAAgB,iBAAiB,CAChC,UAAU,EAAE,SAAS,YAAY,EAAE,GACjC,YAAY,CAsBd"}
|
|
@@ -0,0 +1,105 @@
|
|
|
1
|
+
/**
|
|
2
|
+
* Speaker-embedding encoder — shared contract, model-id constants, and the
|
|
3
|
+
* `averageEmbeddings` centroid helper.
|
|
4
|
+
*
|
|
5
|
+
* The speaker encoder runs EXCLUSIVELY through the fused `libelizainference`
|
|
6
|
+
* `eliza_inference_speaker_*` ABI (`FusedSpeakerEncoder` in `encoder-fused.ts`).
|
|
7
|
+
* The standalone `WespeakerEncoder` (`libvoice_classifier`) binding has been
|
|
8
|
+
* removed — there is one on-device voice runtime.
|
|
9
|
+
*
|
|
10
|
+
* This module retains the cross-cutting pieces every encoder caller shares:
|
|
11
|
+
* - the `SpeakerEncoder` interface the fused encoder implements,
|
|
12
|
+
* - the stored model-id strings (kept stable so existing voice profiles in
|
|
13
|
+
* the database stay valid),
|
|
14
|
+
* - the canonical dims (re-exported from `encoder-ggml`),
|
|
15
|
+
* - the `SpeakerEncoderUnavailableError` the enrollment routes raise,
|
|
16
|
+
* - the pure `averageEmbeddings` centroid helper.
|
|
17
|
+
*/
|
|
18
|
+
|
|
19
|
+
import { normalizeVoiceEmbedding } from "../speaker-imprint";
|
|
20
|
+
import {
|
|
21
|
+
SPEAKER_GGML_EMBEDDING_DIM,
|
|
22
|
+
SPEAKER_GGML_MIN_SAMPLES,
|
|
23
|
+
SPEAKER_GGML_SAMPLE_RATE,
|
|
24
|
+
} from "./encoder-ggml";
|
|
25
|
+
|
|
26
|
+
// ---------------------------------------------------------------------------
|
|
27
|
+
// Stored model id constants.
|
|
28
|
+
// The model id strings are kept stable for compatibility with stored profiles
|
|
29
|
+
// (changing them would invalidate any existing voice profiles in the database).
|
|
30
|
+
// ---------------------------------------------------------------------------
|
|
31
|
+
|
|
32
|
+
export const WESPEAKER_RESNET34_LM_INT8_MODEL_ID =
|
|
33
|
+
"wespeaker-resnet34-lm-int8" as const;
|
|
34
|
+
export const WESPEAKER_RESNET34_LM_FP32_MODEL_ID =
|
|
35
|
+
"wespeaker-resnet34-lm-fp32" as const;
|
|
36
|
+
export type WespeakerModelId =
|
|
37
|
+
| typeof WESPEAKER_RESNET34_LM_INT8_MODEL_ID
|
|
38
|
+
| typeof WESPEAKER_RESNET34_LM_FP32_MODEL_ID;
|
|
39
|
+
|
|
40
|
+
export const WESPEAKER_EMBEDDING_DIM = SPEAKER_GGML_EMBEDDING_DIM;
|
|
41
|
+
export const WESPEAKER_SAMPLE_RATE = SPEAKER_GGML_SAMPLE_RATE;
|
|
42
|
+
export const WESPEAKER_MIN_SAMPLES = SPEAKER_GGML_MIN_SAMPLES;
|
|
43
|
+
|
|
44
|
+
// ---------------------------------------------------------------------------
|
|
45
|
+
// Structured error.
|
|
46
|
+
// ---------------------------------------------------------------------------
|
|
47
|
+
|
|
48
|
+
export class SpeakerEncoderUnavailableError extends Error {
|
|
49
|
+
readonly code:
|
|
50
|
+
| "native-missing"
|
|
51
|
+
| "library-missing"
|
|
52
|
+
| "model-missing"
|
|
53
|
+
| "model-load-failed"
|
|
54
|
+
| "model-shape-mismatch"
|
|
55
|
+
| "forward-not-implemented"
|
|
56
|
+
| "invalid-input";
|
|
57
|
+
constructor(code: SpeakerEncoderUnavailableError["code"], message: string) {
|
|
58
|
+
super(message);
|
|
59
|
+
this.name = "SpeakerEncoderUnavailableError";
|
|
60
|
+
this.code = code;
|
|
61
|
+
}
|
|
62
|
+
}
|
|
63
|
+
|
|
64
|
+
// ---------------------------------------------------------------------------
|
|
65
|
+
// SpeakerEncoder interface.
|
|
66
|
+
// ---------------------------------------------------------------------------
|
|
67
|
+
|
|
68
|
+
/** The minimal contract every speaker encoder honors. */
|
|
69
|
+
export interface SpeakerEncoder {
|
|
70
|
+
readonly embeddingDim: number;
|
|
71
|
+
readonly sampleRate: number;
|
|
72
|
+
readonly modelId?: string;
|
|
73
|
+
encode(pcm: Float32Array): Promise<Float32Array>;
|
|
74
|
+
dispose(): Promise<void>;
|
|
75
|
+
}
|
|
76
|
+
|
|
77
|
+
// ---------------------------------------------------------------------------
|
|
78
|
+
// averageEmbeddings — pure centroid helper.
|
|
79
|
+
// ---------------------------------------------------------------------------
|
|
80
|
+
|
|
81
|
+
export function averageEmbeddings(
|
|
82
|
+
embeddings: readonly Float32Array[],
|
|
83
|
+
): Float32Array {
|
|
84
|
+
if (embeddings.length === 0) {
|
|
85
|
+
throw new SpeakerEncoderUnavailableError(
|
|
86
|
+
"invalid-input",
|
|
87
|
+
"[wespeaker] averageEmbeddings called with no inputs",
|
|
88
|
+
);
|
|
89
|
+
}
|
|
90
|
+
const dim = embeddings[0].length;
|
|
91
|
+
const sum = new Float64Array(dim);
|
|
92
|
+
for (const emb of embeddings) {
|
|
93
|
+
if (emb.length !== dim) {
|
|
94
|
+
throw new SpeakerEncoderUnavailableError(
|
|
95
|
+
"invalid-input",
|
|
96
|
+
`[wespeaker] embedding dim mismatch: ${emb.length} vs ${dim}`,
|
|
97
|
+
);
|
|
98
|
+
}
|
|
99
|
+
for (let i = 0; i < dim; i += 1) sum[i] += emb[i];
|
|
100
|
+
}
|
|
101
|
+
const out = new Float32Array(dim);
|
|
102
|
+
for (let i = 0; i < dim; i += 1) out[i] = sum[i] / embeddings.length;
|
|
103
|
+
const normalized = normalizeVoiceEmbedding(out);
|
|
104
|
+
return Float32Array.from(normalized);
|
|
105
|
+
}
|
|
@@ -0,0 +1,83 @@
|
|
|
1
|
+
import type { VoiceInputSource, VoiceSegment, VoiceSpeaker } from "./types";
|
|
2
|
+
export declare const DEFAULT_VOICE_IMPRINT_MATCH_THRESHOLD = 0.78;
|
|
3
|
+
export interface VoiceImprintProfile {
|
|
4
|
+
id: string;
|
|
5
|
+
centroidEmbedding: ArrayLike<number> | null | undefined;
|
|
6
|
+
embeddingModel?: string | null;
|
|
7
|
+
sampleCount?: number | null;
|
|
8
|
+
confidence?: number | null;
|
|
9
|
+
label?: string | null;
|
|
10
|
+
displayName?: string | null;
|
|
11
|
+
entityId?: string | null;
|
|
12
|
+
sourceKind?: string | null;
|
|
13
|
+
sourceScopeId?: string | null;
|
|
14
|
+
metadata?: Record<string, unknown> | null;
|
|
15
|
+
}
|
|
16
|
+
export interface VoiceImprintMatch {
|
|
17
|
+
profile: VoiceImprintProfile;
|
|
18
|
+
similarity: number;
|
|
19
|
+
confidence: number;
|
|
20
|
+
}
|
|
21
|
+
export interface VoiceImprintCentroidUpdate {
|
|
22
|
+
centroidEmbedding: number[];
|
|
23
|
+
sampleCount: number;
|
|
24
|
+
confidence: number;
|
|
25
|
+
}
|
|
26
|
+
export interface VoiceImprintObservationInput {
|
|
27
|
+
id: string;
|
|
28
|
+
segmentId?: string;
|
|
29
|
+
text: string;
|
|
30
|
+
startMs: number;
|
|
31
|
+
endMs: number;
|
|
32
|
+
embedding: ArrayLike<number>;
|
|
33
|
+
embeddingModel?: string | null;
|
|
34
|
+
confidence?: number | null;
|
|
35
|
+
source?: VoiceInputSource;
|
|
36
|
+
metadata?: Record<string, unknown> | null;
|
|
37
|
+
}
|
|
38
|
+
export interface AttributedVoiceObservation {
|
|
39
|
+
observation: VoiceImprintObservationInput;
|
|
40
|
+
match: VoiceImprintMatch | null;
|
|
41
|
+
speaker: VoiceSpeaker | null;
|
|
42
|
+
segment: VoiceSegment;
|
|
43
|
+
}
|
|
44
|
+
export interface SpeakerAttributionResult {
|
|
45
|
+
observations: AttributedVoiceObservation[];
|
|
46
|
+
segments: VoiceSegment[];
|
|
47
|
+
primarySpeaker?: VoiceSpeaker;
|
|
48
|
+
summary: {
|
|
49
|
+
totalObservations: number;
|
|
50
|
+
matchedObservations: number;
|
|
51
|
+
unmatchedObservations: number;
|
|
52
|
+
meanConfidence: number;
|
|
53
|
+
meanSimilarity: number | null;
|
|
54
|
+
};
|
|
55
|
+
}
|
|
56
|
+
export declare function normalizeVoiceEmbedding(embedding: ArrayLike<number>): number[];
|
|
57
|
+
export declare function cosineSimilarity(left: ArrayLike<number>, right: ArrayLike<number>): number;
|
|
58
|
+
export declare function updateVoiceImprintCentroid(args: {
|
|
59
|
+
centroidEmbedding?: ArrayLike<number> | null;
|
|
60
|
+
sampleCount?: number | null;
|
|
61
|
+
confidence?: number | null;
|
|
62
|
+
observationEmbedding: ArrayLike<number>;
|
|
63
|
+
observationConfidence?: number | null;
|
|
64
|
+
maxObservationWeight?: number;
|
|
65
|
+
}): VoiceImprintCentroidUpdate;
|
|
66
|
+
export declare function matchVoiceImprint(args: {
|
|
67
|
+
embedding: ArrayLike<number>;
|
|
68
|
+
profiles: readonly VoiceImprintProfile[];
|
|
69
|
+
embeddingModel?: string | null;
|
|
70
|
+
threshold?: number;
|
|
71
|
+
}): VoiceImprintMatch | null;
|
|
72
|
+
export declare function voiceSpeakerFromImprintMatch(args: {
|
|
73
|
+
match: VoiceImprintMatch;
|
|
74
|
+
source?: VoiceInputSource;
|
|
75
|
+
observationId?: string;
|
|
76
|
+
}): VoiceSpeaker;
|
|
77
|
+
export declare function attributeVoiceImprintObservations(args: {
|
|
78
|
+
observations: readonly VoiceImprintObservationInput[];
|
|
79
|
+
profiles: readonly VoiceImprintProfile[];
|
|
80
|
+
threshold?: number;
|
|
81
|
+
defaultSource?: VoiceInputSource;
|
|
82
|
+
}): SpeakerAttributionResult;
|
|
83
|
+
//# sourceMappingURL=speaker-imprint.d.ts.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"speaker-imprint.d.ts","sourceRoot":"","sources":["speaker-imprint.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,gBAAgB,EAAE,YAAY,EAAE,YAAY,EAAE,MAAM,SAAS,CAAC;AAE5E,eAAO,MAAM,qCAAqC,OAAO,CAAC;AAE1D,MAAM,WAAW,mBAAmB;IACnC,EAAE,EAAE,MAAM,CAAC;IACX,iBAAiB,EAAE,SAAS,CAAC,MAAM,CAAC,GAAG,IAAI,GAAG,SAAS,CAAC;IACxD,cAAc,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,KAAK,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACtB,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,QAAQ,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACzB,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,aAAa,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC9B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;CAC1C;AAED,MAAM,WAAW,iBAAiB;IACjC,OAAO,EAAE,mBAAmB,CAAC;IAC7B,UAAU,EAAE,MAAM,CAAC;IACnB,UAAU,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,0BAA0B;IAC1C,iBAAiB,EAAE,MAAM,EAAE,CAAC;IAC5B,WAAW,EAAE,MAAM,CAAC;IACpB,UAAU,EAAE,MAAM,CAAC;CACnB;AAED,MAAM,WAAW,4BAA4B;IAC5C,EAAE,EAAE,MAAM,CAAC;IACX,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,IAAI,EAAE,MAAM,CAAC;IACb,OAAO,EAAE,MAAM,CAAC;IAChB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;IAC7B,cAAc,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,MAAM,CAAC,EAAE,gBAAgB,CAAC;IAC1B,QAAQ,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,IAAI,CAAC;CAC1C;AAED,MAAM,WAAW,0BAA0B;IAC1C,WAAW,EAAE,4BAA4B,CAAC;IAC1C,KAAK,EAAE,iBAAiB,GAAG,IAAI,CAAC;IAChC,OAAO,EAAE,YAAY,GAAG,IAAI,CAAC;IAC7B,OAAO,EAAE,YAAY,CAAC;CACtB;AAED,MAAM,WAAW,wBAAwB;IACxC,YAAY,EAAE,0BAA0B,EAAE,CAAC;IAC3C,QAAQ,EAAE,YAAY,EAAE,CAAC;IACzB,cAAc,CAAC,EAAE,YAAY,CAAC;IAC9B,OAAO,EAAE;QACR,iBAAiB,EAAE,MAAM,CAAC;QAC1B,mBAAmB,EAAE,MAAM,CAAC;QAC5B,qBAAqB,EAAE,MAAM,CAAC;QAC9B,cAAc,EAAE,MAAM,CAAC;QACvB,cAAc,EAAE,MAAM,GAAG,IAAI,CAAC;KAC9B,CAAC;CACF;AAOD,wBAAgB,uBAAuB,CACtC,SAAS,EAAE,SAAS,CAAC,MAAM,CAAC,GAC1B,MAAM,EAAE,CAiBV;AAED,wBAAgB,gBAAgB,CAC/B,IAAI,EAAE,SAAS,CAAC,MAAM,CAAC,EACvB,KAAK,EAAE,SAAS,CAAC,MAAM,CAAC,GACtB,MAAM,CAWR;AAED,wBAAgB,0BAA0B,CAAC,IAAI,EAAE;IAChD,iBAAiB,CAAC,EAAE,SAAS,CAAC,MAAM,CAAC,GAAG,IAAI,CAAC;IAC7C,WAAW,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC5B,UAAU,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC3B,oBAAoB,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;IACxC,qBAAqB,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IACtC,oBAAoB,CAAC,EAAE,MAAM,CAAC;CAC9B,GAAG,0BAA0B,CA+B7B;AAED,wBAAgB,iBAAiB,CAAC,IAAI,EAAE;IACvC,SAAS,EAAE,SAAS,CAAC,MAAM,CAAC,CAAC;IAC7B,QAAQ,EAAE,SAAS,mBAAmB,EAAE,CAAC;IACzC,cAAc,CAAC,EAAE,MAAM,GAAG,IAAI,CAAC;IAC/B,SAAS,CAAC,EAAE,MAAM,CAAC;CACnB,GAAG,iBAAiB,GAAG,IAAI,CA+B3B;AAED,wBAAgB,4BAA4B,CAAC,IAAI,EAAE;IAClD,KAAK,EAAE,iBAAiB,CAAC;IACzB,MAAM,CAAC,EAAE,gBAAgB,CAAC;IAC1B,aAAa,CAAC,EAAE,MAAM,CAAC;CACvB,GAAG,YAAY,CAqBf;AAED,wBAAgB,iCAAiC,CAAC,IAAI,EAAE;IACvD,YAAY,EAAE,SAAS,4BAA4B,EAAE,CAAC;IACtD,QAAQ,EAAE,SAAS,mBAAmB,EAAE,CAAC;IACzC,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,aAAa,CAAC,EAAE,gBAAgB,CAAC;CACjC,GAAG,wBAAwB,CA2E3B"}
|
|
@@ -0,0 +1,185 @@
|
|
|
1
|
+
import { describe, expect, it } from "vitest";
|
|
2
|
+
import {
|
|
3
|
+
attributeVoiceImprintObservations,
|
|
4
|
+
cosineSimilarity,
|
|
5
|
+
matchVoiceImprint,
|
|
6
|
+
normalizeVoiceEmbedding,
|
|
7
|
+
updateVoiceImprintCentroid,
|
|
8
|
+
type VoiceImprintProfile,
|
|
9
|
+
voiceSpeakerFromImprintMatch,
|
|
10
|
+
} from "./speaker-imprint";
|
|
11
|
+
|
|
12
|
+
describe("speaker-imprint", () => {
|
|
13
|
+
it("normalizes embeddings and computes cosine similarity", () => {
|
|
14
|
+
const normalized = normalizeVoiceEmbedding([3, 4]);
|
|
15
|
+
expect(normalized[0]).toBeCloseTo(0.6, 6);
|
|
16
|
+
expect(normalized[1]).toBeCloseTo(0.8, 6);
|
|
17
|
+
expect(cosineSimilarity([2, 0], [4, 0])).toBeCloseTo(1, 6);
|
|
18
|
+
expect(cosineSimilarity([1, 0], [0, 1])).toBeCloseTo(0, 6);
|
|
19
|
+
});
|
|
20
|
+
|
|
21
|
+
it("matches the nearest compatible voice imprint above threshold", () => {
|
|
22
|
+
const profiles: VoiceImprintProfile[] = [
|
|
23
|
+
{
|
|
24
|
+
id: "cluster-a",
|
|
25
|
+
label: "Owner",
|
|
26
|
+
centroidEmbedding: [1, 0],
|
|
27
|
+
embeddingModel: "eliza-voice-embed-v1",
|
|
28
|
+
confidence: 0.9,
|
|
29
|
+
entityId: "entity-owner",
|
|
30
|
+
},
|
|
31
|
+
{
|
|
32
|
+
id: "cluster-b",
|
|
33
|
+
label: "Guest",
|
|
34
|
+
centroidEmbedding: [0, 1],
|
|
35
|
+
embeddingModel: "eliza-voice-embed-v1",
|
|
36
|
+
confidence: 0.9,
|
|
37
|
+
},
|
|
38
|
+
];
|
|
39
|
+
|
|
40
|
+
const match = matchVoiceImprint({
|
|
41
|
+
embedding: [0.98, 0.05],
|
|
42
|
+
embeddingModel: "eliza-voice-embed-v1",
|
|
43
|
+
profiles,
|
|
44
|
+
threshold: 0.8,
|
|
45
|
+
});
|
|
46
|
+
|
|
47
|
+
expect(match?.profile.id).toBe("cluster-a");
|
|
48
|
+
expect(match?.similarity).toBeGreaterThan(0.99);
|
|
49
|
+
if (!match) throw new Error("expected voice imprint match");
|
|
50
|
+
const speaker = voiceSpeakerFromImprintMatch({
|
|
51
|
+
match,
|
|
52
|
+
observationId: "obs-1",
|
|
53
|
+
source: { kind: "local_mic", deviceId: "mic-1" },
|
|
54
|
+
});
|
|
55
|
+
expect(speaker.entityId).toBe("entity-owner");
|
|
56
|
+
expect(speaker.imprintClusterId).toBe("cluster-a");
|
|
57
|
+
expect(speaker.imprintObservationId).toBe("obs-1");
|
|
58
|
+
expect(speaker.source?.kind).toBe("local_mic");
|
|
59
|
+
});
|
|
60
|
+
|
|
61
|
+
it("does not match across embedding-model or dimension mismatches", () => {
|
|
62
|
+
const profiles: VoiceImprintProfile[] = [
|
|
63
|
+
{
|
|
64
|
+
id: "cluster-a",
|
|
65
|
+
centroidEmbedding: [1, 0],
|
|
66
|
+
embeddingModel: "other-model",
|
|
67
|
+
},
|
|
68
|
+
{
|
|
69
|
+
id: "cluster-b",
|
|
70
|
+
centroidEmbedding: [1, 0, 0],
|
|
71
|
+
embeddingModel: "eliza-voice-embed-v1",
|
|
72
|
+
},
|
|
73
|
+
];
|
|
74
|
+
|
|
75
|
+
expect(
|
|
76
|
+
matchVoiceImprint({
|
|
77
|
+
embedding: [1, 0],
|
|
78
|
+
embeddingModel: "eliza-voice-embed-v1",
|
|
79
|
+
profiles,
|
|
80
|
+
}),
|
|
81
|
+
).toBeNull();
|
|
82
|
+
});
|
|
83
|
+
|
|
84
|
+
it("updates a centroid with weighted observations", () => {
|
|
85
|
+
const first = updateVoiceImprintCentroid({
|
|
86
|
+
observationEmbedding: [10, 0],
|
|
87
|
+
observationConfidence: 0.8,
|
|
88
|
+
});
|
|
89
|
+
expect(first.centroidEmbedding).toEqual([1, 0]);
|
|
90
|
+
expect(first.sampleCount).toBe(1);
|
|
91
|
+
expect(first.confidence).toBeCloseTo(0.8, 6);
|
|
92
|
+
|
|
93
|
+
const second = updateVoiceImprintCentroid({
|
|
94
|
+
centroidEmbedding: first.centroidEmbedding,
|
|
95
|
+
sampleCount: first.sampleCount,
|
|
96
|
+
confidence: first.confidence,
|
|
97
|
+
observationEmbedding: [0, 10],
|
|
98
|
+
observationConfidence: 0.5,
|
|
99
|
+
});
|
|
100
|
+
expect(second.sampleCount).toBe(2);
|
|
101
|
+
expect(second.centroidEmbedding[0]).toBeGreaterThan(
|
|
102
|
+
second.centroidEmbedding[1],
|
|
103
|
+
);
|
|
104
|
+
expect(second.confidence).toBeCloseTo(0.65, 6);
|
|
105
|
+
});
|
|
106
|
+
|
|
107
|
+
it("attributes diarized observation embeddings without granting synthesis or identity authority", () => {
|
|
108
|
+
const profiles: VoiceImprintProfile[] = [
|
|
109
|
+
{
|
|
110
|
+
id: "cluster-owner",
|
|
111
|
+
label: "Owner",
|
|
112
|
+
displayName: "Owner",
|
|
113
|
+
entityId: "entity-owner",
|
|
114
|
+
centroidEmbedding: [1, 0],
|
|
115
|
+
embeddingModel: "eliza-voice-embed-v1",
|
|
116
|
+
confidence: 0.9,
|
|
117
|
+
},
|
|
118
|
+
{
|
|
119
|
+
id: "cluster-guest",
|
|
120
|
+
label: "Guest",
|
|
121
|
+
entityId: "entity-guest",
|
|
122
|
+
centroidEmbedding: [0, 1],
|
|
123
|
+
embeddingModel: "eliza-voice-embed-v1",
|
|
124
|
+
confidence: 0.8,
|
|
125
|
+
},
|
|
126
|
+
];
|
|
127
|
+
const source = {
|
|
128
|
+
kind: "local_mic" as const,
|
|
129
|
+
deviceId: "default-input",
|
|
130
|
+
roomId: "room-1",
|
|
131
|
+
};
|
|
132
|
+
|
|
133
|
+
const result = attributeVoiceImprintObservations({
|
|
134
|
+
defaultSource: source,
|
|
135
|
+
profiles,
|
|
136
|
+
threshold: 0.8,
|
|
137
|
+
observations: [
|
|
138
|
+
{
|
|
139
|
+
id: "obs-owner-1",
|
|
140
|
+
segmentId: "seg-owner-1",
|
|
141
|
+
text: "I am the owner",
|
|
142
|
+
startMs: 0,
|
|
143
|
+
endMs: 1200,
|
|
144
|
+
embedding: [0.99, 0.02],
|
|
145
|
+
embeddingModel: "eliza-voice-embed-v1",
|
|
146
|
+
},
|
|
147
|
+
{
|
|
148
|
+
id: "obs-guest-1",
|
|
149
|
+
segmentId: "seg-guest-1",
|
|
150
|
+
text: "and I am a guest",
|
|
151
|
+
startMs: 1300,
|
|
152
|
+
endMs: 2100,
|
|
153
|
+
embedding: [0.02, 0.99],
|
|
154
|
+
embeddingModel: "eliza-voice-embed-v1",
|
|
155
|
+
},
|
|
156
|
+
],
|
|
157
|
+
});
|
|
158
|
+
|
|
159
|
+
expect(result.summary).toMatchObject({
|
|
160
|
+
totalObservations: 2,
|
|
161
|
+
matchedObservations: 2,
|
|
162
|
+
unmatchedObservations: 0,
|
|
163
|
+
});
|
|
164
|
+
expect(result.primarySpeaker?.entityId).toBe("entity-owner");
|
|
165
|
+
expect(result.segments.map((segment) => segment.speaker?.entityId)).toEqual(
|
|
166
|
+
["entity-owner", "entity-guest"],
|
|
167
|
+
);
|
|
168
|
+
expect(result.segments[0].metadata).toMatchObject({
|
|
169
|
+
attributionOnly: true,
|
|
170
|
+
evidenceKind: "voice_imprint_attribution",
|
|
171
|
+
identityAuthority: false,
|
|
172
|
+
synthesisAuthorization: false,
|
|
173
|
+
diarizationMode: "attribution_only",
|
|
174
|
+
imprintClusterId: "cluster-owner",
|
|
175
|
+
imprintObservationId: "obs-owner-1",
|
|
176
|
+
entityId: "entity-owner",
|
|
177
|
+
});
|
|
178
|
+
expect(result.segments[0].speaker?.metadata).toMatchObject({
|
|
179
|
+
attributionOnly: true,
|
|
180
|
+
evidenceKind: "voice_imprint_attribution",
|
|
181
|
+
identityAuthority: false,
|
|
182
|
+
synthesisAuthorization: false,
|
|
183
|
+
});
|
|
184
|
+
});
|
|
185
|
+
});
|