Add UniFFI interface for livekit-wakeword

.cargo/config.toml

@@ -2,7 +2,7 @@
 rustflags = ["-C", "target-feature=+crt-static"]
 
 [target.aarch64-pc-windows-msvc]
-rustflags = ["-C", "target-feature=+crt-static"]
+rustflags = []
 
 [target.x86_64-apple-darwin]
 rustflags = ["-C", "link-args=-ObjC"]

.changeset/add_uniffi_interface_for_livekit_wakeword.md

@@ -0,0 +1,10 @@
+---
+livekit-uniffi: minor
+---
+
+# Add UniFFI interface for livekit-wakeword
+
+## Summary
+- Expose `WakeWordDetector` as a UniFFI Object wrapping `WakeWordModel` with `Mutex` for interior mutability
+- Export `new`, `load_model`, and `predict` methods across FFI
+- Use `#[uniffi::remote(Error)]` with `#[uniffi(flat_error)]` for `WakeWordError`, matching the existing `AccessTokenError` pattern

Cargo.toml

@@ -46,6 +46,7 @@ livekit-api = { version = "0.4.14", path = "livekit-api" }
 livekit-ffi = { version = "0.12.48", path = "livekit-ffi" }
 livekit-protocol = { version = "0.7.1", path = "livekit-protocol" }
 livekit-runtime = { version = "0.4.0", path = "livekit-runtime" }
+livekit-wakeword = { version = "0.1.0", path = "livekit-wakeword" }
 soxr-sys = { version = "0.1.2", path = "soxr-sys" }
 webrtc-sys = { version = "0.3.23", path = "webrtc-sys" }
 webrtc-sys-build = { version = "0.3.13", path = "webrtc-sys/build" }

livekit-uniffi/Cargo.toml

@@ -11,9 +11,14 @@ repository.workspace = true
 readme = "README.md"
 publish = false
 
+[features]
+default = ["wakeword"]
+wakeword = ["livekit-wakeword"]
+
 [dependencies]
 livekit-protocol = { workspace = true }
 livekit-api = { workspace = true }
+livekit-wakeword = { workspace = true, optional = true }
 uniffi = { version = "0.30.0", features = ["cli", "scaffolding-ffi-buffer-fns"] }
 log = { workspace = true }
 tokio = { workspace = true, features = ["sync"] }

livekit-uniffi/src/lib.rs

@@ -21,4 +21,8 @@ pub mod log_forward;
 /// Information about the build such as version.
 pub mod build_info;
 
+/// Wake word detection from [`livekit-wakeword`].
+#[cfg(feature = "wakeword")]
+pub mod wakeword;
+
 uniffi::setup_scaffolding!();

livekit-uniffi/src/wakeword.rs

@@ -0,0 +1,74 @@
+// Copyright 2026 LiveKit, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+use livekit_wakeword::{WakeWordError, WakeWordModel};
+use std::collections::HashMap;
+use std::sync::Mutex;
+
+#[uniffi::remote(Error)]
+#[uniffi(flat_error)]
+pub enum WakeWordError {
+    Ort,
+    Shape,
+    Io,
+    ModelNotFound,
+    UnsupportedSampleRate,
+    Resample,
+}
+
+/// Wake word detector backed by ONNX classifier models.
+///
+/// Wraps [`livekit_wakeword::WakeWordModel`] for use across FFI boundaries.
+/// Uses interior mutability since the underlying model requires `&mut self`.
+#[derive(uniffi::Object)]
+pub struct WakeWordDetector {
+    inner: Mutex<WakeWordModel>,
+}
+
+#[uniffi::export]
+impl WakeWordDetector {
+    /// Create a new wake word detector.
+    ///
+    /// `model_paths` are filesystem paths to ONNX classifier models.
+    /// `sample_rate` is the sample rate of audio that will be passed to
+    /// [`predict`](Self::predict). Supported rates: 16000 (recommended),
+    /// 22050, 32000, 44100, 48000, 88200, 96000, 176400, 192000, 384000 Hz.
+    #[uniffi::constructor]
+    pub fn new(model_paths: Vec<String>, sample_rate: u32) -> Result<Self, WakeWordError> {
+        let model = WakeWordModel::new(&model_paths, sample_rate)?;
+        Ok(Self { inner: Mutex::new(model) })
+    }
+
+    /// Load an additional wake word classifier ONNX model from disk.
+    ///
+    /// If `model_name` is `None`, the file stem is used as the classifier name.
+    pub fn load_model(
+        &self,
+        model_path: String,
+        model_name: Option<String>,
+    ) -> Result<(), WakeWordError> {
+        let mut inner = self.inner.lock().unwrap();
+        inner.load_model(&model_path, model_name.as_deref())?;
+        Ok(())
+    }
+
+    /// Get wake word predictions for an audio chunk.
+    ///
+    /// Pass ~2 seconds of i16 PCM audio at the sample rate configured in
+    /// [`new`](Self::new). Returns a map of classifier name to confidence score.
+    pub fn predict(&self, audio_chunk: Vec<i16>) -> Result<HashMap<String, f32>, WakeWordError> {
+        let mut inner = self.inner.lock().unwrap();
+        Ok(inner.predict(&audio_chunk)?)
+    }
+}

livekit-wakeword/Cargo.toml

@@ -7,7 +7,7 @@ license.workspace = true
 
 [dependencies]
 ndarray = "0.17.2"
-ort = { version = "2.0.0-rc.11", default-features = false, features = ["ndarray", "std"] }
+ort = { version = "2.0.0-rc.11", default-features = false, features = ["ndarray", "std", "download-binaries", "tls-native"] }
 resampler = "0.4"
 thiserror = "2"