From 5344aa9e141fa7a45b561d3d58ce1f542cb73ebd Mon Sep 17 00:00:00 2001 From: user Date: Sat, 21 Dec 2024 14:42:18 -0500 Subject: [PATCH] Allow chat mode to do implementation. --- ra_aid/prompts.py | 2 +- ra_aid/tool_configs.py | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/ra_aid/prompts.py b/ra_aid/prompts.py index e8e1623..9391f43 100644 --- a/ra_aid/prompts.py +++ b/ra_aid/prompts.py @@ -129,7 +129,7 @@ No Planning or Problem-Solving You must remain strictly within the bounds of describing what currently exists. If the task requires *ANY* compilation, unit tests, or any other non-trivial changes, call request_implementation. -If this is a trivial task that can be completed in one shot, do the change using tools available and call one_shot_completed. +If this is a trivial task that can be completed in one shot, do the change using tools available, call one_shot_completed, and immediately exit without saying anything. Remember, many tasks are more complex and nuanced than they seem and still require requesting implementation. For one shot tasks, still take some time to consider whether compilation, testing, or additional validation should be done to check your work. If you implement the task yourself, do not request implementation. diff --git a/ra_aid/tool_configs.py b/ra_aid/tool_configs.py index 55b2b04..998ddda 100644 --- a/ra_aid/tool_configs.py +++ b/ra_aid/tool_configs.py @@ -113,7 +113,8 @@ def get_chat_tools(expert_enabled: bool = True) -> list: """ tools = [ ask_human, - request_research + request_research, + request_research_and_implementation ] # Add expert tools if enabled