Skip to content
Merged
Show file tree
Hide file tree
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
21 changes: 21 additions & 0 deletions crates/goose-server/src/commands/agent.rs
Original file line number Diff line number Diff line change
Expand Up @@ -33,6 +33,27 @@ pub async fn run() -> Result<()> {
std::env::var("GOOSE_SERVER__SECRET_KEY").unwrap_or_else(|_| "test".to_string());

let new_agent = Agent::new();

// Initialize provider like the CLI does
let config = goose::config::Config::global();

let provider_name: String = config
Comment thread
michaelneale marked this conversation as resolved.
Outdated
.get_param("GOOSE_PROVIDER")
.expect("No provider configured. Run 'goose configure' first");

let model_name: String = config
.get_param("GOOSE_MODEL")
.expect("No model configured. Run 'goose configure' first");

let model_config = goose::model::ModelConfig::new(&model_name)
.expect("Failed to create model configuration");

let provider = goose::providers::create(&provider_name, model_config)
.expect("Failed to create provider");

new_agent.update_provider(provider).await
.expect("Failed to update agent provider");

let agent_ref = Arc::new(new_agent);

let app_state = state::AppState::new(agent_ref.clone());
Expand Down
26 changes: 25 additions & 1 deletion crates/goose-server/src/state.rs
Original file line number Diff line number Diff line change
Expand Up @@ -54,7 +54,31 @@ impl AppState {

pub async fn reset(&self) {
let mut agent = self.agent.write().await;
*agent = Arc::new(Agent::new());
let new_agent = Agent::new();

// Re-initialize provider like we do at startup
let config = goose::config::Config::global();

if let (Ok(provider_name), Ok(model_name)) = (
config.get_param::<String>("GOOSE_PROVIDER"),
config.get_param::<String>("GOOSE_MODEL")
) {
if let Ok(model_config) = goose::model::ModelConfig::new(&model_name) {
if let Ok(provider) = goose::providers::create(&provider_name, model_config) {
if let Err(e) = new_agent.update_provider(provider).await {
tracing::error!("Failed to update agent provider during reset: {}", e);
}
} else {
tracing::error!("Failed to create provider during reset");
}
} else {
tracing::error!("Failed to create model config during reset");
}
} else {
tracing::warn!("No provider/model configured during reset");
}

*agent = Arc::new(new_agent);
}

pub async fn mark_recipe_run_if_absent(&self, session_id: &str) -> bool {
Expand Down
Loading