hooks.py

from typing import Any, Awaitable, Callable

from iii import ApiRequest, ApiResponse, FunctionInfo, get_context

from .iii import get_iii


def use_api(
    config: dict[str, Any],
    handler: Callable[[ApiRequest[Any], Any], Awaitable[ApiResponse[Any]]],
) -> None:
    iii = get_iii()
    api_path = config["api_path"]
    http_method = config["http_method"]
    function_id = f"api.{http_method.lower()}.{api_path}"

    async def wrapped(data: Any) -> dict[str, Any]:
        req = ApiRequest(**data) if isinstance(data, dict) else data
        ctx = get_context()
        result = await handler(req, ctx)
        return result.model_dump(by_alias=True)

    iii.register_function(function_id, wrapped)
    iii.register_trigger(
        type="http",
        function_id=function_id,
        config={
            "api_path": api_path,
            "http_method": http_method,
            "description": config.get("description"),
            "metadata": config.get("metadata"),
        },
    )


def use_functions_available(callback: Callable[[list[FunctionInfo]], None]) -> Callable[[], None]:
    iii = get_iii()
    return iii.on_functions_available(callback)