diff --git a/src/crewai/agent.py b/src/crewai/agent.py index b7e31a600..a75b9c51c 100644 --- a/src/crewai/agent.py +++ b/src/crewai/agent.py @@ -478,7 +478,7 @@ class Agent(BaseAgent): for tool_result in self.tools_results: # type: ignore # Item "None" of "list[Any] | None" has no attribute "__iter__" (not iterable) if tool_result.get("result_as_answer", False): from crewai.tools.tool_types import ToolAnswerResult - result = ToolAnswerResult(tool_result["result"]) + result = ToolAnswerResult(tool_result["result"]) # type: ignore crewai_event_bus.emit( self, event=AgentExecutionCompletedEvent(agent=self, task=task, output=result), diff --git a/tests/test_result_as_answer_fix.py b/tests/test_result_as_answer_fix.py index 7374a8459..c78a1eca2 100644 --- a/tests/test_result_as_answer_fix.py +++ b/tests/test_result_as_answer_fix.py @@ -1,5 +1,4 @@ -import pytest -from unittest.mock import Mock, patch +from unittest.mock import patch from crewai.agent import Agent from crewai.task import Task from crewai.tools.base_tool import BaseTool