fix tool calling issue (#1467)

* fix tool calling issue

* Update tool type check

* Drop print

tests/tools/test_tool_usage.py

@@ -0,0 +1,143 @@
+import json
+import random
+from unittest.mock import MagicMock, patch
+
+import pytest
+from crewai_tools import BaseTool
+from pydantic import BaseModel, Field
+
+from crewai import Agent, Crew, Task
+from crewai.tools.tool_usage import ToolUsage
+
+
+class RandomNumberToolInput(BaseModel):
+    min_value: int = Field(
+        ..., description="The minimum value of the range (inclusive)"
+    )
+    max_value: int = Field(
+        ..., description="The maximum value of the range (inclusive)"
+    )
+
+
+class RandomNumberTool(BaseTool):
+    name: str = "Random Number Generator"
+    description: str = "Generates a random number within a specified range"
+    args_schema: type[BaseModel] = RandomNumberToolInput
+
+    def _run(self, min_value: int, max_value: int) -> int:
+        return random.randint(min_value, max_value)
+
+
+# Example agent and task
+example_agent = Agent(
+    role="Number Generator",
+    goal="Generate random numbers for various purposes",
+    backstory="You are an AI agent specialized in generating random numbers within specified ranges.",
+    tools=[RandomNumberTool()],
+    verbose=True,
+)
+
+example_task = Task(
+    description="Generate a random number between 1 and 100",
+    expected_output="A random number between 1 and 100",
+    agent=example_agent,
+)
+
+
+def test_random_number_tool_usage():
+    crew = Crew(
+        agents=[example_agent],
+        tasks=[example_task],
+    )
+
+    with patch.object(random, "randint", return_value=42):
+        result = crew.kickoff()
+
+    assert "42" in result.raw
+
+
+def test_random_number_tool_range():
+    tool = RandomNumberTool()
+    result = tool._run(1, 10)
+    assert 1 <= result <= 10
+
+
+def test_random_number_tool_with_crew():
+    crew = Crew(
+        agents=[example_agent],
+        tasks=[example_task],
+    )
+
+    result = crew.kickoff()
+
+    # Check if the result contains a number between 1 and 100
+    assert any(str(num) in result.raw for num in range(1, 101))
+
+
+def test_random_number_tool_invalid_range():
+    tool = RandomNumberTool()
+    with pytest.raises(ValueError):
+        tool._run(10, 1)  # min_value > max_value
+
+
+def test_random_number_tool_schema():
+    tool = RandomNumberTool()
+
+    # Get the schema using model_json_schema()
+    schema = tool.args_schema.model_json_schema()
+
+    # Convert the schema to a string
+    schema_str = json.dumps(schema)
+
+    # Check if the schema string contains the expected fields
+    assert "min_value" in schema_str
+    assert "max_value" in schema_str
+
+    # Parse the schema string back to a dictionary
+    schema_dict = json.loads(schema_str)
+
+    # Check if the schema contains the correct field types
+    assert schema_dict["properties"]["min_value"]["type"] == "integer"
+    assert schema_dict["properties"]["max_value"]["type"] == "integer"
+
+    # Check if the schema contains the field descriptions
+    assert (
+        "minimum value" in schema_dict["properties"]["min_value"]["description"].lower()
+    )
+    assert (
+        "maximum value" in schema_dict["properties"]["max_value"]["description"].lower()
+    )
+
+
+def test_tool_usage_render():
+    tool = RandomNumberTool()
+
+    tool_usage = ToolUsage(
+        tools_handler=MagicMock(),
+        tools=[tool],
+        original_tools=[tool],
+        tools_description="Sample tool for testing",
+        tools_names="random_number_generator",
+        task=MagicMock(),
+        function_calling_llm=MagicMock(),
+        agent=MagicMock(),
+        action=MagicMock(),
+    )
+
+    rendered = tool_usage._render()
+
+    # Updated checks to match the actual output
+    assert "Tool Name: random number generator" in rendered
+    assert (
+        "Random Number Generator(min_value: 'integer', max_value: 'integer') - Generates a random number within a specified range min_value: 'The minimum value of the range (inclusive)', max_value: 'The maximum value of the range (inclusive)'"
+        in rendered
+    )
+    assert "Tool Arguments:" in rendered
+    assert (
+        "'min_value': {'description': 'The minimum value of the range (inclusive)', 'type': 'int'}"
+        in rendered
+    )
+    assert (
+        "'max_value': {'description': 'The maximum value of the range (inclusive)', 'type': 'int'}"
+        in rendered
+    )