logiguard fork v3: full patch set on verified 8c74db0 tree
Includes prior-session patches (carry forward so the app compiles): - crates/gpui/build.rs: cross-compile manifest fix - crates/gpui/src/platform.rs: PlatformWindow::activate_with_token trait method - crates/gpui/src/window.rs: Window::activate_with_token public API - crates/gpui_linux/src/linux/wayland/window.rs: WaylandWindow::activate_with_token + activate() keyboard-serial fix Plus the focus-serial fix: - serial.rs: SerialKind::KeyboardEnter - client.rs: store wl_keyboard.enter serial; latest_serial_of() Co-Authored-By: Claude Fable 5 <noreply@anthropic.com>
This commit is contained in:
Mohamad Khani
28
crates/opencode/Cargo.toml
Normal file
28
crates/opencode/Cargo.toml
Normal file
@@ -0,0 +1,28 @@
|
||||
[package]
|
||||
name = "opencode"
|
||||
version = "0.1.0"
|
||||
edition.workspace = true
|
||||
publish.workspace = true
|
||||
license = "GPL-3.0-or-later"
|
||||
|
||||
[lints]
|
||||
workspace = true
|
||||
|
||||
[lib]
|
||||
path = "src/opencode.rs"
|
||||
test = false
|
||||
|
||||
[features]
|
||||
default = []
|
||||
schemars = ["dep:schemars"]
|
||||
|
||||
[dependencies]
|
||||
anyhow.workspace = true
|
||||
futures.workspace = true
|
||||
google_ai.workspace = true
|
||||
http_client.workspace = true
|
||||
language_model_core.workspace = true
|
||||
schemars = { workspace = true, optional = true }
|
||||
serde.workspace = true
|
||||
serde_json.workspace = true
|
||||
strum.workspace = true
|
||||
1
crates/opencode/LICENSE-GPL
Symbolic link
1
crates/opencode/LICENSE-GPL
Symbolic link
@@ -0,0 +1 @@
|
||||
../../LICENSE-GPL
|
||||
653
crates/opencode/src/opencode.rs
Normal file
653
crates/opencode/src/opencode.rs
Normal file
@@ -0,0 +1,653 @@
|
||||
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,
|
||||
|
||||
// -- 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 = "minimax-m2.5-free")]
|
||||
MiniMaxM2_5Free,
|
||||
#[serde(rename = "glm-5")]
|
||||
Glm5,
|
||||
#[serde(rename = "glm-5.1")]
|
||||
Glm5_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 = "ring-2.6-1t-free")]
|
||||
Ring2_6_1TFree,
|
||||
#[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<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_5
|
||||
}
|
||||
|
||||
pub fn default_go_fast() -> Self {
|
||||
Self::MiniMaxM2_5
|
||||
}
|
||||
|
||||
pub fn default_free() -> Self {
|
||||
Self::BigPickle
|
||||
}
|
||||
|
||||
pub fn default_free_fast() -> Self {
|
||||
Self::MiniMaxM2_5Free
|
||||
}
|
||||
|
||||
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::MiniMaxM2_5Free
|
||||
| Self::Nemotron3SuperFree
|
||||
| Self::BigPickle
|
||||
| Self::Ring2_6_1TFree => &[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::DeepSeekV4Pro => "deepseek-v4-pro",
|
||||
Self::DeepSeekV4Flash => "deepseek-v4-flash",
|
||||
Self::MiniMaxM2_5 => "minimax-m2.5",
|
||||
Self::MiniMaxM2_5Free => "minimax-m2.5-free",
|
||||
Self::Glm5 => "glm-5",
|
||||
Self::Glm5_1 => "glm-5.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::Ring2_6_1TFree => "ring-2.6-1t-free",
|
||||
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::DeepSeekV4Pro => "DeepSeek V4 Pro",
|
||||
Self::DeepSeekV4Flash => "DeepSeek V4 Flash",
|
||||
Self::MiniMaxM2_5 => "MiniMax M2.5",
|
||||
Self::MiniMaxM2_5Free => "MiniMax M2.5 Free",
|
||||
Self::Glm5 => "GLM 5",
|
||||
Self::Glm5_1 => "GLM 5.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::Ring2_6_1TFree => "Ring 2.6 1T Free",
|
||||
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 => ApiProtocol::Google,
|
||||
|
||||
Self::MiniMaxM2_5Free
|
||||
| Self::Glm5
|
||||
| Self::Glm5_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::Ring2_6_1TFree
|
||||
| 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::BigPickle
|
||||
| Self::Ring2_6_1TFree => true,
|
||||
|
||||
Self::Custom {
|
||||
interleaved_reasoning,
|
||||
..
|
||||
} => *interleaved_reasoning,
|
||||
|
||||
_ => false,
|
||||
}
|
||||
}
|
||||
|
||||
pub fn max_token_count(&self) -> 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,
|
||||
|
||||
// OpenAI-compatible models
|
||||
Self::MiniMaxM2_7 => 204_800,
|
||||
Self::MiniMaxM2_5 | Self::MiniMaxM2_5Free => 204_800,
|
||||
Self::Glm5 | Self::Glm5_1 => 202_725,
|
||||
Self::KimiK2_6 | Self::KimiK2_5 => 262_144,
|
||||
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::Ring2_6_1TFree => 262_000,
|
||||
Self::Nemotron3SuperFree => 204_800,
|
||||
Self::DeepSeekV4Pro | Self::DeepSeekV4Flash => 1_000_000,
|
||||
|
||||
Self::Custom { max_tokens, .. } => *max_tokens,
|
||||
}
|
||||
}
|
||||
|
||||
pub fn max_output_tokens(&self) -> Option<u64> {
|
||||
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 => Some(65_536),
|
||||
|
||||
// OpenAI-compatible models
|
||||
Self::MiniMaxM2_7 => Some(131_072),
|
||||
Self::MiniMaxM2_5 | Self::MiniMaxM2_5Free => Some(131_072),
|
||||
Self::Glm5 | Self::Glm5_1 => Some(32_768),
|
||||
Self::BigPickle => Some(128_000),
|
||||
Self::Ring2_6_1TFree => Some(66_000),
|
||||
Self::KimiK2_6 | Self::KimiK2_5 => Some(65_536),
|
||||
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_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 => true,
|
||||
|
||||
// Google models support images
|
||||
Self::Gemini3_1Pro | Self::Gemini3Flash => true,
|
||||
|
||||
// OpenAI-compatible models with image support
|
||||
Self::KimiK2_6
|
||||
| Self::KimiK2_5
|
||||
| Self::MimoV2_5
|
||||
| Self::Qwen3_5Plus
|
||||
| Self::Qwen3_6Plus => true,
|
||||
|
||||
// OpenAI-compatible models without image support
|
||||
Self::MiniMaxM2_5
|
||||
| Self::MiniMaxM2_5Free
|
||||
| Self::Glm5
|
||||
| Self::Glm5_1
|
||||
| Self::MiniMaxM2_7
|
||||
| Self::MimoV2_5Pro
|
||||
| Self::DeepSeekV4Pro
|
||||
| Self::DeepSeekV4Flash
|
||||
| Self::BigPickle
|
||||
| Self::Ring2_6_1TFree
|
||||
| Self::Nemotron3SuperFree => 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 {
|
||||
Self::Ring2_6_1TFree | 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<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_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
|
||||
))
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user