72 lines
2.6 KiB
C
72 lines
2.6 KiB
C
|
#ifndef HUMANUS_AGENT_PLANNING_H
|
||
|
|
#define HUMANUS_AGENT_PLANNING_H
|
||
|
|
|
||
|
|
#include "toolcall.h"
|
||
|
|
#include "../tool/planning.h"
|
||
|
|
|
||
|
|
namespace humanus {
|
||
|
|
|
||
|
|
/**
|
||
|
|
* An agent that creates and manages plans to solve tasks.
|
||
|
|
|
||
|
|
* This agent uses a planning tool to create and manage structured plans,
|
||
|
|
* and tracks progress through individual steps until task completion.
|
||
|
|
*/
|
||
|
|
struct PlanningAgent : ToolCallAgent {
|
||
|
|
std::string active_plan_id;
|
||
|
|
|
||
|
|
// Add a dictionary to track the step status for each tool call
|
||
|
|
std::map<std::string, json> step_execution_tracker;
|
||
|
|
int current_step_index;
|
||
|
|
|
||
|
|
PlanningAgent(
|
||
|
|
const ToolCollection& available_tools = ToolCollection(
|
||
|
|
{
|
||
|
|
std::make_shared<PlanningTool>(),
|
||
|
|
std::make_shared<Terminate>()
|
||
|
|
}
|
||
|
|
),
|
||
|
|
const std::string& tool_choice = "auto",
|
||
|
|
const std::set<std::string>& special_tool_names = {"terminate"},
|
||
|
|
const std::string& name = "planning",
|
||
|
|
const std::string& description = "An agent that creates and manages plans to solve tasks",
|
||
|
|
const std::string& system_prompt = prompt::planning::PLANNING_SYSTEM_PROMPT,
|
||
|
|
const std::string& next_step_prompt = prompt::planning::NEXT_STEP_PROMPT,
|
||
|
|
int max_steps = 20,
|
||
|
|
int current_step = 0,
|
||
|
|
int duplicate_threshold = 2
|
||
|
|
) : ToolCallAgent(available_tools, tool_choice, special_tool_names, name, description, system_prompt, next_step_prompt, max_steps, current_step, duplicate_threshold) {
|
||
|
|
current_step_index = -1; // no plan yet
|
||
|
|
}
|
||
|
|
|
||
|
|
// Initialize the agent with a default plan ID and validate required tools.
|
||
|
|
PlanningAgent* initialize_plan_and_verify_tools();
|
||
|
|
|
||
|
|
// Decide the next action based on plan status.
|
||
|
|
bool think() override;
|
||
|
|
|
||
|
|
// Execute a step and track its completion status.
|
||
|
|
std::string act() override;
|
||
|
|
|
||
|
|
// Retrieve the current plan status.
|
||
|
|
std::string get_plan();
|
||
|
|
|
||
|
|
// Run the agent with an optional initial request.
|
||
|
|
std::string run(const std::string& request = "") override;
|
||
|
|
|
||
|
|
// Update the current plan progress based on completed tool execution.
|
||
|
|
// Only marks a step as completed if the associated tool has been successfully executed.
|
||
|
|
void update_plan_status(const std::string& tool_call_id);
|
||
|
|
|
||
|
|
// Parse the current plan to identify the first non-completed step's index.
|
||
|
|
// Returns None if no active step is found.
|
||
|
|
int _get_current_step_index();
|
||
|
|
|
||
|
|
// Create an initial plan based on the request.
|
||
|
|
void create_initial_plan(const std::string& request);
|
||
|
|
};
|
||
|
|
|
||
|
|
|
||
|
|
} // namespace humanus
|
||
|
|
|
||
|
|
#endif // HUMANUS_AGENT_PLANNING_H
|