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

Fix task completion tracking.

This commit is contained in:
user 2024-12-22 10:05:43 -05:00
parent c73e9ada84
commit 51663dc479
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
ra_aid/tools/agent.py
View File

@ -78,7 +78,7 @@ def request_research(query: str) -> ResearchResult:
# Clear completion state from global memory # Clear completion state from global memory
_global_memory['completion_message'] = '' _global_memory['completion_message'] = ''
_global_memory['completion_state'] = False _global_memory['task_completed'] = False
return { return {
"completion_message": completion_message, "completion_message": completion_message,
@ -129,7 +129,8 @@ def request_research_and_implementation(query: str) -> Dict[str, Any]:
# Clear completion state from global memory # Clear completion state from global memory
_global_memory['completion_message'] = '' _global_memory['completion_message'] = ''
_global_memory['completion_state'] = False _global_memory['task_completed'] = False
_global_memory['plan_completed'] = False
return { return {
"completion_message": completion_message, "completion_message": completion_message,
@ -187,7 +188,7 @@ def request_task_implementation(task_spec: str) -> Dict[str, Any]:
# Clear completion state from global memory # Clear completion state from global memory
_global_memory['completion_message'] = '' _global_memory['completion_message'] = ''
_global_memory['completion_state'] = False _global_memory['task_completed'] = False
return { return {
"key_facts": get_memory_value("key_facts"), "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 # Clear completion state from global memory
_global_memory['completion_message'] = '' _global_memory['completion_message'] = ''
_global_memory['completion_state'] = False _global_memory['task_completed'] = False
_global_memory['plan_completed'] = False
return { return {
"completion_message": completion_message, "completion_message": completion_message,