当前位置：首页 > news >正文

使用 Node.js 快速构建基于 Taotoken 多模型的内容生成服务

news 2026/4/30 18:57:04

使用 Node.js 快速构建基于 Taotoken 多模型的内容生成服务

1. 环境准备与基础配置

在开始构建内容生成服务前，请确保已安装 Node.js 16 或更高版本。创建一个新项目目录并初始化 npm 包管理：

mkdir taotoken-content-service cd taotoken-content-service npm init -y

安装必要的依赖包，包括openai官方 SDK 和dotenv用于管理环境变量：

npm install openai dotenv

在项目根目录创建.env文件，用于安全存储 Taotoken API Key：

TAOTOKEN_API_KEY=your_api_key_here

重要安全提示：请勿将.env文件提交到版本控制系统，建议将其添加到.gitignore中。

2. 初始化 OpenAI 客户端

创建一个taotokenClient.js文件，配置基础客户端：

import { config } from 'dotenv'; import OpenAI from 'openai'; config(); // 加载 .env 文件 const taotokenClient = new OpenAI({ apiKey: process.env.TAOTOKEN_API_KEY, baseURL: 'https://taotoken.net/api', }); export default taotokenClient;

这段代码初始化了一个指向 Taotoken 端点的 OpenAI 兼容客户端。注意baseURL设置为https://taotoken.net/api，这是 Taotoken 平台的标准 OpenAI 兼容接口地址。

3. 实现基础内容生成函数

创建contentGenerator.js文件，实现核心生成逻辑：

import taotokenClient from './taotokenClient.js'; /** * 使用指定模型生成内容 * @param {string} model - 模型ID，如 'claude-sonnet-4-6' * @param {string} prompt - 用户输入提示 * @param {number} [maxRetries=3] - 最大重试次数 * @returns {Promise<string>} 生成的内容 */ export async function generateContent(model, prompt, maxRetries = 3) { let attempt = 0; while (attempt < maxRetries) { try { const completion = await taotokenClient.chat.completions.create({ model, messages: [{ role: 'user', content: prompt }], }); return completion.choices[0]?.message?.content || ''; } catch (error) { attempt++; if (attempt >= maxRetries) { throw new Error(`生成内容失败: ${error.message}`); } // 指数退避重试 await new Promise(resolve => setTimeout(resolve, 1000 * Math.pow(2, attempt)) ); } } }

此函数实现了带重试机制的内容生成逻辑，包含以下关键点：

支持通过model参数指定不同模型
内置指数退避重试策略
返回 Promise 便于异步调用

4. 构建简易 REST API 服务

创建server.js文件，使用 Express 构建 Web 服务：

import express from 'express'; import { generateContent } from './contentGenerator.js'; const app = express(); app.use(express.json()); app.post('/generate', async (req, res) => { try { const { model = 'claude-sonnet-4-6', prompt } = req.body; if (!prompt) { return res.status(400).json({ error: '缺少 prompt 参数' }); } const content = await generateContent(model, prompt); res.json({ model, content }); } catch (error) { res.status(500).json({ error: error.message }); } }); const PORT = process.env.PORT || 3000; app.listen(PORT, () => { console.log(`服务运行在 http://localhost:${PORT}`); });

启动服务后，可通过以下 curl 命令测试：

curl -X POST http://localhost:3000/generate \ -H "Content-Type: application/json" \ -d '{"model":"claude-sonnet-4-6","prompt":"如何用Node.js写一个HTTP服务？"}'

5. 模型切换与扩展建议

Taotoken 平台支持多种模型，您可以通过以下方式扩展服务：

模型列表管理：创建一个models.js文件定义可用模型：

export const SUPPORTED_MODELS = { CLAUDE_SONNET: 'claude-sonnet-4-6', CLAUDE_HAIKU: 'claude-haiku-4-8', // 添加更多支持的模型 };

模型验证中间件：在 Express 中添加模型验证：

import { SUPPORTED_MODELS } from './models.js'; app.use('/generate', (req, res, next) => { if (req.body.model && !Object.values(SUPPORTED_MODELS).includes(req.body.model)) { return res.status(400).json({ error: '不支持的模型', supportedModels: Object.values(SUPPORTED_MODELS) }); } next(); });

性能监控：考虑添加简单的调用统计：

const callStats = new Map(); app.post('/generate', async (req, res) => { const start = Date.now(); const model = req.body.model || 'claude-sonnet-4-6'; try { // ...原有逻辑... const duration = Date.now() - start; callStats.set(model, (callStats.get(model) || 0) + 1); console.log(`模型 ${model} 调用次数: ${callStats.get(model)}, 耗时: ${duration}ms`); res.json({ model, content, duration }); } catch (error) { // ...错误处理... } });