diff --git a/src-tauri/src/services/stream_check.rs b/src-tauri/src/services/stream_check.rs index 3fd7d97cf..b8ce157e1 100644 --- a/src-tauri/src/services/stream_check.rs +++ b/src-tauri/src/services/stream_check.rs @@ -3,7 +3,6 @@ //! 使用流式 API 进行快速健康检查,只需接收首个 chunk 即判定成功。 use futures::StreamExt; -use regex::Regex; use reqwest::Client; use serde::{Deserialize, Serialize}; use serde_json::json; @@ -1412,10 +1411,11 @@ impl StreamCheckService { return None; } - let re = Regex::new(r#"^model\s*=\s*["']([^"']+)["']"#).ok()?; - re.captures(config_text) - .and_then(|caps| caps.get(1)) - .map(|m| m.as_str().trim().to_string()) + let table = toml::from_str::(config_text).ok()?; + table + .get("model") + .and_then(|v| v.as_str()) + .map(|s| s.trim().to_string()) .filter(|value| !value.is_empty()) }