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

View File

@ -297,4 +297,4 @@ class CiaynAgent:
e, self, self.chat_history
)
last_result = self.handle_fallback_response(fallback_response, e)
yield {}
yield {}

View File

@ -36,7 +36,7 @@ from ra_aid.agent_context import (
reset_completion_flags,
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.config import DEFAULT_MAX_TEST_CMD_RETRIES, DEFAULT_RECURSION_LIMIT
from ra_aid.console.formatting import print_error, print_stage_header

View File

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

View File

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

View File

@ -4,7 +4,7 @@ from unittest.mock import Mock
import pytest
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

View File

@ -9,7 +9,7 @@ from langchain_core.messages import HumanMessage
from langchain_google_genai.chat_models import ChatGoogleGenerativeAI
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.llm import (
create_llm_client,