mirror of
https://github.com/farion1231/cc-switch.git
synced 2026-04-27 05:01:06 +08:00
55301abc00
- Create new `proxy/transform/` module with clean architecture: - `traits.rs`: FormatTransformer trait definition - `format.rs`: ApiFormat enum (Anthropic, OpenAI, Gemini) - `registry.rs`: TransformerRegistry with global instance - `config.rs`: TransformConfig from Provider settings - `anthropic_openai/`: Bidirectional Anthropic ↔ OpenAI converters - Move streaming transform from providers/ to transform/ - Remove legacy transform code from providers/adapter.rs and claude.rs - Update handlers.rs to use new transform module This refactor improves maintainability and makes it easier to add new format converters in the future.
86 lines
2.6 KiB
Rust
86 lines
2.6 KiB
Rust
//! Provider Adapter Trait
|
||
//!
|
||
//! 定义供应商适配器的统一接口，抽象不同上游供应商的处理逻辑。
|
||
|
||
use super::auth::AuthInfo;
|
||
use crate::provider::Provider;
|
||
use crate::proxy::error::ProxyError;
|
||
use reqwest::RequestBuilder;
|
||
|
||
/// 供应商适配器 Trait
|
||
///
|
||
/// 所有供应商适配器都需要实现此 trait，提供统一的接口来处理：
|
||
/// - URL 构建
|
||
/// - 认证信息提取和头部注入
|
||
/// - 请求/响应格式转换（可选）
|
||
///
|
||
/// # 示例
|
||
///
|
||
/// ```ignore
|
||
/// pub struct ClaudeAdapter;
|
||
///
|
||
/// impl ProviderAdapter for ClaudeAdapter {
|
||
/// fn name(&self) -> &'static str { "Claude" }
|
||
///
|
||
/// fn extract_base_url(&self, provider: &Provider) -> Result<String, ProxyError> {
|
||
/// // 从 provider 配置中提取 base_url
|
||
/// }
|
||
///
|
||
/// fn extract_auth(&self, provider: &Provider) -> Option<AuthInfo> {
|
||
/// // 从 provider 配置中提取认证信息
|
||
/// }
|
||
///
|
||
/// fn build_url(&self, base_url: &str, endpoint: &str) -> String {
|
||
/// format!("{}{}", base_url.trim_end_matches('/'), endpoint)
|
||
/// }
|
||
///
|
||
/// fn add_auth_headers(&self, request: RequestBuilder, auth: &AuthInfo) -> RequestBuilder {
|
||
/// // 添加认证头
|
||
/// }
|
||
/// }
|
||
/// ```
|
||
pub trait ProviderAdapter: Send + Sync {
|
||
/// 适配器名称（用于日志和调试）
|
||
fn name(&self) -> &'static str;
|
||
|
||
/// 从 Provider 配置中提取 base_url
|
||
///
|
||
/// # Arguments
|
||
/// * `provider` - Provider 配置
|
||
///
|
||
/// # Returns
|
||
/// * `Ok(String)` - 提取到的 base_url（已去除尾部斜杠）
|
||
/// * `Err(ProxyError)` - 提取失败
|
||
fn extract_base_url(&self, provider: &Provider) -> Result<String, ProxyError>;
|
||
|
||
/// 从 Provider 配置中提取认证信息
|
||
///
|
||
/// # Arguments
|
||
/// * `provider` - Provider 配置
|
||
///
|
||
/// # Returns
|
||
/// * `Some(AuthInfo)` - 提取到的认证信息
|
||
/// * `None` - 未找到认证信息
|
||
fn extract_auth(&self, provider: &Provider) -> Option<AuthInfo>;
|
||
|
||
/// 构建请求 URL
|
||
///
|
||
/// # Arguments
|
||
/// * `base_url` - 基础 URL
|
||
/// * `endpoint` - 请求端点（如 `/v1/messages`）
|
||
///
|
||
/// # Returns
|
||
/// 完整的请求 URL
|
||
fn build_url(&self, base_url: &str, endpoint: &str) -> String;
|
||
|
||
/// 添加认证头到请求
|
||
///
|
||
/// # Arguments
|
||
/// * `request` - reqwest RequestBuilder
|
||
/// * `auth` - 认证信息
|
||
///
|
||
/// # Returns
|
||
/// 添加了认证头的 RequestBuilder
|
||
fn add_auth_headers(&self, request: RequestBuilder, auth: &AuthInfo) -> RequestBuilder;
|
||
}
|