111 lines
2.9 KiB
JavaScript
111 lines
2.9 KiB
JavaScript
|
const Router = require('koa-router');
|
||
|
|
const fetch = require('node-fetch');
|
||
|
|
|
||
|
|
const router = new Router();
|
||
|
|
|
||
|
|
// NVIDIA API 配置
|
||
|
|
const NVIDIA_API_KEY = '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';
|
||
|
|
const NVIDIA_API_URL = 'https://integrate.api.nvidia.com/v1/chat/completions';
|
||
|
|
|
||
|
|
// 生成商品信息的 API
|
||
|
|
router.post('/generate-product', async (ctx) => {
|
||
|
|
try {
|
||
|
|
const { imageUrl, keywords } = ctx.request.body;
|
||
|
|
|
||
|
|
if (!imageUrl && !keywords) {
|
||
|
|
ctx.status = 400;
|
||
|
|
ctx.body = {
|
||
|
|
code: 400,
|
||
|
|
message: '请提供商品图片或关键词'
|
||
|
|
};
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
|
||
|
|
// 构建提示词
|
||
|
|
let prompt = '你是一个专业的便利店商品管理助手。';
|
||
|
|
|
||
|
|
if (imageUrl) {
|
||
|
|
prompt += `\n请分析这张商品图片：${imageUrl}`;
|
||
|
|
}
|
||
|
|
|
||
|
|
if (keywords) {
|
||
|
|
prompt += `\n关键词：${keywords}`;
|
||
|
|
}
|
||
|
|
|
||
|
|
prompt += `
|
||
|
|
请生成商品的详细信息，返回JSON格式，不要包含其他内容：
|
||
|
|
{
|
||
|
|
"name": "商品名称（简洁明了，2-10字）",
|
||
|
|
"category": "商品分类（请从以下选择：饮料,零食,日用品,食品,烟酒,其他）",
|
||
|
|
"description": "商品详细描述（50-100字，突出产品特点）",
|
||
|
|
"suggestedPrice": 建议售价（数字）
|
||
|
|
}`;
|
||
|
|
|
||
|
|
// 调用 NVIDIA API
|
||
|
|
const response = await fetch(NVIDIA_API_URL, {
|
||
|
|
method: 'POST',
|
||
|
|
headers: {
|
||
|
|
'Authorization': `Bearer ${NVIDIA_API_KEY}`,
|
||
|
|
'Content-Type': 'application/json'
|
||
|
|
},
|
||
|
|
body: JSON.stringify({
|
||
|
|
model: 'deepseek-ai/deepseek-v4-pro',
|
||
|
|
messages: [
|
||
|
|
{
|
||
|
|
role: 'user',
|
||
|
|
content: prompt
|
||
|
|
}
|
||
|
|
],
|
||
|
|
temperature: 0.7,
|
||
|
|
max_tokens: 500
|
||
|
|
})
|
||
|
|
});
|
||
|
|
|
||
|
|
if (!response.ok) {
|
||
|
|
const errorText = await response.text();
|
||
|
|
console.error('NVIDIA API Error:', response.status, errorText);
|
||
|
|
throw new Error(`API 调用失败: ${response.status}`);
|
||
|
|
}
|
||
|
|
|
||
|
|
const data = await response.json();
|
||
|
|
const aiResponse = data.choices[0].message.content;
|
||
|
|
|
||
|
|
// 解析 JSON 响应
|
||
|
|
let productInfo;
|
||
|
|
try {
|
||
|
|
// 清理响应文本，只保留 JSON 部分
|
||
|
|
const jsonMatch = aiResponse.match(/\{[\s\S]*\}/);
|
||
|
|
if (jsonMatch) {
|
||
|
|
productInfo = JSON.parse(jsonMatch[0]);
|
||
|
|
} else {
|
||
|
|
throw new Error('无法解析 AI 响应');
|
||
|
|
}
|
||
|
|
} catch (parseError) {
|
||
|
|
console.error('JSON 解析失败:', aiResponse);
|
||
|
|
// 如果解析失败，提供默认值
|
||
|
|
productInfo = {
|
||
|
|
name: keywords || '新商品',
|
||
|
|
category: '其他',
|
||
|
|
description: '商品描述',
|
||
|
|
suggestedPrice: 0
|
||
|
|
};
|
||
|
|
}
|
||
|
|
|
||
|
|
ctx.body = {
|
||
|
|
code: 200,
|
||
|
|
message: '生成成功',
|
||
|
|
data: productInfo
|
||
|
|
};
|
||
|
|
|
||
|
|
} catch (error) {
|
||
|
|
console.error('生成商品信息失败:', error);
|
||
|
|
ctx.status = 500;
|
||
|
|
ctx.body = {
|
||
|
|
code: 500,
|
||
|
|
message: error.message || '生成失败，请稍后重试'
|
||
|
|
};
|
||
|
|
}
|
||
|
|
});
|
||
|
|
|
||
|
|
module.exports = router;
|