use anyhow::{Result, anyhow}; use futures::{AsyncBufReadExt, AsyncReadExt, StreamExt, io::BufReader, stream::BoxStream}; use http_client::{AsyncBody, HttpClient, Method, Request as HttpRequest}; 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, Free, } impl OpenCodeSubscription { pub fn display_name(&self) -> &'static str { match self { Self::Zen => "Zen", Self::Go => "Go", Self::Free => "Free", } } pub fn id_prefix(&self) -> &'static str { match self { Self::Zen => "zen", Self::Go => "go", Self::Free => "free", } } pub fn api_path_suffix(&self) -> &'static str { match self { Self::Zen | Self::Free => "", 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-4-7")] ClaudeOpus4_7, #[serde(rename = "claude-opus-4-6")] ClaudeOpus4_6, #[serde(rename = "claude-opus-4-5")] ClaudeOpus4_5, #[serde(rename = "claude-opus-4-1")] ClaudeOpus4_1, #[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, // -- OpenAI Responses API models -- #[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")] Gemini3_5Flash, // -- OpenAI Chat Completions protocol models -- #[serde(rename = "deepseek-v4-pro")] DeepSeekV4Pro, #[serde(rename = "deepseek-v4-flash")] DeepSeekV4Flash, #[serde(rename = "minimax-m2.5")] MiniMaxM2_5, #[serde(rename = "glm-5")] Glm5, #[serde(rename = "glm-5.1")] Glm5_1, #[serde(rename = "grok-build-0.1")] GrokBuild0_1, #[serde(rename = "kimi-k2.5")] KimiK2_5, #[serde(rename = "kimi-k2.6")] KimiK2_6, #[serde(rename = "minimax-m2.7")] MiniMaxM2_7, #[serde(rename = "mimo-v2.5-pro")] MimoV2_5Pro, #[serde(rename = "mimo-v2.5")] MimoV2_5, #[serde(rename = "big-pickle")] BigPickle, #[serde(rename = "nemotron-3-super-free")] Nemotron3SuperFree, #[serde(rename = "qwen3.5-plus")] Qwen3_5Plus, #[serde(rename = "qwen3.6-plus")] Qwen3_6Plus, // -- Custom model -- #[serde(rename = "custom")] Custom { name: String, display_name: Option, max_tokens: u64, max_output_tokens: Option, protocol: ApiProtocol, reasoning_effort_levels: Option>, custom_model_api_url: Option, interleaved_reasoning: bool, }, } impl Model { pub fn default_fast() -> Self { Self::ClaudeHaiku4_5 } pub fn default_go() -> Self { Self::KimiK2_5 } pub fn default_go_fast() -> Self { Self::MiniMaxM2_5 } pub fn default_free() -> Self { Self::BigPickle } pub fn default_free_fast() -> Self { Self::Nemotron3SuperFree } pub fn available_subscriptions(&self) -> &'static [OpenCodeSubscription] { match self { // Models available in both Zen and Go Self::Glm5 | Self::Glm5_1 | Self::KimiK2_6 | Self::KimiK2_5 | Self::MiniMaxM2_5 | Self::Qwen3_5Plus | Self::Qwen3_6Plus => &[OpenCodeSubscription::Zen, OpenCodeSubscription::Go], // Go-only models Self::MiniMaxM2_7 | Self::MimoV2_5Pro | Self::MimoV2_5 | Self::DeepSeekV4Pro | Self::DeepSeekV4Flash => &[OpenCodeSubscription::Go], // Free models Self::Nemotron3SuperFree | Self::BigPickle => &[OpenCodeSubscription::Free], // 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::ClaudeOpus4_7 => "claude-opus-4-7", Self::ClaudeOpus4_6 => "claude-opus-4-6", Self::ClaudeOpus4_5 => "claude-opus-4-5", Self::ClaudeOpus4_1 => "claude-opus-4-1", 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::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_5Flash => "gemini-3.5-flash", Self::DeepSeekV4Pro => "deepseek-v4-pro", Self::DeepSeekV4Flash => "deepseek-v4-flash", Self::MiniMaxM2_5 => "minimax-m2.5", Self::Glm5 => "glm-5", Self::Glm5_1 => "glm-5.1", Self::GrokBuild0_1 => "grok-build-0.1", Self::KimiK2_5 => "kimi-k2.5", Self::KimiK2_6 => "kimi-k2.6", Self::MiniMaxM2_7 => "minimax-m2.7", 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::BigPickle => "big-pickle", Self::Nemotron3SuperFree => "nemotron-3-super-free", Self::Custom { name, .. } => name, } } pub fn display_name(&self) -> &str { match self { Self::ClaudeOpus4_7 => "Claude Opus 4.7", Self::ClaudeOpus4_6 => "Claude Opus 4.6", Self::ClaudeOpus4_5 => "Claude Opus 4.5", Self::ClaudeOpus4_1 => "Claude Opus 4.1", 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::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_5Flash => "Gemini 3.5 Flash", Self::DeepSeekV4Pro => "DeepSeek V4 Pro", Self::DeepSeekV4Flash => "DeepSeek V4 Flash", Self::MiniMaxM2_5 => "MiniMax M2.5", Self::Glm5 => "GLM 5", Self::Glm5_1 => "GLM 5.1", Self::GrokBuild0_1 => "Grok Build 0.1", Self::KimiK2_5 => "Kimi K2.5", Self::KimiK2_6 => "Kimi K2.6", Self::MiniMaxM2_7 => "MiniMax M2.7", 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::BigPickle => "Big Pickle", Self::Nemotron3SuperFree => "Nemotron 3 Super Free", 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::MiniMaxM2_7 | Self::MiniMaxM2_5 => { if subscription == OpenCodeSubscription::Zen { ApiProtocol::OpenAiChat } else { ApiProtocol::Anthropic } } Self::ClaudeOpus4_7 | Self::ClaudeOpus4_6 | Self::ClaudeOpus4_5 | Self::ClaudeOpus4_1 | Self::ClaudeSonnet4_6 | Self::ClaudeSonnet4_5 | Self::ClaudeSonnet4 | Self::ClaudeHaiku4_5 => ApiProtocol::Anthropic, 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_5Flash => ApiProtocol::Google, Self::Glm5 | Self::Glm5_1 | Self::GrokBuild0_1 | Self::KimiK2_5 | Self::KimiK2_6 | Self::MimoV2_5Pro | Self::MimoV2_5 | Self::Qwen3_5Plus | Self::Qwen3_6Plus | Self::DeepSeekV4Pro | Self::DeepSeekV4Flash | Self::BigPickle | Self::Nemotron3SuperFree => ApiProtocol::OpenAiChat, Self::Custom { protocol, .. } => *protocol, } } pub fn interleaved_reasoning(&self) -> bool { match self { Self::DeepSeekV4Pro | Self::DeepSeekV4Flash | Self::KimiK2_5 | Self::KimiK2_6 | Self::MimoV2_5 | Self::MimoV2_5Pro | Self::Glm5 | Self::Glm5_1 | Self::Nemotron3SuperFree | Self::BigPickle => true, Self::Custom { interleaved_reasoning, .. } => *interleaved_reasoning, _ => false, } } pub fn max_token_count(&self, subscription: OpenCodeSubscription) -> u64 { match self { // Anthropic models 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::ClaudeOpus4_1 => 200_000, Self::ClaudeSonnet4 => 1_000_000, // OpenAI models 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 => 1_048_576, Self::Gemini3Flash => 1_048_576, Self::Gemini3_5Flash => 1_048_576, // OpenAI-compatible models Self::MiniMaxM2_7 => 204_800, Self::MiniMaxM2_5 => 204_800, Self::Glm5 | Self::Glm5_1 => { if subscription == OpenCodeSubscription::Go { 202_752 } else { 204_800 } } Self::KimiK2_6 | Self::KimiK2_5 => 262_144, Self::GrokBuild0_1 => 256_000, Self::MimoV2_5Pro => 1_048_576, Self::MimoV2_5 => 1_000_000, Self::Qwen3_5Plus | Self::Qwen3_6Plus => 262_144, Self::BigPickle => 200_000, Self::Nemotron3SuperFree => 204_800, Self::DeepSeekV4Pro | Self::DeepSeekV4Flash => 1_000_000, Self::Custom { max_tokens, .. } => *max_tokens, } } pub fn max_output_tokens(&self, subscription: OpenCodeSubscription) -> Option { match self { // Anthropic models 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::ClaudeOpus4_1 => Some(32_000), // OpenAI models 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_5Flash => Some(65_536), // OpenAI-compatible models Self::MiniMaxM2_7 => Some(131_072), 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::BigPickle => Some(128_000), Self::KimiK2_6 | Self::KimiK2_5 => Some(65_536), Self::GrokBuild0_1 => Some(256_000), Self::Qwen3_5Plus | Self::Qwen3_6Plus => Some(65_536), Self::DeepSeekV4Pro | Self::DeepSeekV4Flash => Some(384_000), Self::Nemotron3SuperFree => Some(128_000), Self::MimoV2_5Pro | Self::MimoV2_5 => Some(128_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::ClaudeOpus4_7 | Self::ClaudeOpus4_6 | Self::ClaudeOpus4_5 | Self::ClaudeOpus4_1 | Self::ClaudeSonnet4_6 | Self::ClaudeSonnet4_5 | Self::ClaudeSonnet4 | Self::ClaudeHaiku4_5 => true, // OpenAI models support images 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_5Flash => true, // OpenAI-compatible models with image support Self::KimiK2_6 | Self::KimiK2_5 | Self::GrokBuild0_1 | Self::MimoV2_5 | Self::Qwen3_5Plus | Self::Qwen3_6Plus => true, // OpenAI-compatible models without image support Self::MiniMaxM2_5 | Self::Glm5 | Self::Glm5_1 | Self::MiniMaxM2_7 | Self::MimoV2_5Pro | Self::DeepSeekV4Pro | Self::DeepSeekV4Flash | Self::BigPickle | Self::Nemotron3SuperFree => false, Self::Custom { protocol, .. } => matches!( protocol, ApiProtocol::Anthropic | ApiProtocol::OpenAiResponses | ApiProtocol::OpenAiChat | ApiProtocol::Google ), } } pub fn supported_reasoning_effort_levels(&self) -> Option> { match self { Self::MimoV2_5Pro | Self::MimoV2_5 => Some(vec![ ReasoningEffort::Low, ReasoningEffort::Medium, ReasoningEffort::High, ]), Self::DeepSeekV4Pro | Self::DeepSeekV4Flash => Some(vec![ 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, ) -> Result>> { 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_builder = HttpRequest::builder() .method(Method::POST) .uri(uri) .header("Content-Type", "application/json") .header("Authorization", format!("Bearer {api_key}")); let request = request_builder.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 )) } }