Skip to content

Commit 2b7d8b5

Browse files
fix: remove last embedchain mocks
1 parent fb99414 commit 2b7d8b5

File tree

2 files changed

+7
-14
lines changed

2 files changed

+7
-14
lines changed

tests/agents/test_agent.py

Lines changed: 6 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -2361,13 +2361,11 @@ def mock_get_auth_token():
23612361

23622362
@patch("crewai.cli.plus_api.PlusAPI.get_agent")
23632363
def test_agent_from_repository(mock_get_agent, mock_get_auth_token):
2364-
# Mock embedchain initialization to prevent race conditions in parallel CI execution
2365-
with patch("embedchain.client.Client.setup"):
2366-
from crewai_tools import (
2367-
EnterpriseActionTool,
2368-
FileReadTool,
2369-
SerperDevTool,
2370-
)
2364+
from crewai_tools import (
2365+
EnterpriseActionTool,
2366+
FileReadTool,
2367+
SerperDevTool,
2368+
)
23712369

23722370
mock_get_response = MagicMock()
23732371
mock_get_response.status_code = 200
@@ -2423,9 +2421,7 @@ def test_agent_from_repository(mock_get_agent, mock_get_auth_token):
24232421

24242422
@patch("crewai.cli.plus_api.PlusAPI.get_agent")
24252423
def test_agent_from_repository_override_attributes(mock_get_agent, mock_get_auth_token):
2426-
# Mock embedchain initialization to prevent race conditions in parallel CI execution
2427-
with patch("embedchain.client.Client.setup"):
2428-
from crewai_tools import SerperDevTool
2424+
from crewai_tools import SerperDevTool
24292425

24302426
mock_get_response = MagicMock()
24312427
mock_get_response.status_code = 200

tests/test_crew.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -3818,10 +3818,7 @@ def test_task_tools_preserve_code_execution_tools():
38183818
"""
38193819
Test that task tools don't override code execution tools when allow_code_execution=True
38203820
"""
3821-
3822-
# Mock embedchain initialization to prevent race conditions in parallel CI execution
3823-
with patch("embedchain.client.Client.setup"):
3824-
from crewai_tools import CodeInterpreterTool
3821+
from crewai_tools import CodeInterpreterTool
38253822
from pydantic import BaseModel, Field
38263823

38273824
from crewai.tools import BaseTool

0 commit comments

Comments
 (0)