取消 mock
This commit is contained in:
+22
-61
@@ -7,40 +7,6 @@ const router = new Router();
|
|||||||
const NVIDIA_API_KEY = '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';
|
const NVIDIA_API_KEY = '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';
|
||||||
const NVIDIA_API_URL = 'https://integrate.api.nvidia.com/v1/chat/completions';
|
const NVIDIA_API_URL = 'https://integrate.api.nvidia.com/v1/chat/completions';
|
||||||
|
|
||||||
|
|
||||||
// 启用/禁用 AI API(当网络不可用时设为 false)
|
|
||||||
const ENABLE_AI_API = false; // 由于服务器网络无法访问 NVIDIA,暂时禁用
|
|
||||||
|
|
||||||
// Mock 数据(用于测试或网络不可用时)
|
|
||||||
const mockProducts = [
|
|
||||||
{ name: '可口可乐', category: '饮料', description: '经典碳酸饮料,清爽解渴,330ml罐装', suggestedPrice: 3.5 },
|
|
||||||
{ name: '乐事薯片', category: '零食', description: '香脆可口,原味薯片,75g包装', suggestedPrice: 8.9 },
|
|
||||||
{ name: '维达纸巾', category: '日用品', description: '三层加厚,柔软亲肤,10包家庭装', suggestedPrice: 25.9 },
|
|
||||||
{ name: '康师傅方便面', category: '食品', description: '红烧牛肉面口味,12桶整箱装', suggestedPrice: 39.9 },
|
|
||||||
{ name: '农夫山泉', category: '饮料', description: '天然矿泉水,550ml瓶装,12瓶/箱', suggestedPrice: 24.0 },
|
|
||||||
{ name: '奥利奥饼干', category: '零食', description: '夹心饼干,巧克力口味,388g家庭装', suggestedPrice: 18.9 },
|
|
||||||
{ name: '旺仔牛奶', category: '饮料', description: '香甜可口,儿童喜爱,245ml罐装', suggestedPrice: 5.5 },
|
|
||||||
{ name: '益达口香糖', category: '零食', description: '清新口气,薄荷口味,40粒装', suggestedPrice: 12.9 },
|
|
||||||
{ name: '舒肤佳香皂', category: '日用品', description: '有效除菌,纯白清香,125g', suggestedPrice: 6.9 },
|
|
||||||
{ name: '金龙鱼食用油', category: '食品', description: '调和油,健康烹饪,1.8L', suggestedPrice: 45.9 },
|
|
||||||
];
|
|
||||||
|
|
||||||
// 获取随机 Mock 数据
|
|
||||||
function getMockProduct(keywords) {
|
|
||||||
let product = mockProducts[Math.floor(Math.random() * mockProducts.length)];
|
|
||||||
|
|
||||||
if (keywords) {
|
|
||||||
const keyword = keywords.toLowerCase();
|
|
||||||
const matched = mockProducts.find(p =>
|
|
||||||
p.name.toLowerCase().includes(keyword) ||
|
|
||||||
p.category.toLowerCase().includes(keyword)
|
|
||||||
);
|
|
||||||
if (matched) product = matched;
|
|
||||||
}
|
|
||||||
|
|
||||||
return { ...product };
|
|
||||||
}
|
|
||||||
|
|
||||||
// 生成商品信息的 API
|
// 生成商品信息的 API
|
||||||
router.post('/generate-product', async (ctx) => {
|
router.post('/generate-product', async (ctx) => {
|
||||||
try {
|
try {
|
||||||
@@ -55,11 +21,6 @@ router.post('/generate-product', async (ctx) => {
|
|||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
let productInfo;
|
|
||||||
let useMock = !ENABLE_AI_API; // 如果禁用 API,直接使用 Mock
|
|
||||||
|
|
||||||
if (!useMock) {
|
|
||||||
// 构建提示词
|
|
||||||
let prompt = '你是一个专业的便利店商品管理助手。';
|
let prompt = '你是一个专业的便利店商品管理助手。';
|
||||||
|
|
||||||
if (imageUrl) {
|
if (imageUrl) {
|
||||||
@@ -79,8 +40,6 @@ router.post('/generate-product', async (ctx) => {
|
|||||||
"suggestedPrice": 建议售价(数字)
|
"suggestedPrice": 建议售价(数字)
|
||||||
}`;
|
}`;
|
||||||
|
|
||||||
try {
|
|
||||||
// 调用 NVIDIA NVCF 云 API(OpenAI 兼容格式)
|
|
||||||
const response = await fetch(NVIDIA_API_URL, {
|
const response = await fetch(NVIDIA_API_URL, {
|
||||||
method: 'POST',
|
method: 'POST',
|
||||||
headers: {
|
headers: {
|
||||||
@@ -98,45 +57,47 @@ router.post('/generate-product', async (ctx) => {
|
|||||||
temperature: 0.7,
|
temperature: 0.7,
|
||||||
max_tokens: 500
|
max_tokens: 500
|
||||||
}),
|
}),
|
||||||
timeout: 30000 // 30秒超时
|
timeout: 60000
|
||||||
});
|
});
|
||||||
|
|
||||||
if (!response.ok) {
|
if (!response.ok) {
|
||||||
const errorText = await response.text();
|
const errorText = await response.text();
|
||||||
console.error('NVIDIA API Error:', response.status, errorText);
|
console.error('NVIDIA API Error:', response.status, errorText);
|
||||||
throw new Error(`API 调用失败: ${response.status}`);
|
ctx.status = 502;
|
||||||
|
ctx.body = {
|
||||||
|
code: 502,
|
||||||
|
message: 'AI 服务调用失败,请稍后重试'
|
||||||
|
};
|
||||||
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
const data = await response.json();
|
const data = await response.json();
|
||||||
const aiResponse = data.choices?.[0]?.message?.content;
|
const aiResponse = data.choices?.[0]?.message?.content;
|
||||||
|
|
||||||
if (!aiResponse) {
|
if (!aiResponse) {
|
||||||
throw new Error('API 返回为空');
|
ctx.status = 500;
|
||||||
|
ctx.body = {
|
||||||
|
code: 500,
|
||||||
|
message: 'AI 服务返回为空'
|
||||||
|
};
|
||||||
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
// 解析 JSON 响应
|
|
||||||
const jsonMatch = aiResponse.match(/\{[\s\S]*\}/);
|
const jsonMatch = aiResponse.match(/\{[\s\S]*\}/);
|
||||||
if (jsonMatch) {
|
if (!jsonMatch) {
|
||||||
productInfo = JSON.parse(jsonMatch[0]);
|
ctx.status = 500;
|
||||||
} else {
|
ctx.body = {
|
||||||
throw new Error('无法解析 AI 响应');
|
code: 500,
|
||||||
|
message: '无法解析 AI 响应'
|
||||||
|
};
|
||||||
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
} catch (apiError) {
|
const productInfo = JSON.parse(jsonMatch[0]);
|
||||||
console.error('NVIDIA API 调用异常:', apiError.message);
|
|
||||||
console.log('切换到 Mock 模式');
|
|
||||||
useMock = true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// 使用 Mock 数据
|
|
||||||
if (useMock) {
|
|
||||||
productInfo = getMockProduct(keywords);
|
|
||||||
}
|
|
||||||
|
|
||||||
ctx.body = {
|
ctx.body = {
|
||||||
code: 200,
|
code: 200,
|
||||||
message: useMock ? '使用模拟数据生成' : '生成成功',
|
message: '生成成功',
|
||||||
data: productInfo
|
data: productInfo
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user