64 lines
1.4 KiB
Python
64 lines
1.4 KiB
Python
"""Tool factory - decorator registration"""
|
|
from typing import Callable
|
|
from .core import ToolDefinition, registry
|
|
|
|
|
|
def tool(
|
|
name: str,
|
|
description: str,
|
|
parameters: dict,
|
|
category: str = "general"
|
|
) -> Callable:
|
|
"""
|
|
Tool registration decorator
|
|
|
|
Usage:
|
|
@tool(
|
|
name="web_search",
|
|
description="Search the web",
|
|
parameters={"type": "object", "properties": {...}},
|
|
category="crawler"
|
|
)
|
|
def web_search(arguments: dict) -> dict:
|
|
...
|
|
"""
|
|
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 register_tool(
|
|
name: str,
|
|
handler: Callable,
|
|
description: str,
|
|
parameters: dict,
|
|
category: str = "general"
|
|
) -> None:
|
|
"""
|
|
Register a tool directly (without decorator)
|
|
|
|
Usage:
|
|
register_tool(
|
|
name="my_tool",
|
|
handler=my_function,
|
|
description="Description",
|
|
parameters={...}
|
|
)
|
|
"""
|
|
tool_def = ToolDefinition(
|
|
name=name,
|
|
description=description,
|
|
parameters=parameters,
|
|
handler=handler,
|
|
category=category
|
|
)
|
|
registry.register(tool_def)
|