ai-space/goose
3
0
Fork 0
mirror of https://github.com/aaif-goose/goose.git synced 2026-07-03 14:10:03 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix(extension_manager): set TCP_USER_TIMEOUT on streamable HTTP clients (#9207)

Browse Source 
Signed-off-by: Hugues Clouâtre <hugues@linux.com>
This commit is contained in:
Hugues Clouâtre
2026-05-18 10:15:42 -04:00
committed by GitHub
parent 7bce91e5eb
commit c3b4f78daf
1 changed files with 16 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
crates/goose/src/agents/extension_manager.rs
+16 -6
View File
@@ -517,8 +517,13 @@ async fn connect_with_auth(
) -> ExtensionResult<Box<dyn McpClientTrait>> {
let mut auth_headers = HeaderMap::new();
auth_headers.insert(reqwest::header::USER_AGENT, GOOSE_USER_AGENT);
let auth_http_client = reqwest::Client::builder()
.default_headers(auth_headers)
#[allow(unused_mut)]
let mut auth_client_builder = reqwest::Client::builder().default_headers(auth_headers);
#[cfg(target_os = "linux")]
{
auth_client_builder = auth_client_builder.tcp_user_timeout(Some(timeout));
}
let auth_http_client = auth_client_builder
.build()
.map_err(|_| ExtensionError::ConfigError("could not construct http client".to_string()))?;
let auth_client = AuthClient::new(auth_http_client, auth_manager);
@@ -587,8 +592,15 @@ async fn create_streamable_http_client(
);
}
let http_client = reqwest::Client::builder()
.default_headers(default_headers)
let timeout_duration = Duration::from_secs(resolve_timeout(timeout));
#[allow(unused_mut)]
let mut http_client_builder = reqwest::Client::builder().default_headers(default_headers);
#[cfg(target_os = "linux")]
{
http_client_builder = http_client_builder.tcp_user_timeout(Some(timeout_duration));
}
let http_client = http_client_builder
.build()
.map_err(|_| ExtensionError::ConfigError("could not construct http client".to_string()))?;
@@ -597,8 +609,6 @@ async fn create_streamable_http_client(
StreamableHttpClientTransportConfig::with_uri(uri),
);
let timeout_duration = Duration::from_secs(resolve_timeout(timeout));
// If we have stored OAuth credentials, try refreshing and connecting directly.
// This avoids the unnecessary 401 → browser re-auth cycle on every new session.
let credential_store = GooseCredentialStore::new(name.to_string());