// retoor #include "tool.h" #include "agent.h" #include "messages.h" #include "r_config.h" #include #include #include #include typedef struct { tool_t tool; } tool_agent_t; static struct json_object *tool_spawn_agent_get_description(void) { struct json_object *obj = json_object_new_object(); json_object_object_add(obj, "name", json_object_new_string("spawn_agent")); json_object_object_add(obj, "description", json_object_new_string("Spawn a specialized sub-agent to handle a specific task.")); struct json_object *params = json_object_new_object(); json_object_object_add(params, "type", json_object_new_string("object")); struct json_object *props = json_object_new_object(); struct json_object *persona = json_object_new_object(); json_object_object_add(persona, "type", json_object_new_string("string")); json_object_object_add(persona, "description", json_object_new_string("The persona of the agent (researcher, developer, security, fetcher).")); struct json_object *persona_enum = json_object_new_array(); json_object_array_add(persona_enum, json_object_new_string("researcher")); json_object_array_add(persona_enum, json_object_new_string("developer")); json_object_array_add(persona_enum, json_object_new_string("security")); json_object_array_add(persona_enum, json_object_new_string("fetcher")); json_object_object_add(persona, "enum", persona_enum); json_object_object_add(props, "persona", persona); struct json_object *goal = json_object_new_object(); json_object_object_add(goal, "type", json_object_new_string("string")); json_object_object_add(goal, "description", json_object_new_string("The specific task or goal for the sub-agent.")); json_object_object_add(props, "goal", goal); struct json_object *max_subagents = json_object_new_object(); json_object_object_add(max_subagents, "type", json_object_new_string("integer")); json_object_object_add(max_subagents, "description", json_object_new_string("Remaining budget for spawning recursive sub-agents. Decrement this by 1 when spawning a sub-agent. Default is 2.")); json_object_object_add(max_subagents, "default", json_object_new_int(2)); json_object_object_add(props, "max_subagents", max_subagents); json_object_object_add(params, "properties", props); struct json_object *required = json_object_new_array(); json_object_array_add(required, json_object_new_string("persona")); json_object_array_add(required, json_object_new_string("goal")); json_object_array_add(required, json_object_new_string("max_subagents")); json_object_object_add(params, "required", required); json_object_object_add(params, "additionalProperties", json_object_new_boolean(0)); json_object_object_add(obj, "parameters", params); r_config_handle cfg = r_config_get_instance(); if (r_config_use_strict(cfg)) { json_object_object_add(obj, "strict", json_object_new_boolean(1)); } struct json_object *full_obj = json_object_new_object(); json_object_object_add(full_obj, "type", json_object_new_string("function")); json_object_object_add(full_obj, "function", obj); return full_obj; } static char *tool_spawn_agent_execute(tool_t *self, struct json_object *args) { (void)self; struct json_object *persona_obj, *goal_obj, *max_subagents_obj; if (!json_object_object_get_ex(args, "persona", &persona_obj) || !json_object_object_get_ex(args, "goal", &goal_obj)) { return strdup("Error: Missing persona or goal"); } int max_subagents = 2; if (json_object_object_get_ex(args, "max_subagents", &max_subagents_obj)) { max_subagents = json_object_get_int(max_subagents_obj); } if (max_subagents <= 0) { return strdup("Error: Spawning limit reached. You are not allowed to spawn more sub-agents. Perform the task yourself using existing tools."); } const char *persona_str = json_object_get_string(persona_obj); const char *goal_str = json_object_get_string(goal_obj); tool_registry_type_t type = TOOL_TYPE_ALL; const char *system_prompt_base = NULL; if (strcmp(persona_str, "researcher") == 0) { type = TOOL_TYPE_RESEARCHER; system_prompt_base = "You are a specialized Research Agent. Your goal is to find, extract, and summarize information. " "Do not attempt to write or execute code unless it's for data analysis. " "Focus on using web search, http fetch, and reading files. " "## Sub-Agent Rules\n" "- Your output is for a master agent, not the final user.\n" "- DO NOT ask questions or for permission.\n" "- Provide RAW DATA and summaries.\n" "- Do not say 'task complete'.\n" "## Hierarchical Research Workflow\n" "When web_search returns URLs with content:\n" "1. Spawn 'fetcher' agents in parallel to fetch URL contents\n" "2. Each fetcher should use http_fetch tool for individual URLs\n" "3. Aggregate results and synthesize with citations\n" "Citation format: [Source N] Title (URL)\n" "Use spawn_agent extensively for URL fetching from search results."; } else if (strcmp(persona_str, "developer") == 0) { type = TOOL_TYPE_DEVELOPER; system_prompt_base = "You are a specialized Developer Agent. Your goal is to write, test, and debug code. " "## Sub-Agent Rules\n" "- Your output is for a master agent, not the final user.\n" "- DO NOT ask questions or for permission.\n" "- Just perform the requested development task and report results.\n" "Use the terminal, file editing tools, and python execution to fulfill your task. " "Always verify your changes by running tests or the code itself."; } else if (strcmp(persona_str, "security") == 0) { type = TOOL_TYPE_SECURITY; system_prompt_base = "You are a specialized Security Auditor Agent. Your goal is to find vulnerabilities and perform security analysis. " "## Sub-Agent Rules\n" "- Your output is for a master agent, not the final user.\n" "- DO NOT ask questions or for permission.\n" "Be pedantic and thorough. Use fuzzing, port scanning, and code analysis tools. " "Report any findings clearly with potential impact."; } else if (strcmp(persona_str, "fetcher") == 0) { type = TOOL_TYPE_ALL; system_prompt_base = "You are a specialized URL Fetcher Agent. Your goal is to fetch and extract content from multiple URLs. " "## Sub-Agent Rules\n" "- Just return the content of the URLs.\n" "- Do not provide commentary unless necessary.\n" "Use http_fetch tool to retrieve content from each URL. " "Handle errors gracefully and continue with other URLs. " "Clean HTML/extract text suitable for LLM analysis. " "Truncate content to ~10K chars per URL to stay within token limits."; } else { return strdup("Error: Invalid persona"); } time_t now = time(NULL); struct tm *tm_info = localtime(&now); char datetime[64]; strftime(datetime, sizeof(datetime), "%Y-%m-%d %H:%M:%S %Z", tm_info); size_t prompt_size = strlen(system_prompt_base) + 1024; char *system_prompt = malloc(prompt_size); if (!system_prompt) return strdup("Error: Out of memory"); snprintf(system_prompt, prompt_size, "Current date/time: %s\n\n%s\n\n" "CRITICAL: It is currently %s.\n" "ORCHESTRATION BUDGET: You are allowed to spawn up to %d more levels of sub-agents. " "When using spawn_agent, you MUST pass 'max_subagents' as %d.\n", datetime, system_prompt_base, datetime, max_subagents, max_subagents - 1); char session_id[256]; snprintf(session_id, sizeof(session_id), "subagent-%s-%u", persona_str, (unsigned int)time(NULL)); messages_handle msgs = messages_create(session_id); messages_add(msgs, "system", system_prompt); free(system_prompt); agent_handle agent = agent_create(goal_str, msgs); if (!agent) { messages_destroy(msgs); return strdup("Error: Failed to create sub-agent"); } agent_set_is_subagent(agent, true); tool_registry_t *specialized_tools = tool_registry_get_specialized(type); if (specialized_tools) { agent_set_tool_registry(agent, specialized_tools); } agent_set_max_iterations(agent, 50); // Sub-agents have lower limit char *agent_response = agent_run(agent, goal_str); char *result = messages_to_json_string(msgs); agent_destroy(agent); free(agent_response); if (specialized_tools) { tool_registry_destroy(specialized_tools); } if (!result) { return strdup("Sub-agent failed to provide a result."); } return result; } static void tool_spawn_agent_print_action(const char *name, struct json_object *args) { struct json_object *persona_obj, *goal_obj; const char *persona = "unknown"; const char *goal = "unknown"; if (json_object_object_get_ex(args, "persona", &persona_obj)) persona = json_object_get_string(persona_obj); if (json_object_object_get_ex(args, "goal", &goal_obj)) goal = json_object_get_string(goal_obj); printf("\033[1;34m[Agent] Spawning %s agent for: %s\033[0m\n", persona, goal); } static const tool_vtable_t tool_spawn_agent_vtable = { .get_description = tool_spawn_agent_get_description, .execute = tool_spawn_agent_execute, .print_action = tool_spawn_agent_print_action }; tool_t *tool_spawn_agent_create(void) { tool_agent_t *tool = calloc(1, sizeof(tool_agent_t)); if (!tool) return NULL; tool->tool.vtable = &tool_spawn_agent_vtable; tool->tool.name = "spawn_agent"; return &tool->tool; }