TinyFishTool

def __init__(
    self,
    *,
    llm_config: LLMConfig | dict[str, Any] | None = None,
    tinyfish_api_key: str | None = None,
):
    """Initializes the TinyFishTool.

    Args:
        llm_config (Optional[Union[LLMConfig, dict[str, Any]]]): LLM configuration.
            (Currently unused but kept for potential future integration).
        tinyfish_api_key (Optional[str]): The API key for the TinyFish API. If not provided,
            it attempts to read from the ``TINYFISH_API_KEY`` environment variable.

    Raises:
        ValueError: If ``tinyfish_api_key`` is not provided either directly or via the environment variable.
    """
    self.tinyfish_api_key = tinyfish_api_key or os.getenv("TINYFISH_API_KEY")

    if self.tinyfish_api_key is None:
        raise ValueError("tinyfish_api_key must be provided either as an argument or via TINYFISH_API_KEY env var")

    def tinyfish_scrape(
        url: Annotated[str, "The URL to scrape."],
        goal: Annotated[str, "A natural language description of what information to extract from the page."],
        tinyfish_api_key: Annotated[str | None, Depends(on(self.tinyfish_api_key))],
    ) -> dict[str, Any]:
        """Deep-scrape a URL using TinyFish with a natural language goal.

        Pass a URL and describe what you want to extract. TinyFish will navigate the page
        and return structured data matching your goal.

        Args:
            url: The URL to scrape.
            goal: A natural language description of what to extract (e.g.,
                "Find all team members and their roles",
                "Extract pricing information and plan details").
            tinyfish_api_key: The API key for TinyFish (injected dependency).

        Returns:
            A dictionary containing the URL, goal, and extracted data.

        Raises:
            ValueError: If the TinyFish API key is not available.
        """
        if tinyfish_api_key is None:
            raise ValueError("TinyFish API key is missing.")
        return _tinyfish_scrape(
            url=url,
            goal=goal,
            tinyfish_api_key=tinyfish_api_key,
        )

    super().__init__(
        name="tinyfish_scrape",
        description="Deep-scrape a URL using TinyFish. Pass a URL and a natural language goal describing what to extract.",
        func_or_tool=tinyfish_scrape,
    )