From 01f89ac4a9ca65e104e295c1bd5eb27f4e2edfdb Mon Sep 17 00:00:00 2001
From: TaoXieSZ <497592613@qq.com>
Date: Thu, 6 Feb 2025 22:22:33 +0800
Subject: [PATCH] compatible with local deepseek-r1

---
 src-tauri/src/ai_translator.rs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/src-tauri/src/ai_translator.rs b/src-tauri/src/ai_translator.rs
index fa08723..715b270 100644
--- a/src-tauri/src/ai_translator.rs
+++ b/src-tauri/src/ai_translator.rs
@@ -212,7 +212,8 @@ pub async fn translate_with_gpt(app: &AppHandle, original: &str) -> Result<Strin
 
     let client = Client::new();
 
-    let request_body = if settings.model_type == "deepseek-R1" {
+    let request_body = if settings.model_type.to_lowercase().starts_with("deepseek-r1") || 
+                                settings.custom_model.model_name.to_lowercase().starts_with("deepseek-r1") {
         json!({
             "model": model_config.model_name,
             "messages": [
@@ -249,7 +250,7 @@ pub async fn translate_with_gpt(app: &AppHandle, original: &str) -> Result<Strin
             "frequency_penalty": -0.3
         })
     };
-
+    
     let response = match client
         .post(&model_config.api_url)
         .header("Content-Type", "application/json")