diff --git a/ra_aid/__main__.py b/ra_aid/__main__.py index 4c7246d..bdbbed9 100644 --- a/ra_aid/__main__.py +++ b/ra_aid/__main__.py @@ -8,7 +8,7 @@ from langgraph.prebuilt import create_react_agent from ra_aid.env import validate_environment from ra_aid.tools.memory import _global_memory from ra_aid.tools.human import ask_human -from ra_aid import print_stage_header, print_error, print_interrupt +from ra_aid import print_stage_header, print_error from ra_aid.tools.human import ask_human from ra_aid.agent_utils import ( AgentInterrupt, diff --git a/ra_aid/agent_utils.py b/ra_aid/agent_utils.py index 2354d18..b8ea3cb 100644 --- a/ra_aid/agent_utils.py +++ b/ra_aid/agent_utils.py @@ -3,7 +3,7 @@ import sys import time import uuid -from typing import Optional, Any, List +from typing import Optional, Any import signal import threading diff --git a/ra_aid/tool_configs.py b/ra_aid/tool_configs.py index 9a6fc2e..1a1d7cd 100644 --- a/ra_aid/tool_configs.py +++ b/ra_aid/tool_configs.py @@ -1,4 +1,3 @@ -from typing import List from ra_aid.tools import ( ask_expert, ask_human, run_shell_command, run_programming_task, emit_research_notes, emit_plan, emit_related_files, emit_task, diff --git a/ra_aid/tools/expert.py b/ra_aid/tools/expert.py index 115f67f..0532f96 100644 --- a/ra_aid/tools/expert.py +++ b/ra_aid/tools/expert.py @@ -5,7 +5,7 @@ from rich.console import Console from rich.panel import Panel from rich.markdown import Markdown from ..llm import initialize_expert_llm -from .memory import get_memory_value, get_related_files, _global_memory +from .memory import get_memory_value, _global_memory console = Console() _model = None diff --git a/ra_aid/tools/list_directory.py b/ra_aid/tools/list_directory.py index 7d89a0c..52a0553 100644 --- a/ra_aid/tools/list_directory.py +++ b/ra_aid/tools/list_directory.py @@ -1,5 +1,5 @@ from pathlib import Path -from typing import List, Optional, Dict, Any +from typing import List, Optional import datetime from dataclasses import dataclass import pathspec diff --git a/ra_aid/tools/programmer.py b/ra_aid/tools/programmer.py index ce6843c..effaf72 100644 --- a/ra_aid/tools/programmer.py +++ b/ra_aid/tools/programmer.py @@ -1,9 +1,8 @@ import os -from typing import List, Optional, Dict, Union, Set +from typing import List, Optional, Dict, Union from langchain_core.tools import tool from rich.console import Console from rich.panel import Panel -from rich.syntax import Syntax from rich.markdown import Markdown from rich.text import Text from ra_aid.proc.interactive import run_interactive_command diff --git a/ra_aid/tools/read_file.py b/ra_aid/tools/read_file.py index edf72eb..e515cee 100644 --- a/ra_aid/tools/read_file.py +++ b/ra_aid/tools/read_file.py @@ -1,11 +1,10 @@ import os.path import logging import time -from typing import Dict, Optional, Tuple +from typing import Dict from langchain_core.tools import tool from rich.console import Console from rich.panel import Panel -from rich.markdown import Markdown from ra_aid.text.processing import truncate_output console = Console() diff --git a/ra_aid/tools/ripgrep.py b/ra_aid/tools/ripgrep.py index 04ae790..dfa534f 100644 --- a/ra_aid/tools/ripgrep.py +++ b/ra_aid/tools/ripgrep.py @@ -1,4 +1,4 @@ -from typing import Dict, Union, Optional, List +from typing import Dict, Union, List from langchain_core.tools import tool from rich.console import Console from rich.panel import Panel