Fix modelInfo typing (#998)

Fix #996
huggingface · Oct 30, 2024 · c565b6b · c565b6b
1 parent a965b74
commit c565b6b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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
@@ -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;