From dc90a80468a347c86a861501ae5c4e75e5be5e8c Mon Sep 17 00:00:00 2001
From: t8y2 <1156263951@qq.com>
Date: Fri, 19 Jun 2026 11:36:29 +0800
Subject: [PATCH] fix(ai): enable temperature for Kimi models and default to
 1.0

---
 crates/dbx-core/src/ai.rs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/crates/dbx-core/src/ai.rs b/crates/dbx-core/src/ai.rs
index 4478bbef7..0495ef876 100644
--- a/crates/dbx-core/src/ai.rs
+++ b/crates/dbx-core/src/ai.rs
@@ -436,12 +436,13 @@ fn is_kimi_model(model: &str) -> bool {
 }
 
 pub fn supports_temperature(config: &AiConfig) -> bool {
-    !(is_openai_api_config(config) && is_openai_reasoning_model(&config.model)) && !is_kimi_model(&config.model)
+    !(is_openai_api_config(config) && is_openai_reasoning_model(&config.model))
 }
 
 pub fn add_temperature_if_supported(body: &mut serde_json::Value, request: &AiCompletionRequest) {
     if supports_temperature(&request.config) {
-        body["temperature"] = json!(request.temperature.unwrap_or(0.2));
+        let default_temp = if is_kimi_model(&request.config.model) { 1.0 } else { 0.2 };
+        body["temperature"] = json!(request.temperature.unwrap_or(default_temp));
     }
 }