npm install

# Hugging Face API 密钥（如果使用 Hugging Face 服务）
HUGGINGFACE_API_KEY=your_huggingface_api_key

# Dify API 配置（如果使用 Dify 服务）
DIFY_API_KEY=your_dify_api_key
DIFY_BASE_URL=your_dify_base_url

npm run start:dify

import { createHuggingFaceMcpTool } from './src/hf_mcp';

// 创建工具实例
const hfTool = createHuggingFaceMcpTool('your_huggingface_api_key');

// 使用工具调用模型
async function example() {
  const result = await hfTool.textGeneration(
    'gpt2',
    '人工智能正在改变世界，',
    { max_new_tokens: 50 }
  );
  console.log(result);
}

example();

// 向 Dify 发送聊天消息
const response = await difyClient.chatMessage(
  "你好，请介绍一下自己", // 查询内容
  {}, // 输入参数
  "conversation_id_123", // 对话 ID（可选）
  "user_123" // 用户标识（可选）
);

console.log(response.answer); // 输出 Dify 的回复

// 使用 Dify 生成文本
const response = await difyClient.completionMessage(
  { prompt: "写一篇关于人工智能的短文" }, // 输入参数
  "user_123" // 用户标识（可选）
);

console.log(response.answer); // 输出生成的文本

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);