Hermes对话系统原理与实践

// 配置意图分类器
const intentClassifier = new IntentClassifier({
model: 'transformers',
modelName: 'bert-base-chinese',
intents: [
{ name: 'greeting', patterns: ['你好', '嗨', '早上好'] },
{ name: 'weather', patterns: ['天气', '温度', '下雨'] },
{ name: 'news', patterns: ['新闻', '资讯', '头条'] }
]
});

async function classifyIntent(text) {
// 1. 文本预处理
const cleanedText = preprocess(text);
// 2. 规则匹配（快速路径）
const ruleMatch = await ruleMatcher.match(cleanedText);
if (ruleMatch) return ruleMatch;
// 3. 机器学习模型预测
const prediction = await model.predict(cleanedText);
return prediction.intent;
}

// 实体提取示例
const entityExtractor = new EntityExtractor();
const result = await entityExtractor.extract('帮我查询北京明天的天气');
// 输出：
// {
// location: '北京',
// date: '明天',
// intent: 'weather'
// }

class SlotFiller {
constructor(slots) {
this.slots = slots; // 需要填充的槽位
this.filledSlots = {};
}
async fill(userInput) {
// 提取实体
const entities = await this.entityExtractor.extract(userInput);
// 更新已填充的槽位
Object.keys(entities).forEach(key => {
if (this.slots.includes(key)) {
this.filledSlots[key] = entities[key];
}
});
// 检查是否所有槽位都已填充
const missingSlots = this.slots.filter(s => !this.filledSlots[s]);
if (missingSlots.length > 0) {
return `请问${missingSlots.join('和')}是什么？`;
}
return null; // 所有槽位已填充
}
}

class DialogManager {
constructor() {
this.state = 'idle'; // idle, asking, processing, finished
this.context = {};
}
async handleInput(input) {
switch (this.state) {
case 'idle':
const intent = await this.intentClassifier.classify(input);
this.state = 'asking';
return await this.getResponse(intent);
case 'asking':
const slotResponse = await this.slotFiller.fill(input);
if (slotResponse) {
return slotResponse;
}
this.state = 'processing';
return await this.executeAction();
case 'processing':
this.state = 'finished';
return '任务已完成！';
}
}
}

class ContextManager {
constructor(maxHistory = 10) {
this.history = [];
this.maxHistory = maxHistory;
}
addTurn(userInput, systemResponse) {
this.history.push({
user: userInput,
system: systemResponse,
timestamp: Date.now()
});
// 保持历史记录数量限制
while (this.history.length > this.maxHistory) {
this.history.shift();
}
}
getHistory() {
return this.history;
}
}

const responseTemplates = {
weather: {
template: '{location}明天的天气是{weather}，温度{temperature}度',
fallback: '抱歉，无法获取天气信息'
},
greeting: {
template: '你好！我是Hermes，有什么可以帮助你的？',
fallback: '你好！'
}
};
function generateResponse(intent, data) {
const template = responseTemplates[intent];
if (!template) return '我不太明白你的意思'; let response = template.template;
Object.keys(data).forEach(key => {
response = response.replace(`{${key}}`, data[key]);
});
return response;
}

class LLMResponseGenerator {
constructor(modelName = 'gpt-2') {
this.model = this.loadModel(modelName);
}
async generate(context, maxLength = 50) {
// 构建prompt
const prompt = this.buildPrompt(context);
// 生成响应
const response = await this.model.generate({
prompt: prompt,
maxLength: maxLength,
temperature: 0.7
});
return response.text.trim();
}
}

// 创建对话系统实例
const hermes = new Hermes({
intents: [
{ name: 'weather', patterns: ['天气', '温度', '预报'] }
],
slots: ['location', 'date']
});
// 对话流程
async function chat() {
console.log('Hermes: 你好！请问需要查询什么？');
// 第一轮：用户表达意图
const response1 = await hermes.handle('明天天气怎么样');
console.log(`Hermes: ${response1}`); // "请问地点是什么？"
// 第二轮：用户提供地点
const response2 = await hermes.handle('北京');
console.log(`Hermes: ${response2}`); // "北京明天天气晴朗，温度25度"
}
chat();