mirror of
https://github.com/fawney19/Aether.git
synced 2026-10-08 18:37:46 +08:00
80 lines
2.8 KiB
Rust
80 lines
2.8 KiB
Rust
|
use std::collections::BTreeMap;
|
||
|
|
|
||
|
|
use aether_data_contracts::repository::provider_catalog::StoredProviderCatalogEndpoint;
|
||
|
|
|
||
|
|
use crate::capability::ProviderPoolCapabilities;
|
||
|
|
use crate::provider::{
|
||
|
|
provider_pool_endpoint_format_matches, provider_pool_matching_endpoint, ProviderPoolAdapter,
|
||
|
|
};
|
||
|
|
use crate::quota_refresh::ProviderPoolQuotaRequestSpec;
|
||
|
|
|
||
|
|
pub const CLAUDE_CODE_OAUTH_USAGE_URL: &str = "https://api.anthropic.com/api/oauth/usage";
|
||
|
|
pub const CLAUDE_CODE_OAUTH_BETA: &str = "oauth-2025-04-20";
|
||
|
|
pub const CLAUDE_CODE_USAGE_USER_AGENT: &str = "claude-code/2.1.284";
|
||
|
|
|
||
|
|
#[derive(Debug, Clone, Default)]
|
||
|
|
pub struct ClaudeCodeProviderPoolAdapter;
|
||
|
|
|
||
|
|
impl ProviderPoolAdapter for ClaudeCodeProviderPoolAdapter {
|
||
|
|
fn provider_type(&self) -> &'static str {
|
||
|
|
"claude_code"
|
||
|
|
}
|
||
|
|
|
||
|
|
fn capabilities(&self) -> ProviderPoolCapabilities {
|
||
|
|
ProviderPoolCapabilities {
|
||
|
|
quota_refresh: true,
|
||
|
|
..ProviderPoolCapabilities::default()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
fn quota_refresh_endpoint(
|
||
|
|
&self,
|
||
|
|
endpoints: &[StoredProviderCatalogEndpoint],
|
||
|
|
include_inactive: bool,
|
||
|
|
) -> Option<StoredProviderCatalogEndpoint> {
|
||
|
|
provider_pool_matching_endpoint(endpoints, include_inactive, |endpoint| {
|
||
|
|
provider_pool_endpoint_format_matches(endpoint, "claude:messages")
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
fn quota_refresh_missing_endpoint_message(&self) -> String {
|
||
|
|
"找不到有效的 claude:messages 端点".to_string()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
/// Builds the `GET /api/oauth/usage` request that reports the account's 5h / 7d
|
||
|
|
/// utilization windows. The URL is fixed (the origin allowlist only accepts
|
||
|
|
/// `api.anthropic.com`), independent of the inference endpoint's base URL.
|
||
|
|
pub fn build_claude_code_pool_quota_request(
|
||
|
|
key_id: &str,
|
||
|
|
authorization: (String, String),
|
||
|
|
) -> ProviderPoolQuotaRequestSpec {
|
||
|
|
let headers = BTreeMap::from([
|
||
|
|
("authorization".to_string(), authorization.1),
|
||
|
|
("accept".to_string(), "application/json".to_string()),
|
||
|
|
("content-type".to_string(), "application/json".to_string()),
|
||
|
|
(
|
||
|
|
"anthropic-beta".to_string(),
|
||
|
|
CLAUDE_CODE_OAUTH_BETA.to_string(),
|
||
|
|
),
|
||
|
|
(
|
||
|
|
"user-agent".to_string(),
|
||
|
|
CLAUDE_CODE_USAGE_USER_AGENT.to_string(),
|
||
|
|
),
|
||
|
|
]);
|
||
|
|
|
||
|
|
ProviderPoolQuotaRequestSpec {
|
||
|
|
request_id: format!("claude-code-quota:{key_id}"),
|
||
|
|
provider_name: "claude_code".to_string(),
|
||
|
|
quota_kind: "claude_code".to_string(),
|
||
|
|
method: "GET".to_string(),
|
||
|
|
url: CLAUDE_CODE_OAUTH_USAGE_URL.to_string(),
|
||
|
|
headers,
|
||
|
|
content_type: None,
|
||
|
|
json_body: None,
|
||
|
|
client_api_format: "claude:messages".to_string(),
|
||
|
|
provider_api_format: "claude_code:oauth_usage".to_string(),
|
||
|
|
model_name: Some("oauth_usage".to_string()),
|
||
|
|
}
|
||
|
|
}
|