tjdify/api/core/tools/utils/tool_parameter_converter.py

from typing import Any

from core.tools.entities.tool_entities import ToolParameter


class ToolParameterConverter:
    @staticmethod
    def get_parameter_type(parameter_type: str | ToolParameter.ToolParameterType) -> str:
        match parameter_type:
            case ToolParameter.ToolParameterType.STRING \
                 | ToolParameter.ToolParameterType.SECRET_INPUT \
                 | ToolParameter.ToolParameterType.SELECT:
                return 'string'

            case ToolParameter.ToolParameterType.BOOLEAN:
                return 'boolean'

            case ToolParameter.ToolParameterType.NUMBER:
                return 'number'

            case _:
                raise ValueError(f"Unsupported parameter type {parameter_type}")

    @staticmethod
    def cast_parameter_by_type(value: Any, parameter_type: str) -> Any:
        # convert tool parameter config to correct type
        try:
            match parameter_type:
                case ToolParameter.ToolParameterType.STRING \
                     | ToolParameter.ToolParameterType.SECRET_INPUT \
                     | ToolParameter.ToolParameterType.SELECT:
                    if value is None:
                        return ''
                    else:
                        return value if isinstance(value, str) else str(value)

                case ToolParameter.ToolParameterType.BOOLEAN:
                    if value is None:
                        return False
                    elif isinstance(value, str):
                        # Allowed YAML boolean value strings: https://yaml.org/type/bool.html
                        # and also '0' for False and '1' for True
                        match value.lower():
                            case 'true' | 'yes' | 'y' | '1':
                                return True
                            case 'false' | 'no' | 'n' | '0':
                                return False
                            case _:
                                return bool(value)
                    else:
                        return value if isinstance(value, bool) else bool(value)

                case ToolParameter.ToolParameterType.NUMBER:
                    if isinstance(value, int) | isinstance(value, float):
                        return value
                    elif isinstance(value, str):
                        if '.' in value:
                            return float(value)
                        else:
                            return int(value)
                case ToolParameter.ToolParameterType.FILE:
                    return value
                case _:
                    return str(value)

        except Exception:
            raise ValueError(f"The tool parameter value {value} is not in correct type of {parameter_type}.")