MCP 服务工具

这是一个 MCP（模型上下文协议）服务工具集合，用于调用各种 AI 服务 API。

功能特点

Hugging Face MCP 服务

• 支持文本生成、分类、问答、填充遮罩、摘要生成、翻译等任务
• 支持图像分类、图像分割、图像到文本转换以及零样本图像分类
• 提供语音识别功能

Dify MCP 服务

• 提供多种 AI 应用接口，具体参数和返回值可能因配置不同而有所差异
• 具体功能请参考相关文档

安装与配置

1. 安装依赖 使用 Node.js 的包管理工具 npm 进行安装：

   npm install mcp-service-tools
   

2. 环境变量配置 设置相应的 API 密钥：

   export HUGGING_FACE_KEY='your-api-key'
   export DIFY_API_KEY='your-api-key'
   

使用方法

Hugging Face 工具示例

const { hfTool } = require('mcp-service-tools');

// 文本生成
await hfTool.textGeneration('gpt2', '输入文本', { max_new_tokens: 50 });

// 文本分类
await hfTool.textClassification('distilbert-base-uncased-finetuned-sst-2-english', '这是一段文本');

// 问答
await hfTool.questionAnswering('deepset/roberta-base-squad2', {
  question: '问题?',
  context: '包含答案的上下文'
});

// 填充遮罩
await hfTool.fillMask('bert-base-uncased', 'Paris is the [MASK] of France.');

// 摘要生成
await hfTool.summarization('facebook/bart-large-cnn', '长文本内容');

// 文本翻译
await hfTool.translation('t5-base', '要翻译的文本');

// 图像分类
await hfTool.imageClassification('google/vit-base-patch16-224', imageBlob);

// 图像分割
await hfTool.imageSegmentation('facebook/detr-resnet-50-panoptic', imageBlob);

// 图像到文本
await hfTool.imageToText('nlpconnect/vit-gpt2-image-captioning', imageBlob);

// 零样本图像分类
await hfTool.zeroShotImageClassification(
  'openai/clip-vit-base-patch32',
  imageBlob,
  ['猫', '狗', '鸟']
);

// 零样本文本分类
await hfTool.zeroShotClassification(
  'facebook/bart-large-mnli',
  '这是一段文本',
  ['类别1', '类别2', '类别3'],
  false
);

// 语音识别
await hfTool.automaticSpeechRecognition('facebook/wav2vec2-base-960h', audioBlob);

Dify 工具示例

const { difyTool } = require('mcp-service-tools');

// 示例调用（具体参数请参考官方文档）
await difyTool.someFunction('parameter1', 'parameter2');

注意事项

• 使用前需要先获取相应的 API 密钥
• 请注意 API 调用限制和费用
• Dify API 的具体参数和返回值可能因应用配置不同而有所差异

许可证

MIT