モデル性能のハードコードを消し飛し、Codexのフォーマットの修正

2026-04-21 18:35:56 +09:00
parent 2914800673
commit 47da4a03cb
25 changed files with 628 additions and 246 deletions
@@ -0,0 +1,211 @@
+//! `SchemeKind` × `model_id` → [`ModelCapability`] の既知モデル静的テーブル。
+//!
+//! このテーブルは「モデル ID の知識」であり、wire 実装(`llm-worker`)の
+//! 責務ではなく高レベル構築層(`crates/provider`)の責務として置く。
+//! llm-worker の scheme には `default_capability()` のみ残し、未知モデル
+//! 時は scheme 既定にフォールバックする。
+//!
+//! 解決順(`build_client` が呼ぶ):
+//!   1. `ModelConfig.capability` 明示指定
+//!   2. [`lookup`] (本モジュール)
+//!   3. `Scheme::default_capability()` (llm-worker)
+
+use llm_worker::llm_client::capability::{
+    CacheStrategy, ModelCapability, ReasoningSupport, StructuredOutput, ToolCallingSupport,
+};
+use manifest::SchemeKind;
+
+/// `scheme` と `model_id` から既知モデルの capability を返す。
+/// 未知なら `None`(呼び出し側が `Scheme::default_capability()` へフォールバック)。
+pub fn lookup(scheme: SchemeKind, model_id: &str) -> Option<ModelCapability> {
+    match scheme {
+        SchemeKind::Anthropic => anthropic_lookup(model_id),
+        SchemeKind::OpenaiChat => openai_chat_lookup(model_id),
+        SchemeKind::OpenaiResponses => openai_responses_lookup(model_id),
+        SchemeKind::Gemini => gemini_lookup(model_id),
+    }
+}
+
+// --- Anthropic --------------------------------------------------------------
+
+fn anthropic_lookup(model_id: &str) -> Option<ModelCapability> {
+    if !model_id.starts_with("claude-") {
+        return None;
+    }
+    Some(ModelCapability {
+        tool_calling: ToolCallingSupport::Parallel,
+        structured_output: StructuredOutput::JsonSchema,
+        reasoning: Some(ReasoningSupport::BudgetTokens),
+        vision: true,
+        prompt_caching: CacheStrategy::Explicit { max_breakpoints: 4 },
+    })
+}
+
+// --- OpenAI (chat / responses で共有の family 判定) --------------------------
+
+#[derive(Debug, Clone, Copy, PartialEq, Eq)]
+enum OpenAiFamily {
+    /// GPT-5 / o1 / o3 / o4 系 — reasoning 対応
+    Reasoning,
+    /// GPT-4o / GPT-4 系
+    Gpt4,
+    /// GPT-3.5 系(旧式)
+    Gpt35,
+}
+
+fn openai_classify(model_id: &str) -> Option<OpenAiFamily> {
+    if model_id.starts_with("gpt-5")
+        || model_id.starts_with("o1")
+        || model_id.starts_with("o3")
+        || model_id.starts_with("o4")
+    {
+        return Some(OpenAiFamily::Reasoning);
+    }
+    if model_id.starts_with("gpt-4") {
+        return Some(OpenAiFamily::Gpt4);
+    }
+    if model_id.starts_with("gpt-3.5") {
+        return Some(OpenAiFamily::Gpt35);
+    }
+    None
+}
+
+fn openai_chat_lookup(model_id: &str) -> Option<ModelCapability> {
+    openai_classify(model_id).map(|family| match family {
+        OpenAiFamily::Reasoning => ModelCapability {
+            tool_calling: ToolCallingSupport::Parallel,
+            structured_output: StructuredOutput::JsonSchema,
+            reasoning: Some(ReasoningSupport::Effort),
+            vision: true,
+            prompt_caching: CacheStrategy::Auto,
+        },
+        OpenAiFamily::Gpt4 => ModelCapability {
+            tool_calling: ToolCallingSupport::Parallel,
+            structured_output: StructuredOutput::JsonSchema,
+            reasoning: None,
+            vision: true,
+            prompt_caching: CacheStrategy::Auto,
+        },
+        OpenAiFamily::Gpt35 => ModelCapability {
+            tool_calling: ToolCallingSupport::Parallel,
+            structured_output: StructuredOutput::JsonObject,
+            reasoning: None,
+            vision: false,
+            prompt_caching: CacheStrategy::Auto,
+        },
+    })
+}
+
+fn openai_responses_lookup(model_id: &str) -> Option<ModelCapability> {
+    // `codex-` prefix は ChatGPT backend 経由(CodexOAuth)でのみ使える
+    // Reasoning モデル family。`gpt-5` 系と同じ扱い。
+    let family = openai_classify(model_id).or_else(|| {
+        if model_id.starts_with("codex-") {
+            Some(OpenAiFamily::Reasoning)
+        } else {
+            None
+        }
+    })?;
+    Some(match family {
+        OpenAiFamily::Reasoning => ModelCapability {
+            tool_calling: ToolCallingSupport::Parallel,
+            structured_output: StructuredOutput::JsonSchema,
+            reasoning: Some(ReasoningSupport::Effort),
+            vision: true,
+            prompt_caching: CacheStrategy::Auto,
+        },
+        OpenAiFamily::Gpt4 => ModelCapability {
+            tool_calling: ToolCallingSupport::Parallel,
+            structured_output: StructuredOutput::JsonSchema,
+            reasoning: None,
+            vision: true,
+            prompt_caching: CacheStrategy::Auto,
+        },
+        OpenAiFamily::Gpt35 => ModelCapability {
+            tool_calling: ToolCallingSupport::Parallel,
+            structured_output: StructuredOutput::JsonObject,
+            reasoning: None,
+            vision: false,
+            prompt_caching: CacheStrategy::Auto,
+        },
+    })
+}
+
+// --- Gemini -----------------------------------------------------------------
+
+fn gemini_lookup(model_id: &str) -> Option<ModelCapability> {
+    if !model_id.starts_with("gemini-") {
+        return None;
+    }
+    // 2.5 系以降は thinking / reasoning を持つ
+    let reasoning = if model_id.starts_with("gemini-2.5") || model_id.starts_with("gemini-3") {
+        Some(ReasoningSupport::BudgetTokens)
+    } else {
+        None
+    };
+    Some(ModelCapability {
+        tool_calling: ToolCallingSupport::Parallel,
+        structured_output: StructuredOutput::JsonSchema,
+        reasoning,
+        vision: true,
+        prompt_caching: CacheStrategy::Auto,
+    })
+}
+
+#[cfg(test)]
+mod tests {
+    use super::*;
+
+    #[test]
+    fn anthropic_known_claude() {
+        let cap = lookup(SchemeKind::Anthropic, "claude-sonnet-4-6").unwrap();
+        assert!(matches!(
+            cap.prompt_caching,
+            CacheStrategy::Explicit { max_breakpoints: 4 }
+        ));
+        assert!(cap.reasoning.is_some());
+    }
+
+    #[test]
+    fn anthropic_unknown_is_none() {
+        assert!(lookup(SchemeKind::Anthropic, "llama3").is_none());
+    }
+
+    #[test]
+    fn openai_chat_gpt5_is_reasoning() {
+        let cap = lookup(SchemeKind::OpenaiChat, "gpt-5-xyz").unwrap();
+        assert!(cap.reasoning.is_some());
+    }
+
+    #[test]
+    fn openai_responses_codex_prefix_is_reasoning() {
+        let cap = lookup(SchemeKind::OpenaiResponses, "codex-mini-latest").unwrap();
+        assert!(cap.reasoning.is_some());
+    }
+
+    #[test]
+    fn openai_responses_gpt5_codex_is_reasoning() {
+        // gpt-5 prefix 経由で classify される
+        let cap = lookup(SchemeKind::OpenaiResponses, "gpt-5-codex").unwrap();
+        assert!(cap.reasoning.is_some());
+    }
+
+    #[test]
+    fn gemini_25_has_reasoning() {
+        let cap = lookup(SchemeKind::Gemini, "gemini-2.5-pro").unwrap();
+        assert!(matches!(cap.reasoning, Some(ReasoningSupport::BudgetTokens)));
+    }
+
+    #[test]
+    fn gemini_legacy_has_no_reasoning() {
+        let cap = lookup(SchemeKind::Gemini, "gemini-1.5-pro").unwrap();
+        assert!(cap.reasoning.is_none());
+    }
+
+    #[test]
+    fn unknown_model_is_none_across_schemes() {
+        assert!(lookup(SchemeKind::OpenaiChat, "foo-bar").is_none());
+        assert!(lookup(SchemeKind::OpenaiResponses, "foo-bar").is_none());
+        assert!(lookup(SchemeKind::Gemini, "foo-bar").is_none());
+    }
+}
@@ -143,7 +143,7 @@ impl CodexAuthProvider {
    }

    fn build_headers(snap: &AuthSnapshot) -> Result<Vec<(HeaderName, HeaderValue)>, CodexAuthError> {
-        let mut out = Vec::with_capacity(3);
+        let mut out = Vec::with_capacity(5);

        let auth_val = HeaderValue::from_str(&format!("Bearer {}", snap.access_token))
            .map_err(|e| CodexAuthError::InvalidHeader(format!("Authorization: {e}")))?;
@@ -156,6 +156,18 @@ impl CodexAuthProvider {
            acc_val,
        ));

+        // Cloudflare WAF は ChatGPT backend アクセス元を `originator` /
+        // `User-Agent` で識別する。Codex CLI が送る固定値を流用しないと
+        // HTML challenge (403) を返されて SSE に到達できない。
+        out.push((
+            HeaderName::from_static("originator"),
+            HeaderValue::from_static("codex_cli_rs"),
+        ));
+        out.push((
+            HeaderName::from_static("user-agent"),
+            HeaderValue::from_static("codex_cli_rs/0.60.0"),
+        ));
+
        // FedRAMP 組織は id_token JWT 内の claim で判定
        if jwt::parse_chatgpt_claims(&snap.id_token)
            .map(|c| c.is_fedramp)
@@ -10,6 +10,7 @@
 //! なる認証ストア解決（Codex OAuth の `~/.codex/auth.json` 読取等）は
 //! このクレートに追加する。

+pub mod capability;
 pub mod codex_oauth;

 use std::sync::Arc;
@@ -87,12 +88,14 @@ fn resolve_auth(
 }

 /// `AuthRef::CodexOAuth` 指定時、`base_url` 未指定なら ChatGPT backend を既定とする。
+/// Codex CLI が使う `/backend-api/codex` を base に取り、scheme 側の `/responses`
+/// path と結合して `https://chatgpt.com/backend-api/codex/responses` になる。
 fn effective_base_url<S: Scheme>(scheme: &S, config: &ModelConfig) -> String {
    if let Some(b) = &config.base_url {
        return b.clone();
    }
    if matches!(config.auth, AuthRef::CodexOAuth) {
-        return "https://chatgpt.com/backend-api".to_string();
+        return "https://chatgpt.com/backend-api/codex".to_string();
    }
    scheme.default_base_url().to_string()
 }
@@ -108,14 +111,15 @@ fn build_transport<S: Scheme>(
        });
    }
    // capability の優先順位:
-    //   1. `ModelConfig.capability` の明示指定（OpenAI 互換ルーターの
-    //      未知モデル等、マニフェストで完全に上書きしたいケース）
-    //   2. scheme 静的テーブル（既知モデル）
-    //   3. `Scheme::default_capability()`（scheme ごとの安全側デフォルト）
+    //   1. `ModelConfig.capability` の明示指定(OpenAI 互換ルーターの
+    //      未知モデル等、マニフェストで完全に上書きしたいケース)
+    //   2. `provider::capability::lookup` の既知モデルテーブル
+    //      (モデル ID の知識は高レベル構築層(ここ)の責務)
+    //   3. `Scheme::default_capability()`(scheme ごとの wire-level 安全側)
    let capability: ModelCapability = config
        .capability
        .clone()
-        .or_else(|| scheme.capability_for(&config.model_id))
+        .or_else(|| capability::lookup(config.scheme, &config.model_id))
        .unwrap_or_else(|| scheme.default_capability());
    let base_url = effective_base_url(&scheme, config);
    Ok(Box::new(HttpTransport::new(