src/agent/utils/tooling.py

from functools import wraps
import json
import inspect
import re

def generate_tools_json(functions):
    """
    Generates a JSON representation of the tools based on the provided functions.
    """
    tools = []
    for func in functions:
        tools.append(func.tool_info)

    with open('tools.json', 'w') as file:
        json.dump(tools, file, indent=4)

    return {
        "tools": tools
    }

def tool(func):
    """
    Decorator to get function information for tool generation.
    Args:
        func: The function to be decorated.
    """
    @wraps(func)
    def wrapper(*args, **kwargs):
        return func(*args, **kwargs)

    description = func.__doc__.strip() if func.__doc__ else "No description available."

    sig = inspect.signature(func)
    parameters = {}
    required = []

    for param_name, param in sig.parameters.items():
        param_type = param.annotation
        if param_type == param.empty:
            param_type = "string"
        else:
            param_type = param_type.__name__
            if param_type == 'str':
                param_type = 'string'
            elif param_type == 'int':
                param_type = 'integer'
            elif param_type == 'list':
                param_type = 'array'
            elif param_type == 'float':
                param_type = 'number'
            elif param_type == 'bool':
                param_type = 'boolean'
            elif param_type == 'dict':
                param_type = 'object'

        param_description = f"The {param_name}."

        docstring_lines = description.split('\n')
        for line in docstring_lines:
            match = re.match(rf"\s*{param_name}\s*\(.*\):\s*(.*)", line)
            if match:
                param_description = match.group(1).strip()
                break

        parameters[param_name] = {
            "type": param_type,
            "description": param_description,
        }

        if param.default == param.empty:
            required.append(param_name)

    wrapper.tool_info = {
        "type": "function",
        "function": {
            "name": func.__name__,
            "description": description.split('\n')[0],
            "parameters": {
                "type": "object",
                "properties": parameters,
                "required": required,
            },
        },
    }
    return wrapper