tool_entrypoints.py

#!/usr/bin/env python3
"""File-based function tool entrypoints for the production Monty runtime."""

from __future__ import annotations

import sys
from pathlib import Path
from typing import Any

_PACKAGE_DIR = Path(__file__).resolve().parent
_ROOT_DIR = _PACKAGE_DIR.parent
for candidate in (_ROOT_DIR, _PACKAGE_DIR):
    candidate_str = str(candidate)
    if candidate_str not in sys.path:
        sys.path.insert(0, candidate_str)

from monty_api import (  # noqa: E402
    HELPER_EXTERNALS,
    hf_hub_query as _hf_hub_query,
    hf_hub_query_raw as _hf_hub_query_raw,
    main,
)


async def hf_hub_query(
    code: str,
    query: str | None = None,
    max_calls: int | None = None,
    timeout_sec: int | None = None,
) -> dict[str, Any]:
    return await _hf_hub_query(
        code=code,
        query=query,
        max_calls=max_calls,
        timeout_sec=timeout_sec,
    )


async def hf_hub_query_raw(
    code: str,
    query: str | None = None,
    max_calls: int | None = None,
    timeout_sec: int | None = None,
) -> Any:
    return await _hf_hub_query_raw(
        code=code,
        query=query,
        max_calls=max_calls,
        timeout_sec=timeout_sec,
    )

__all__ = [
    "HELPER_EXTERNALS",
    "hf_hub_query",
    "hf_hub_query_raw",
    "main",
]

if __name__ == "__main__":
    raise SystemExit(main())