yenru0/zed
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
ebf342c60b4b6b13d7b375ba6676bc8435dcdbcd
zed/crates/opencode/src/opencode.rs
T
Xiaobo LiuBennet Bo FennerBennet Bo Fenner
 1469507d1b deepseek: Update DeepSeek Flash to V4.1 (#64014) 
# Objective

deepseek API: https://api-docs.deepseek.com/updates/

## Solution

deepseek rename deepseek v4 flash to deepseek flash



https://github.com/user-attachments/assets/72c473c9-a4bf-49e0-a7d6-0217a6dcc65b


---

Release Notes:

- agent: Added support for DeepSeek Flash 4.1

---------

Signed-off-by: Xiaobo Liu <cppcoffee@gmail.com>
Co-authored-by: Bennet Bo Fenner <bennetbo@gmx.de>
Co-authored-by: Bennet Bo Fenner <bennet@zed.dev>
2026-09-16 09:49:13 +00:00

986 lines
32 KiB
Rust
Raw Blame History

use anyhow::{Result, anyhow};
use futures::{AsyncBufReadExt, AsyncReadExt, StreamExt, io::BufReader, stream::BoxStream};
use http_client::{
AsyncBody, CustomHeaders, HttpClient, Method, Request as HttpRequest, RequestBuilderExt,
};
use language_model_core::ReasoningEffort;
use serde::{Deserialize, Serialize};
use strum::EnumIter;
pub const OPENCODE_API_URL: &str = "https://opencode.ai/zen";
#[derive(Clone, Copy, Debug, Default, PartialEq, Eq, Serialize, Deserialize)]
#[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))]
#[serde(rename_all = "snake_case")]
pub enum ApiProtocol {
#[default]
Anthropic,
OpenAiResponses,
OpenAiChat,
Google,
}
#[derive(Clone, Copy, Debug, PartialEq, Eq, Hash, Serialize, Deserialize)]
#[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))]
#[serde(rename_all = "snake_case")]
pub enum OpenCodeSubscription {
Zen,
Go,
}
impl OpenCodeSubscription {
pub fn display_name(&self) -> &'static str {
match self {
Self::Zen => "Zen",
Self::Go => "Go",
}
}
pub fn id_prefix(&self) -> &'static str {
match self {
Self::Zen => "zen",
Self::Go => "go",
}
}
pub fn api_path_suffix(&self) -> &'static str {
match self {
Self::Zen => "",
Self::Go => "/go",
}
}
}
#[cfg_attr(feature = "schemars", derive(schemars::JsonSchema))]
#[derive(Clone, Debug, Default, Serialize, Deserialize, PartialEq, EnumIter)]
pub enum Model {
// -- Anthropic protocol models --
#[serde(rename = "claude-opus-5")]
ClaudeOpus5,
#[serde(rename = "claude-opus-4-8")]
ClaudeOpus4_8,
#[serde(rename = "claude-opus-4-7")]
ClaudeOpus4_7,
#[serde(rename = "claude-opus-4-6")]
ClaudeOpus4_6,
#[serde(rename = "claude-opus-4-5")]
ClaudeOpus4_5,
#[default]
#[serde(rename = "claude-sonnet-4-6")]
ClaudeSonnet4_6,
#[serde(rename = "claude-sonnet-4-5")]
ClaudeSonnet4_5,
#[serde(rename = "claude-sonnet-4")]
ClaudeSonnet4,
#[serde(rename = "claude-haiku-4-5")]
ClaudeHaiku4_5,
#[serde(rename = "claude-sonnet-5")]
ClaudeSonnet5,
#[serde(rename = "claude-fable-5")]
ClaudeFable5,
// -- OpenAI Responses API models --
#[serde(rename = "gpt-5.6-sol")]
Gpt5_6Sol,
#[serde(rename = "gpt-5.6-terra")]
Gpt5_6Terra,
#[serde(rename = "gpt-5.6-luna")]
Gpt5_6Luna,
#[serde(rename = "gpt-5.5")]
Gpt5_5,
#[serde(rename = "gpt-5.5-pro")]
Gpt5_5Pro,
#[serde(rename = "gpt-5.4")]
Gpt5_4,
#[serde(rename = "gpt-5.4-pro")]
Gpt5_4Pro,
#[serde(rename = "gpt-5.4-mini")]
Gpt5_4Mini,
#[serde(rename = "gpt-5.4-nano")]
Gpt5_4Nano,
#[serde(rename = "gpt-5.3-codex")]
Gpt5_3Codex,
#[serde(rename = "gpt-5.3-codex-spark")]
Gpt5_3Spark,
#[serde(rename = "gpt-5.2")]
Gpt5_2,
#[serde(rename = "gpt-5.2-codex")]
Gpt5_2Codex,
#[serde(rename = "gpt-5.1")]
Gpt5_1,
#[serde(rename = "gpt-5.1-codex")]
Gpt5_1Codex,
#[serde(rename = "gpt-5.1-codex-max")]
Gpt5_1CodexMax,
#[serde(rename = "gpt-5.1-codex-mini")]
Gpt5_1CodexMini,
#[serde(rename = "gpt-5")]
Gpt5,
#[serde(rename = "gpt-5-codex")]
Gpt5Codex,
#[serde(rename = "gpt-5-nano")]
Gpt5Nano,
// -- Google protocol models --
#[serde(rename = "gemini-3.1-pro")]
Gemini3_1Pro,
#[serde(rename = "gemini-3-flash")]
Gemini3Flash,
#[serde(rename = "gemini-3.5-flash-lite")]
Gemini3_5FlashLite,
#[serde(rename = "gemini-3.5-flash")]
Gemini3_5Flash,
#[serde(rename = "gemini-3.6-flash")]
Gemini3_6Flash,
#[serde(rename = "gemini-3.7-flash")]
Gemini3_7Flash,
// -- OpenAI Chat Completions protocol models --
#[serde(rename = "deepseek-v4-pro")]
DeepSeekV4Pro,
#[serde(rename = "deepseek-flash")]
DeepSeekV4_1Flash,
#[serde(rename = "minimax-m2.5")]
MiniMaxM2_5,
#[serde(rename = "glm-5")]
Glm5,
#[serde(rename = "glm-5.1")]
Glm5_1,
#[serde(rename = "glm-5.2")]
Glm5_2,
#[serde(rename = "glm-5.3")]
Glm5_3,
#[serde(rename = "grok-build-0.1")]
GrokBuild0_1,
#[serde(rename = "grok-4.5")]
Grok4_5,
#[serde(rename = "grok-4.6")]
Grok4_6,
#[serde(rename = "muse-spark-1.2")]
MuseSpark1_2,
#[serde(rename = "kimi-k2.5")]
KimiK2_5,
#[serde(rename = "kimi-k2.6")]
KimiK2_6,
#[serde(rename = "kimi-k2.7-code")]
KimiK2_7Code,
#[serde(rename = "kimi-k3")]
KimiK3,
#[serde(rename = "minimax-m2.7")]
MiniMaxM2_7,
#[serde(rename = "minimax-m3")]
MiniMaxM3,
#[serde(rename = "mimo-v2.5-pro")]
MimoV2_5Pro,
#[serde(rename = "mimo-v2.5")]
MimoV2_5,
#[serde(rename = "qwen3.5-plus")]
Qwen3_5Plus,
#[serde(rename = "qwen3.6-plus")]
Qwen3_6Plus,
#[serde(rename = "qwen3.7-plus")]
Qwen3_7Plus,
#[serde(rename = "qwen3.7-max")]
Qwen3_7Max,
#[serde(rename = "qwen3.8-max")]
Qwen3_8Max,
#[serde(rename = "hy3")]
Hy3,
// -- Custom model --
#[serde(rename = "custom")]
Custom {
name: String,
display_name: Option<String>,
max_tokens: u64,
max_output_tokens: Option<u64>,
protocol: ApiProtocol,
reasoning_effort_levels: Option<Vec<ReasoningEffort>>,
custom_model_api_url: Option<String>,
interleaved_reasoning: bool,
},
}
impl Model {
pub fn default_fast() -> Self {
Self::ClaudeHaiku4_5
}
pub fn default_go() -> Self {
Self::KimiK2_6
}
pub fn default_go_fast() -> Self {
Self::MiniMaxM2_7
}
pub fn available_subscriptions(&self) -> &'static [OpenCodeSubscription] {
match self {
// Models available in both Zen and Go
Self::Glm5_1
| Self::Glm5_2
| Self::Grok4_5
| Self::KimiK2_6
| Self::KimiK2_7Code
| Self::KimiK3
| Self::MiniMaxM2_7
| Self::MiniMaxM3
| Self::DeepSeekV4Pro
| Self::Gpt5_6Luna
| Self::Qwen3_6Plus => &[OpenCodeSubscription::Zen, OpenCodeSubscription::Go],
// Go-only models
Self::MimoV2_5Pro
| Self::MimoV2_5
| Self::Glm5_3
| Self::Qwen3_7Plus
| Self::Qwen3_7Max
| Self::Qwen3_8Max
| Self::DeepSeekV4_1Flash
| Self::Hy3 => &[OpenCodeSubscription::Go],
// Deprecated on Go (per models.dev); still offered on Zen
Self::Glm5 | Self::KimiK2_5 | Self::MiniMaxM2_5 | Self::Qwen3_5Plus => {
&[OpenCodeSubscription::Zen]
}
// Custom models get their subscription from settings, not from here
Self::Custom { .. } => &[],
// All other built-in models are Zen-only
_ => &[OpenCodeSubscription::Zen],
}
}
pub fn id(&self) -> &str {
match self {
Self::ClaudeOpus5 => "claude-opus-5",
Self::ClaudeOpus4_8 => "claude-opus-4-8",
Self::ClaudeOpus4_7 => "claude-opus-4-7",
Self::ClaudeOpus4_6 => "claude-opus-4-6",
Self::ClaudeOpus4_5 => "claude-opus-4-5",
Self::ClaudeSonnet4_6 => "claude-sonnet-4-6",
Self::ClaudeSonnet4_5 => "claude-sonnet-4-5",
Self::ClaudeSonnet4 => "claude-sonnet-4",
Self::ClaudeHaiku4_5 => "claude-haiku-4-5",
Self::ClaudeSonnet5 => "claude-sonnet-5",
Self::ClaudeFable5 => "claude-fable-5",
Self::Gpt5_6Sol => "gpt-5.6-sol",
Self::Gpt5_6Terra => "gpt-5.6-terra",
Self::Gpt5_6Luna => "gpt-5.6-luna",
Self::Gpt5_5 => "gpt-5.5",
Self::Gpt5_5Pro => "gpt-5.5-pro",
Self::Gpt5_4 => "gpt-5.4",
Self::Gpt5_4Pro => "gpt-5.4-pro",
Self::Gpt5_4Mini => "gpt-5.4-mini",
Self::Gpt5_4Nano => "gpt-5.4-nano",
Self::Gpt5_3Codex => "gpt-5.3-codex",
Self::Gpt5_3Spark => "gpt-5.3-codex-spark",
Self::Gpt5_2 => "gpt-5.2",
Self::Gpt5_2Codex => "gpt-5.2-codex",
Self::Gpt5_1 => "gpt-5.1",
Self::Gpt5_1Codex => "gpt-5.1-codex",
Self::Gpt5_1CodexMax => "gpt-5.1-codex-max",
Self::Gpt5_1CodexMini => "gpt-5.1-codex-mini",
Self::Gpt5 => "gpt-5",
Self::Gpt5Codex => "gpt-5-codex",
Self::Gpt5Nano => "gpt-5-nano",
Self::Gemini3_1Pro => "gemini-3.1-pro",
Self::Gemini3Flash => "gemini-3-flash",
Self::Gemini3_5FlashLite => "gemini-3.5-flash-lite",
Self::Gemini3_5Flash => "gemini-3.5-flash",
Self::Gemini3_6Flash => "gemini-3.6-flash",
Self::Gemini3_7Flash => "gemini-3.7-flash",
Self::DeepSeekV4Pro => "deepseek-v4-pro",
Self::DeepSeekV4_1Flash => "deepseek-flash",
Self::MiniMaxM2_5 => "minimax-m2.5",
Self::Glm5 => "glm-5",
Self::Glm5_1 => "glm-5.1",
Self::Glm5_2 => "glm-5.2",
Self::Glm5_3 => "glm-5.3",
Self::GrokBuild0_1 => "grok-build-0.1",
Self::Grok4_5 => "grok-4.5",
Self::Grok4_6 => "grok-4.6",
Self::MuseSpark1_2 => "muse-spark-1.2",
Self::KimiK2_5 => "kimi-k2.5",
Self::KimiK2_6 => "kimi-k2.6",
Self::KimiK2_7Code => "kimi-k2.7-code",
Self::KimiK3 => "kimi-k3",
Self::MiniMaxM2_7 => "minimax-m2.7",
Self::MiniMaxM3 => "minimax-m3",
Self::MimoV2_5Pro => "mimo-v2.5-pro",
Self::MimoV2_5 => "mimo-v2.5",
Self::Qwen3_5Plus => "qwen3.5-plus",
Self::Qwen3_6Plus => "qwen3.6-plus",
Self::Qwen3_7Plus => "qwen3.7-plus",
Self::Qwen3_7Max => "qwen3.7-max",
Self::Qwen3_8Max => "qwen3.8-max",
Self::Hy3 => "hy3",
Self::Custom { name, .. } => name,
}
}
pub fn display_name(&self) -> &str {
match self {
Self::ClaudeOpus5 => "Claude Opus 5",
Self::ClaudeOpus4_8 => "Claude Opus 4.8",
Self::ClaudeOpus4_7 => "Claude Opus 4.7",
Self::ClaudeOpus4_6 => "Claude Opus 4.6",
Self::ClaudeOpus4_5 => "Claude Opus 4.5",
Self::ClaudeSonnet4_6 => "Claude Sonnet 4.6",
Self::ClaudeSonnet4_5 => "Claude Sonnet 4.5",
Self::ClaudeSonnet4 => "Claude Sonnet 4",
Self::ClaudeHaiku4_5 => "Claude Haiku 4.5",
Self::ClaudeSonnet5 => "Claude Sonnet 5",
Self::ClaudeFable5 => "Claude Fable 5",
Self::Gpt5_6Sol => "GPT 5.6 Sol",
Self::Gpt5_6Terra => "GPT 5.6 Terra",
Self::Gpt5_6Luna => "GPT 5.6 Luna",
Self::Gpt5_5 => "GPT 5.5",
Self::Gpt5_5Pro => "GPT 5.5 Pro",
Self::Gpt5_4 => "GPT 5.4",
Self::Gpt5_4Pro => "GPT 5.4 Pro",
Self::Gpt5_4Mini => "GPT 5.4 Mini",
Self::Gpt5_4Nano => "GPT 5.4 Nano",
Self::Gpt5_3Codex => "GPT 5.3 Codex",
Self::Gpt5_3Spark => "GPT 5.3 Codex Spark",
Self::Gpt5_2 => "GPT 5.2",
Self::Gpt5_2Codex => "GPT 5.2 Codex",
Self::Gpt5_1 => "GPT 5.1",
Self::Gpt5_1Codex => "GPT 5.1 Codex",
Self::Gpt5_1CodexMax => "GPT 5.1 Codex Max",
Self::Gpt5_1CodexMini => "GPT 5.1 Codex Mini",
Self::Gpt5 => "GPT 5",
Self::Gpt5Codex => "GPT 5 Codex",
Self::Gpt5Nano => "GPT 5 Nano",
Self::Gemini3_1Pro => "Gemini 3.1 Pro",
Self::Gemini3Flash => "Gemini 3 Flash",
Self::Gemini3_5FlashLite => "Gemini 3.5 Flash Lite",
Self::Gemini3_5Flash => "Gemini 3.5 Flash",
Self::Gemini3_6Flash => "Gemini 3.6 Flash",
Self::Gemini3_7Flash => "Gemini 3.7 Flash",
Self::DeepSeekV4Pro => "DeepSeek V4 Pro",
Self::DeepSeekV4_1Flash => "DeepSeek V4.1 Flash",
Self::MiniMaxM2_5 => "MiniMax M2.5",
Self::Glm5 => "GLM 5",
Self::Glm5_1 => "GLM 5.1",
Self::Glm5_2 => "GLM 5.2",
Self::Glm5_3 => "GLM 5.3",
Self::GrokBuild0_1 => "Grok Build 0.1",
Self::Grok4_5 => "Grok 4.5",
Self::Grok4_6 => "Grok 4.6",
Self::MuseSpark1_2 => "Muse Spark 1.2",
Self::KimiK2_5 => "Kimi K2.5",
Self::KimiK2_6 => "Kimi K2.6",
Self::KimiK2_7Code => "Kimi K2.7 Code",
Self::KimiK3 => "Kimi K3",
Self::MiniMaxM2_7 => "MiniMax M2.7",
Self::MiniMaxM3 => "MiniMax M3",
Self::MimoV2_5Pro => "MiMo V2.5 Pro",
Self::MimoV2_5 => "MiMo V2.5",
Self::Qwen3_5Plus => "Qwen3.5 Plus",
Self::Qwen3_6Plus => "Qwen3.6 Plus",
Self::Qwen3_7Plus => "Qwen3.7 Plus",
Self::Qwen3_7Max => "Qwen3.7 Max",
Self::Qwen3_8Max => "Qwen3.8 Max",
Self::Hy3 => "Hy3",
Self::Custom {
name, display_name, ..
} => display_name.as_deref().unwrap_or(name),
}
}
pub fn protocol(&self, subscription: OpenCodeSubscription) -> ApiProtocol {
match self {
// Models offered by OpenCode have the same configuration across subscriptions
// with one outlier: non-free MiniMax models
Self::MiniMaxM3 | Self::MiniMaxM2_7 | Self::MiniMaxM2_5 => {
if subscription == OpenCodeSubscription::Zen {
ApiProtocol::OpenAiChat
} else {
ApiProtocol::Anthropic
}
}
Self::ClaudeFable5
| Self::ClaudeOpus5
| Self::ClaudeOpus4_8
| Self::ClaudeOpus4_7
| Self::ClaudeOpus4_6
| Self::ClaudeOpus4_5
| Self::ClaudeSonnet5
| Self::ClaudeSonnet4_6
| Self::ClaudeSonnet4_5
| Self::ClaudeSonnet4
| Self::ClaudeHaiku4_5 => ApiProtocol::Anthropic,
Self::Gpt5_6Sol
| Self::Gpt5_6Terra
| Self::Gpt5_6Luna
| Self::Gpt5_5
| Self::Gpt5_5Pro
| Self::Gpt5_4
| Self::Gpt5_4Pro
| Self::Gpt5_4Mini
| Self::Gpt5_4Nano
| Self::Gpt5_3Codex
| Self::Gpt5_3Spark
| Self::Gpt5_2
| Self::Gpt5_2Codex
| Self::Gpt5_1
| Self::Gpt5_1Codex
| Self::Gpt5_1CodexMax
| Self::Gpt5_1CodexMini
| Self::Gpt5
| Self::Gpt5Codex
| Self::Gpt5Nano => ApiProtocol::OpenAiResponses,
Self::Gemini3_1Pro
| Self::Gemini3Flash
| Self::Gemini3_5FlashLite
| Self::Gemini3_5Flash
| Self::Gemini3_6Flash
| Self::Gemini3_7Flash => ApiProtocol::Google,
Self::Qwen3_8Max
| Self::Qwen3_7Max
| Self::Qwen3_7Plus
| Self::Qwen3_6Plus
| Self::Qwen3_5Plus => ApiProtocol::Anthropic,
Self::Glm5
| Self::Glm5_1
| Self::Glm5_2
| Self::Glm5_3
| Self::GrokBuild0_1
| Self::KimiK2_5
| Self::KimiK2_6
| Self::KimiK2_7Code
| Self::KimiK3
| Self::MimoV2_5Pro
| Self::MimoV2_5
| Self::DeepSeekV4Pro
| Self::DeepSeekV4_1Flash
| Self::Hy3 => ApiProtocol::OpenAiChat,
Self::Grok4_6 | Self::Grok4_5 | Self::MuseSpark1_2 => ApiProtocol::OpenAiResponses,
Self::Custom { protocol, .. } => *protocol,
}
}
pub fn interleaved_reasoning(&self) -> bool {
match self {
Self::DeepSeekV4Pro
| Self::DeepSeekV4_1Flash
| Self::KimiK2_5
| Self::KimiK2_6
| Self::KimiK2_7Code
| Self::KimiK3
| Self::MimoV2_5
| Self::MimoV2_5Pro
| Self::Glm5
| Self::Glm5_1
| Self::Glm5_2
| Self::Glm5_3
| Self::MiniMaxM2_5
| Self::MiniMaxM2_7
| Self::MiniMaxM3 => true,
Self::Custom {
interleaved_reasoning,
..
} => *interleaved_reasoning,
_ => false,
}
}
pub fn max_token_count(&self, subscription: OpenCodeSubscription) -> u64 {
match self {
// Anthropic models
Self::ClaudeOpus5 | Self::ClaudeOpus4_8 | Self::ClaudeOpus4_7 => 1_000_000,
Self::ClaudeOpus4_6 | Self::ClaudeSonnet4_6 => 1_000_000,
Self::ClaudeSonnet4_5 => 1_000_000,
Self::ClaudeOpus4_5 | Self::ClaudeHaiku4_5 => 200_000,
Self::ClaudeSonnet4 => 1_000_000,
Self::ClaudeSonnet5 => 1_000_000,
Self::ClaudeFable5 => 1_000_000,
// OpenAI models
Self::Gpt5_6Sol | Self::Gpt5_6Terra | Self::Gpt5_6Luna => 1_050_000,
Self::Gpt5_5 | Self::Gpt5_5Pro => 1_050_000,
Self::Gpt5_4 | Self::Gpt5_4Pro => 1_050_000,
Self::Gpt5_4Mini | Self::Gpt5_4Nano => 400_000,
Self::Gpt5_3Codex => 400_000,
Self::Gpt5_3Spark => 128_000,
Self::Gpt5_2 | Self::Gpt5_2Codex => 400_000,
Self::Gpt5_1 | Self::Gpt5_1Codex | Self::Gpt5_1CodexMax | Self::Gpt5_1CodexMini => {
400_000
}
Self::Gpt5 | Self::Gpt5Codex | Self::Gpt5Nano => 400_000,
// Google models
Self::Gemini3_1Pro
| Self::Gemini3Flash
| Self::Gemini3_5FlashLite
| Self::Gemini3_5Flash
| Self::Gemini3_6Flash
| Self::Gemini3_7Flash => 1_048_576,
// OpenAI-compatible models
Self::MiniMaxM2_7 => 204_800,
Self::MiniMaxM3 => {
if subscription == OpenCodeSubscription::Go {
1_000_000
} else {
512_000
}
}
Self::MiniMaxM2_5 => 204_800,
Self::Glm5 | Self::Glm5_1 => {
if subscription == OpenCodeSubscription::Go {
202_752
} else {
204_800
}
}
Self::Glm5_3 | Self::Glm5_2 => 1_000_000,
Self::KimiK2_6 | Self::KimiK2_5 | Self::KimiK2_7Code => 262_144,
Self::KimiK3 => 1_048_576,
Self::GrokBuild0_1 => 256_000,
Self::Grok4_6 | Self::Grok4_5 => 500_000,
Self::MuseSpark1_2 => 1_048_576,
Self::MimoV2_5Pro => 1_048_576,
Self::MimoV2_5 => 1_000_000,
Self::Qwen3_5Plus => 262_144,
Self::Qwen3_6Plus => {
if subscription == OpenCodeSubscription::Go {
1_000_000
} else {
262_144
}
}
Self::Qwen3_8Max | Self::Qwen3_7Max | Self::Qwen3_7Plus => 1_000_000,
Self::Hy3 => 256_000,
Self::DeepSeekV4Pro | Self::DeepSeekV4_1Flash => 1_000_000,
Self::Custom { max_tokens, .. } => *max_tokens,
}
}
pub fn max_output_tokens(&self, subscription: OpenCodeSubscription) -> Option<u64> {
match self {
// Anthropic models
Self::ClaudeOpus5 | Self::ClaudeOpus4_8 | Self::ClaudeOpus4_7 | Self::ClaudeOpus4_6 => {
Some(128_000)
}
Self::ClaudeOpus4_5
| Self::ClaudeSonnet4_6
| Self::ClaudeSonnet4_5
| Self::ClaudeHaiku4_5
| Self::ClaudeSonnet4 => Some(64_000),
Self::ClaudeSonnet5 => Some(128_000),
Self::ClaudeFable5 => Some(128_000),
// OpenAI models
Self::Gpt5_6Sol
| Self::Gpt5_6Terra
| Self::Gpt5_6Luna
| Self::Gpt5_5
| Self::Gpt5_5Pro
| Self::Gpt5_4
| Self::Gpt5_4Pro
| Self::Gpt5_4Mini
| Self::Gpt5_4Nano
| Self::Gpt5_3Codex
| Self::Gpt5_3Spark
| Self::Gpt5_2
| Self::Gpt5_2Codex
| Self::Gpt5_1
| Self::Gpt5_1Codex
| Self::Gpt5_1CodexMax
| Self::Gpt5_1CodexMini
| Self::Gpt5
| Self::Gpt5Codex
| Self::Gpt5Nano => Some(128_000),
// Google models
Self::Gemini3_1Pro
| Self::Gemini3Flash
| Self::Gemini3_5FlashLite
| Self::Gemini3_5Flash
| Self::Gemini3_6Flash
| Self::Gemini3_7Flash => Some(65_536),
// OpenAI-compatible models
Self::MiniMaxM2_7 => Some(131_072),
Self::MiniMaxM3 => {
if subscription == OpenCodeSubscription::Go {
Some(131_072)
} else {
Some(128_000)
}
}
Self::MiniMaxM2_5 => {
if subscription == OpenCodeSubscription::Go {
Some(65_536)
} else {
Some(131_072)
}
}
Self::Glm5 | Self::Glm5_1 => {
if subscription == OpenCodeSubscription::Go {
Some(32_768)
} else {
Some(131_072)
}
}
Self::Glm5_3 | Self::Glm5_2 => Some(131_072),
Self::KimiK2_6 | Self::KimiK2_5 => Some(65_536),
Self::KimiK2_7Code => Some(262_144),
Self::KimiK3 => Some(131_072),
Self::GrokBuild0_1 => Some(256_000),
Self::Grok4_6 | Self::Grok4_5 => Some(500_000),
Self::MuseSpark1_2 => Some(131_072),
Self::Qwen3_7Max | Self::Qwen3_7Plus | Self::Qwen3_6Plus | Self::Qwen3_5Plus => {
Some(65_536)
}
Self::Qwen3_8Max => Some(131_072),
Self::DeepSeekV4Pro | Self::DeepSeekV4_1Flash => Some(384_000),
Self::MimoV2_5Pro | Self::MimoV2_5 => Some(128_000),
Self::Hy3 => Some(64_000),
Self::Custom {
max_output_tokens, ..
} => *max_output_tokens,
}
}
pub fn supports_tools(&self) -> bool {
true
}
pub fn supports_images(&self) -> bool {
match self {
// Anthropic models support images
Self::ClaudeOpus5
| Self::ClaudeOpus4_8
| Self::ClaudeOpus4_7
| Self::ClaudeOpus4_6
| Self::ClaudeOpus4_5
| Self::ClaudeSonnet4_6
| Self::ClaudeSonnet4_5
| Self::ClaudeSonnet4
| Self::ClaudeHaiku4_5
| Self::ClaudeSonnet5
| Self::ClaudeFable5 => true,
// OpenAI models support images
Self::Gpt5_6Sol
| Self::Gpt5_6Terra
| Self::Gpt5_6Luna
| Self::Gpt5_5
| Self::Gpt5_5Pro
| Self::Gpt5_4
| Self::Gpt5_4Pro
| Self::Gpt5_4Mini
| Self::Gpt5_4Nano
| Self::Gpt5_3Codex
| Self::Gpt5_2
| Self::Gpt5_2Codex
| Self::Gpt5_1
| Self::Gpt5_1Codex
| Self::Gpt5_1CodexMax
| Self::Gpt5_1CodexMini
| Self::Gpt5
| Self::Gpt5Codex
| Self::Gpt5Nano => true,
// OpenAI models without image support
Self::Gpt5_3Spark => false,
// Google models support images
Self::Gemini3_1Pro
| Self::Gemini3Flash
| Self::Gemini3_5FlashLite
| Self::Gemini3_5Flash
| Self::Gemini3_6Flash
| Self::Gemini3_7Flash => true,
// OpenAI-compatible models with image support
Self::KimiK2_6
| Self::KimiK3
| Self::KimiK2_7Code
| Self::KimiK2_5
| Self::GrokBuild0_1
| Self::Grok4_5
| Self::Grok4_6
| Self::MuseSpark1_2
| Self::MimoV2_5
| Self::Qwen3_5Plus
| Self::Qwen3_6Plus
| Self::Qwen3_7Plus
| Self::Qwen3_8Max
| Self::MiniMaxM3
| Self::DeepSeekV4_1Flash => true,
// OpenAI-compatible models without image support
Self::MiniMaxM2_5
| Self::Glm5
| Self::Glm5_1
| Self::Glm5_2
| Self::Glm5_3
| Self::MiniMaxM2_7
| Self::MimoV2_5Pro
| Self::DeepSeekV4Pro
| Self::Qwen3_7Max
| Self::Hy3 => false,
Self::Custom { protocol, .. } => matches!(
protocol,
ApiProtocol::Anthropic
| ApiProtocol::OpenAiResponses
| ApiProtocol::OpenAiChat
| ApiProtocol::Google
),
}
}
pub fn supported_reasoning_effort_levels(&self) -> Option<Vec<ReasoningEffort>> {
match self {
// Anthropic models
Self::ClaudeFable5
| Self::ClaudeOpus5
| Self::ClaudeOpus4_8
| Self::ClaudeOpus4_7
| Self::ClaudeSonnet5 => Some(vec![
ReasoningEffort::Low,
ReasoningEffort::Medium,
ReasoningEffort::High,
ReasoningEffort::XHigh,
ReasoningEffort::Max,
]),
Self::ClaudeOpus4_6 | Self::ClaudeSonnet4_6 => Some(vec![
ReasoningEffort::Low,
ReasoningEffort::Medium,
ReasoningEffort::High,
ReasoningEffort::Max,
]),
Self::ClaudeOpus4_5 => Some(vec![
ReasoningEffort::Low,
ReasoningEffort::Medium,
ReasoningEffort::High,
]),
// OpenAI models
Self::Gpt5_5
| Self::Gpt5_4
| Self::Gpt5_4Mini
| Self::Gpt5_4Nano
| Self::Gpt5_3Codex
| Self::Gpt5_2 => Some(vec![
ReasoningEffort::None,
ReasoningEffort::Low,
ReasoningEffort::Medium,
ReasoningEffort::High,
ReasoningEffort::XHigh,
]),
Self::Gpt5_5Pro | Self::Gpt5_4Pro => Some(vec![
ReasoningEffort::Medium,
ReasoningEffort::High,
ReasoningEffort::XHigh,
]),
Self::Gpt5_2Codex | Self::Gpt5_3Spark | Self::Gpt5_1CodexMax => Some(vec![
ReasoningEffort::Low,
ReasoningEffort::Medium,
ReasoningEffort::High,
ReasoningEffort::XHigh,
]),
Self::Gpt5_1 => Some(vec![
ReasoningEffort::None,
ReasoningEffort::Low,
ReasoningEffort::Medium,
ReasoningEffort::High,
]),
Self::Gpt5Codex | Self::Gpt5_1Codex | Self::Gpt5_1CodexMini => Some(vec![
ReasoningEffort::Low,
ReasoningEffort::Medium,
ReasoningEffort::High,
]),
Self::Gpt5 | Self::Gpt5Nano => Some(vec![
ReasoningEffort::Minimal,
ReasoningEffort::Low,
ReasoningEffort::Medium,
ReasoningEffort::High,
]),
Self::Gpt5_6Sol | Self::Gpt5_6Terra | Self::Gpt5_6Luna => Some(vec![
ReasoningEffort::None,
ReasoningEffort::Low,
ReasoningEffort::Medium,
ReasoningEffort::High,
ReasoningEffort::XHigh,
ReasoningEffort::Max,
]),
// Google models
Self::Gemini3Flash
| Self::Gemini3_5FlashLite
| Self::Gemini3_5Flash
| Self::Gemini3_6Flash => Some(vec![
ReasoningEffort::Minimal,
ReasoningEffort::Low,
ReasoningEffort::Medium,
ReasoningEffort::High,
]),
Self::Gemini3_7Flash | Self::Gemini3_1Pro => Some(vec![
ReasoningEffort::Low,
ReasoningEffort::Medium,
ReasoningEffort::High,
]),
// DeepSeek models
Self::DeepSeekV4Pro | Self::DeepSeekV4_1Flash => Some(vec![
// OpenCode also supports Low&Medium but as per DeepSeek those are mapped to High
ReasoningEffort::High,
ReasoningEffort::Max,
]),
// Moonshot AI models
Self::KimiK3 => Some(vec![ReasoningEffort::Max]),
// MiniMax models
Self::MiniMaxM3 => Some(vec![ReasoningEffort::None]),
// Xiaomi MiMo models
Self::MimoV2_5Pro | Self::MimoV2_5 => Some(vec![
ReasoningEffort::Low,
ReasoningEffort::Medium,
ReasoningEffort::High,
]),
// Z AI models
Self::Glm5_2 => Some(vec![ReasoningEffort::High, ReasoningEffort::Max]),
Self::Glm5_3 => Some(vec![
ReasoningEffort::Low,
ReasoningEffort::High,
ReasoningEffort::Max,
]),
// Tencent models
Self::Hy3 => Some(vec![
ReasoningEffort::None,
ReasoningEffort::Low,
ReasoningEffort::High,
]),
// SpaceXAI models
Self::Grok4_6 => Some(vec![
ReasoningEffort::Low,
ReasoningEffort::Medium,
ReasoningEffort::High,
ReasoningEffort::XHigh,
]),
Self::Grok4_5 => Some(vec![
ReasoningEffort::Low,
ReasoningEffort::Medium,
ReasoningEffort::High,
]),
// Meta AI models
Self::MuseSpark1_2 => Some(vec![
ReasoningEffort::Minimal,
ReasoningEffort::Low,
ReasoningEffort::Medium,
ReasoningEffort::High,
ReasoningEffort::XHigh,
]),
Self::Custom {
reasoning_effort_levels,
..
} => reasoning_effort_levels.clone(),
_ => None,
}
}
}
/// Stream generate content for Google models via OpenCode.
///
/// Unlike `google_ai::stream_generate_content()`, this uses:
/// - `/v1/models/{model}` path (not `/v1beta/models/{model}`)
/// - `Authorization: Bearer` header (not `key=` query param)
pub async fn stream_generate_content(
client: &dyn HttpClient,
api_url: &str,
api_key: &str,
request: google_ai::GenerateContentRequest,
extra_headers: &CustomHeaders,
) -> Result<BoxStream<'static, Result<google_ai::GenerateContentResponse>>> {
let api_key = api_key.trim();
let model_id = &request.model.model_id;
let uri = format!("{api_url}/v1/models/{model_id}:streamGenerateContent?alt=sse");
let request = HttpRequest::builder()
.method(Method::POST)
.uri(uri)
.header("Content-Type", "application/json")
.header("Authorization", format!("Bearer {api_key}"))
.extra_headers(extra_headers)
.body(AsyncBody::from(serde_json::to_string(&request)?))?;
let mut response = client.send(request).await?;
if response.status().is_success() {
let reader = BufReader::new(response.into_body());
Ok(reader
.lines()
.filter_map(|line| async move {
match line {
Ok(line) => {
if let Some(line) = line.strip_prefix("data: ") {
match serde_json::from_str(line) {
Ok(response) => Some(Ok(response)),
Err(error) => {
Some(Err(anyhow!("Error parsing JSON: {error:?}\n{line:?}")))
}
}
} else {
None
}
}
Err(error) => Some(Err(anyhow!(error))),
}
})
.boxed())
} else {
let mut text = String::new();
response.body_mut().read_to_string(&mut text).await?;
Err(anyhow!(
"error during streamGenerateContent via OpenCode, status code: {:?}, body: {}",
response.status(),
text
))
}
}
Reference in new issue View Git Blame Copy Permalink