在Node.js中借助腾讯云SDK调用混元大模型

原创

Front_Yue

发布于 2025-01-08 21:59:26

13400

代码可运行

文章被收录于专栏：云探索云探索

运行总次数：0

代码可运行

前言

随着人工智能技术的不断发展，强大的大模型如腾讯的混元大模型（HunYuan）为开发者提供了丰富的智能服务能力。本文将详细介绍如何在 Node.js 环境中使用腾讯云提供的 SDK 调用混元大模型，构建一个简单的 API 接口来实现与模型的交互。

一、环境准备

首先，确保你已经安装了 Node.js 环境。如果尚未安装，可以访问 Node.js 官网下载并安装最新版本。

二、项目初始化

创建一个新的项目目录并初始化 package.json：

mkdir hunyuan-api
cd hunyuan-api
npm init -y

安装必要的依赖包：

npm install express body-parser cors tencentcloud-sdk-nodejs-hunyuan

这些包分别是：

express：构建服务器的流行框架。
body-parser：解析请求体的中间件。
cors：处理跨域资源共享问题。
tencentcloud-sdk-nodejs-hunyuan：腾讯云混元大模型的 Node.js SDK。

三、编写服务器代码

新建 app.js 文件，并添加以下代码：

const express = require("express");
const bodyParser = require("body-parser");
const cors = require("cors");
const tencentcloud = require("tencentcloud-sdk-nodejs-hunyuan");
const HunyuanClient = tencentcloud.hunyuan.v20230901.Client;

const app = express();
const port = 3000;

// 使用 CORS 中间件，允许所有来源
app.use(cors());

// 解析 JSON 请求体
app.use(bodyParser.json());

// 定义 POST /hunyuan 接口
app.post("/hunyuan", async (req, res) => {
  console.log("Received request body:", req.body);

  const clientConfig = {
    credential: {
      secretId: "YOUR_SECRET_ID",      // 替换为你的 SecretId
      secretKey: "YOUR_SECRET_KEY",    // 替换为你的 SecretKey
    },
    region: "your-region",               // 替换为你的区域，如 "ap-guangzhou"
    profile: {
      httpProfile: {
        endpoint: "hunyuan.tencentcloudapi.com",
      },
    },
  };

  const client = new HunyuanClient(clientConfig);

  const params = {
    Model: "hunyuan-pro",
    Messages: req.body,
    Stream: false,
  };

  try {
    const response = await client.ChatCompletions(params);
    res.json(response);
  } catch (err) {
    console.error("Error calling HunYuan:", err);
    res.status(500).json({ error: '请求失败', details: err.message });
  }
});

app.listen(port, () => {
  console.log(`Server is running on port ${port}`);
});

四、代码解析

中间件配置：
- cors 允许跨域请求，避免前端请求时的跨域问题。
- bodyParser.json() 用于解析 JSON 格式的请求体。
接口定义：
- 定义了一个 POST /hunyuan 接口，接收前端发送的消息体。
- 配置腾讯云的 clientConfig，包括 secretId 和 secretKey，请确保这些凭证信息的安全性，建议使用环境变量或配置文件管理。
调用混元大模型：
- 使用 HunyuanClient 创建客户端实例。
- 构建请求参数，包括模型类型、消息体和是否流式传输。
- 调用 ChatCompletions 方法发送请求，并返回模型的响应。
错误处理：
- 在调用过程中，如果发生错误，捕获并返回 500 状态码及错误信息。

五、启动服务器

在终端中运行以下命令启动服务器：

node app.js

你应该会看到：

Server is running on port 3000

六、通过Apifox工具进行测试

我们准备测试数据如下：

[
    {
        "Role": "user",
        "Content": "今天星期几？"
    }
]

在Apifox编写接口地址以及测试数据

然后我们开始执行，获取最终数据。

七、前端调用示例

以下是一个简单的前端 JavaScript 示例，展示如何通过 fetch 调用上述接口：

async function sendMessage(message) {
  const response = await fetch('http://localhost:3000/hunyuan', {
    method: 'POST',
    headers: {
      'Content-Type': 'application/json'
    },
    body: JSON.stringify([{ content: message, role: "user" }])
  });

  const data = await response.json();

  if (response.ok) {
    console.log("HunYuan Response:", data.Response.Messages);
  } else {
    console.error("Error:", data.error);
  }
}

// 示例调用
sendMessage("你好，混元大模型！");

六、注意事项

凭证安全：切勿将 secretId 和 secretKey 直接暴露在代码中，建议使用环境变量进行管理。例如，可以使用 dotenv 库来加载 .env 文件中的配置。
区域设置：确保 region 参数设置为腾讯云混元大模型所在的区域。具体区域信息请参考腾讯云区域列表。
请求频率限制：根据腾讯云的使用规范，注意调用频率限制，以避免被限制或产生额外费用。
错误处理：在生产环境中，建议添加更完善的错误处理机制，例如日志记录、错误分类等，以便更好地维护和排查问题。