From 51663dc479ee741bed34fedab8af7007978f1bab Mon Sep 17 00:00:00 2001
From: user <user@localhost>
Date: Sun, 22 Dec 2024 10:05:43 -0500
Subject: [PATCH] Fix task completion tracking.

---
 ra_aid/tools/agent.py | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/ra_aid/tools/agent.py b/ra_aid/tools/agent.py
index 3501f5d..6e395c2 100644
--- a/ra_aid/tools/agent.py
+++ b/ra_aid/tools/agent.py
@@ -78,7 +78,7 @@ def request_research(query: str) -> ResearchResult:
     
     # Clear completion state from global memory
     _global_memory['completion_message'] = ''
-    _global_memory['completion_state'] = False
+    _global_memory['task_completed'] = False
         
     return {
         "completion_message": completion_message,
@@ -129,7 +129,8 @@ def request_research_and_implementation(query: str) -> Dict[str, Any]:
     
     # Clear completion state from global memory
     _global_memory['completion_message'] = ''
-    _global_memory['completion_state'] = False
+    _global_memory['task_completed'] = False
+    _global_memory['plan_completed'] = False
         
     return {
         "completion_message": completion_message,
@@ -187,7 +188,7 @@ def request_task_implementation(task_spec: str) -> Dict[str, Any]:
     
     # Clear completion state from global memory
     _global_memory['completion_message'] = ''
-    _global_memory['completion_state'] = False
+    _global_memory['task_completed'] = False
         
     return {
         "key_facts": get_memory_value("key_facts"),
@@ -236,7 +237,8 @@ def request_implementation(task_spec: str) -> Dict[str, Any]:
     
     # Clear completion state from global memory
     _global_memory['completion_message'] = ''
-    _global_memory['completion_state'] = False
+    _global_memory['task_completed'] = False
+    _global_memory['plan_completed'] = False
         
     return {
         "completion_message": completion_message,