From 0498d266c9d70d9f004b5012bd3323c4b58d9411 Mon Sep 17 00:00:00 2001 From: Jay4242 <96741732+Jay4242@users.noreply.github.com> Date: Tue, 25 Mar 2025 12:55:26 -0700 Subject: [PATCH] feat: Allow setting OpenAI timeout from config (#1819) --- crates/goose/src/providers/openai.rs | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/crates/goose/src/providers/openai.rs b/crates/goose/src/providers/openai.rs index 6a6c45f0..5031e490 100644 --- a/crates/goose/src/providers/openai.rs +++ b/crates/goose/src/providers/openai.rs @@ -54,8 +54,9 @@ impl OpenAiProvider { .unwrap_or_else(|_| "v1/chat/completions".to_string()); let organization: Option = config.get_param("OPENAI_ORGANIZATION").ok(); let project: Option = config.get_param("OPENAI_PROJECT").ok(); + let timeout_secs: u64 = config.get_param("OPENAI_TIMEOUT").unwrap_or(600); let client = Client::builder() - .timeout(Duration::from_secs(600)) + .timeout(Duration::from_secs(timeout_secs)) .build()?; Ok(Self { @@ -116,6 +117,7 @@ impl Provider for OpenAiProvider { ConfigKey::new("OPENAI_BASE_PATH", true, false, Some("v1/chat/completions")), ConfigKey::new("OPENAI_ORGANIZATION", false, false, None), ConfigKey::new("OPENAI_PROJECT", false, false, None), + ConfigKey::new("OPENAI_TIMEOUT", false, false, Some("600")), ], ) }