mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-10 08:38:30 +00:00
fix: remove unnecessary wraps parameter in test patch
This commit is contained in:
@@ -7,10 +7,8 @@ from pydantic import Field
|
|||||||
from crewai.agent import Agent
|
from crewai.agent import Agent
|
||||||
from crewai.agents.crew_agent_executor import CrewAgentExecutor
|
from crewai.agents.crew_agent_executor import CrewAgentExecutor
|
||||||
from crewai.crew import Crew
|
from crewai.crew import Crew
|
||||||
from crewai.flow.flow import Flow, listen, start
|
from crewai.events.event_bus import crewai_event_bus
|
||||||
from crewai.llm import LLM
|
from crewai.events.event_listener import EventListener
|
||||||
from crewai.task import Task
|
|
||||||
from crewai.tools.base_tool import BaseTool
|
|
||||||
from crewai.events.types.agent_events import (
|
from crewai.events.types.agent_events import (
|
||||||
AgentExecutionCompletedEvent,
|
AgentExecutionCompletedEvent,
|
||||||
AgentExecutionErrorEvent,
|
AgentExecutionErrorEvent,
|
||||||
@@ -24,9 +22,6 @@ from crewai.events.types.crew_events import (
|
|||||||
CrewTestResultEvent,
|
CrewTestResultEvent,
|
||||||
CrewTestStartedEvent,
|
CrewTestStartedEvent,
|
||||||
)
|
)
|
||||||
from crewai.events.event_bus import crewai_event_bus
|
|
||||||
from crewai.events.event_listener import EventListener
|
|
||||||
from crewai.events.types.tool_usage_events import ToolUsageFinishedEvent
|
|
||||||
from crewai.events.types.flow_events import (
|
from crewai.events.types.flow_events import (
|
||||||
FlowCreatedEvent,
|
FlowCreatedEvent,
|
||||||
FlowFinishedEvent,
|
FlowFinishedEvent,
|
||||||
@@ -47,7 +42,12 @@ from crewai.events.types.task_events import (
|
|||||||
)
|
)
|
||||||
from crewai.events.types.tool_usage_events import (
|
from crewai.events.types.tool_usage_events import (
|
||||||
ToolUsageErrorEvent,
|
ToolUsageErrorEvent,
|
||||||
|
ToolUsageFinishedEvent,
|
||||||
)
|
)
|
||||||
|
from crewai.flow.flow import Flow, listen, start
|
||||||
|
from crewai.llm import LLM
|
||||||
|
from crewai.task import Task
|
||||||
|
from crewai.tools.base_tool import BaseTool
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture(scope="module")
|
@pytest.fixture(scope="module")
|
||||||
@@ -327,9 +327,9 @@ def test_agent_emits_execution_error_event(base_agent, base_task):
|
|||||||
|
|
||||||
error_message = "Error happening while sending prompt to model."
|
error_message = "Error happening while sending prompt to model."
|
||||||
base_agent.max_retry_limit = 0
|
base_agent.max_retry_limit = 0
|
||||||
with patch.object(
|
|
||||||
CrewAgentExecutor, "invoke", wraps=base_agent.agent_executor.invoke
|
# Patch the invoke method on the CrewAgentExecutor class directly
|
||||||
) as invoke_mock:
|
with patch.object(CrewAgentExecutor, "invoke") as invoke_mock:
|
||||||
invoke_mock.side_effect = Exception(error_message)
|
invoke_mock.side_effect = Exception(error_message)
|
||||||
|
|
||||||
with pytest.raises(Exception):
|
with pytest.raises(Exception):
|
||||||
|
|||||||
Reference in New Issue
Block a user