openclaw/extensions/huggingface/index.test.ts

import { describe, expect, it, vi } from "vitest";
import { createTestPluginApi } from "../../test/helpers/plugins/plugin-api.js";

const buildHuggingfaceProviderMock = vi.hoisted(() =>
  vi.fn(async () => ({
    baseUrl: "https://router.huggingface.co/v1",
    api: "openai-completions",
    models: [],
  })),
);

vi.mock("./provider-catalog.js", () => ({
  buildHuggingfaceProvider: buildHuggingfaceProviderMock,
}));

vi.mock("./onboard.js", () => ({
  applyHuggingfaceConfig: vi.fn((cfg) => cfg),
  HUGGINGFACE_DEFAULT_MODEL_REF: "huggingface/deepseek-ai/DeepSeek-R1",
}));

import plugin from "./index.js";

function registerProvider() {
  const registerProviderMock = vi.fn();

  plugin.register(
    createTestPluginApi({
      id: "huggingface",
      name: "Hugging Face",
      source: "test",
      config: {},
      runtime: {} as never,
      registerProvider: registerProviderMock,
    }),
  );

  expect(registerProviderMock).toHaveBeenCalledTimes(1);
  return registerProviderMock.mock.calls[0]?.[0];
}

describe("huggingface plugin", () => {
  it("skips catalog discovery when models.huggingfaceDiscovery.enabled is false", async () => {
    const provider = registerProvider();

    const result = await provider.catalog.run({
      config: { models: { huggingfaceDiscovery: { enabled: false } } },
      resolveProviderApiKey: () => ({
        apiKey: "hf_test_token",
        discoveryApiKey: "hf_test_token",
      }),
    } as never);

    expect(result).toBeNull();
    expect(buildHuggingfaceProviderMock).not.toHaveBeenCalled();
  });
});