from swiftclaw import Agent

agent = Agent(name="test-agent")

@agent.tool
async def search_database(query: str):
    """Search the database"""
    return await db.search(query)

# Test the tool
result = await agent.test_tool(
    "search_database",
    query="test query"
)
print(result)

swiftclaw dev --debug

agent = Agent(
    name="my-agent",
    debug=True
)

swiftclaw logs my-agent --filter tool-execution

@agent.tool(trace=True)
async def complex_tool(param: str):
    """Complex tool with tracing"""
    with agent.trace("step1"):
        result1 = await step1(param)
    
    with agent.trace("step2"):
        result2 = await step2(result1)
    
    return result2

@agent.tool
async def risky_tool(param: str):
    try:
        return await external_api.call(param)
    except APIError as e:
        agent.log_error(f"API error: {e}")
        return {"error": str(e)}
    except Exception as e:
        agent.log_error(f"Unexpected error: {e}")
        raise

from pydantic import BaseModel, validator

class SearchParams(BaseModel):
    query: str
    limit: int = 10
    
    @validator('limit')
    def validate_limit(cls, v):
        if v < 1 or v > 100:
            raise ValueError('Limit must be between 1 and 100')
        return v

@agent.tool
async def search(params: SearchParams):
    return await db.search(params.query, params.limit)

@agent.tool(profile=True)
async def slow_tool(param: str):
    """Tool with performance profiling"""
    result = await expensive_operation(param)
    return result

swiftclaw tools profile my-agent slow_tool

# Mock external API
@agent.tool(mock=True)
async def external_api_call(param: str):
    """Mocked API call"""
    return {"status": "success", "data": "mock data"}

import pytest
from swiftclaw.testing import AgentTest

@pytest.mark.asyncio
async def test_search_tool():
    agent = AgentTest("my-agent")
    
    result = await agent.call_tool(
        "search_database",
        query="test"
    )
    
    assert result["status"] == "success"
    assert len(result["results"]) > 0

swiftclaw tools list my-agent

swiftclaw logs my-agent --filter tool-calls --tail 100

result = await agent.test_tool(
    "my_tool",
    param1="value1",
    param2="value2"
)

swiftclaw metrics my-agent --metric tool-execution-time