v3.0.1, 改成tools调用方式

This commit is contained in:
2026-08-29 23:36:59 +08:00
parent 45cf00bf14
commit 738ef1697c
6 changed files with 160 additions and 82 deletions
+5 -3
View File
@@ -19,10 +19,11 @@ function getErrorMsg(status, data) {
} }
} }
async function callDeepSeek(messages, userId, temperature = 0.1, jsonMode = true) { async function callDeepSeek(messages, userId, temperature = 0.1, jsonMode = true, tools) {
console.log('调用 DeepSeek...'); console.log('调用 DeepSeek...');
var body = { model: 'deepseek-v4-flash', messages: messages, temperature: temperature, stream: false, thinking: { type: 'disabled' } }; var body = { model: 'deepseek-v4-flash', messages: messages, temperature: temperature, stream: false, thinking: { type: 'disabled' } };
if (jsonMode) body.response_format = { type: 'json_object' }; if (tools && tools.length) body.tools = tools;
if (jsonMode && !(tools && tools.length)) body.response_format = { type: 'json_object' };
if (userId) body.user_id = sanitizeUserId(userId); if (userId) body.user_id = sanitizeUserId(userId);
var res = await fetch('https://api.deepseek.com/chat/completions', { var res = await fetch('https://api.deepseek.com/chat/completions', {
method: 'POST', method: 'POST',
@@ -50,7 +51,8 @@ async function callDeepSeek(messages, userId, temperature = 0.1, jsonMode = true
// 输入消息大小日志 // 输入消息大小日志
//var inputChars = JSON.stringify(messages).length; //var inputChars = JSON.stringify(messages).length;
//console.log('输入大小: ' + inputChars + ' 字符, ' + messages.length + ' 条消息'); //console.log('输入大小: ' + inputChars + ' 字符, ' + messages.length + ' 条消息');
return data.choices[0].message.content; // 返回整个 message 对象,可能包含 content 或 tool_calls(供上层决定走工具还是 JSON)
return data.choices[0].message;
} }
module.exports = { callDeepSeek }; module.exports = { callDeepSeek };
+30 -65
View File
@@ -1,75 +1,40 @@
// AI Prompt 模板 — 简化版(增 / 查 / 清空 / 聊天 / 忽略 // AI Prompt 模板 — 工具模式(阶段一
// 增加:价格必须存在,缺失时自动追问 // 新建采购 / 查询 / 聊天 / 忽略 走 Function Tool 调用;
// 删除 / 清空 保留 JSON 输出路径(旧逻辑兜底,见 messages.js)。
// 图片关联由后端决定,不在 prompt 中引导模型输出 use_recent_image。
function buildSystemPrompt(username, isAdmin) { function buildSystemPrompt(username, isAdmin) {
const role = '你是智能财务助手小财”。你只输出一个合法的 JSON 对象,且必须包含 "action" 字段。'; const role = '你是智能财务助手"小财"。当前用户:' + username + ',管理员状态:' + isAdmin + 'true=管理员,false=普通用户)。';
const actions = [ const guidance = [
'动作定义(action 字段必须为以下之一)', '请根据用户消息选择合适的方式回应',
'1. purchase:新建一条采购记录。', '',
'一、调用工具(适用于以下场景):',
'1. 新建采购记录 → 调用工具 create_purchase。',
' 触发条件:用户明确表达了购买/采购/下单/付款/买了/花了等消费意图,且不是在查询历史。', ' 触发条件:用户明确表达了购买/采购/下单/付款/买了/花了等消费意图,且不是在查询历史。',
' 需要提取的字段(能提取多少就提取多少,不确定的用默认值):', ' 物品名、数量、单价、运费、支付方式、发票类型、状态、备注等字段含义见工具参数定义。',
' - purchase_item (必填, 字符串, 物品名)', ' 关键规则1:如果用户没有明确提供单价(价格),不要调用 create_purchase,而应调用 reply_chat 追问价格。',
' - quantity (数字, 默认1)', ' 关键规则2:用户如提及"交了""付了""已付款""已收货""已完成"等,必须把 status 填为对应状态,不要漏填。',
' - unit_price (必填, 数字, 单价) ← 必须明确或能从语句中推断出大于0的数值', ' 关键规则3:一次消息中包含多笔采购时,必须为每一笔分别调用一次 create_purchase,不要合并或遗漏。',
' - freight (数字, 默认0)', '2. 查询历史采购记录 → 调用工具 query_purchases,参数 keywords 为物品名或关键词(查全部则用空字符串)。',
' - payment_method (字符串, 如"支付宝""微信",未提及则默认"未指定")', ' 触发条件:用户询问"买过什么/有没有买过/查一下/找找看/搜索/列表/我买过/看看"等查询意图。',
' - invoice_type (字符串, 可选值"无票"/"电子票",默认"无票")', ' 即使物品名听起来奇怪或可能不存在,也必须用 query_purchases,绝不能调 create_purchase。',
' - status (字符串, 只能从"待付款"/"已付款"/"已收货"/"已完成"中选择,默认"待付款"。如果用户明确说已付款、已收货、已完成等,请映射到对应状态)', '3. 纯闲聊、引导手动修改、追问价格、权限不足拒绝 → 调用工具 reply_chat,参数 reply 为要显示给用户的回复。',
' - applicant (字符串, 默认"' + username + '")', '4. 完全无关的消息(纯符号、纯表情、明显不是对助手说的话)→ 调用工具 ignore。',
' - remarks (字符串, 能从语句中提取的备注信息,如购买平台等)',
' ⚠️ 关键规则:如果用户没有提供价格(unit_price 为 0 或无法提取),你必须返回 action="chat"reply 内容礼貌地询问价格,例如:“好的,请问这个 ‘插排’ 是多少钱呢?”',
' 示例输出(价格齐全):{"action":"purchase","purchase_item":"扩展坞","quantity":1,"unit_price":89,"freight":0,"payment_method":"未指定","invoice_type":"无票","status":"已付款","applicant":"' + username + '","remarks":"淘宝"}',
'', '',
'2. query:查询历史采购记录。', '二、直接输出 JSON(不要调用任何工具):',
' 触发条件:用户询问“买过什么/有没有买过/查一下/找找看/搜索/列表/我买过/看看”等查询历史采购的意图。', '5. 删除采购记录 → 输出 {"action":"delete","delete_item":"物品名","quantity":数量,"amount":金额},其中 quantity/amount 可选。',
' 即使物品名听起来奇怪或可能不存在,也必须用 query,绝不能返回 ignore。', '6. 清空所有采购数据 → 输出 {"action":"clear_all"}。仅管理员(isAdmin=true)可执行;',
' 字段:', ' 非管理员必须输出 {"action":"chat","reply":"仅管理员可执行此操作"}。',
' - keywords (字符串, 提取用户想查找的物品名或关键词;如果问的是全部记录则设为空字符串 "")',
' 示例:{"action":"query","keywords":"记账本"}',
' 注意:query 会触发系统在本地数据库中搜索,然后再次调用 AI 生成回答,你无需直接给出查询结果。',
'', '',
'3. clear_all:清空所有采购数据。', '三、通用规则:',
' 触发条件:用户要求“清空所有记录”“全部删掉”“重置采购数据”等。', '- 用户要求修改、更正、调整、更新已有采购记录的任何意图(改数量、改价格、改状态、补发票、加备注等),一律用 reply_chat 引导用户手动操作,不要尝试更新记录。',
' 权限:如果当前用户是管理员(isAdmin=true),你可以返回 {"action":"clear_all"}。', '- 修改意图的统一回复模板:"如需修改记录,请前往采购清单页面,找到对应条目后手动编辑。"',
' 如果当前用户不是管理员(isAdmin=false),你必须返回 action="chat"reply 说明“仅管理员可执行此操作”。', '- 如果用户同时表达了购买和查询,优先判断为购买(create_purchase),除非提问明显是查询历史。',
'', '- 若无法确定用户意图,默认调用 reply_chat 并给出友好回复,绝不忽略。'
'4. chat:直接回复文本。',
' 适用于以下情况:',
' - 纯闲聊(如“你好”“今天天气如何”)',
' - 用户要求修改、更正、调整、更新已有采购记录的任何意图(如“把扩展坞价格改成90”“已付款”“已收货”等),你必须引导用户手动操作,不得尝试更新记录。',
' - 权限不足时的拒绝提示(如非管理员要求清空)',
' - 采购意图但缺少价格信息时,追问价格',
' - 其他无法归入上述动作但需要回复的情况',
' 字段:reply (字符串,显示给用户)',
' 修改意图的统一回复模板:"如需修改记录,请前往采购清单页面,找到对应条目后手动编辑。"',
'',
'5. ignore:完全无关的消息。',
' 触发条件:纯符号、纯表情、明显不是对助手说的话等。',
' 不需要额外字段,仅返回 {"action":"ignore"} 即可。'
].join('\n'); ].join('\n');
const rules = [ return [role, guidance].join('\n\n');
'⚠️ 重要规则:',
'- 所有输出必须是严格的 JSON,且顶层必须包含 "action" 字段。',
'- 不要输出除 JSON 外的任何文字。',
'- 当前用户:' + username + ',管理员状态:' + isAdmin + 'true=管理员,false=普通用户)。',
'- 查询意图(包含“查”“找”“看”“搜索”“有没有”“买过”等)必须走 query,绝不可判为 ignore 或 chat。',
'- 任何希望修改已有记录的说法(改数量、改价格、改状态、补发票、加备注等),一律作为 chat 引导手动操作,不要试图更新记录。',
'- 如果用户同时表达了购买和查询,优先判断为购买(purchase),除非提问明显是查询历史。',
'- 当用户说出购买意图但缺少价格时,必须返回 chat 追问价格,绝不允许带着 unit_price=0 直接创建记录。',
'- 如果无法确定 action,默认走 chat 并给出友好回复。'
].join('\n');
const forceOutputRule = [
'',
'🔴 防呆底线(最高优先级):',
'- 你必须始终返回一个合法 JSON 对象,且必须包含 "action" 字段。',
'- 即使你完全不确定用户意图,也要返回 {"action":"ignore"}。',
'- 绝不允许返回空字符串、纯文本、只有大括号无内容或任何非 JSON 格式的输出。',
'- 如果你的推理碰到矛盾,优先选择 chat 并给出引导,而不是放弃输出。'
].join('\n');
return [role, actions, rules, forceOutputRule].join('\n\n');
} }
module.exports = { buildSystemPrompt }; module.exports = { buildSystemPrompt };
+85
View File
@@ -0,0 +1,85 @@
// DeepSeek Function Tools 定义
// 阶段一:迁移 purchase / chat / ignore 为工具调用;query 也提供工具入口(避免查询意图误判成新建),
// 但其汇总逻辑仍复用 handlers.js 的 query 分支(不做二次调用回填)。
// delete / clear_all 保留旧 JSON 输出路径(在 prompt 中引导走 JSON,见 prompt.js)。
// 图片关联不在工具参数中暴露,由后端根据消息文本/附件自行决定(见 handlers.js 的 create_purchase 分支)。
const STATUS_ENUM = ['待付款', '已付款', '已收货', '已完成'];
const INVOICE_ENUM = ['无票', '电子票'];
const createPurchaseTool = {
type: 'function',
function: {
name: 'create_purchase',
description:
'新建一条采购记录。仅在用户明确表达购买/采购/下单/付款/买了/花了等消费意图且不是查询历史时调用。' +
'注意:如果用户没有明确提供单价(价格),不要调用本工具,应改为调用 reply_chat 追问价格。',
parameters: {
type: 'object',
properties: {
purchase_item: { type: 'string', description: '物品名称,必填' },
quantity: { type: 'number', description: '数量,默认 1' },
unit_price: { type: 'number', description: '单价。必须大于 0,若用户未明确给出价格请不要调用本工具,改为 reply_chat 追问。' },
freight: { type: 'number', description: '运费,默认 0' },
payment_method: { type: 'string', description: '支付方式,如 支付宝/微信,未提及则默认未指定' },
invoice_type: { type: 'string', enum: INVOICE_ENUM, description: '发票类型,默认无票' },
status: { type: 'string', enum: STATUS_ENUM, description: '付款/收货状态。若用户明确说了已付款、交了钱、付了款、已收、已到货、已完成等,必须主动填对应值(如已付款),不要省略。仅当完全没提状态时才用默认待付款。' },
remarks: { type: 'string', description: '备注,如购买平台等' }
},
required: ['purchase_item']
}
}
};
const replyChatTool = {
type: 'function',
function: {
name: 'reply_chat',
description:
'直接回复一段文本给用户。用于以下情况:' +
'1) 纯闲聊(如打招呼、问天气等);' +
'2) 用户要求修改/更正/调整/更新已有采购记录时,引导其去采购清单页面手动编辑;' +
'3) 用户有购买意图但缺少价格信息时,追问价格;' +
'4) 权限不足需要拒绝(如非管理员要求清空记录);' +
'5) 其他需要回复但不属于新建采购或查询/删除/清空的情况。',
parameters: {
type: 'object',
properties: {
reply: { type: 'string', description: '显示给用户的回复内容' }
},
required: ['reply']
}
}
};
const queryPurchasesTool = {
type: 'function',
function: {
name: 'query_purchases',
description:
'查询历史采购记录。当用户询问"买过什么/有没有买过/查一下/找找看/搜索/列表/我买过/看看"等查询历史采购的意图时调用。' +
'即使物品名听起来奇怪或可能不存在,也必须调用本工具,绝不能调用 create_purchase 或 ignore。' +
'调用后系统会搜索本地数据库并生成回答,本工具不直接返回查询结果。',
parameters: {
type: 'object',
properties: {
keywords: { type: 'string', description: '用户想查找的物品名或关键词;若查询全部记录则设为空字符串 ""' }
},
required: ['keywords']
}
}
};
const ignoreTool = {
type: 'function',
function: {
name: 'ignore',
description:
'忽略当前消息,不产生任何回复。用于纯符号、纯表情、明显不是对助手说的话等无关消息。',
parameters: { type: 'object', properties: {} }
}
};
const TOOLS = [createPurchaseTool, replyChatTool, queryPurchasesTool, ignoreTool];
module.exports = { TOOLS };
+5 -12
View File
@@ -96,10 +96,6 @@ function handleAIResult(aiResult, roomId, username, originalText, attachments) {
} }
if (aiResult.action === 'purchase') { if (aiResult.action === 'purchase') {
if (aiResult.use_recent_image && (!aiResult.purchase_item || aiResult.purchase_item === 'null')) {
const lastPurchase = db.prepare("SELECT item FROM purchases WHERE room_id = ? AND item IS NOT NULL AND item != '' ORDER BY created_at DESC LIMIT 1").get(roomId);
if (lastPurchase) { aiResult.purchase_item = lastPurchase.item; }
}
const item = aiResult.purchase_item; const item = aiResult.purchase_item;
if (!item || item === 'null' || item === 'undefined' || item.trim().length < 2) { if (!item || item === 'null' || item === 'undefined' || item.trim().length < 2) {
storeAndBroadcastText(roomId, '小财', '请提供具体的物品名称。'); storeAndBroadcastText(roomId, '小财', '请提供具体的物品名称。');
@@ -108,13 +104,9 @@ function handleAIResult(aiResult, roomId, username, originalText, attachments) {
} }
let usedRecentImage = false; let usedRecentImage = false;
if (aiResult.use_recent_image) { // 图片关联:由后端根据消息文本判断,不再依赖模型返回 use_recent_image。
const recentMsg = db.prepare("SELECT attachments FROM messages WHERE room_id = ? AND attachments IS NOT NULL AND attachments != '' AND attachments != '[]' ORDER BY id DESC LIMIT 1").get(roomId); // 用户消息提及图片/附件等词且本次未带附件时,自动关联最近的图片消息。
if (recentMsg) { if (!attachments?.length && /(图片|附件|加到|存入|关联|作为附件)/.test(originalText)) {
try { const files = JSON.parse(recentMsg.attachments); if (files.length) { attachments = files; usedRecentImage = true; } } catch(e) {}
}
}
if (!usedRecentImage && !attachments?.length && /(图片|附件|加到|存入|关联|作为附件)/.test(originalText)) {
const recentMsg = db.prepare("SELECT attachments FROM messages WHERE room_id = ? AND attachments IS NOT NULL AND attachments != '' AND attachments != '[]' ORDER BY id DESC LIMIT 1").get(roomId); const recentMsg = db.prepare("SELECT attachments FROM messages WHERE room_id = ? AND attachments IS NOT NULL AND attachments != '' AND attachments != '[]' ORDER BY id DESC LIMIT 1").get(roomId);
if (recentMsg) { if (recentMsg) {
try { const files = JSON.parse(recentMsg.attachments); if (files.length) { attachments = files; usedRecentImage = true; } } catch(e) {} try { const files = JSON.parse(recentMsg.attachments); if (files.length) { attachments = files; usedRecentImage = true; } } catch(e) {}
@@ -190,7 +182,8 @@ async function callDeepSeekForSummary(prompt, userId) {
{ role: 'system', content: '你是一个财务助手,请根据采购记录生成简洁回复。' }, { role: 'system', content: '你是一个财务助手,请根据采购记录生成简洁回复。' },
{ role: 'user', content: prompt } { role: 'user', content: prompt }
]; ];
return callDeepSeek(messages, userId, 0.3, false); const msg = await callDeepSeek(messages, userId, 0.3, false);
return msg && typeof msg === 'object' ? (msg.content || '') : (msg || '');
} }
module.exports = { module.exports = {
+1 -1
View File
@@ -237,7 +237,7 @@
<script> <script>
// 版本号 // 版本号
(function() { (function() {
document.getElementById('version-text').textContent = 'v2.8.5-260729'; document.getElementById('version-text').textContent = 'v3.0.1-260829';
})(); })();
document.addEventListener('input', function(e) { document.addEventListener('input', function(e) {
if (e.target.id === 'msg-input' || e.target.id === 'purchase-msg-input') { if (e.target.id === 'msg-input' || e.target.id === 'purchase-msg-input') {
+34 -1
View File
@@ -5,6 +5,7 @@ const { broadcastToRoomExcludeSelf, broadcastToRoom } = require('../ws');
const { buildSystemPrompt } = require('../ai/prompt'); const { buildSystemPrompt } = require('../ai/prompt');
const { validate, normalize } = require('../ai/validator'); const { validate, normalize } = require('../ai/validator');
const { callDeepSeek } = require('../ai/client'); const { callDeepSeek } = require('../ai/client');
const { TOOLS } = require('../ai/tools');
const { getHistory, addToHistory, hasPendingAction, handleAIResult, executePendingAction } = require('../handlers'); const { getHistory, addToHistory, hasPendingAction, handleAIResult, executePendingAction } = require('../handlers');
const db = require('../db'); const db = require('../db');
const { timestamp } = require('../utils'); const { timestamp } = require('../utils');
@@ -69,7 +70,39 @@ async function analyzeWithDeepSeek(text, historyMessages, username, isAdmin, roo
]; ];
historyMessages.forEach(function(m) { messages.push(m); }); historyMessages.forEach(function(m) { messages.push(m); });
messages.push({ role: 'system', content: '当前服务器时间:' + new Date().toLocaleString('zh-CN', { timeZone: 'Asia/Shanghai', hour12: false }) }); messages.push({ role: 'system', content: '当前服务器时间:' + new Date().toLocaleString('zh-CN', { timeZone: 'Asia/Shanghai', hour12: false }) });
const content = await callDeepSeek(messages, username); const message = await callDeepSeek(messages, username, 0.1, true, TOOLS);
// 1) 工具调用路径:模型调用了 create_purchase / query_purchases / reply_chat / ignore
if (message && Array.isArray(message.tool_calls) && message.tool_calls.length) {
message.tool_calls.forEach(function(tc, i) {
const fn = tc.function || {};
console.log('🤖 AI 工具调用[' + i + ']: ' + (fn.name || '?') + ' args=' + (fn.arguments || '').substring(0, 300));
});
const results = [];
message.tool_calls.forEach(function(tc) {
const fn = tc.function || {};
const name = fn.name || '';
let args = {};
try { args = JSON.parse(fn.arguments || '{}'); } catch(e) { console.warn('⚠️ 工具参数解析失败:', name, (fn.arguments || '').substring(0, 100)); args = {}; }
if (name === 'create_purchase') {
// 工具参数已是结构化对象,补上 action 字段以复用现有 handleAIResult 的 purchase 分支
results.push(Object.assign({ action: 'purchase' }, args));
} else if (name === 'query_purchases') {
// 复用现有 query 分支(本地查库 + 二次汇总),不依赖模型返回结果
results.push({ action: 'query', keywords: args.keywords || '' });
} else if (name === 'reply_chat') {
results.push({ action: 'chat', reply: args.reply || '好的。' });
} else {
// ignore 或未知工具 → 忽略
console.warn('⚠️ 未识别的工具名:', name);
results.push({ action: 'ignore' });
}
});
return results;
}
// 2) 纯文本/JSON 路径:query / delete / clear_all 等按旧逻辑解析
const content = (message && typeof message === 'object') ? (message.content || '') : (message || '');
console.log('🤖 AI 返回:', content.substring(0, 200)); console.log('🤖 AI 返回:', content.substring(0, 200));
try { try {
var cleaned = content.replace(/```json|```/g, '').trim(); var cleaned = content.replace(/```json|```/g, '').trim();