Fix Rig tool continuation and plan creation

This commit is contained in:
2026-08-14 09:01:53 -05:00
parent 5737f8342c
commit f4a04d0240
12 changed files with 642 additions and 155 deletions
+25
View File
@@ -124,6 +124,31 @@ fn builds_a_rig_turn_directly_from_galaxy_request_state() {
));
}
#[test]
fn normal_turn_advertises_plan_creation_and_corrects_false_unavailability_claims() {
let mut params = RequestParams::new_for_test();
params.planning_enabled = true;
params.input = vec![user_query("Please create a plan, and let's review.")];
let prepared = prepare_rig_turn(
&config(),
params,
vec![ToolType::CreateDocuments],
Vec::new(),
);
let prompt = prepared.request.system_prompt.expect("system prompt");
assert!(prepared
.request
.tools
.iter()
.any(|tool| tool.name == "create_plan"));
assert!(prompt.contains("Plan document creation is available through `create_plan`"));
assert!(prompt.contains("do not merely return the plan as prose"));
assert!(prompt.contains("no plan-creation tool is available"));
assert!(prompt.contains("do not implement it until they approve"));
}
#[test]
fn no_tools_turn_flattens_historical_tool_protocol_messages() {
let mut params = RequestParams::new_for_test();