chore: apply linting fixes to crewai-tools

This commit is contained in:
Greyson LaLonde
2025-10-02 11:06:38 -04:00
committed by GitHub
parent df754dbcc8
commit 5456c80556
222 changed files with 3482 additions and 2552 deletions

View File

@@ -1,5 +1,4 @@
import json
from typing import List, Optional, Type
from unittest import mock
from crewai.tools.base_tool import BaseTool, EnvVar
@@ -11,23 +10,23 @@ import pytest
class MockToolSchema(BaseModel):
query: str = Field(..., description="The query parameter")
count: int = Field(5, description="Number of results to return")
filters: Optional[List[str]] = Field(None, description="Optional filters to apply")
filters: list[str] | None = Field(None, description="Optional filters to apply")
class MockTool(BaseTool):
name: str = "Mock Search Tool"
description: str = "A tool that mocks search functionality"
args_schema: Type[BaseModel] = MockToolSchema
args_schema: type[BaseModel] = MockToolSchema
another_parameter: str = Field(
"Another way to define a default value", description=""
)
my_parameter: str = Field("This is default value", description="What a description")
my_parameter_bool: bool = Field(False)
package_dependencies: List[str] = Field(
package_dependencies: list[str] = Field(
["this-is-a-required-package", "another-required-package"], description=""
)
env_vars: List[EnvVar] = [
env_vars: list[EnvVar] = [
EnvVar(
name="SERPER_API_KEY",
description="API key for Serper",
@@ -110,7 +109,7 @@ def test_extract_init_params_schema(mock_tool_extractor):
assert my_parameter["type"] == "string"
my_parameter_bool = init_params_schema["properties"]["my_parameter_bool"]
assert my_parameter_bool["default"] == False
assert not my_parameter_bool["default"]
assert my_parameter_bool["type"] == "boolean"
@@ -121,12 +120,12 @@ def test_extract_env_vars(mock_tool_extractor):
api_key_var, rate_limit_var = tool_info["env_vars"]
assert api_key_var["name"] == "SERPER_API_KEY"
assert api_key_var["description"] == "API key for Serper"
assert api_key_var["required"] == True
assert api_key_var["default"] == None
assert api_key_var["required"]
assert api_key_var["default"] is None
assert rate_limit_var["name"] == "API_RATE_LIMIT"
assert rate_limit_var["description"] == "API rate limit"
assert rate_limit_var["required"] == False
assert not rate_limit_var["required"]
assert rate_limit_var["default"] == "100"
@@ -151,7 +150,7 @@ def test_extract_run_params_schema(mock_tool_extractor):
filters_param = run_params_schema["properties"]["filters"]
assert filters_param["description"] == "Optional filters to apply"
assert filters_param["default"] == None
assert filters_param["default"] is None
assert filters_param["anyOf"] == [
{"items": {"type": "string"}, "type": "array"},
{"type": "null"},