download_huggingface_model
Download Hugging Face models to your local machine with options for file filtering, revision selection, and custom directories.
Instructions
⚡ 高速下载 HuggingFace 模型到本地
Input Schema
TableJSON Schema
| Name | Required | Description | Default |
|---|---|---|---|
| repo_id | Yes | HuggingFace 仓库 ID（格式：owner/repo） | |
| download_dir | No | 下载目录（默认：~/Downloads/huggingface_models） | |
| revision | No | Git 分支或标签 | main |
| pattern | No | 下载文件模式（例：*.safetensors） | |
| exclude | No | 排除文件模式（例：*.bin） | |
| files | No | 指定下载的文件列表 | |
| max_files | No | 最大下载文件数 | |
| force | No | 强制重新下载 | |
| max_concurrent | No | 最大并发下载数 |
Implementation Reference
- src/mcp/tools.js:215-287 (handler)Primary execution logic for the download_huggingface_model tool. Validates arguments, prepares download options, invokes the downloader, and returns formatted success/error results.async callDownloadTool(args) { try { // 基础验证 if (!args.repo_id) { return CallToolResult.error( ToolContent.text('缺少必需参数: repo_id') ); } logger.info(`开始下载: ${args.repo_id}`); // 获取配置 const config = getConfig(); const downloadDir = args.download_dir || config.download_dir; // 构建目标目录 const targetDir = require('path').join( downloadDir, args.repo_id.replace('/', '_') ); // 构建选项 const options = { revision: args.revision, pattern: args.pattern, exclude: args.exclude, files: args.files, maxFiles: args.max_files, force: args.force, maxConcurrent: args.max_concurrent, token: args.token || process.env.HF_TOKEN }; // 执行下载 const result = await this.downloader.download( args.repo_id, targetDir, options ); if (result.success) { logger.info(`下载完成: ${result.files} 个文件`); const response = { success: true, repo_id: args.repo_id, path: result.path, files: result.files, size: result.size, duration: result.duration, suggestions: result.suggestions }; return CallToolResult.success( ToolContent.text(JSON.stringify(response, null, 2)) ); } else { logger.error('下载失败:', result.error); return CallToolResult.error( ToolContent.text(JSON.stringify({ success: false, error: result.error, suggestions: result.suggestions }, null, 2)) ); } } catch (error) { logger.error('工具调用失败:', error); return CallToolResult.error( ToolContent.text(`工具调用失败: ${error.message}`) ); } }
- src/mcp/tools.js:157-210 (schema)Tool schema definition for 'download_huggingface_model', specifying input parameters, descriptions, defaults, and required fields.getDownloadTool() { return new Tool( 'download_huggingface_model', '⚡ 高速下载 HuggingFace 模型到本地', { type: 'object', properties: { repo_id: { type: 'string', description: 'HuggingFace 仓库 ID（格式：owner/repo）', examples: ['2Noise/ChatTTS', 'microsoft/DialoGPT-medium'] }, download_dir: { type: 'string', description: '下载目录（默认：~/Downloads/huggingface_models）' }, revision: { type: 'string', description: 'Git 分支或标签', default: 'main' }, pattern: { type: 'string', description: '下载文件模式（例：*.safetensors）' }, exclude: { type: 'string', description: '排除文件模式（例：*.bin）' }, files: { type: 'array', items: { type: 'string' }, description: '指定下载的文件列表' }, max_files: { type: 'integer', description: '最大下载文件数', default: 100 }, force: { type: 'boolean', description: '强制重新下载', default: false }, max_concurrent: { type: 'integer', description: '最大并发下载数', default: 5 } }, required: ['repo_id'] } ); }
- src/mcp/tools.js:446-461 (registration)Tool dispatch registration in callTool method, mapping 'download_huggingface_model' to its handler function.async callTool(name, args) { switch (name) { case 'download_huggingface_model': return await this.callDownloadTool(args); case 'list_huggingface_files': return await this.callListTool(args); case 'explore_huggingface_repo': return await this.callExploreTool(args); case 'search_huggingface_files': return await this.callSearchTool(args); default: return CallToolResult.error( ToolContent.text(`未知工具: ${name}`) ); } }
- src/mcp/tools.js:24-31 (registration)Registers the download tool by including getDownloadTool() in the list returned by getTools(), used by the MCP server.getTools() { return [ this.getDownloadTool(), this.getListTool(), this.getExploreTool(), this.getSearchTool() ]; }