From 17ab6d2a508b8b2dabc65211f4aad7b869ef8acf Mon Sep 17 00:00:00 2001
From: AI Christianson <ai.christianson@christianson.ai>
Date: Sun, 2 Mar 2025 14:16:15 -0500
Subject: [PATCH] refactor: agents/ -> agent_backends/

---
 ra_aid/{agents => agent_backends}/__init__.py    | 0
 ra_aid/{agents => agent_backends}/ciayn_agent.py | 2 +-
 ra_aid/agent_utils.py                            | 2 +-
 ra_aid/agents_alias.py                           | 2 +-
 tests/ra_aid/test_agent_utils.py                 | 2 +-
 tests/ra_aid/test_ciayn_agent.py                 | 2 +-
 tests/ra_aid/test_llm.py                         | 2 +-
 7 files changed, 6 insertions(+), 6 deletions(-)
 rename ra_aid/{agents => agent_backends}/__init__.py (100%)
 rename ra_aid/{agents => agent_backends}/ciayn_agent.py (99%)

diff --git a/ra_aid/agents/__init__.py b/ra_aid/agent_backends/__init__.py
similarity index 100%
rename from ra_aid/agents/__init__.py
rename to ra_aid/agent_backends/__init__.py
diff --git a/ra_aid/agents/ciayn_agent.py b/ra_aid/agent_backends/ciayn_agent.py
similarity index 99%
rename from ra_aid/agents/ciayn_agent.py
rename to ra_aid/agent_backends/ciayn_agent.py
index 848bc1c..b6ed0f6 100644
--- a/ra_aid/agents/ciayn_agent.py
+++ b/ra_aid/agent_backends/ciayn_agent.py
@@ -297,4 +297,4 @@ class CiaynAgent:
                     e, self, self.chat_history
                 )
                 last_result = self.handle_fallback_response(fallback_response, e)
-                yield {}
+                yield {}
\ No newline at end of file
diff --git a/ra_aid/agent_utils.py b/ra_aid/agent_utils.py
index 1dc4643..540d2b3 100644
--- a/ra_aid/agent_utils.py
+++ b/ra_aid/agent_utils.py
@@ -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
diff --git a/ra_aid/agents_alias.py b/ra_aid/agents_alias.py
index 2cf6077..b3597ac 100644
--- a/ra_aid/agents_alias.py
+++ b/ra_aid/agents_alias.py
@@ -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:
diff --git a/tests/ra_aid/test_agent_utils.py b/tests/ra_aid/test_agent_utils.py
index 09b6fea..4b5dd2c 100644
--- a/tests/ra_aid/test_agent_utils.py
+++ b/tests/ra_aid/test_agent_utils.py
@@ -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
 
diff --git a/tests/ra_aid/test_ciayn_agent.py b/tests/ra_aid/test_ciayn_agent.py
index 4cd9dc2..2412523 100644
--- a/tests/ra_aid/test_ciayn_agent.py
+++ b/tests/ra_aid/test_ciayn_agent.py
@@ -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
 
 
diff --git a/tests/ra_aid/test_llm.py b/tests/ra_aid/test_llm.py
index d43824f..04f20b9 100644
--- a/tests/ra_aid/test_llm.py
+++ b/tests/ra_aid/test_llm.py
@@ -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,