from typing import Any from nanobot.agent.tools.base import Tool from nanobot.agent.tools.registry import ToolRegistry from nanobot.agent.tools.shell import ExecTool class SampleTool(Tool): @property def name(self) -> str: return "sample" @property def description(self) -> str: return "sample tool" @property def parameters(self) -> dict[str, Any]: return { "type": "object", "properties": { "query": {"type": "string", "minLength": 2}, "count": {"type": "integer", "minimum": 1, "maximum": 10}, "mode": {"type": "string", "enum": ["fast", "full"]}, "meta": { "type": "object", "properties": { "tag": {"type": "string"}, "flags": { "type": "array", "items": {"type": "string"}, }, }, "required": ["tag"], }, }, "required": ["query", "count"], } async def execute(self, **kwargs: Any) -> str: return "ok" def test_validate_params_missing_required() -> None: tool = SampleTool() errors = tool.validate_params({"query": "hi"}) assert "missing required count" in "; ".join(errors) def test_validate_params_type_and_range() -> None: tool = SampleTool() errors = tool.validate_params({"query": "hi", "count": 0}) assert any("count must be >= 1" in e for e in errors) errors = tool.validate_params({"query": "hi", "count": "2"}) assert any("count should be integer" in e for e in errors) def test_validate_params_enum_and_min_length() -> None: tool = SampleTool() errors = tool.validate_params({"query": "h", "count": 2, "mode": "slow"}) assert any("query must be at least 2 chars" in e for e in errors) assert any("mode must be one of" in e for e in errors) def test_validate_params_nested_object_and_array() -> None: tool = SampleTool() errors = tool.validate_params( { "query": "hi", "count": 2, "meta": {"flags": [1, "ok"]}, } ) assert any("missing required meta.tag" in e for e in errors) assert any("meta.flags[0] should be string" in e for e in errors) def test_validate_params_ignores_unknown_fields() -> None: tool = SampleTool() errors = tool.validate_params({"query": "hi", "count": 2, "extra": "x"}) assert errors == [] async def test_registry_returns_validation_error() -> None: reg = ToolRegistry() reg.register(SampleTool()) result = await reg.execute("sample", {"query": "hi"}) assert "Invalid parameters" in result def test_exec_extract_absolute_paths_keeps_full_windows_path() -> None: cmd = r"type C:\user\workspace\txt" paths = ExecTool._extract_absolute_paths(cmd) assert paths == [r"C:\user\workspace\txt"] def test_exec_extract_absolute_paths_ignores_relative_posix_segments() -> None: cmd = ".venv/bin/python script.py" paths = ExecTool._extract_absolute_paths(cmd) assert "/bin/python" not in paths def test_exec_extract_absolute_paths_captures_posix_absolute_paths() -> None: cmd = "cat /tmp/data.txt > /tmp/out.txt" paths = ExecTool._extract_absolute_paths(cmd) assert "/tmp/data.txt" in paths assert "/tmp/out.txt" in paths # --- cast_params tests --- class CastTestTool(Tool): """Minimal tool for testing cast_params.""" def __init__(self, schema: dict[str, Any]) -> None: self._schema = schema @property def name(self) -> str: return "cast_test" @property def description(self) -> str: return "test tool for casting" @property def parameters(self) -> dict[str, Any]: return self._schema async def execute(self, **kwargs: Any) -> str: return "ok" def test_cast_params_string_to_int() -> None: tool = CastTestTool( { "type": "object", "properties": {"count": {"type": "integer"}}, } ) result = tool.cast_params({"count": "42"}) assert result["count"] == 42 assert isinstance(result["count"], int) def test_cast_params_string_to_number() -> None: tool = CastTestTool( { "type": "object", "properties": {"rate": {"type": "number"}}, } ) result = tool.cast_params({"rate": "3.14"}) assert result["rate"] == 3.14 assert isinstance(result["rate"], float) def test_cast_params_string_to_bool() -> None: tool = CastTestTool( { "type": "object", "properties": {"enabled": {"type": "boolean"}}, } ) assert tool.cast_params({"enabled": "true"})["enabled"] is True assert tool.cast_params({"enabled": "false"})["enabled"] is False assert tool.cast_params({"enabled": "1"})["enabled"] is True def test_cast_params_array_items() -> None: tool = CastTestTool( { "type": "object", "properties": { "nums": {"type": "array", "items": {"type": "integer"}}, }, } ) result = tool.cast_params({"nums": ["1", "2", "3"]}) assert result["nums"] == [1, 2, 3] def test_cast_params_nested_object() -> None: tool = CastTestTool( { "type": "object", "properties": { "config": { "type": "object", "properties": { "port": {"type": "integer"}, "debug": {"type": "boolean"}, }, }, }, } ) result = tool.cast_params({"config": {"port": "8080", "debug": "true"}}) assert result["config"]["port"] == 8080 assert result["config"]["debug"] is True def test_cast_params_bool_not_cast_to_int() -> None: """Booleans should not be silently cast to integers.""" tool = CastTestTool( { "type": "object", "properties": {"count": {"type": "integer"}}, } ) # Bool input should remain bool (validation will catch it) result = tool.cast_params({"count": True}) assert result["count"] is True # Not cast to 1 def test_cast_params_preserves_empty_string() -> None: """Empty strings should be preserved for string type.""" tool = CastTestTool( { "type": "object", "properties": {"name": {"type": "string"}}, } ) result = tool.cast_params({"name": ""}) assert result["name"] == "" def test_cast_params_bool_string_false() -> None: """Test that 'false', '0', 'no' strings convert to False.""" tool = CastTestTool( { "type": "object", "properties": {"flag": {"type": "boolean"}}, } ) assert tool.cast_params({"flag": "false"})["flag"] is False assert tool.cast_params({"flag": "False"})["flag"] is False assert tool.cast_params({"flag": "0"})["flag"] is False assert tool.cast_params({"flag": "no"})["flag"] is False assert tool.cast_params({"flag": "NO"})["flag"] is False def test_cast_params_bool_string_invalid() -> None: """Invalid boolean strings should not be cast.""" tool = CastTestTool( { "type": "object", "properties": {"flag": {"type": "boolean"}}, } ) # Invalid strings should be preserved (validation will catch them) result = tool.cast_params({"flag": "random"}) assert result["flag"] == "random" result = tool.cast_params({"flag": "maybe"}) assert result["flag"] == "maybe" def test_cast_params_invalid_string_to_int() -> None: """Invalid strings should not be cast to integer.""" tool = CastTestTool( { "type": "object", "properties": {"count": {"type": "integer"}}, } ) result = tool.cast_params({"count": "abc"}) assert result["count"] == "abc" # Original value preserved result = tool.cast_params({"count": "12.5.7"}) assert result["count"] == "12.5.7" def test_cast_params_invalid_string_to_number() -> None: """Invalid strings should not be cast to number.""" tool = CastTestTool( { "type": "object", "properties": {"rate": {"type": "number"}}, } ) result = tool.cast_params({"rate": "not_a_number"}) assert result["rate"] == "not_a_number" def test_cast_params_none_values() -> None: """Test None handling for different types.""" tool = CastTestTool( { "type": "object", "properties": { "name": {"type": "string"}, "count": {"type": "integer"}, "items": {"type": "array"}, "config": {"type": "object"}, }, } ) result = tool.cast_params( { "name": None, "count": None, "items": None, "config": None, } ) # None should be preserved for all types assert result["name"] is None assert result["count"] is None assert result["items"] is None assert result["config"] is None def test_cast_params_single_value_not_auto_wrapped_to_array() -> None: """Single values should NOT be automatically wrapped into arrays.""" tool = CastTestTool( { "type": "object", "properties": {"items": {"type": "array"}}, } ) # Non-array values should be preserved (validation will catch them) result = tool.cast_params({"items": 5}) assert result["items"] == 5 # Not wrapped to [5] result = tool.cast_params({"items": "text"}) assert result["items"] == "text" # Not wrapped to ["text"] def test_cast_params_empty_string_to_array() -> None: """Empty string should convert to empty array.""" tool = CastTestTool( { "type": "object", "properties": {"items": {"type": "array"}}, } ) result = tool.cast_params({"items": ""}) assert result["items"] == [] def test_cast_params_empty_string_to_object() -> None: """Empty string should convert to empty object.""" tool = CastTestTool( { "type": "object", "properties": {"config": {"type": "object"}}, } ) result = tool.cast_params({"config": ""}) assert result["config"] == {} def test_cast_params_float_to_int() -> None: """Float values should be cast to integers.""" tool = CastTestTool( { "type": "object", "properties": {"count": {"type": "integer"}}, } ) result = tool.cast_params({"count": 42.7}) assert result["count"] == 42 assert isinstance(result["count"], int)