From 58622b8357e466c03e741e18f6b58c33f5dbcb40 Mon Sep 17 00:00:00 2001 From: AI Christianson Date: Thu, 19 Dec 2024 12:20:38 -0500 Subject: [PATCH] Make delete_tasks tool available to planning agent. --- CHANGELOG.md | 4 ++++ ra_aid/__main__.py | 5 +++-- ra_aid/tools/__init__.py | 3 ++- 3 files changed, 9 insertions(+), 3 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index a63a8a2..58a1367 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -5,6 +5,10 @@ All notable changes to this project will be documented in this file. The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html). +## [Unreleased] + +- Make delete_tasks tool available to planning agent. + ## [0.6.4] - 2024-12-19 - Added monorepo_detected, existing_project_detected, and ui_detected tools so the agent can take specific actions. diff --git a/ra_aid/__main__.py b/ra_aid/__main__.py index 0f5dfb5..c850294 100644 --- a/ra_aid/__main__.py +++ b/ra_aid/__main__.py @@ -10,7 +10,7 @@ from ra_aid.tools import ( ask_expert, run_shell_command, run_programming_task, emit_research_notes, emit_plan, emit_related_files, emit_task, emit_expert_context, get_memory_value, emit_key_facts, delete_key_facts, - emit_key_snippets, delete_key_snippets, + emit_key_snippets, delete_key_snippets, delete_tasks, request_research_subtask, request_implementation, read_file_tool, fuzzy_find_project_files, ripgrep_search, list_directory_tree, swap_task_order, monorepo_detected, existing_project_detected, ui_detected ) @@ -162,8 +162,9 @@ def get_planning_tools(expert_enabled: bool = True) -> list: # Add planning-specific tools planning_tools = [ + delete_tasks, emit_plan, - emit_task, + emit_task, swap_task_order ] tools.extend(planning_tools) diff --git a/ra_aid/tools/__init__.py b/ra_aid/tools/__init__.py index f979cf4..cc25235 100644 --- a/ra_aid/tools/__init__.py +++ b/ra_aid/tools/__init__.py @@ -9,7 +9,7 @@ from .fuzzy_find import fuzzy_find_project_files from .list_directory import list_directory_tree from .ripgrep import ripgrep_search from .memory import ( - emit_research_notes, emit_plan, emit_task, get_memory_value, emit_key_facts, + delete_tasks, emit_research_notes, emit_plan, emit_task, get_memory_value, emit_key_facts, request_implementation, skip_implementation, delete_key_facts, request_research_subtask, emit_key_snippets, delete_key_snippets, emit_related_files, swap_task_order ) @@ -37,6 +37,7 @@ __all__ = [ 'request_research_subtask', 'ripgrep_search', 'file_str_replace', + 'delete_tasks', 'swap_task_order', 'monorepo_detected', 'existing_project_detected',