zach
/
RA.Aid
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix bug where completion message was wiped too early

This commit is contained in:
AI Christianson 2025-02-12 16:58:39 -05:00
parent a169ed8517
commit 0c8a4009dc
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ra_aid/agent_utils.py
View File

@ -809,11 +809,9 @@ def run_agent_with_retry(agent, prompt: str, config: dict) -> Optional[str]:
if _global_memory["plan_completed"]: if _global_memory["plan_completed"]:
_global_memory["plan_completed"] = False _global_memory["plan_completed"] = False
_global_memory["task_completed"] = False _global_memory["task_completed"] = False
_global_memory["completion_message"] = ""
break break
if _global_memory["task_completed"]: if _global_memory["task_completed"]:
_global_memory["task_completed"] = False _global_memory["task_completed"] = False
_global_memory["completion_message"] = ""
break break
# Execute test command if configured # Execute test command if configured

2
ra_aid/tools/agent.py
View File

@ -275,6 +275,7 @@ def request_task_implementation(task_spec: str) -> Dict[str, Any]:
print_task_header(task_spec) print_task_header(task_spec)
# Run implementation agent # Run implementation agent
from ..agent_utils import run_task_implementation_agent from ..agent_utils import run_task_implementation_agent
_global_memory["completion_message"] = ""
_result = run_task_implementation_agent( _result = run_task_implementation_agent(
base_task=_global_memory.get("base_task", ""), base_task=_global_memory.get("base_task", ""),
@ -344,6 +345,7 @@ def request_implementation(task_spec: str) -> Dict[str, Any]:
try: try:
# Run planning agent # Run planning agent
from ..agent_utils import run_planning_agent from ..agent_utils import run_planning_agent
_global_memory["completion_message"] = ""
_result = run_planning_agent( _result = run_planning_agent(
task_spec, task_spec,