refactor: agents/ -> agent_backends/
This commit is contained in:
parent
5736144d89
commit
17ab6d2a50
|
|
@ -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
|
||||
|
|
|
|||
|
|
@ -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:
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
||||
|
|
|
|||
|
|
@ -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
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -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,
|
||||
|
|
|
|||
Loading…
Reference in New Issue