71 lines
3.0 KiB
JavaScript
71 lines
3.0 KiB
JavaScript
|
const path = require('path');
|
|||
|
|
const https = require('https');
|
|||
|
|
const fs = require('fs');
|
|||
|
|
|
|||
|
|
// 加载 .env
|
|||
|
|
const envPath = path.join(__dirname, '../.env');
|
|||
|
|
if (fs.existsSync(envPath)) {
|
|||
|
|
fs.readFileSync(envPath, 'utf8').split('\n').forEach(line => {
|
|||
|
|
const t = line.trim();
|
|||
|
|
if (!t || t.startsWith('#')) return;
|
|||
|
|
const i = t.indexOf('=');
|
|||
|
|
if (i > 0) {
|
|||
|
|
const k = t.slice(0, i).trim();
|
|||
|
|
let v = t.slice(i + 1).trim();
|
|||
|
|
if ((v.startsWith('"') && v.endsWith('"')) || (v.startsWith("'") && v.endsWith("'")))
|
|||
|
|
v = v.slice(1, -1);
|
|||
|
|
if (!process.env[k]) process.env[k] = v;
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
const PROMPT = '你是大沃的智能助手，负责回答与PM-FitLine德国产品相关的问题。知识库涵盖近50款PM-FitLine产品的完整资料（成分、用法、剂量、价格、规格、搭配方案、好转反应等）及117个常见问答。回答规则：产品相关具体信息必须严格依据知识库，不得猜测或自行补充；公司背景、健康常识可适当补充。产品常有别名（小红=艾特维、大白=倍适、小白=维适多等），请注意识别。不得编造产品名或数据。PM是营养品非药物，涉及疾病建议咨询医生。若知识库无相关内容，坦诚说明并建议咨询推荐人。回答口语化、简洁，1-3句给结论，150字内。';
|
|||
|
|
|
|||
|
|
const QUERY = process.argv[2] || '一成系统 赋能 德国PM';
|
|||
|
|
|
|||
|
|
const m = process.env.VOLC_ARK_KB_MODEL || process.env.VOLC_ARK_KNOWLEDGE_ENDPOINT_ID;
|
|||
|
|
const ak = process.env.VOLC_ARK_API_KEY || process.env.VOLC_ACCESS_KEY_ID;
|
|||
|
|
const ids = (process.env.VOLC_ARK_KNOWLEDGE_BASE_IDS || '').split(',').map(s => s.trim()).filter(Boolean);
|
|||
|
|
const th = parseFloat(process.env.VOLC_ARK_KNOWLEDGE_THRESHOLD) || 0.3;
|
|||
|
|
|
|||
|
|
const body = JSON.stringify({
|
|||
|
|
model: m,
|
|||
|
|
messages: [
|
|||
|
|
{ role: 'system', content: PROMPT },
|
|||
|
|
{ role: 'user', content: QUERY },
|
|||
|
|
],
|
|||
|
|
metadata: { knowledge_base: { dataset_ids: ids, top_k: 3, threshold: th } },
|
|||
|
|
stream: false,
|
|||
|
|
max_tokens: 80,
|
|||
|
|
thinking: { type: 'disabled' },
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
const t0 = Date.now();
|
|||
|
|
const req = https.request({
|
|||
|
|
hostname: 'ark.cn-beijing.volces.com',
|
|||
|
|
path: '/api/v3/chat/completions',
|
|||
|
|
method: 'POST',
|
|||
|
|
headers: {
|
|||
|
|
'Content-Type': 'application/json',
|
|||
|
|
'Authorization': `Bearer ${ak}`,
|
|||
|
|
'Content-Length': Buffer.byteLength(body),
|
|||
|
|
},
|
|||
|
|
timeout: 15000,
|
|||
|
|
}, (res) => {
|
|||
|
|
let chunks = [];
|
|||
|
|
res.on('data', c => chunks.push(c));
|
|||
|
|
res.on('end', () => {
|
|||
|
|
const j = JSON.parse(Buffer.concat(chunks).toString());
|
|||
|
|
const content = j.choices?.[0]?.message?.content || '(empty)';
|
|||
|
|
const u = j.usage || {};
|
|||
|
|
const ms = Date.now() - t0;
|
|||
|
|
const isNoHit = /未找到|未提及|暂无|没有相关|无法|不确定/.test(content);
|
|||
|
|
console.log(`\n检索词: "${QUERY}"`);
|
|||
|
|
console.log(`延迟: ${ms}ms | tokens: ${u.prompt_tokens}->${u.completion_tokens} | ${isNoHit ? '❌NO-HIT' : '✅HIT'}`);
|
|||
|
|
console.log(`\n回答:\n${content}\n`);
|
|||
|
|
});
|
|||
|
|
});
|
|||
|
|
req.on('error', e => console.error('ERROR:', e.message));
|
|||
|
|
req.write(body);
|
|||
|
|
req.end();
|