From f13539896383fa0ab1f4f15df94394c42a63c9b0 Mon Sep 17 00:00:00 2001
From: coso <wutongci@hotmail.com>
Date: Sat, 31 Jan 2026 11:48:18 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E5=A4=8D=E6=B5=8B=E8=AF=95?=
 =?UTF-8?q?=E4=BB=A3=E7=A0=81=E7=BC=96=E8=AF=91=E9=94=99=E8=AF=AF=20-=20?=
 =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E7=BC=BA=E5=A4=B1=E7=9A=84=20asr=20=E5=AD=97?=
 =?UTF-8?q?=E6=AE=B5=E5=92=8C=20import?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src-tauri/src/config/tests.rs                          | 2 ++
 src-tauri/src/config/types.rs                          | 1 +
 src-tauri/src/services/general_chat/session_service.rs | 1 +
 src-tauri/src/websocket/tests.rs                       | 1 +
 4 files changed, 5 insertions(+)

diff --git a/src-tauri/src/config/tests.rs b/src-tauri/src/config/tests.rs
index ba5533df3..55c75ed2d 100644
--- a/src-tauri/src/config/tests.rs
+++ b/src-tauri/src/config/tests.rs
@@ -1228,6 +1228,7 @@ fn arb_credential_pool_config() -> impl Strategy<Value = CredentialPoolConfig> {
                 gemini_api_keys: vec![],
                 vertex_api_keys: vec![],
                 codex: vec![],
+                asr: vec![],
             },
         )
 }
@@ -2245,6 +2246,7 @@ fn arb_extended_credential_pool_config() -> impl Strategy<Value = CredentialPool
                     gemini_api_keys,
                     vertex_api_keys,
                     codex,
+                    asr: vec![],
                 }
             },
         )
diff --git a/src-tauri/src/config/types.rs b/src-tauri/src/config/types.rs
index 3b7679e94..d37394051 100644
--- a/src-tauri/src/config/types.rs
+++ b/src-tauri/src/config/types.rs
@@ -1749,6 +1749,7 @@ mod unit_tests {
             gemini_api_keys: vec![],
             vertex_api_keys: vec![],
             codex: vec![],
+            asr: vec![],
         };
 
         let yaml = serde_yaml::to_string(&pool).unwrap();
diff --git a/src-tauri/src/services/general_chat/session_service.rs b/src-tauri/src/services/general_chat/session_service.rs
index a2baeae7d..8ed0bdb85 100644
--- a/src-tauri/src/services/general_chat/session_service.rs
+++ b/src-tauri/src/services/general_chat/session_service.rs
@@ -186,6 +186,7 @@ impl SessionService {
 #[cfg(test)]
 mod tests {
     use super::*;
+    use crate::services::general_chat::MessageRole;
 
     #[test]
     fn test_create_session_with_name() {
diff --git a/src-tauri/src/websocket/tests.rs b/src-tauri/src/websocket/tests.rs
index 8f6284312..26a2ee38f 100644
--- a/src-tauri/src/websocket/tests.rs
+++ b/src-tauri/src/websocket/tests.rs
@@ -2,6 +2,7 @@
 
 #![allow(dead_code)]
 
+use super::types::{WsConnectionStatus, WsErrorCode};
 use super::*;
 
 #[test]