diff --git a/src/crates/assembly/core/src/agentic/agents/definitions/modes/cowork.rs b/src/crates/assembly/core/src/agentic/agents/definitions/modes/cowork.rs index d94da2f30..2fb8f9458 100644 --- a/src/crates/assembly/core/src/agentic/agents/definitions/modes/cowork.rs +++ b/src/crates/assembly/core/src/agentic/agents/definitions/modes/cowork.rs @@ -30,6 +30,9 @@ impl CoworkMode { // Clarification + planning helpers "AskUserQuestion".to_string(), "TodoWrite".to_string(), + "get_goal".to_string(), + "create_goal".to_string(), + "update_goal".to_string(), "Task".to_string(), "ListModels".to_string(), "AgentWait".to_string(), @@ -109,6 +112,14 @@ mod tests { use super::CoworkMode; use crate::agentic::agents::Agent; + #[test] + fn cowork_mode_includes_goal_lifecycle_tools_in_defaults() { + let tools = CoworkMode::new().default_tools(); + for tool in ["get_goal", "create_goal", "update_goal"] { + assert!(tools.contains(&tool.to_string())); + } + } + #[test] fn cowork_mode_includes_miniapp_lifecycle_tools_in_defaults() { let tools = CoworkMode::new().default_tools();