aljaz/goose
1
0
Fork 0
mirror of https://github.com/aljazceru/goose.git synced 2025-12-18 22:54:24 +01:00
Code Issues Packages Projects Releases Wiki Activity

Add -with-remote-extension (#2062)

Browse Source
This commit is contained in:
Jim Bennett
2025-04-07 13:42:38 -07:00
committed by GitHub
parent 63623733dc
commit 050a8f2f42
21 changed files with 121 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
crates/goose-cli/src/cli.rs
View File

@@ -128,6 +128,16 @@ enum Command {
)]
extension: Vec<String>,
/// Add remote extensions with a URL
#[arg(
long = "with-remote-extension",
value_name = "URL",
help = "Add remote extensions (can be specified multiple times)",
long_help = "Add remote extensions from a URL. Can be specified multiple times. Format: 'url...'",
action = clap::ArgAction::Append
)]
remote_extension: Vec<String>,
/// Add builtin extensions by name
#[arg(
long = "with-builtin",
@@ -203,6 +213,16 @@ enum Command {
)]
extension: Vec<String>,
/// Add remote extensions
#[arg(
long = "with-remote-extension",
value_name = "URL",
help = "Add remote extensions (can be specified multiple times)",
long_help = "Add remote extensions. Can be specified multiple times. Format: 'url...'",
action = clap::ArgAction::Append
)]
remote_extension: Vec<String>,
/// Add builtin extensions by name
#[arg(
long = "with-builtin",
@@ -323,6 +343,7 @@ pub async fn cli() -> Result<()> {
resume,
debug,
extension,
remote_extension,
builtin,
}) => {
match command {
@@ -336,6 +357,7 @@ pub async fn cli() -> Result<()> {
identifier.map(extract_identifier),
resume,
extension,
remote_extension,
builtin,
debug,
)
@@ -357,6 +379,7 @@ pub async fn cli() -> Result<()> {
resume,
debug,
extension,
remote_extension,
builtin,
}) => {
let contents = match (instructions, input_text) {
@@ -385,6 +408,7 @@ pub async fn cli() -> Result<()> {
identifier.map(extract_identifier),
resume,
extension,
remote_extension,
builtin,
debug,
)
@@ -468,7 +492,7 @@ pub async fn cli() -> Result<()> {
return Ok(());
} else {
// Run session command by default
let mut session = build_session(None, false, vec![], vec![], false).await;
let mut session = build_session(None, false, vec![], vec![], vec![], false).await;
setup_logging(
session.session_file().file_stem().and_then(|s| s.to_str()),
None,

1
crates/goose-cli/src/commands/bench.rs
View File

@@ -85,6 +85,7 @@ async fn run_eval(
None,
false,
requirements.external,
requirements.remote,
requirements.builtin,
false,
)

9
crates/goose-cli/src/session/builder.rs
View File

@@ -14,6 +14,7 @@ pub async fn build_session(
identifier: Option<Identifier>,
resume: bool,
extensions: Vec<String>,
remote_extensions: Vec<String>,
builtins: Vec<String>,
debug: bool,
) -> Session {
@@ -126,6 +127,14 @@ pub async fn build_session(
}
}
// Add remote extensions if provided
for extension_str in remote_extensions {
if let Err(e) = session.add_remote_extension(extension_str).await {
eprintln!("Failed to start extension: {}", e);
process::exit(1);
}
}
// Add builtin extensions
for builtin in builtins {
if let Err(e) = session.add_builtin(builtin).await {

31
crates/goose-cli/src/session/mod.rs
View File

@@ -172,6 +172,37 @@ impl Session {
Ok(())
}
/// Add a remote extension to the session
///
/// # Arguments
/// * `extension_url` - URL of the server
pub async fn add_remote_extension(&mut self, extension_url: String) -> Result<()> {
let name: String = rand::thread_rng()
.sample_iter(&Alphanumeric)
.take(8)
.map(char::from)
.collect();
let config = ExtensionConfig::Sse {
name,
uri: extension_url,
envs: Envs::new(HashMap::new()),
description: Some(goose::config::DEFAULT_EXTENSION_DESCRIPTION.to_string()),
// TODO: should set timeout
timeout: Some(goose::config::DEFAULT_EXTENSION_TIMEOUT),
};
self.agent
.add_extension(config)
.await
.map_err(|e| anyhow::anyhow!("Failed to start extension: {}", e))?;
// Invalidate the completion cache when a new extension is added
self.invalidate_completion_cache().await;
Ok(())
}
/// Add a builtin extension to the session
///
/// # Arguments