122 lines
5.7 KiB
Rust
122 lines
5.7 KiB
Rust
//! LLM モデル宣言型
|
||
//!
|
||
//! Pod マニフェストの `[model]` セクションで記述する型。`ref`（プロバイダ
|
||
//! とモデルを両方指し示す短縮形）と inline 指定（`scheme` / `model_id`
|
||
//! 直書き）の両方を受け入れるため、すべてのフィールドを `Option` として
|
||
//! 持つ 1 つの型 [`ModelManifest`] に統合している。実解決（ref をプロバイダ
|
||
//! カタログ / モデルカタログから引いて `scheme` や `model_id` を埋める）
|
||
//! は `crates/provider` の責務で、本モジュールはデータ表現のみを提供する。
|
||
//!
|
||
//! 同じ型を partial（カスケード層）と完成形（最終マニフェスト）の両方で
|
||
//! 使うことで、merge と最終変換の重複を避ける。
|
||
|
||
use std::path::PathBuf;
|
||
|
||
use serde::{Deserialize, Serialize};
|
||
|
||
// `ModelCapability` は `llm-worker` 側に定義される runtime 構造だが、
|
||
// マニフェストで任意に override できるよう型だけ再エクスポートする。
|
||
pub use llm_worker::llm_client::capability::{ModelCapability, ReasoningControl, ReasoningEffort};
|
||
|
||
/// Pod マニフェストの `[model]` セクション。
|
||
///
|
||
/// - ref だけ書く: `[model] ref = "anthropic/claude-sonnet-4-6"`
|
||
/// - ref + 一部 override: ref で基底を引き、`auth` 等だけ書き換え
|
||
/// - 完全 inline: `ref` を省略して `scheme` / `model_id` / `auth` を直書き
|
||
///
|
||
/// どの形が有効かの判定は `provider::resolve_model_manifest` が担う。
|
||
/// 本クレートは「どこから取るか」を表現するだけで、未設定かどうかを
|
||
/// 理由にした hard error は出さない。
|
||
#[derive(Debug, Clone, Default, Serialize, Deserialize, PartialEq)]
|
||
pub struct ModelManifest {
|
||
/// `<provider_id>/<model_id_in_ref>` 形式のカタログ参照。`/` の
|
||
/// 最初の 1 文字目で split し provider カタログを引く。
|
||
/// OpenRouter の `anthropic/claude-sonnet-4` のように `/` を含む
|
||
/// model_id は `openrouter/anthropic/claude-sonnet-4` と書く
|
||
/// （provider 側で最初の `/` のみ split するため）。
|
||
#[serde(default, rename = "ref", skip_serializing_if = "Option::is_none")]
|
||
pub ref_: Option<String>,
|
||
/// wire format の明示指定。ref 未指定時は必須、ref 指定時は override。
|
||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||
pub scheme: Option<SchemeKind>,
|
||
/// API のベース URL。scheme の既定値を override する。
|
||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||
pub base_url: Option<String>,
|
||
/// プロバイダが受け付けるモデル ID。ref 未指定時は必須。
|
||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||
pub model_id: Option<String>,
|
||
/// 認証方式。ref 未指定時は必須、ref 指定時は override。
|
||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||
pub auth: Option<AuthRef>,
|
||
/// モデル能力の明示指定。未指定時はモデルカタログ → provider
|
||
/// `default_capability` → scheme 既定の順で解決される。
|
||
#[serde(default, skip_serializing_if = "Option::is_none")]
|
||
pub capability: Option<ModelCapability>,
|
||
}
|
||
|
||
impl ModelManifest {
|
||
/// `upper` を `self` に上書きマージする。マニフェスト cascade 向け
|
||
/// （builtin → user → project → overlay の優先順位で呼ばれる）。
|
||
pub fn merge(self, upper: Self) -> Self {
|
||
Self {
|
||
ref_: upper.ref_.or(self.ref_),
|
||
scheme: upper.scheme.or(self.scheme),
|
||
base_url: upper.base_url.or(self.base_url),
|
||
model_id: upper.model_id.or(self.model_id),
|
||
auth: upper.auth.or(self.auth),
|
||
capability: upper.capability.or(self.capability),
|
||
}
|
||
}
|
||
}
|
||
|
||
/// サポートする wire scheme の種類。
|
||
#[derive(Debug, Clone, Copy, Serialize, Deserialize, PartialEq, Eq)]
|
||
#[serde(rename_all = "snake_case")]
|
||
pub enum SchemeKind {
|
||
/// Anthropic Messages API (`/v1/messages`)。Ollama `/v1/messages` もこれで扱う
|
||
Anthropic,
|
||
/// OpenAI Chat Completions (`/v1/chat/completions`)。OpenAI 互換ルーター共通枠
|
||
OpenaiChat,
|
||
/// OpenAI Responses API (`/v1/responses`)。別チケットで scheme 新設予定
|
||
OpenaiResponses,
|
||
/// Google Gemini (`/v1beta/models/...:streamGenerateContent`)
|
||
Gemini,
|
||
}
|
||
|
||
/// 認証の参照。
|
||
///
|
||
/// 実際のトークン値の解決（env / file 読取、OAuth refresh 等）は
|
||
/// `crates/provider` で行う。ここはあくまで「どこから取るか」の宣言。
|
||
#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
|
||
#[serde(tag = "kind", rename_all = "snake_case")]
|
||
pub enum AuthRef {
|
||
/// 認証不要（ローカル Ollama 等）
|
||
#[default]
|
||
None,
|
||
/// API key。env / file のいずれか（両方指定された場合は env が優先）
|
||
ApiKey {
|
||
/// 環境変数名。未指定のときは scheme ごとの既定（`INSOMNIA_API_KEY_*`）
|
||
#[serde(default)]
|
||
env: Option<String>,
|
||
/// key を書き込んだファイル（絶対パス）
|
||
#[serde(default)]
|
||
file: Option<PathBuf>,
|
||
},
|
||
/// ChatGPT OAuth（`~/.codex/auth.json`）。実装は `llm-auth-codex-oauth` チケット
|
||
#[serde(rename = "codex_oauth")]
|
||
CodexOAuth,
|
||
}
|
||
|
||
impl SchemeKind {
|
||
/// 既定の環境変数名（`INSOMNIA_API_KEY_*`）。
|
||
///
|
||
/// `AuthRef::ApiKey { env: None, .. }` の env 未指定時に使う。
|
||
pub fn default_env_var(self) -> &'static str {
|
||
match self {
|
||
Self::Anthropic => "INSOMNIA_API_KEY_ANTHROPIC",
|
||
Self::OpenaiChat | Self::OpenaiResponses => "INSOMNIA_API_KEY_OPENAI",
|
||
Self::Gemini => "INSOMNIA_API_KEY_GEMINI",
|
||
}
|
||
}
|
||
}
|