diff --git a/src/internal/ai/runtime/services.rs b/src/internal/ai/runtime/services.rs index 49aab9fb8..8a8e3bd3d 100644 --- a/src/internal/ai/runtime/services.rs +++ b/src/internal/ai/runtime/services.rs @@ -29,7 +29,8 @@ use crate::internal::ai::{ handlers::{ ApplyPatchHandler, GrepFilesHandler, ListDirHandler, PlanHandler, ReadFileHandler, RequestUserInputHandler, SearchFilesHandler, ShellHandler, SubmitIntentDraftHandler, - SubmitPlanDraftHandler, WebSearchHandler, register_semantic_handlers, + SubmitPlanDraftHandler, SubmitTaskCompleteHandler, WebSearchHandler, + register_semantic_handlers, }, }, }; @@ -126,7 +127,8 @@ impl CodeAgentServicesBuilder { Arc::new(RequestUserInputHandler::new(user_input_tx)), ) .register("submit_intent_draft", Arc::new(SubmitIntentDraftHandler)) - .register("submit_plan_draft", Arc::new(SubmitPlanDraftHandler)); + .register("submit_plan_draft", Arc::new(SubmitPlanDraftHandler)) + .register("submit_task_complete", Arc::new(SubmitTaskCompleteHandler)); CodeAgentServices { profile: CodeAgentLaunchProfile::WebHeadless, diff --git a/tests/code_tool_acl_test.rs b/tests/code_tool_acl_test.rs index b7d227a60..901a05b94 100644 --- a/tests/code_tool_acl_test.rs +++ b/tests/code_tool_acl_test.rs @@ -86,6 +86,7 @@ fn dev_context_filter_keeps_all_registered_tools() { "update_plan", "submit_intent_draft", "submit_plan_draft", + "submit_task_complete", ] { assert!( allowed.iter().any(|name| name == required),