@llmindset/hf-mcp 0.3.9 → 0.3.11

package/src/gradio-files.ts

@@ -108,7 +108,7 @@ function matchesFileType(file: FileWithUrl, fileType: FileTypeFilter): boolean {
 const FILE_TYPE_SCHEMA = z.enum(['all', 'image', 'audio', 'text']).optional().default('all').describe('Filter by type');
 
 export const LIST_FILES_TOOL_CONFIG = {
-	name: 'list_files',
+	name: 'gradio_files',
 	description:
 		'List files available to use as Gradio File/Image/Audio inputs. Prefer these URLs when a Space asks for a file input and the user has not provided an explicit URL.',
 	schema: z.object({
@@ -122,29 +122,7 @@ export const LIST_FILES_TOOL_CONFIG = {
 	},
 } as const;
 
-export const GRADIO_FILES_TOOL_CONFIG = {
-	name: 'gradio_files',
-	description:
-		'List available URLs that can be used for Gradio File Inputs. Use when an input is requested and an explicit URL has not been provided.', // This will be dynamically set with username
-	schema: z.object({
-		fileType: FILE_TYPE_SCHEMA,
-	}),
-	annotations: {
-		title: 'Gradio Files List',
-		destructiveHint: false,
-		readOnlyHint: true,
-		openWorldHint: true,
-	},
-} as const;
-
-export const GRADIO_FILES_PROMPT_CONFIG = {
-	name: 'Available Gradio Input Files',
-	description: 'Returns a list of files and their URLs to use as Gradio File Inputs.',
-	schema: z.object({}),
-};
-
 export type ListFilesParams = z.infer<typeof LIST_FILES_TOOL_CONFIG.schema>;
-export type GradioFilesParams = z.infer<typeof GRADIO_FILES_TOOL_CONFIG.schema>;
 
 /**
  * Service for listing files from a Hugging Face Bucket or the legacy gradio-files dataset.

package/src/index.browser.ts

@@ -44,6 +44,19 @@ export const REPO_SEARCH_TOOL_CONFIG: BrowserToolConfig = {
 	},
 };
 
+export const CREATE_REPO_TOOL_CONFIG: BrowserToolConfig = {
+	name: 'create_repo',
+	description:
+		'Create a Hugging Face model, dataset, Space, or bucket repository. ' +
+		"name must be fully qualified, for example 'username/repo-name'.",
+	annotations: {
+		title: 'Create Hugging Face Repository',
+		destructiveHint: false,
+		readOnlyHint: false,
+		openWorldHint: true,
+	},
+};
+
 export const PAPER_SEARCH_TOOL_CONFIG: BrowserToolConfig = {
 	name: 'paper_search',
 	description:
@@ -121,6 +134,7 @@ export const DOC_FETCH_CONFIG: BrowserToolConfig = {
 export const SPACE_SEARCH_TOOL_ID = SEMANTIC_SEARCH_TOOL_CONFIG.name;
 export const MODEL_SEARCH_TOOL_ID = 'model_search';
 export const REPO_SEARCH_TOOL_ID = REPO_SEARCH_TOOL_CONFIG.name;
+export const CREATE_REPO_TOOL_ID = CREATE_REPO_TOOL_CONFIG.name;
 export const MODEL_DETAIL_TOOL_ID = 'model_details';
 export const PAPER_SEARCH_TOOL_ID = PAPER_SEARCH_TOOL_CONFIG.name;
 export const DATASET_SEARCH_TOOL_ID = 'dataset_search';
@@ -139,6 +153,7 @@ export const ALL_BUILTIN_TOOL_IDS = [
 	SPACE_SEARCH_TOOL_ID,
 	MODEL_SEARCH_TOOL_ID,
 	REPO_SEARCH_TOOL_ID,
+	CREATE_REPO_TOOL_ID,
 	MODEL_DETAIL_TOOL_ID,
 	PAPER_SEARCH_TOOL_ID,
 	DATASET_SEARCH_TOOL_ID,
@@ -168,6 +183,7 @@ export const TOOL_ID_GROUPS = {
 	hf_api: [
 		SPACE_SEARCH_TOOL_ID,
 		REPO_SEARCH_TOOL_ID,
+		CREATE_REPO_TOOL_ID,
 		PAPER_SEARCH_TOOL_ID,
 		HUB_REPO_DETAILS_TOOL_ID,
 		DOCS_SEMANTIC_SEARCH_TOOL_ID,

package/src/index.ts

@@ -8,6 +8,7 @@ export * from './utilities.js';
 export * from './paper-search.js';
 export * from './dataset-search.js';
 export * from './repo-search.js';
+export * from './create-repo.js';
 export * from './dataset-detail.js';
 export * from './hub-inspect.js';
 export * from './duplicate-space.js';

package/src/jobs/types.ts

@@ -1,3 +1,4 @@
+import type { SpaceHardwareFlavor } from '@huggingface/hub';
 import { z } from 'zod';
 
 /**
@@ -20,8 +21,8 @@ export const GPU_FLAVORS = [
 	'a10g-largex2',
 	'a10g-largex4',
 	'a100-large',
-	'h100',
-	'h100x8',
+	'a100x4',
+	'a100x8',
 ] as const;
 
 export const SPECIALIZED_FLAVORS = ['inf2x6'] as const;
@@ -30,6 +31,13 @@ export const ALL_FLAVORS = [...CPU_FLAVORS, ...GPU_FLAVORS, ...SPECIALIZED_FLAVO
 
 export type JobFlavor = (typeof ALL_FLAVORS)[number];
 
+function assertExhaustiveHardwareUnion<T extends never>(_value?: T): void {
+	void _value;
+}
+
+assertExhaustiveHardwareUnion<Exclude<SpaceHardwareFlavor, JobFlavor>>();
+assertExhaustiveHardwareUnion<Exclude<JobFlavor, SpaceHardwareFlavor>>();
+
 /**
  * Job status stages (from OpenAPI spec)
  */

package/src/tool-ids.ts

@@ -10,6 +10,7 @@ import {
 	MODEL_DETAIL_PROMPT_CONFIG,
 	PAPER_SEARCH_TOOL_CONFIG,
 	REPO_SEARCH_TOOL_CONFIG,
+	CREATE_REPO_TOOL_CONFIG,
 	DATASET_SEARCH_TOOL_CONFIG,
 	DATASET_DETAIL_TOOL_CONFIG,
 	DATASET_DETAIL_PROMPT_CONFIG,
@@ -30,6 +31,7 @@ import {
 export const SPACE_SEARCH_TOOL_ID = SEMANTIC_SEARCH_TOOL_CONFIG.name;
 export const MODEL_SEARCH_TOOL_ID = MODEL_SEARCH_TOOL_CONFIG.name;
 export const REPO_SEARCH_TOOL_ID = REPO_SEARCH_TOOL_CONFIG.name;
+export const CREATE_REPO_TOOL_ID = CREATE_REPO_TOOL_CONFIG.name;
 export const MODEL_DETAIL_TOOL_ID = MODEL_DETAIL_TOOL_CONFIG.name;
 export const PAPER_SEARCH_TOOL_ID = PAPER_SEARCH_TOOL_CONFIG.name;
 export const DATASET_SEARCH_TOOL_ID = DATASET_SEARCH_TOOL_CONFIG.name;
@@ -53,6 +55,7 @@ export const ALL_BUILTIN_TOOL_IDS = [
 	SPACE_SEARCH_TOOL_ID,
 	MODEL_SEARCH_TOOL_ID,
 	REPO_SEARCH_TOOL_ID,
+	CREATE_REPO_TOOL_ID,
 	MODEL_DETAIL_TOOL_ID,
 	PAPER_SEARCH_TOOL_ID,
 	DATASET_SEARCH_TOOL_ID,
@@ -82,6 +85,7 @@ export const TOOL_ID_GROUPS = {
 	hf_api: [
 		SPACE_SEARCH_TOOL_ID,
 		REPO_SEARCH_TOOL_ID,
+		CREATE_REPO_TOOL_ID,
 		PAPER_SEARCH_TOOL_ID,
 		HUB_REPO_DETAILS_TOOL_ID,
 		DOCS_SEMANTIC_SEARCH_TOOL_ID,

package/test/list-files.spec.ts

@@ -1,7 +1,7 @@
 import { describe, it, expect, vi, beforeEach } from 'vitest';
 import * as hub from '@huggingface/hub';
 import type { ListFileEntry } from '@huggingface/hub';
-import { LIST_FILES_TOOL_CONFIG, ListFilesTool, GRADIO_FILES_TOOL_CONFIG } from '../src/gradio-files.js';
+import { LIST_FILES_TOOL_CONFIG, ListFilesTool } from '../src/gradio-files.js';
 
 vi.mock('@huggingface/hub', () => ({
 	listFiles: vi.fn(),
@@ -113,9 +113,8 @@ describe('ListFilesTool', () => {
 		expect(markdown).not.toContain('private bucket URLs require authorization');
 	});
 
-	it('exposes list_files while keeping gradio_files compatibility config', () => {
-		expect(LIST_FILES_TOOL_CONFIG.name).toBe('list_files');
+	it('exposes the gradio_files tool config', () => {
+		expect(LIST_FILES_TOOL_CONFIG.name).toBe('gradio_files');
 		expect(LIST_FILES_TOOL_CONFIG.annotations.readOnlyHint).toBe(true);
-		expect(GRADIO_FILES_TOOL_CONFIG.name).toBe('gradio_files');
 	});
 });