File size: 424 Bytes
88c4c60
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
import { describe, it, expect } from "vitest";
import { parseModel } from "../../open-sse/services/model.js";

describe("HuggingFace model alias parsing", () => {
  it("resolves hf alias to huggingface provider", () => {
    expect(parseModel("hf/black-forest-labs/FLUX.1-schnell")).toMatchObject({
      provider: "huggingface",
      model: "black-forest-labs/FLUX.1-schnell",
      providerAlias: "hf",
    });
  });
});