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

Cleanup.

This commit is contained in:
user 2024-12-21 13:56:18 -05:00
parent f92c30b4e6
commit 2b0efe5fd2
2 changed files with 4 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
ra_aid/__main__.py
View File

@ -7,8 +7,8 @@ from langgraph.checkpoint.memory import MemorySaver
from langgraph.prebuilt import create_react_agent from langgraph.prebuilt import create_react_agent
from ra_aid.env import validate_environment from ra_aid.env import validate_environment
from ra_aid.tools.memory import _global_memory, get_related_files, get_memory_value from ra_aid.tools.memory import _global_memory, get_related_files, get_memory_value
from ra_aid import print_stage_header, print_task_header, print_error from ra_aid import print_stage_header, print_error
from ra_aid.agent_utils import run_agent_with_retry, run_task_implementation_agent from ra_aid.agent_utils import run_agent_with_retry
from ra_aid.agent_utils import run_research_agent from ra_aid.agent_utils import run_research_agent
from ra_aid.prompts import ( from ra_aid.prompts import (
PLANNING_PROMPT, PLANNING_PROMPT,
@ -20,7 +20,6 @@ from ra_aid.llm import initialize_llm
from ra_aid.tool_configs import ( from ra_aid.tool_configs import (
get_planning_tools, get_planning_tools,
get_implementation_tools,
get_chat_tools get_chat_tools
) )

7
ra_aid/tools/memory.py
View File

@ -287,17 +287,14 @@ def one_shot_completed(message: str) -> str:
Args: Args:
message: Completion message to display message: Completion message to display
Returns:
Original message if task can be completed, or error message if there are
pending subtasks or implementation requests
""" """
if _global_memory.get('implementation_requested', False): if _global_memory.get('implementation_requested', False):
return "Cannot complete in one shot - implementation was requested" return "Cannot complete in one shot - implementation was requested"
_global_memory['task_completed'] = True _global_memory['task_completed'] = True
_global_memory['completion_message'] = message _global_memory['completion_message'] = message
return message console.print(Panel(Markdown(message), title="✅ One-Shot Task Completed"))
return "Completion noted."
@tool("task_completed") @tool("task_completed")
def task_completed(message: str) -> str: def task_completed(message: str) -> str: