diff --git a/src/crewai/tools/human_tool.py b/src/crewai/tools/human_tool.py index 8e4cb00f5..03c19718b 100644 --- a/src/crewai/tools/human_tool.py +++ b/src/crewai/tools/human_tool.py @@ -1,6 +1,7 @@ """Tool for handling human input using LangGraph's interrupt mechanism.""" from typing import Any, Dict + from pydantic import Field from crewai.tools import BaseTool diff --git a/tests/tools/test_human_tool.py b/tests/tools/test_human_tool.py index dbdd7529e..67274b3c9 100644 --- a/tests/tools/test_human_tool.py +++ b/tests/tools/test_human_tool.py @@ -1,8 +1,9 @@ """Test HumanTool functionality.""" -import pytest from unittest.mock import patch +import pytest + from crewai.tools import HumanTool def test_human_tool_basic(): diff --git a/tests/tools/test_tool_usage.py b/tests/tools/test_tool_usage.py index 76ec5eefb..1218ce9ca 100644 --- a/tests/tools/test_tool_usage.py +++ b/tests/tools/test_tool_usage.py @@ -1,14 +1,14 @@ import json import random -from unittest.mock import MagicMock +from unittest.mock import MagicMock, patch import pytest from pydantic import BaseModel, Field from crewai import Agent, Task from crewai.tools import BaseTool -from crewai.tools.tool_usage import ToolUsage from crewai.tools.tool_calling import ToolCalling +from crewai.tools.tool_usage import ToolUsage class RandomNumberToolInput(BaseModel):