

class HuggingFaceMock:
  """Mock Hugging Face API client for testing."""

  def generate_content(
      self,
      messages: list[dict[str, str]],
      model: str,
      max_tokens: int | None=None,
      temperature: float | None=None,
      stop: list[str] | None=None) -> str:
    return 'mock response'
