chores

Author: yujonglee

crates/am/src/lib.rs

@@ -20,8 +20,8 @@ mod tests {
         client
             .init(InitRequest {
                 api_key: "".to_string(),
-                model: None,
-                model_repo: None,
+                model: Some("nvidia_parakeet-v2_476MB".to_string()),
+                model_repo: Some("argmaxinc/parakeetkit-pro".to_string()),
                 model_folder: None,
                 custom_vocabulary: Some(vec![]),
             })

crates/audio/src/lib.rs

@@ -93,6 +93,14 @@ impl AudioInput {
         name
     }
 
+    pub fn sample_rate(&self) -> u32 {
+        match &self.source {
+            AudioSource::RealtimeMic => self.mic.as_ref().unwrap().sample_rate(),
+            AudioSource::RealtimeSpeaker => self.speaker.as_ref().unwrap().sample_rate(),
+            AudioSource::Recorded => 16000,
+        }
+    }
+
     pub fn list_mic_devices() -> Vec<String> {
         let host = cpal::default_host();