diff --git a/packages/hub/src/lib/model-info.spec.ts b/packages/hub/src/lib/model-info.spec.ts index cdbad71c16..81e6f82467 100644 --- a/packages/hub/src/lib/model-info.spec.ts +++ b/packages/hub/src/lib/model-info.spec.ts @@ -5,10 +5,12 @@ describe("modelInfo", () => { it("should return the model info", async () => { const info = await modelInfo({ name: "openai-community/gpt2", + additionalFields: ["author"], }); expect(info).toEqual({ id: "621ffdc036468d709f17434d", downloads: expect.any(Number), + author: "openai-community", gated: false, name: "openai-community/gpt2", updatedAt: expect.any(Date), diff --git a/packages/hub/src/lib/model-info.ts b/packages/hub/src/lib/model-info.ts index 2744905c05..02f41d717e 100644 --- a/packages/hub/src/lib/model-info.ts +++ b/packages/hub/src/lib/model-info.ts @@ -7,7 +7,7 @@ import { pick } from "../utils/pick"; import { MODEL_EXPAND_KEYS, type MODEL_EXPANDABLE_KEYS, type ModelEntry } from "./list-models"; export async function modelInfo< - const T extends Exclude<(typeof MODEL_EXPANDABLE_KEYS)[number], (typeof MODEL_EXPANDABLE_KEYS)[number]> = never, + const T extends Exclude<(typeof MODEL_EXPANDABLE_KEYS)[number], (typeof MODEL_EXPAND_KEYS)[number]> = never, >( params: { name: string;