Yehowshua/ZIm
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Merge 5d433cbaa1 into 0e575b2809

Browse source
This commit is contained in:
沈瑗杰 2025-08-27 00:39:17 +08:00 • committed by GitHub
commit a20814771a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 15 additions and 15 deletions
Download patch file Download diff file Expand all files Collapse all files

18
crates/deepseek/src/deepseek.rs
View file

@ -48,10 +48,10 @@ impl From<Role> for String {
#[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))] #[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))]
#[derive(Clone, Debug, Default, Serialize, Deserialize, PartialEq)] #[derive(Clone, Debug, Default, Serialize, Deserialize, PartialEq)]
pub enum Model { pub enum Model {
#[serde(rename = "deepseek-chat")] #[serde(rename = "deepseek-v3.1")]
#[default] #[default]
Chat, Chat,
#[serde(rename = "deepseek-reasoner")] #[serde(rename = "deepseek-v3.1(think)")]
Reasoner, Reasoner,
#[serde(rename = "custom")] #[serde(rename = "custom")]
Custom { Custom {
@ -70,24 +70,24 @@ impl Model {
pub fn from_id(id: &str) -> Result<Self> { pub fn from_id(id: &str) -> Result<Self> {
match id { match id {
"deepseek-chat" => Ok(Self::Chat), "deepseek-v3.1" => Ok(Self::Chat),
"deepseek-reasoner" => Ok(Self::Reasoner), "deepseek-v3.1(think)" => Ok(Self::Reasoner),
_ => anyhow::bail!("invalid model id {id}"), _ => anyhow::bail!("invalid model id {id}"),
} }
} }
pub fn id(&self) -> &str { pub fn id(&self) -> &str {
match self { match self {
Self::Chat => "deepseek-chat", Self::Chat => "deepseek-v3.1",
Self::Reasoner => "deepseek-reasoner", Self::Reasoner => "deepseek-v3.1(think)",
Self::Custom { name, .. } => name, Self::Custom { name, .. } => name,
} }
} }
pub fn display_name(&self) -> &str { pub fn display_name(&self) -> &str {
match self { match self {
Self::Chat => "DeepSeek Chat", Self::Chat => "DeepSeek V3.1",
Self::Reasoner => "DeepSeek Reasoner", Self::Reasoner => "DeepSeek V3.1 (think)",
Self::Custom { Self::Custom {
name, display_name, .. name, display_name, ..
} => display_name.as_ref().unwrap_or(name).as_str(), } => display_name.as_ref().unwrap_or(name).as_str(),
@ -96,7 +96,7 @@ impl Model {
pub fn max_token_count(&self) -> u64 { pub fn max_token_count(&self) -> u64 {
match self { match self {
Self::Chat | Self::Reasoner => 64_000, Self::Chat | Self::Reasoner => 128_000,
Self::Custom { max_tokens, .. } => *max_tokens, Self::Custom { max_tokens, .. } => *max_tokens,
} }
} }

4
crates/language_models/src/provider/deepseek.rs
View file

@ -196,8 +196,8 @@ impl LanguageModelProvider for DeepSeekLanguageModelProvider {
fn provided_models(&self, cx: &App) -> Vec<Arc<dyn LanguageModel>> { fn provided_models(&self, cx: &App) -> Vec<Arc<dyn LanguageModel>> {
let mut models = BTreeMap::default(); let mut models = BTreeMap::default();
models.insert("deepseek-chat", deepseek::Model::Chat); models.insert("deepseek-v3.1", deepseek::Model::Chat);
models.insert("deepseek-reasoner", deepseek::Model::Reasoner); models.insert("deepseek-v3.1(think)", deepseek::Model::Reasoner);
for available_model in AllLanguageModelSettings::get_global(cx) for available_model in AllLanguageModelSettings::get_global(cx)
.deepseek .deepseek

8
docs/src/ai/llm-providers.md
View file

@ -183,13 +183,13 @@ If you wish to use alternate models or customize the API endpoint, you can do so
"api_url": "https://api.deepseek.com", "api_url": "https://api.deepseek.com",
"available_models": [ "available_models": [
{ {
"name": "deepseek-chat", "name": "deepseek-v3.1",
"display_name": "DeepSeek Chat", "display_name": "DeepSeek V3.1",
"max_tokens": 64000 "max_tokens": 64000
}, },
{ {
"name": "deepseek-reasoner", "name": "deepseek-v3.1(think)",
"display_name": "DeepSeek Reasoner", "display_name": "DeepSeek V3.1(think)",
"max_tokens": 64000, "max_tokens": 64000,
"max_output_tokens": 4096 "max_output_tokens": 4096
} }