fix: address compilation issue in main (#6776)

This commit is contained in:
Alex Hancock
2026-01-28 12:39:53 -05:00
committed by GitHub
parent 96dc2fd9b1
commit be4e6635d3
2 changed files with 10 additions and 9 deletions
+6 -4
View File
@@ -9,7 +9,6 @@ mod prompt;
mod task_execution_display;
mod thinking;
use crate::cli::StreamableHttpOptions;
use crate::session::task_execution_display::{
format_task_execution_notification, TASK_EXECUTION_NOTIFICATION_TYPE,
};
@@ -292,7 +291,7 @@ impl CliSession {
})
}
pub fn parse_streamable_http_extension(extension_url: &str) -> ExtensionConfig {
pub fn parse_streamable_http_extension(extension_url: &str, timeout: u64) -> ExtensionConfig {
ExtensionConfig::StreamableHttp {
name: String::new(),
uri: extension_url.to_string(),
@@ -300,7 +299,7 @@ impl CliSession {
env_keys: Vec::new(),
headers: HashMap::new(),
description: goose::config::DEFAULT_EXTENSION_DESCRIPTION.to_string(),
timeout: Some(goose::config::DEFAULT_EXTENSION_TIMEOUT),
timeout: Some(timeout),
bundled: None,
available_tools: Vec::new(),
}
@@ -357,7 +356,10 @@ impl CliSession {
}
pub async fn add_streamable_http_extension(&mut self, extension_url: String) -> Result<()> {
let config = Self::parse_streamable_http_extension(&extension_url);
let config = Self::parse_streamable_http_extension(
&extension_url,
goose::config::DEFAULT_EXTENSION_TIMEOUT,
);
self.add_and_persist_extensions(vec![config]).await
}