"""工具装饰器工厂"""
from typing import Callable, Any, Dict
from alcor.tools.core import ToolDefinition, registry


def tool(
    name: str,
    description: str,
    parameters: Dict[str, Any],
    category: str = "general"
) -> Callable:
    """
    工具注册装饰器
    
    用法示例:
    ```python
    @tool(
        name="web_search",
        description="Search the internet for information",
        parameters={
            "type": "object",
            "properties": {
                "query": {"type": "string", "description": "Search keywords"},
                "max_results": {"type": "integer", "description": "Max results", "default": 5}
            },
            "required": ["query"]
        },
        category="crawler"
    )
    def web_search(arguments: dict) -> dict:
        # 实现...
        return {"results": []}
    ```
    """
    def decorator(func: Callable) -> Callable:
        tool_def = ToolDefinition(
            name=name,
            description=description,
            parameters=parameters,
            handler=func,
            category=category
        )
        registry.register(tool_def)
        return func
    return decorator


def tool_function(
    name: str,
    description: str,
    parameters: Dict[str, Any],
    category: str = "general"
):
    """
    工具装饰器的别名，提供更语义化的命名
    """
    return tool(name=name, description=description, parameters=parameters, category=category)