andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-01 23:32:39 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

refactor: reorganize agent executor imports and introduce CrewAgentExecutorFlow

Browse Source 
- Removed the old import of CrewAgentExecutorFlow and replaced it with the new import from the experimental module.
- Updated relevant references in the codebase to ensure compatibility with the new structure.
- Enhanced the organization of imports in core.py and base_agent.py for better clarity and maintainability.
This commit is contained in:
lorenzejay
2025-12-10 09:40:00 -08:00
parent e70fbb8898
commit 563280cc9b
12 changed files with 62 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
lib/crewai/tests/agents/test_crew_agent_executor_flow.py
View File

@@ -8,7 +8,7 @@ from unittest.mock import Mock, patch
import pytest
from crewai.agents.crew_agent_executor_flow import (
from crewai.experimental.crew_agent_executor_flow import (
AgentReActState,
CrewAgentExecutorFlow,
)
@@ -233,7 +233,7 @@ class TestCrewAgentExecutorFlow:
AgentFinish(thought="thinking", output="test", text="final")
)
@patch("crewai.agents.crew_agent_executor_flow.handle_output_parser_exception")
@patch("crewai.experimental.crew_agent_executor_flow.handle_output_parser_exception")
def test_recover_from_parser_error(
self, mock_handle_exception, mock_dependencies
):
@@ -252,7 +252,7 @@ class TestCrewAgentExecutorFlow:
assert executor.state.iterations == initial_iterations + 1
mock_handle_exception.assert_called_once()
@patch("crewai.agents.crew_agent_executor_flow.handle_context_length")
@patch("crewai.experimental.crew_agent_executor_flow.handle_context_length")
def test_recover_from_context_length(
self, mock_handle_context, mock_dependencies
):
@@ -321,8 +321,8 @@ class TestFlowErrorHandling:
"tools_handler": Mock(),
}
@patch("crewai.agents.crew_agent_executor_flow.get_llm_response")
@patch("crewai.agents.crew_agent_executor_flow.enforce_rpm_limit")
@patch("crewai.experimental.crew_agent_executor_flow.get_llm_response")
@patch("crewai.experimental.crew_agent_executor_flow.enforce_rpm_limit")
def test_call_llm_parser_error(
self, mock_enforce_rpm, mock_get_llm, mock_dependencies
):
@@ -338,9 +338,9 @@ class TestFlowErrorHandling:
assert result == "parser_error"
assert executor._last_parser_error is not None
@patch("crewai.agents.crew_agent_executor_flow.get_llm_response")
@patch("crewai.agents.crew_agent_executor_flow.enforce_rpm_limit")
@patch("crewai.agents.crew_agent_executor_flow.is_context_length_exceeded")
@patch("crewai.experimental.crew_agent_executor_flow.get_llm_response")
@patch("crewai.experimental.crew_agent_executor_flow.enforce_rpm_limit")
@patch("crewai.experimental.crew_agent_executor_flow.is_context_length_exceeded")
def test_call_llm_context_error(
self,
mock_is_context_exceeded,