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

refactor: agents/ -> agent_backends/

This commit is contained in:
AI Christianson 2025-03-02 14:16:15 -05:00
parent 5736144d89
commit 17ab6d2a50
7 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

0
ra_aid/agents/__init__.py → ra_aid/agent_backends/__init__.py
View File

0
ra_aid/agents/ciayn_agent.py → ra_aid/agent_backends/ciayn_agent.py
View File

2
ra_aid/agent_utils.py
View File

@ -36,7 +36,7 @@ from ra_aid.agent_context import (
reset_completion_flags, reset_completion_flags,
should_exit, should_exit,
) )
from ra_aid.agents.ciayn_agent import CiaynAgent from ra_aid.agent_backends.ciayn_agent import CiaynAgent
from ra_aid.agents_alias import RAgents from ra_aid.agents_alias import RAgents
from ra_aid.config import DEFAULT_MAX_TEST_CMD_RETRIES, DEFAULT_RECURSION_LIMIT from ra_aid.config import DEFAULT_MAX_TEST_CMD_RETRIES, DEFAULT_RECURSION_LIMIT
from ra_aid.console.formatting import print_error, print_stage_header from ra_aid.console.formatting import print_error, print_stage_header

2
ra_aid/agents_alias.py
View File

@ -4,7 +4,7 @@ from langgraph.graph.graph import CompiledGraph
# Unfortunately need this to avoid Circular Imports # Unfortunately need this to avoid Circular Imports
if TYPE_CHECKING: if TYPE_CHECKING:
from ra_aid.agents.ciayn_agent import CiaynAgent from ra_aid.agent_backends.ciayn_agent import CiaynAgent
RAgents = CompiledGraph | CiaynAgent RAgents = CompiledGraph | CiaynAgent
else: else:

2
tests/ra_aid/test_agent_utils.py
View File

@ -194,7 +194,7 @@ def test_state_modifier(mock_messages):
state = AgentState(messages=mock_messages) state = AgentState(messages=mock_messages)
with patch( with patch(
"ra_aid.agents.ciayn_agent.CiaynAgent._estimate_tokens" "ra_aid.agent_backends.ciayn_agent.CiaynAgent._estimate_tokens"
) as mock_estimate: ) as mock_estimate:
mock_estimate.side_effect = lambda msg: 100 if msg else 0 mock_estimate.side_effect = lambda msg: 100 if msg else 0

2
tests/ra_aid/test_ciayn_agent.py
View File

@ -4,7 +4,7 @@ from unittest.mock import Mock
import pytest import pytest
from langchain_core.messages import AIMessage, BaseMessage, HumanMessage from langchain_core.messages import AIMessage, BaseMessage, HumanMessage
from ra_aid.agents.ciayn_agent import CiaynAgent, validate_function_call_pattern from ra_aid.agent_backends.ciayn_agent import CiaynAgent, validate_function_call_pattern
from ra_aid.exceptions import ToolExecutionError from ra_aid.exceptions import ToolExecutionError

2
tests/ra_aid/test_llm.py
View File

@ -9,7 +9,7 @@ from langchain_core.messages import HumanMessage
from langchain_google_genai.chat_models import ChatGoogleGenerativeAI from langchain_google_genai.chat_models import ChatGoogleGenerativeAI
from langchain_openai.chat_models import ChatOpenAI from langchain_openai.chat_models import ChatOpenAI
from ra_aid.agents.ciayn_agent import CiaynAgent from ra_aid.agent_backends.ciayn_agent import CiaynAgent
from ra_aid.env import validate_environment from ra_aid.env import validate_environment
from ra_aid.llm import ( from ra_aid.llm import (
create_llm_client, create_llm_client,