Ling0925
/
buster
mirror of https://github.com/buster-so/buster.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

low reasoning with conditions

This commit is contained in:
dal 2025-04-16 09:29:24 -06:00
parent 4a832d984a
commit 3dacf6ff1f
No known key found for this signature in database
GPG Key ID: 16F4B0E1E9F61122
2 changed files with 24 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/libs/agents/src/agent.rs
View File

@ -618,7 +618,7 @@ impl Agent {
session_id: thread.id.to_string(),
trace_id: Uuid::new_v4().to_string(),
}),
reasoning_effort: Some("medium".to_string()),
reasoning_effort: Some("low".to_string()),
..Default::default()
};

24
api/libs/agents/src/agents/buster_multi_agent.rs
View File

@ -139,6 +139,28 @@ impl BusterMultiAgent {
&& state.contains_key("data_context")
});
// Define the new condition for the Done tool
let done_condition = Some(|state: &HashMap<String, Value>| -> bool {
let review_needed = state.get("review_needed")
.and_then(Value::as_bool)
.unwrap_or(false);
// Check if all todos are marked as complete
// Assumes 'todos' is an array of objects, each with a 'completed' boolean field
let all_todos_complete = state.get("todos")
.and_then(Value::as_array)
.map(|todos| {
todos.iter().all(|todo| {
todo.get("completed")
.and_then(Value::as_bool)
.unwrap_or(false) // Treat missing 'completed' as false
})
})
.unwrap_or(false); // If 'todos' doesn't exist or isn't an array, assume not all complete
review_needed || all_todos_complete
});
// Add tools to the agent with their conditions
self.agent
.add_tool(
@ -207,7 +229,7 @@ impl BusterMultiAgent {
.add_tool(
done_tool.get_name(),
done_tool.into_tool_call_executor(),
after_search_condition.clone(), // Use after_search_condition instead of None
done_condition, // Use the new done_condition
)
.await;
self.agent