347 lines
18 KiB
JavaScript
347 lines
18 KiB
JavaScript
const ToolExecutor = require('./toolExecutor');
|
||
const db = require('../db');
|
||
const { hasKnowledgeRouteKeyword } = require('./knowledgeKeywords');
|
||
|
||
function normalizeTextForSpeech(text) {
|
||
return (text || '')
|
||
.replace(/^#{1,6}\s*/gm, '')
|
||
.replace(/\*\*([^*]*)\*\*/g, '$1')
|
||
.replace(/__([^_]*)__/g, '$1')
|
||
.replace(/\*([^*]+)\*/g, '$1')
|
||
.replace(/_([^_]+)_/g, '$1')
|
||
.replace(/~~([^~]*)~~/g, '$1')
|
||
.replace(/`{1,3}[^`]*`{1,3}/g, '')
|
||
.replace(/^[-*]{3,}\s*$/gm, '')
|
||
.replace(/^>\s*/gm, '')
|
||
.replace(/!\[[^\]]*\]\([^)]*\)/g, '')
|
||
.replace(/\[([^\]]*)\]\([^)]*\)/g, '$1')
|
||
.replace(/^[\s]*[-*+]\s+/gm, ' ')
|
||
.replace(/^[\s]*\d+[.)]\s+/gm, ' ')
|
||
.replace(/---\s*来源[：:]/g, '来源：')
|
||
.replace(/\r/g, ' ')
|
||
.replace(/\n{2,}/g, '。')
|
||
.replace(/\n/g, ' ')
|
||
.replace(/。{2,}/g, '。')
|
||
.replace(/([！？；，])\1+/g, '$1')
|
||
.replace(/([。！？；，])\s*([。！？；，])/g, '$2')
|
||
.replace(/\s+/g, ' ')
|
||
.trim();
|
||
}
|
||
|
||
function splitTextForSpeech(text, maxLen = 180) {
|
||
const content = normalizeTextForSpeech(text);
|
||
if (!content) return [];
|
||
if (content.length <= maxLen) return [content];
|
||
const chunks = [];
|
||
let remaining = content;
|
||
while (remaining.length > maxLen) {
|
||
const currentMaxLen = chunks.length === 0 ? Math.min(90, maxLen) : maxLen;
|
||
let splitIndex = Math.max(
|
||
remaining.lastIndexOf('。', currentMaxLen),
|
||
remaining.lastIndexOf('！', currentMaxLen),
|
||
remaining.lastIndexOf('？', currentMaxLen),
|
||
remaining.lastIndexOf('；', currentMaxLen),
|
||
remaining.lastIndexOf('，', currentMaxLen),
|
||
remaining.lastIndexOf(',', currentMaxLen)
|
||
);
|
||
if (splitIndex < Math.floor(currentMaxLen / 2)) {
|
||
splitIndex = currentMaxLen;
|
||
} else {
|
||
splitIndex += 1;
|
||
}
|
||
chunks.push(remaining.slice(0, splitIndex).trim());
|
||
remaining = remaining.slice(splitIndex).trim();
|
||
}
|
||
if (remaining) chunks.push(remaining);
|
||
return chunks.filter(Boolean);
|
||
}
|
||
|
||
function estimateSpeechDurationMs(text) {
|
||
const plainText = normalizeTextForSpeech(text).replace(/\s+/g, '');
|
||
const length = plainText.length;
|
||
return Math.max(4000, Math.min(60000, length * 180));
|
||
}
|
||
|
||
function normalizeKnowledgeAlias(text) {
|
||
return String(text || '')
|
||
.replace(/X{2}系统/gi, '一成系统')
|
||
.replace(/一城系统|逸城系统|一程系统|易成系统|一诚系统|亦成系统|艺成系统|溢成系统|义成系统|毅成系统|怡成系统|以成系统|已成系统|亿成系统|忆成系统|益成系统|益生系统|易诚系统|义诚系统|忆诚系统|以诚系统|一声系统|亿生系统|易乘系统/g, '一成系统')
|
||
.replace(/(?<![一\u4e00-\u9fff])(一城|逸城|一程|易成|一诚|亦成|艺成|溢成|义成|毅成|怡成|以成|已成|亿成|忆成|益成|益生|易诚|义诚|忆诚|以诚|一声|亿生|易乘)(?=系统)/g, '一成')
|
||
.replace(/大窝|大握|大我|大卧/g, '大沃')
|
||
.replace(/盛咖学院|圣咖学愿|盛咖学院|圣咖学院|盛卡学愿/g, '盛咖学愿')
|
||
.replace(/AI众享|Ai众享|爱众享|艾众享|哎众享/gi, 'Ai众享')
|
||
.replace(/暖炉原理/g, '火炉原理');
|
||
}
|
||
|
||
function hasKnowledgeKeyword(text) {
|
||
const normalized = normalizeKnowledgeAlias(text).replace(/\s+/g, '');
|
||
return hasKnowledgeRouteKeyword(normalized);
|
||
}
|
||
|
||
function isKnowledgeFollowUp(text) {
|
||
const normalized = String(text || '').trim().replace(/[，,。！？?~～\s]+$/g, '').replace(/^(那你|那再|那|你再|再来|再|麻烦你|帮我)[，,、\s]*/g, '');
|
||
if (!normalized) return false;
|
||
if (/^(详细|详细说说|详细查一下|展开说说|继续说|继续讲|介绍一下|给我介绍一下|详细介绍一下|继续介绍一下|怎么用|怎么操作|怎么配置|适合谁|有什么区别|费用多少|价格多少|怎么申请|怎么开通|是什么|什么意思|地址在哪|公司地址在哪|电话多少|公司电话多少|联系方式|公司联系方式|具体政策|具体内容|怎么吃|功效是什么|有什么功效|成分是什么|有什么成分|多少钱|哪里买|怎么买|配方|原理是什么|有什么好处|怎么服用|适合什么人)$/.test(normalized)) {
|
||
return true;
|
||
}
|
||
return /^(这个|那个|它|该系统|这个系统|那个系统|这个功能|那个功能|这个产品|那个产品|这个公司|那家公司|这个政策|那个政策|这个培训|那个培训)(的)?(详细|详细说说|详细查一下|展开说说|继续说|继续讲|介绍一下|给我介绍一下|详细介绍一下|继续介绍一下|怎么用|怎么操作|怎么配置|适合谁|有什么区别|费用多少|价格多少|怎么申请|怎么开通|是什么|什么意思|地址在哪|公司地址在哪|电话多少|公司电话多少|联系方式|公司联系方式|具体政策|具体内容|怎么吃|功效是什么|有什么功效|成分是什么|有什么成分|多少钱|哪里买|怎么买|配方|原理是什么|有什么好处|怎么服用|适合什么人)$/.test(normalized);
|
||
}
|
||
|
||
function shouldForceKnowledgeRoute(userText, context = []) {
|
||
const text = (userText || '').trim();
|
||
if (!text) return false;
|
||
if (hasKnowledgeKeyword(text)) return true;
|
||
if (!isKnowledgeFollowUp(text)) return false;
|
||
const recentContextText = (Array.isArray(context) ? context : [])
|
||
.slice(-6)
|
||
.map((item) => String(item?.content || '').trim())
|
||
.join('\n');
|
||
return hasKnowledgeKeyword(recentContextText);
|
||
}
|
||
|
||
function withHandoffSummary(session, context) {
|
||
const summary = String(session?.handoffSummary || '').trim();
|
||
if (!summary || session?.handoffSummaryUsed) {
|
||
return context;
|
||
}
|
||
return [
|
||
{ role: 'assistant', content: `会话交接摘要：${summary}` },
|
||
...(Array.isArray(context) ? context : []),
|
||
];
|
||
}
|
||
|
||
function getRuleBasedDirectRouteDecision(userText) {
|
||
const text = (userText || '').trim();
|
||
if (!text) return { route: 'chat', args: {} };
|
||
if (/(几点|几号|日期|星期|周几|现在时间|当前时间)/.test(text)) return { route: 'get_current_time', args: {} };
|
||
if (/(天气|气温|下雨|晴天|阴天|温度)/.test(text)) {
|
||
return { route: 'query_weather', args: { city: text.replace(/.*?(北京|上海|广州|深圳|杭州|成都|重庆|武汉|西安|南京|苏州|天津|长沙|郑州|青岛|宁波|无锡)/, '$1') || '北京' } };
|
||
}
|
||
if (/(订单|物流|快递|单号)/.test(text)) return { route: 'query_order', args: { order_id: text } };
|
||
if (/^[\d\s+\-*/().=%]+$/.test(text) || /(等于多少|帮我算|计算一下|算一下)/.test(text)) {
|
||
return { route: 'calculate', args: { expression: text.replace(/(帮我算|计算一下|算一下|等于多少)/g, '').trim() || text } };
|
||
}
|
||
if (/(传销|骗局|骗子|正规吗|合法吗|正不正规|合不合法|是不是传销|直销还是传销|层级分销|非法集资|拉人头|下线|发展下线|报单|人头费)/.test(text)) {
|
||
return { route: 'search_knowledge', args: { query: text } };
|
||
}
|
||
if (/(产品|公司.*产品|产品.*有哪些|有什么产品|产品介绍|产品列表|公司介绍|我们公司|你们公司|你们的产品|我们的产品|都有什么|有些什么)/.test(text)) {
|
||
return { route: 'search_knowledge', args: { query: text } };
|
||
}
|
||
if (/(培训|新人入门|新人.*起步|团队培训|会议组织|起步三关|成长上总裁|精品会议)/.test(text)) {
|
||
return { route: 'search_knowledge', args: { query: text } };
|
||
}
|
||
if (/(招商|代理|加盟|招募|事业机会|创业|代理商|招商稿|合作.*加盟|事业.*合作)/.test(text)) {
|
||
return { route: 'search_knowledge', args: { query: text } };
|
||
}
|
||
if (/(一成系统|Ai众享|AI众享|数字化工作室|盛咖学愿|邀约话术|文化解析|AI赋能|ai赋能|团队发展|一成AI|一成Ai|AI落地|ai落地|转观念)/.test(text)) {
|
||
return { route: 'search_knowledge', args: { query: text } };
|
||
}
|
||
if (/(地址|电话|联系方式|总部|分公司|公司.*实力|公司.*背景|PM公司|德国PM)/.test(text)) {
|
||
return { route: 'search_knowledge', args: { query: text } };
|
||
}
|
||
if (/(成分|功效|怎么吃|怎么服用|吃法|用法|服用方法|副作用|好转反应|排毒反应|整应反应|搭配|原料|配方)/.test(text)) {
|
||
return { route: 'search_knowledge', args: { query: text } };
|
||
}
|
||
if (/(能吃吗|可以吃吗|能喝吗|可以喝吗|能用吗|可以用吗|一起吃|同时吃|混着吃|搭配吃|跟药一起)/.test(text)) {
|
||
return { route: 'search_knowledge', args: { query: text } };
|
||
}
|
||
if (/(发痒|头晕|便秘|腹泻|拉肚子|恶心|呕吐|长痘|出疹子|红肿|上火|胃痛|抽筋|浮肿|失眠|出汗)/.test(text)) {
|
||
return { route: 'search_knowledge', args: { query: text } };
|
||
}
|
||
if (/(高血压|糖尿病|胆固醇|心脏病|肾病|肝病|痛风|贫血|甲亢|甲减|胃炎|肠炎|哮喘|湿疹|过敏|癌症|肿瘤|尿酸|结石)/.test(text)) {
|
||
return { route: 'search_knowledge', args: { query: text } };
|
||
}
|
||
if (/(孕妇|哺乳期|怀孕|孕期|儿童|小孩|老人|老年人|手术后|化疗|放疗|术后|月子)/.test(text)) {
|
||
return { route: 'search_knowledge', args: { query: text } };
|
||
}
|
||
if (/(多少钱|价格|贵不贵|性价比|划算|值得买|保质期|储存|哪里买|怎么买|多久见效|适合谁|适用人群)/.test(text)) {
|
||
return { route: 'search_knowledge', args: { query: text } };
|
||
}
|
||
if (/(营养素|营养品|保健品|保健食品|营养补充|细胞营养|NTC|暖炉原理|火炉原理|阿育吠陀)/.test(text)) {
|
||
return { route: 'search_knowledge', args: { query: text } };
|
||
}
|
||
if (/(科普|误区|认证|检测|检测报告|安全认证|GMP|Halal|临床试验)/.test(text)) {
|
||
return { route: 'search_knowledge', args: { query: text } };
|
||
}
|
||
if (/(免疫力|抗疲劳|排毒|减肥|瘦身|护肤|护发|脱发|掉发|抗氧化|抗衰老|美容|美白|关节|骨密度|胶原蛋白)/.test(text)) {
|
||
return { route: 'search_knowledge', args: { query: text } };
|
||
}
|
||
if (/(信用评级|行业排名|获奖|慈善|社会责任|不上市|汽车奖励|退休金|旅行|福利|发展历程|全球布局|各国地址|各国电话|多少个国家)/.test(text)) {
|
||
return { route: 'search_knowledge', args: { query: text } };
|
||
}
|
||
if (/(奖金制度|事业机会|创业|副业|兼职|投资多少|门槛|起步费|怎么赚钱|能赚钱吗|值得做吗)/.test(text)) {
|
||
return { route: 'search_knowledge', args: { query: text } };
|
||
}
|
||
if (/^(喂|你好|您好|嗨|哈喽|hello|hi|在吗|在不在|早上好|中午好|下午好|晚上好|早安|晚安|谢谢|感谢|再见|拜拜|嗯|哦|好的|对|是的|没有了|没事了|可以了|行|OK|ok)[，,！。？?~～\s]*[啊呀吧呢哦嗯嘛哈的了]*[！。？?~～]*$/.test(text)) {
|
||
return { route: 'chat', args: {} };
|
||
}
|
||
if (/^(喂[，,\s]*)?(你好|您好)[，,！。？?\s]*(在吗|请问)?[！。？?]*$/.test(text)) {
|
||
return { route: 'chat', args: {} };
|
||
}
|
||
return { route: 'chat', args: {} };
|
||
}
|
||
|
||
function extractToolResultText(toolName, toolResult) {
|
||
if (!toolResult) return '';
|
||
if (toolName === 'search_knowledge') {
|
||
if (toolResult.errorType === 'timeout') {
|
||
return '知识库查询超时了，请稍后重试，或换一种更具体的问法再试。';
|
||
}
|
||
if (toolResult.errorType === 'not_configured') {
|
||
return '知识库当前未配置完成，请先检查知识库配置。';
|
||
}
|
||
if (toolResult.errorType === 'endpoint_not_configured') {
|
||
return '知识库已配置但方舟LLM端点未就绪，暂时无法检索，请稍后再试。';
|
||
}
|
||
if (toolResult.results && Array.isArray(toolResult.results)) {
|
||
return toolResult.results.map((item) => item.content || JSON.stringify(item)).join('\n');
|
||
}
|
||
if (typeof toolResult === 'string') return toolResult;
|
||
if (toolResult.error) return toolResult.error;
|
||
}
|
||
if (toolName === 'query_weather' && !toolResult.error) return `${toolResult.city}今天${toolResult.weather}，气温${toolResult.temp}，湿度${toolResult.humidity}，${toolResult.wind}。${toolResult.tips || ''}`.trim();
|
||
if (toolName === 'query_order' && !toolResult.error) return `订单${toolResult.order_id}当前状态是${toolResult.status}，预计送达时间${toolResult.estimated_delivery}，快递单号${toolResult.tracking_number}。`;
|
||
if (toolName === 'get_current_time' && !toolResult.error) return `现在是${toolResult.datetime}，${toolResult.weekday}。`;
|
||
if (toolName === 'calculate' && !toolResult.error) return `${toolResult.expression} 的计算结果是 ${toolResult.formatted}。`;
|
||
if (toolResult.error) return toolResult.error;
|
||
return typeof toolResult === 'string' ? toolResult : JSON.stringify(toolResult);
|
||
}
|
||
|
||
async function resolveReply(sessionId, session, text) {
|
||
const recentMessages = await db.getRecentMessages(sessionId, 20).catch(() => []);
|
||
const scopedMessages = session?.handoffSummaryUsed
|
||
? recentMessages.filter((item) => !/^chat_/i.test(String(item?.source || '')))
|
||
: recentMessages;
|
||
const baseContext = scopedMessages
|
||
.filter((item) => item && (item.role === 'user' || item.role === 'assistant'))
|
||
.map((item) => ({ role: item.role, content: item.content }));
|
||
const context = withHandoffSummary(session, baseContext);
|
||
const originalText = text.trim();
|
||
let routeDecision = getRuleBasedDirectRouteDecision(text.trim());
|
||
if (routeDecision.route === 'chat' && shouldForceKnowledgeRoute(text.trim(), context)) {
|
||
routeDecision = { route: 'search_knowledge', args: { query: text.trim() } };
|
||
}
|
||
let replyText = '';
|
||
let source = 'voice_bot';
|
||
let toolName = null;
|
||
let responseMeta = {
|
||
route: routeDecision.route,
|
||
original_text: originalText,
|
||
};
|
||
if (routeDecision.route === 'chat') {
|
||
session.handoffSummaryUsed = true;
|
||
return {
|
||
delivery: 'upstream_chat',
|
||
speechText: '',
|
||
ragItems: [],
|
||
source,
|
||
toolName,
|
||
routeDecision,
|
||
responseMeta,
|
||
};
|
||
} else {
|
||
toolName = routeDecision.route;
|
||
source = 'voice_tool';
|
||
const toolArgs = toolName === 'search_knowledge'
|
||
? { ...(routeDecision.args || {}), response_mode: 'answer', session_id: sessionId }
|
||
: routeDecision.args;
|
||
const toolResult = await ToolExecutor.execute(routeDecision.route, toolArgs, context);
|
||
replyText = extractToolResultText(toolName, toolResult);
|
||
responseMeta = {
|
||
...responseMeta,
|
||
tool_name: toolName,
|
||
tool_args: toolArgs || {},
|
||
source: toolResult?.source || null,
|
||
original_query: toolResult?.original_query || routeDecision.args?.query || originalText,
|
||
rewritten_query: toolResult?.rewritten_query || null,
|
||
selected_dataset_ids: toolResult?.selected_dataset_ids || null,
|
||
selected_kb_routes: toolResult?.selected_kb_routes || null,
|
||
hit: typeof toolResult?.hit === 'boolean' ? toolResult.hit : null,
|
||
reason: toolResult?.reason || null,
|
||
error_type: toolResult?.errorType || null,
|
||
latency_ms: toolResult?.latency_ms || null,
|
||
};
|
||
|
||
const ragItems = toolName === 'search_knowledge'
|
||
? (toolResult?.hit && Array.isArray(toolResult?.results)
|
||
? toolResult.results
|
||
.filter((item) => item && item.content)
|
||
.map((item) => ({
|
||
title: item.title || '知识库结果',
|
||
content: item.content,
|
||
}))
|
||
: [])
|
||
: (!toolResult?.error && replyText
|
||
? [{ title: `${toolName}结果`, content: replyText }]
|
||
: []);
|
||
|
||
if (ragItems.length > 0) {
|
||
let speechText = normalizeTextForSpeech(replyText);
|
||
session.handoffSummaryUsed = true;
|
||
if (toolName === 'search_knowledge' && speechText) {
|
||
const cleanedText = speechText.replace(/^(根据知识库信息[，,：:\s]*|根据.*?[，,]\s*)/i, '');
|
||
return {
|
||
delivery: 'external_rag',
|
||
speechText: '',
|
||
ragItems: [{ title: '知识库结果', content: cleanedText || speechText }],
|
||
source,
|
||
toolName,
|
||
routeDecision,
|
||
responseMeta,
|
||
};
|
||
}
|
||
return {
|
||
delivery: 'external_rag',
|
||
speechText: '',
|
||
ragItems,
|
||
source,
|
||
toolName,
|
||
routeDecision,
|
||
responseMeta,
|
||
};
|
||
}
|
||
|
||
if (toolName === 'search_knowledge' && !toolResult?.hit) {
|
||
session.handoffSummaryUsed = true;
|
||
// 敏感问题（传销/正规性）知识库未命中时，不交给S2S自由发挥，直接返回安全回复
|
||
if (/(传销|骗局|骗子|正规吗|合法吗|正不正规|合不合法|是不是传销|直销还是传销|层级分销|非法集资|拉人头|下线|发展下线|报单|人头费)/.test(originalText)) {
|
||
const safeReply = '德国PM是一家1993年成立于德国的合法直销公司，获得邓白氏AAA+认证，业务覆盖100多个国家和地区。它不是传销，是正规的直销企业哦。如果你想了解更多，可以问我关于PM公司或产品的详细介绍。';
|
||
return {
|
||
delivery: 'external_rag',
|
||
speechText: '',
|
||
ragItems: [{ title: '品牌保护', content: safeReply }],
|
||
source: 'voice_tool',
|
||
toolName: 'search_knowledge',
|
||
routeDecision,
|
||
responseMeta: { ...responseMeta, hit: true, reason: 'brand_protection' },
|
||
};
|
||
}
|
||
return {
|
||
delivery: 'upstream_chat',
|
||
speechText: '',
|
||
ragItems: [],
|
||
source: 'voice_bot',
|
||
toolName: null,
|
||
routeDecision,
|
||
responseMeta,
|
||
};
|
||
}
|
||
}
|
||
const speechText = normalizeTextForSpeech(replyText);
|
||
session.handoffSummaryUsed = true;
|
||
if (!speechText) {
|
||
return { delivery: 'local_tts', speechText: '', ragItems: [], source, toolName, routeDecision, responseMeta };
|
||
}
|
||
return { delivery: 'local_tts', speechText, ragItems: [], source, toolName, routeDecision, responseMeta };
|
||
}
|
||
|
||
module.exports = {
|
||
getRuleBasedDirectRouteDecision,
|
||
normalizeKnowledgeAlias,
|
||
normalizeTextForSpeech,
|
||
splitTextForSpeech,
|
||
estimateSpeechDurationMs,
|
||
shouldForceKnowledgeRoute,
|
||
resolveReply,
|
||
};
|