feat(goose): support customizing extension timeout (#1428)
This commit is contained in:
@@ -1,6 +1,7 @@
|
||||
use dotenv::dotenv;
|
||||
use futures::StreamExt;
|
||||
use goose::agents::{AgentFactory, ExtensionConfig};
|
||||
use goose::config::DEFAULT_EXTENSION_TIMEOUT;
|
||||
use goose::message::Message;
|
||||
use goose::providers::databricks::DatabricksProvider;
|
||||
|
||||
@@ -14,7 +15,11 @@ async fn main() {
|
||||
// Setup an agent with the developer extension
|
||||
let mut agent = AgentFactory::create("reference", provider).expect("default should exist");
|
||||
|
||||
let config = ExtensionConfig::stdio("developer", "./target/debug/developer");
|
||||
let config = ExtensionConfig::stdio(
|
||||
"developer",
|
||||
"./target/debug/developer",
|
||||
DEFAULT_EXTENSION_TIMEOUT,
|
||||
);
|
||||
agent.add_extension(config).await.unwrap();
|
||||
|
||||
println!("Extensions:");
|
||||
|
||||
Reference in New Issue
Block a user