Fix Rig tool continuation and plan creation
This commit is contained in:
@@ -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();
|
||||
|
||||
Reference in New Issue
Block a user