Fix Rig tool continuation and plan creation

2026-08-14 09:01:53 -05:00
parent 5737f8342c
commit f4a04d0240
12 changed files with 642 additions and 155 deletions
@@ -79,6 +79,16 @@ fn supported_tools_expose_local_subagents_without_hosted_orchestration_tools() {
    assert!(!supported_tools.contains(&api::ToolType::StartAgentV2));
 }

+#[test]
+fn supported_tools_include_plan_document_capabilities() {
+    let params = request_params_with_ask_user_question_enabled(false);
+    let supported_tools = get_supported_tools(&params);
+
+    assert!(supported_tools.contains(&api::ToolType::ReadDocuments));
+    assert!(supported_tools.contains(&api::ToolType::CreateDocuments));
+    assert!(supported_tools.contains(&api::ToolType::EditDocuments));
+}
+
 #[test]
 fn supported_tools_omit_subagents_when_orchestration_is_disabled() {
    let params = request_params_with_ask_user_question_enabled(false);