SidneyZhang/QuiCommit
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 2 Wiki Activity

feat(deepseek): 添加 DeepSeek reasoning 模式支持

Browse Source
This commit is contained in:
Sidney Zhang
2026-05-26 16:27:49 +08:00
parent 3a57d25a76
commit 1063369d96
6 changed files with 103 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Cargo.toml
View File

@@ -1,6 +1,6 @@
[package]
name = "quicommit"
version = "0.2.0"
version = "0.2.1"
edition = "2024"
authors = ["Sidney Zhang <zly@lyzhang.me>"]
description = "A powerful Git assistant tool with AI-powered commit/tag/changelog generation(alpha version)"

2
src/commands/config.rs
View File

@@ -993,7 +993,7 @@ impl ConfigCommand {
"deepseek" => {
println!("DeepSeek models:");
println!(" deepseek-chat");
println!(" deepseek-coder");
println!(" deepseek-reasoner");
}
"openrouter" => {
println!("OpenRouter models (examples):");

5
src/config/mod.rs
View File

@@ -110,6 +110,10 @@ pub struct LlmConfig {
/// API key (stored in config for fallback, encrypted if encrypt_sensitive is true)
#[serde(default)]
pub api_key: Option<String>,
/// Enable thinking/reasoning mode (deepseek, kimi)
#[serde(default)]
pub thinking_enabled: bool,
}
fn default_api_key_storage() -> String {
@@ -127,6 +131,7 @@ impl Default for LlmConfig {
timeout: default_timeout(),
api_key_storage: default_api_key_storage(),
api_key: None,
thinking_enabled: false,
}
}
}

34
src/llm/deepseek.rs
View File

@@ -10,6 +10,7 @@ pub struct DeepSeekClient {
api_key: String,
model: String,
client: reqwest::Client,
thinking_enabled: bool,
}
#[derive(Debug, Serialize)]
@@ -21,6 +22,14 @@ struct ChatCompletionRequest {
#[serde(skip_serializing_if = "Option::is_none")]
temperature: Option<f32>,
stream: bool,
#[serde(skip_serializing_if = "Option::is_none")]
thinking: Option<ThinkingConfig>,
}
#[derive(Debug, Serialize)]
struct ThinkingConfig {
#[serde(rename = "type")]
thinking_type: String,
}
#[derive(Debug, Serialize, Deserialize)]
@@ -37,6 +46,8 @@ struct ChatCompletionResponse {
#[derive(Debug, Deserialize)]
struct Choice {
message: Message,
#[serde(default)]
reasoning_content: Option<String>,
}
#[derive(Debug, Deserialize)]
@@ -57,10 +68,11 @@ impl DeepSeekClient {
let client = create_http_client(Duration::from_secs(60))?;
Ok(Self {
base_url: "https://api.deepseek.com/v1".to_string(),
base_url: "https://api.deepseek.com/".to_string(),
api_key: api_key.to_string(),
model: model.to_string(),
client,
thinking_enabled: false,
})
}
@@ -73,6 +85,7 @@ impl DeepSeekClient {
api_key: api_key.to_string(),
model: model.to_string(),
client,
thinking_enabled: false,
})
}
@@ -82,6 +95,12 @@ impl DeepSeekClient {
Ok(self)
}
/// Enable or disable thinking mode
pub fn with_thinking(mut self, enabled: bool) -> Self {
self.thinking_enabled = enabled;
self
}
/// List available models
pub async fn list_models(&self) -> Result<Vec<String>> {
let url = format!("{}/models", self.base_url);
@@ -177,12 +196,21 @@ impl DeepSeekClient {
async fn chat_completion(&self, messages: Vec<Message>) -> Result<String> {
let url = format!("{}/chat/completions", self.base_url);
let thinking = if self.thinking_enabled {
Some(ThinkingConfig {
thinking_type: "enabled".to_string(),
})
} else {
None
};
let request = ChatCompletionRequest {
model: self.model.clone(),
messages,
max_tokens: Some(500),
temperature: Some(0.7),
stream: false,
thinking,
};
let response = self.client
@@ -199,7 +227,6 @@ impl DeepSeekClient {
if !status.is_success() {
let text = response.text().await.unwrap_or_default();
// Try to parse error
if let Ok(error) = serde_json::from_str::<ErrorResponse>(&text) {
bail!("DeepSeek API error: {} ({})", error.error.message, error.error.error_type);
}
@@ -223,7 +250,7 @@ impl DeepSeekClient {
/// Available DeepSeek models
pub const DEEPSEEK_MODELS: &[&str] = &[
"deepseek-chat",
"deepseek-coder",
"deepseek-reasoner",
];
/// Check if a model name is valid
@@ -238,6 +265,7 @@ mod tests {
#[test]
fn test_model_validation() {
assert!(is_valid_model("deepseek-chat"));
assert!(is_valid_model("deepseek-reasoner"));
assert!(!is_valid_model("invalid-model"));
}
}

31
src/llm/kimi.rs
View File

@@ -10,6 +10,7 @@ pub struct KimiClient {
api_key: String,
model: String,
client: reqwest::Client,
thinking_enabled: bool,
}
#[derive(Debug, Serialize)]
@@ -21,6 +22,14 @@ struct ChatCompletionRequest {
#[serde(skip_serializing_if = "Option::is_none")]
temperature: Option<f32>,
stream: bool,
#[serde(skip_serializing_if = "Option::is_none")]
thinking: Option<ThinkingConfig>,
}
#[derive(Debug, Serialize)]
struct ThinkingConfig {
#[serde(rename = "type")]
thinking_type: String,
}
#[derive(Debug, Serialize, Deserialize)]
@@ -37,6 +46,8 @@ struct ChatCompletionResponse {
#[derive(Debug, Deserialize)]
struct Choice {
message: Message,
#[serde(default)]
reasoning_content: Option<String>,
}
#[derive(Debug, Deserialize)]
@@ -61,6 +72,7 @@ impl KimiClient {
api_key: api_key.to_string(),
model: model.to_string(),
client,
thinking_enabled: false,
})
}
@@ -73,6 +85,7 @@ impl KimiClient {
api_key: api_key.to_string(),
model: model.to_string(),
client,
thinking_enabled: false,
})
}
@@ -82,6 +95,12 @@ impl KimiClient {
Ok(self)
}
/// Enable or disable thinking mode
pub fn with_thinking(mut self, enabled: bool) -> Self {
self.thinking_enabled = enabled;
self
}
/// List available models
pub async fn list_models(&self) -> Result<Vec<String>> {
let url = format!("{}/models", self.base_url);
@@ -177,12 +196,21 @@ impl KimiClient {
async fn chat_completion(&self, messages: Vec<Message>) -> Result<String> {
let url = format!("{}/chat/completions", self.base_url);
let thinking = if self.thinking_enabled {
Some(ThinkingConfig {
thinking_type: "enabled".to_string(),
})
} else {
None
};
let request = ChatCompletionRequest {
model: self.model.clone(),
messages,
max_tokens: Some(500),
temperature: Some(0.7),
temperature: Some(1.0),
stream: false,
thinking,
};
let response = self.client
@@ -199,7 +227,6 @@ impl KimiClient {
if !status.is_success() {
let text = response.text().await.unwrap_or_default();
// Try to parse error
if let Ok(error) = serde_json::from_str::<ErrorResponse>(&text) {
bail!("Kimi API error: {} ({})", error.error.message, error.error.error_type);
}

8
src/llm/mod.rs
View File

@@ -44,6 +44,7 @@ pub struct LlmClientConfig {
pub max_tokens: u32,
pub temperature: f32,
pub timeout: Duration,
pub thinking_enabled: bool,
}
impl Default for LlmClientConfig {
@@ -52,6 +53,7 @@ impl Default for LlmClientConfig {
max_tokens: 500,
temperature: 0.7,
timeout: Duration::from_secs(30),
thinking_enabled: false,
}
}
}
@@ -64,12 +66,14 @@ impl LlmClient {
max_tokens: config.llm.max_tokens,
temperature: config.llm.temperature,
timeout: Duration::from_secs(config.llm.timeout),
thinking_enabled: config.llm.thinking_enabled,
};
let provider = config.llm.provider.as_str();
let model = config.llm.model.as_str();
let base_url = manager.llm_base_url();
let api_key = manager.get_api_key();
let thinking_enabled = config.llm.thinking_enabled;
let provider: Box<dyn LlmProvider> = match provider {
"ollama" => {
@@ -88,12 +92,12 @@ impl LlmClient {
"kimi" => {
let key = api_key.as_ref()
.ok_or_else(|| anyhow::anyhow!("Kimi API key not configured"))?;
Box::new(KimiClient::with_base_url(key, model, &base_url)?)
Box::new(KimiClient::with_base_url(key, model, &base_url)?.with_thinking(thinking_enabled))
}
"deepseek" => {
let key = api_key.as_ref()
.ok_or_else(|| anyhow::anyhow!("DeepSeek API key not configured"))?;
Box::new(DeepSeekClient::with_base_url(key, model, &base_url)?)
Box::new(DeepSeekClient::with_base_url(key, model, &base_url)?.with_thinking(thinking_enabled))
}
"openrouter" => {
let key = api_key.as_ref()