fix: improve error handling in ASR audio processing and result retrieval

Author: L-jasmine

src/ai/bailian/realtime_asr.rs

@@ -157,7 +157,7 @@ impl ParaformerRealtimeV2Asr {
 
     pub async fn next_result(&mut self) -> anyhow::Result<Option<ResponsePayloadOutputSentence>> {
         while let Some(message) = self.websocket.next().await {
-            match message? {
+            match message.map_err(|e| anyhow::anyhow!("Paraformer ASR WebSocket error: {}", e))? {
                 reqwest_websocket::Message::Binary(_) => {
                     log::debug!("Received unexpected binary message");
                 }

src/services/ws.rs

@@ -457,7 +457,9 @@ async fn get_paraformer_v2_text(
                 ClientMsg::AudioChunk(data) => {
                     samples.extend_from_slice(&data);
                     if let Some(asr) = asr.as_mut() {
-                        asr.send_audio(data).await?;
+                        asr.send_audio(data).await.map_err(|e| {
+                            anyhow::anyhow!("`{id}` error sending paraformer asr audio: {e}")
+                        })?;
                     }
                 }
                 ClientMsg::Submit => {
@@ -474,7 +476,9 @@ async fn get_paraformer_v2_text(
                         )
                         .await?;
 
-                    paraformer_asr.start_pcm_recognition().await?;
+                    paraformer_asr.start_pcm_recognition().await.map_err(|e| {
+                        anyhow::anyhow!("`{id}` error starting paraformer asr: {e}")
+                    })?;
                     asr = Some(paraformer_asr);
                     continue;
                 }
@@ -498,9 +502,15 @@ async fn get_paraformer_v2_text(
                 log::error!("`{id}` error writing asr file {id}: {e}");
             };
             samples.clear();
-            asr.finish_task().await?;
+            asr.finish_task()
+                .await
+                .map_err(|e| anyhow::anyhow!("`{id}` error finishing paraformer asr task: {e}"))?;
             let mut text = String::new();
-            while let Some(sentence) = asr.next_result().await? {
+            while let Some(sentence) = asr
+                .next_result()
+                .await
+                .map_err(|e| anyhow::anyhow!("`{id}` error getting paraformer asr result: {e}"))?
+            {
                 if sentence.sentence_end {
                     text = sentence.text;
                     log::info!("ASR final result: {:?}", text);