diff --git a/rust/crates/api/src/error.rs b/rust/crates/api/src/error.rs index aa8e5fc1..60ae1f5f 100644 --- a/rust/crates/api/src/error.rs +++ b/rust/crates/api/src/error.rs @@ -582,6 +582,7 @@ mod tests { body: String::new(), retryable: false, suggested_action: None, + retry_after: None, }; assert!(error.is_context_window_failure()); diff --git a/rust/crates/api/src/lib.rs b/rust/crates/api/src/lib.rs index a5b7d726..e96e92f8 100644 --- a/rust/crates/api/src/lib.rs +++ b/rust/crates/api/src/lib.rs @@ -12,9 +12,8 @@ pub use client::{ }; pub use error::ApiError; pub use http_client::{ - TimeoutConfig, build_http_client, build_http_client_or_default, build_http_client_with, - build_http_client_with_opts, ProxyConfig, + build_http_client_with_opts, ProxyConfig, TimeoutConfig, }; pub use prompt_cache::{ CacheBreakEvent, PromptCache, PromptCacheConfig, PromptCachePaths, PromptCacheRecord, diff --git a/rust/crates/api/src/providers/anthropic.rs b/rust/crates/api/src/providers/anthropic.rs index 72001780..dd458195 100644 --- a/rust/crates/api/src/providers/anthropic.rs +++ b/rust/crates/api/src/providers/anthropic.rs @@ -467,7 +467,8 @@ impl AnthropicClient { break; } - let delay = if let Some(retry_after) = last_error.as_ref().and_then(|e| e.retry_after()) { + let delay = if let Some(retry_after) = last_error.as_ref().and_then(|e| e.retry_after()) + { retry_after } else { self.jittered_backoff_for_attempt(attempts)? @@ -909,7 +910,10 @@ async fn expect_success(response: reqwest::Response) -> Result Option { +fn parse_retry_after( + headers: &reqwest::header::HeaderMap, + status: reqwest::StatusCode, +) -> Option { if status != reqwest::StatusCode::TOO_MANY_REQUESTS { return None; } diff --git a/rust/crates/api/src/providers/openai_compat.rs b/rust/crates/api/src/providers/openai_compat.rs index 00326f52..cfeb4c3c 100644 --- a/rust/crates/api/src/providers/openai_compat.rs +++ b/rust/crates/api/src/providers/openai_compat.rs @@ -1667,7 +1667,10 @@ async fn expect_success(response: reqwest::Response) -> Result Option { +fn parse_retry_after( + headers: &reqwest::header::HeaderMap, + status: reqwest::StatusCode, +) -> Option { if status != reqwest::StatusCode::TOO_MANY_REQUESTS { return None; } diff --git a/rust/crates/runtime/src/config.rs b/rust/crates/runtime/src/config.rs index 46639b75..b31c0168 100644 --- a/rust/crates/runtime/src/config.rs +++ b/rust/crates/runtime/src/config.rs @@ -1184,10 +1184,8 @@ fn parse_optional_api_timeout_config(root: &JsonValue) -> Result