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

Improve tool return values.

This commit is contained in:
user 2024-12-21 15:40:19 -05:00
parent 5053b36d8b
commit 5be2792612
1 changed files with 14 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
ra_aid/tools/agent.py
View File

@ -53,9 +53,10 @@ def request_research(query: str) -> Dict[str, Any]:
return { return {
"completion_message": completion_message, "completion_message": completion_message,
"facts": get_memory_value("key_facts"), "key_facts": get_memory_value("key_facts"),
"files": list(get_related_files()), "related_files": list(get_related_files()),
"notes": get_memory_value("research_notes"), "research_notes": get_memory_value("research_notes"),
"key_snippets": get_memory_value("key_snippets"),
"success": success, "success": success,
"reason": reason "reason": reason
} }
@ -103,9 +104,10 @@ def request_research_and_implementation(query: str) -> Dict[str, Any]:
return { return {
"completion_message": completion_message, "completion_message": completion_message,
"facts": get_memory_value("key_facts"), "key_facts": get_memory_value("key_facts"),
"files": list(get_related_files()), "related_files": list(get_related_files()),
"notes": get_memory_value("research_notes"), "research_notes": get_memory_value("research_notes"),
"key_snippets": get_memory_value("key_snippets"),
"success": success, "success": success,
"reason": reason "reason": reason
} }
@ -159,8 +161,9 @@ def request_task_implementation(task_spec: str) -> Dict[str, Any]:
_global_memory['completion_state'] = False _global_memory['completion_state'] = False
return { return {
"facts": get_memory_value("key_facts"), "key_facts": get_memory_value("key_facts"),
"files": list(get_related_files()), "related_files": list(get_related_files()),
"key_snippets": get_memory_value("key_snippets"),
"completion_message": completion_message, "completion_message": completion_message,
"success": success, "success": success,
"reason": reason "reason": reason
@ -208,8 +211,9 @@ def request_implementation(task_spec: str) -> Dict[str, Any]:
return { return {
"completion_message": completion_message, "completion_message": completion_message,
"facts": get_memory_value("key_facts"), "key_facts": get_memory_value("key_facts"),
"files": list(get_related_files()), "related_files": list(get_related_files()),
"key_snippets": get_memory_value("key_snippets"),
"success": success, "success": success,
"reason": reason "reason": reason
} }