feat: 合同审查异步方案(定时任务触发、非流式轮询扣子API、token计费、HTML报告下载)
This commit is contained in:
@@ -0,0 +1,12 @@
|
||||
{
|
||||
"triggers": [
|
||||
{
|
||||
"name": "http",
|
||||
"type": "http",
|
||||
"config": {
|
||||
"path": "/review-stream",
|
||||
"method": ["POST"]
|
||||
}
|
||||
}
|
||||
]
|
||||
}
|
||||
@@ -0,0 +1,106 @@
|
||||
'use strict';
|
||||
|
||||
exports.main = async (event, context) => {
|
||||
// uniCloud HTTP 触发器:event 本身就是请求信息
|
||||
// event.httpMethod = 'GET'/'POST'/'OPTIONS'
|
||||
// event.body = 请求体
|
||||
// event.headers = 请求头
|
||||
// context.callbackWaitsForEmptyEventLoop = true 保持连接
|
||||
|
||||
const corsHeaders = {
|
||||
'Access-Control-Allow-Origin': '*',
|
||||
'Access-Control-Allow-Methods': 'GET, POST, OPTIONS',
|
||||
'Access-Control-Allow-Headers': 'Content-Type, Authorization',
|
||||
'Access-Control-Max-Age': '86400'
|
||||
};
|
||||
|
||||
// 处理 CORS 预检
|
||||
if (event.httpMethod === 'OPTIONS') {
|
||||
return {
|
||||
statusCode: 204,
|
||||
headers: corsHeaders,
|
||||
body: ''
|
||||
};
|
||||
}
|
||||
|
||||
// 设置等待空事件循环(保持 SSE 连接)
|
||||
context.callbackWaitsForEmptyEventLoop = false;
|
||||
|
||||
// 解析请求体
|
||||
let body = {};
|
||||
try {
|
||||
const raw = event.body || '';
|
||||
if (event.isBase64Encoded) {
|
||||
body = JSON.parse(Buffer.from(raw, 'base64').toString());
|
||||
} else {
|
||||
try { body = JSON.parse(raw); } catch (e) {
|
||||
const params = new URLSearchParams(raw);
|
||||
for (const [k, v] of params) body[k] = v;
|
||||
}
|
||||
}
|
||||
} catch (e) {}
|
||||
|
||||
const { file_id, user_id, file_name, file_size, record_id } = body;
|
||||
|
||||
if (!file_id) {
|
||||
return {
|
||||
statusCode: 400,
|
||||
headers: { ...corsHeaders, 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ error: '缺少 file_id' })
|
||||
};
|
||||
}
|
||||
|
||||
const COZE_TOKEN = 'sat_OiXlI4Da5Q99fxmul6pNfExmA6hR898UbxBDwrvjZ638APxRSlFnXbz6zyFedKNG';
|
||||
const BOT_ID = '7689821072762011698';
|
||||
|
||||
const objectString = JSON.stringify([
|
||||
{ type: 'file', file_id: file_id },
|
||||
{ type: 'text', text: '请审查这份合同' }
|
||||
]);
|
||||
|
||||
// 调用扣子(流式)
|
||||
try {
|
||||
const chatRes = await uniCloud.httpclient.request('https://api.coze.cn/v3/chat', {
|
||||
method: 'POST',
|
||||
dataType: 'text',
|
||||
contentType: 'json',
|
||||
timeout: 300000,
|
||||
headers: { 'Authorization': 'Bearer ' + COZE_TOKEN },
|
||||
data: {
|
||||
bot_id: BOT_ID,
|
||||
user_id: user_id || 'anonymous',
|
||||
stream: true,
|
||||
auto_save_history: true,
|
||||
additional_messages: [{
|
||||
role: 'user',
|
||||
content: objectString,
|
||||
content_type: 'object_string'
|
||||
}]
|
||||
}
|
||||
});
|
||||
|
||||
// 返回 SSE 格式
|
||||
const text = chatRes.data || '';
|
||||
const sseBody = text.split('\n').map(line => {
|
||||
if (line.startsWith('data:')) return line + '\n\n';
|
||||
return '';
|
||||
}).join('') + 'data: [DONE]\n\n';
|
||||
|
||||
return {
|
||||
statusCode: 200,
|
||||
headers: {
|
||||
...corsHeaders,
|
||||
'Content-Type': 'text/event-stream',
|
||||
'Cache-Control': 'no-cache',
|
||||
'Connection': 'keep-alive'
|
||||
},
|
||||
body: sseBody
|
||||
};
|
||||
} catch (e) {
|
||||
return {
|
||||
statusCode: 500,
|
||||
headers: { ...corsHeaders, 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ error: e.message })
|
||||
};
|
||||
}
|
||||
};
|
||||
@@ -0,0 +1,6 @@
|
||||
{
|
||||
"name": "jjxReviewStream",
|
||||
"version": "1.0.0",
|
||||
"description": "合同审查流式输出云函数",
|
||||
"main": "index.js"
|
||||
}
|
||||
@@ -0,0 +1,239 @@
|
||||
'use strict';
|
||||
|
||||
// 定时任务:每分钟检查 reviewing 状态的记录,触发审查
|
||||
const db = uniCloud.database();
|
||||
const _ = db.command;
|
||||
|
||||
// 导入审查执行函数(从 router 的 service 目录)
|
||||
// 注意:定时任务是独立云函数,不能直接 require router 的 service
|
||||
// 所以把审查逻辑直接写在这里
|
||||
|
||||
const COZE_TOKEN = 'sat_OiXlI4Da5Q99fxmul6pNfExmA6hR898UbxBDwrvjZ638APxRSlFnXbz6zyFedKNG';
|
||||
const BOT_ID = '7689821072762011698';
|
||||
|
||||
async function doReview(record) {
|
||||
const { _id: record_id, file_id, user_id, file_name, file_size } = record;
|
||||
|
||||
const objectString = JSON.stringify([
|
||||
{ type: 'file', file_id: file_id },
|
||||
{ type: 'text', text: '请审查这份合同' }
|
||||
]);
|
||||
|
||||
try {
|
||||
// 1. 非流式发起对话(立即返回 chat_id)
|
||||
const chatRes = await uniCloud.httpclient.request('https://api.coze.cn/v3/chat', {
|
||||
method: 'POST',
|
||||
dataType: 'json',
|
||||
contentType: 'json',
|
||||
timeout: 30000,
|
||||
headers: { 'Authorization': 'Bearer ' + COZE_TOKEN },
|
||||
data: {
|
||||
bot_id: BOT_ID,
|
||||
user_id: user_id || 'anonymous',
|
||||
stream: false,
|
||||
auto_save_history: true,
|
||||
additional_messages: [{
|
||||
role: 'user',
|
||||
content: objectString,
|
||||
content_type: 'object_string'
|
||||
}]
|
||||
}
|
||||
});
|
||||
|
||||
const chatData = chatRes.data && chatRes.data.data;
|
||||
const chatId = chatData && chatData.id;
|
||||
const conversationId = chatData && chatData.conversation_id;
|
||||
console.log('[审查] chat_id:', chatId, 'conversation_id:', conversationId);
|
||||
|
||||
if (!chatId) {
|
||||
throw new Error('未获取到 chat_id: ' + JSON.stringify(chatRes.data));
|
||||
}
|
||||
|
||||
// 保存 chat_id 和 conversation_id 到数据库
|
||||
await db.collection('jjx-review-records').doc(record_id).update({
|
||||
chat_id: chatId,
|
||||
conversation_id: conversationId || '',
|
||||
update_time: new Date()
|
||||
});
|
||||
console.log('[审查] 已保存 chat_id:', chatId);
|
||||
|
||||
// 2. 轮询对话状态(每5秒查一次,最多5分钟)
|
||||
let answer = '';
|
||||
let tokenCount = 0;
|
||||
for (let i = 0; i < 60; i++) {
|
||||
await new Promise(r => setTimeout(r, 5000));
|
||||
|
||||
const statusRes = await uniCloud.httpclient.request(
|
||||
'https://api.coze.cn/v3/chat/retrieve?chat_id=' + chatId + '&conversation_id=' + conversationId,
|
||||
{ method: 'GET', dataType: 'json', headers: { 'Authorization': 'Bearer ' + COZE_TOKEN }, timeout: 15000 }
|
||||
);
|
||||
|
||||
const status = statusRes.data && statusRes.data.data && statusRes.data.data.status;
|
||||
console.log('[审查] 第', i + 1, '次轮询, 状态:', status);
|
||||
|
||||
if (status === 'completed') {
|
||||
// 保存 token 消耗信息
|
||||
const usage = statusRes.data && statusRes.data.data && statusRes.data.data.usage;
|
||||
tokenCount = (usage && usage.token_count) || 0;
|
||||
console.log('[审查] Token 消耗:', tokenCount);
|
||||
|
||||
// 3. 获取消息列表
|
||||
const msgRes = await uniCloud.httpclient.request(
|
||||
'https://api.coze.cn/v3/chat/message/list?chat_id=' + chatId + '&conversation_id=' + conversationId,
|
||||
{ method: 'GET', dataType: 'json', headers: { 'Authorization': 'Bearer ' + COZE_TOKEN }, timeout: 15000 }
|
||||
);
|
||||
|
||||
const messages = msgRes.data && msgRes.data.data;
|
||||
if (Array.isArray(messages)) {
|
||||
for (const msg of messages) {
|
||||
if (msg.type === 'answer' && msg.role === 'assistant' && msg.content) {
|
||||
answer = msg.content; // answer 类型的完整内容
|
||||
}
|
||||
if (msg.type === 'tool_response' && msg.content) {
|
||||
try {
|
||||
const toolData = JSON.parse(msg.content);
|
||||
if (toolData.url) answer += '\n\n📎 报告下载:' + toolData.url;
|
||||
} catch (e) {}
|
||||
}
|
||||
}
|
||||
}
|
||||
break;
|
||||
}
|
||||
|
||||
if (status === 'failed') {
|
||||
throw new Error('对话失败');
|
||||
}
|
||||
}
|
||||
|
||||
if (!answer) answer = '审查未能获取结果,请稍后重试';
|
||||
console.log('[审查] 结果长度:', answer.length);
|
||||
|
||||
// 计算消耗积分
|
||||
// 积分 = token消耗 / 100(最低1积分)
|
||||
const pointsUsed = Math.max(1, Math.round(tokenCount / 100));
|
||||
console.log('[积分] token_count:', tokenCount, '→ 扣除积分:', pointsUsed);
|
||||
|
||||
// 扣积分
|
||||
const userPoints = await db.collection('jjx-user-points').doc(user_id).get();
|
||||
if (userPoints.data.length && userPoints.data[0].balance >= pointsUsed) {
|
||||
await db.collection('jjx-user-points').doc(user_id).update({
|
||||
balance: _.inc(-pointsUsed),
|
||||
total_spent: _.inc(pointsUsed),
|
||||
update_time: new Date()
|
||||
});
|
||||
const updatedPoints = await db.collection('jjx-user-points').doc(user_id).get();
|
||||
await db.collection('jjx-points-log').add({
|
||||
user_id: user_id,
|
||||
type: 'spend',
|
||||
amount: -pointsUsed,
|
||||
balance_after: updatedPoints.data[0] ? updatedPoints.data[0].balance : 0,
|
||||
reason: '合同审查 — ' + (file_name || '') + '(' + tokenCount + ' token)',
|
||||
biz_type: 'contract_review',
|
||||
biz_id: record_id,
|
||||
_add_time: new Date()
|
||||
});
|
||||
}
|
||||
|
||||
// 保存 HTML 报告到云存储
|
||||
let htmlUrl = '';
|
||||
try {
|
||||
let htmlContent = answer;
|
||||
const htmlMatch = answer.match(/```html\s*([\s\S]*?)```/);
|
||||
if (htmlMatch && htmlMatch[1]) htmlContent = htmlMatch[1].trim();
|
||||
|
||||
if (htmlContent.includes('<!DOCTYPE') || htmlContent.includes('<html')) {
|
||||
const cloudPath = 'review-reports/' + Date.now() + '_' + (file_name || 'report').replace(/\.[^.]+$/, '') + '.html';
|
||||
const uploadRes = await uniCloud.uploadFile({
|
||||
cloudPath: cloudPath,
|
||||
fileContent: Buffer.from(htmlContent, 'utf-8')
|
||||
});
|
||||
if (uploadRes.fileID) {
|
||||
const urlRes = await uniCloud.getTempFileURL({ fileList: [uploadRes.fileID] });
|
||||
if (urlRes.fileList && urlRes.fileList[0]) htmlUrl = urlRes.fileList[0].tempFileURL;
|
||||
}
|
||||
}
|
||||
} catch (e) {
|
||||
console.log('HTML报告上传失败:', e.message);
|
||||
}
|
||||
|
||||
// 更新记录为完成
|
||||
await db.collection('jjx-review-records').doc(record_id).update({
|
||||
status: 'completed',
|
||||
result: answer,
|
||||
html_url: htmlUrl,
|
||||
token_count: tokenCount,
|
||||
points_used: pointsUsed,
|
||||
conversation_id: conversationId || '',
|
||||
update_time: new Date()
|
||||
});
|
||||
|
||||
console.log('审查完成:', record_id, file_name);
|
||||
return { code: 0, msg: '审查完成' };
|
||||
|
||||
} catch (e) {
|
||||
await db.collection('jjx-review-records').doc(record_id).update({
|
||||
status: 'failed',
|
||||
result: '审查失败: ' + e.message,
|
||||
error: e.message,
|
||||
update_time: new Date()
|
||||
});
|
||||
console.error('审查失败:', record_id, e.message);
|
||||
return { code: -1, msg: '审查失败: ' + e.message };
|
||||
}
|
||||
}
|
||||
|
||||
exports.main = async (event, context) => {
|
||||
// 先把卡在 processing 超过10分钟的记录重置为 reviewing(超时保护)
|
||||
const tenMinutesAgo = new Date(Date.now() - 10 * 60 * 1000);
|
||||
const stuckRecords = await db.collection('jjx-review-records')
|
||||
.where({ status: 'processing', update_time: _.lt(tenMinutesAgo) })
|
||||
.get();
|
||||
for (const stuck of stuckRecords.data) {
|
||||
await db.collection('jjx-review-records').doc(stuck._id).update({
|
||||
status: 'reviewing',
|
||||
update_time: new Date()
|
||||
});
|
||||
}
|
||||
|
||||
// 查询所有 reviewing 状态的记录(超过1分钟的,避免重复触发)
|
||||
const oneMinuteAgo = new Date(Date.now() - 60 * 1000);
|
||||
|
||||
const pendingRecords = await db.collection('jjx-review-records')
|
||||
.where({
|
||||
status: 'reviewing',
|
||||
review_started_at: _.lt(oneMinuteAgo)
|
||||
})
|
||||
.limit(5) // 每次最多处理5条
|
||||
.get();
|
||||
|
||||
if (!pendingRecords.data.length) {
|
||||
return { code: 0, msg: '无待处理记录' };
|
||||
}
|
||||
|
||||
console.log('发现待处理审查记录:', pendingRecords.data.length, '条');
|
||||
|
||||
// 逐条处理(串行,避免并发问题)
|
||||
const results = [];
|
||||
for (const record of pendingRecords.data) {
|
||||
// 先标记为 processing,避免重复触发
|
||||
await db.collection('jjx-review-records').doc(record._id).update({
|
||||
status: 'processing',
|
||||
update_time: new Date()
|
||||
});
|
||||
|
||||
try {
|
||||
const result = await doReview(record);
|
||||
results.push({ record_id: record._id, ...result });
|
||||
} catch (e) {
|
||||
// 审查失败,重置为 reviewing 以便下次重试
|
||||
await db.collection('jjx-review-records').doc(record._id).update({
|
||||
status: 'reviewing',
|
||||
update_time: new Date()
|
||||
});
|
||||
console.error('审查异常,已重置状态:', record._id, e.message);
|
||||
results.push({ record_id: record._id, code: -1, msg: e.message });
|
||||
}
|
||||
}
|
||||
|
||||
return { code: 0, msg: '处理完成', data: results };
|
||||
};
|
||||
@@ -0,0 +1,4 @@
|
||||
// 本文件中的json内容将在云函数【运行】时作为参数传给云函数。
|
||||
// 配置教程参考:https://uniapp.dcloud.net.cn/uniCloud/rundebug.html#runparam
|
||||
{
|
||||
}
|
||||
@@ -0,0 +1,19 @@
|
||||
{
|
||||
"name": "jjxReviewTimer",
|
||||
"version": "1.0.0",
|
||||
"description": "合同审查定时任务",
|
||||
"main": "index.js",
|
||||
"cloudfunction-config": {
|
||||
"memorySize": 256,
|
||||
"timeout": 600,
|
||||
"triggers": [
|
||||
{
|
||||
"name": "myTrigger",
|
||||
"type": "timer",
|
||||
"config": "0 */1 * * * * *"
|
||||
}
|
||||
],
|
||||
"path": "",
|
||||
"runtime": "Nodejs16"
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,53 @@
|
||||
// client/ai/pub/doReviewTimer.js
|
||||
module.exports = {
|
||||
main: async (event) => {
|
||||
let { data = {} } = event;
|
||||
const { record_id, file_id, user_id, file_name, file_size } = data;
|
||||
|
||||
if (!record_id || !file_id) return { code: -1, msg: '参数错误' };
|
||||
|
||||
const db = uniCloud.database();
|
||||
|
||||
// 检查是否已经在审查中(防止重复触发)
|
||||
const record = await db.collection('jjx-review-records').doc(record_id).get();
|
||||
if (record.data.length && record.data[0].status === 'reviewing') {
|
||||
// 检查是否刚触发(5分钟内不重复触发)
|
||||
const lastUpdate = record.data[0].update_time;
|
||||
if (lastUpdate && (Date.now() - new Date(lastUpdate).getTime() < 5 * 60 * 1000)) {
|
||||
return { code: 0, msg: '已在审查中,跳过' };
|
||||
}
|
||||
}
|
||||
|
||||
// 更新状态为审查中
|
||||
await db.collection('jjx-review-records').doc(record_id).update({
|
||||
status: 'reviewing',
|
||||
review_started_at: new Date(),
|
||||
update_time: new Date()
|
||||
});
|
||||
|
||||
// 调用实际的审查逻辑
|
||||
try {
|
||||
const doReview = require('./jjxDoReview.js');
|
||||
await doReview.main({
|
||||
data: {
|
||||
record_id,
|
||||
file_id,
|
||||
user_id: user_id || 'anonymous',
|
||||
file_name: file_name || '',
|
||||
file_size: file_size || 0
|
||||
}
|
||||
});
|
||||
return { code: 0, msg: '审查完成' };
|
||||
} catch (e) {
|
||||
console.error('审查失败:', e);
|
||||
// 更新状态为失败
|
||||
await db.collection('jjx-review-records').doc(record_id).update({
|
||||
status: 'failed',
|
||||
result: '审查失败: ' + e.message,
|
||||
error: e.message,
|
||||
update_time: new Date()
|
||||
});
|
||||
return { code: -1, msg: '审查失败: ' + e.message };
|
||||
}
|
||||
}
|
||||
};
|
||||
@@ -1,3 +1,4 @@
|
||||
// client/ai/pub/jjxDoReview.js
|
||||
module.exports = {
|
||||
main: async (event) => {
|
||||
let { data = {}, userInfo } = event;
|
||||
@@ -17,13 +18,16 @@ module.exports = {
|
||||
{ type: 'text', text: '请审查这份合同' }
|
||||
]);
|
||||
|
||||
// 调用 Coze 接口(流式模式,等待完整响应,和 Demo 一致)
|
||||
// 调用 Coze 接口(流式模式)
|
||||
const chatRes = await uniCloud.httpclient.request('https://api.coze.cn/v3/chat', {
|
||||
method: 'POST',
|
||||
dataType: 'text',
|
||||
dataType: 'stream', // 关键:改成 stream
|
||||
contentType: 'json',
|
||||
timeout: 300000, // 5分钟超时
|
||||
headers: { 'Authorization': 'Bearer ' + COZE_TOKEN },
|
||||
headers: {
|
||||
'Authorization': 'Bearer ' + COZE_TOKEN,
|
||||
'Accept': 'text/event-stream'
|
||||
},
|
||||
data: {
|
||||
bot_id: BOT_ID,
|
||||
user_id: user_id || 'anonymous',
|
||||
@@ -37,92 +41,168 @@ module.exports = {
|
||||
}
|
||||
});
|
||||
|
||||
// 从 SSE 流中提取回答
|
||||
// 从流中逐步解析
|
||||
const stream = chatRes.data || chatRes.res;
|
||||
let buffer = '';
|
||||
let answer = '';
|
||||
let conversationId = '';
|
||||
const text = chatRes.data || '';
|
||||
const lines = text.split('\n');
|
||||
for (const line of lines) {
|
||||
if (!line.startsWith('data:')) continue;
|
||||
const d = line.slice(5).trim();
|
||||
if (d === '[DONE]' || d === '"[DONE]"') continue;
|
||||
try {
|
||||
const p = JSON.parse(d);
|
||||
if (p.conversation_id) conversationId = p.conversation_id;
|
||||
if (p.type === 'answer' && p.content) answer += p.content;
|
||||
if (p.type === 'tool_response' && p.content) {
|
||||
try {
|
||||
const toolData = JSON.parse(p.content);
|
||||
if (toolData.url) answer += '\n\n📎 报告下载:' + toolData.url;
|
||||
} catch (e) {
|
||||
const urlMatch = p.content.match(/https?:\/\/[^\s\)\]\"\<]+\.pdf[^\s\)\]\"\<]*/i);
|
||||
if (urlMatch) answer += '\n\n📎 报告下载:' + urlMatch[0];
|
||||
let lastUpdateLen = 0;
|
||||
|
||||
// 返回 Promise,等待流结束
|
||||
return await new Promise((resolve, reject) => {
|
||||
|
||||
stream.on('data', (chunk) => {
|
||||
buffer += chunk.toString();
|
||||
const blocks = buffer.split('\n\n');
|
||||
buffer = blocks.pop();
|
||||
|
||||
for (const block of blocks) {
|
||||
const lines = block.split('\n');
|
||||
let eventName = '';
|
||||
let dataStr = '';
|
||||
|
||||
for (const line of lines) {
|
||||
if (line.startsWith('event:')) eventName = line.slice(6).trim();
|
||||
if (line.startsWith('data:')) dataStr = line.slice(5).trim();
|
||||
}
|
||||
|
||||
if (!dataStr || dataStr === '[DONE]' || dataStr === '"[DONE]"') continue;
|
||||
|
||||
try {
|
||||
const p = JSON.parse(dataStr);
|
||||
if (p.conversation_id) conversationId = p.conversation_id;
|
||||
|
||||
// 处理回答内容
|
||||
if (p.type === 'answer' && p.content) {
|
||||
answer += p.content;
|
||||
|
||||
// 每积累 200 字更新一次数据库(前端轮询能看到进度)
|
||||
if (answer.length - lastUpdateLen >= 200) {
|
||||
lastUpdateLen = answer.length;
|
||||
db.collection('jjx-review-records').doc(record_id).update({
|
||||
result: answer,
|
||||
status: 'reviewing',
|
||||
update_time: new Date()
|
||||
}).catch(() => {});
|
||||
}
|
||||
}
|
||||
|
||||
// 处理工具响应(报告下载链接)
|
||||
if (p.type === 'tool_response' && p.content) {
|
||||
try {
|
||||
const toolData = JSON.parse(p.content);
|
||||
if (toolData.url) answer += '\n\n📎 报告下载:' + toolData.url;
|
||||
} catch (e) {
|
||||
const urlMatch = p.content.match(/https?:\/\/[^\s\)\]\"\<]+\.pdf[^\s\)\]\"\<]*/i);
|
||||
if (urlMatch) answer += '\n\n📎 报告下载:' + urlMatch[0];
|
||||
}
|
||||
}
|
||||
} catch (e) {}
|
||||
}
|
||||
} catch (e) {}
|
||||
}
|
||||
|
||||
if (!answer) {
|
||||
answer = '审查未能获取结果,请稍后重试';
|
||||
}
|
||||
|
||||
// 计算消耗积分
|
||||
const pointsUsed = Math.max(150, Math.ceil((file_size || 0) / (100 * 1024)) * 150);
|
||||
|
||||
// 扣积分
|
||||
const userPoints = await db.collection('jjx-user-points').doc(user_id).get();
|
||||
if (userPoints.data.length && userPoints.data[0].balance >= pointsUsed) {
|
||||
await db.collection('jjx-user-points').doc(user_id).update({
|
||||
balance: _.inc(-pointsUsed),
|
||||
total_spent: _.inc(pointsUsed),
|
||||
update_time: new Date()
|
||||
});
|
||||
|
||||
const updatedPoints = await db.collection('jjx-user-points').doc(user_id).get();
|
||||
await db.collection('jjx-points-log').add({
|
||||
user_id: user_id,
|
||||
type: 'spend',
|
||||
amount: -pointsUsed,
|
||||
balance_after: updatedPoints.data[0] ? updatedPoints.data[0].balance : 0,
|
||||
reason: '合同审查 — ' + (file_name || ''),
|
||||
biz_type: 'contract_review',
|
||||
biz_id: record_id,
|
||||
_add_time: new Date()
|
||||
stream.on('end', async () => {
|
||||
if (!answer) {
|
||||
answer = '审查未能获取结果,请稍后重试';
|
||||
}
|
||||
|
||||
try {
|
||||
// 计算消耗积分
|
||||
const pointsUsed = Math.max(150, Math.ceil((file_size || 0) / (100 * 1024)) * 150);
|
||||
|
||||
// 扣积分
|
||||
const userPoints = await db.collection('jjx-user-points').doc(user_id).get();
|
||||
if (userPoints.data.length && userPoints.data[0].balance >= pointsUsed) {
|
||||
await db.collection('jjx-user-points').doc(user_id).update({
|
||||
balance: _.inc(-pointsUsed),
|
||||
total_spent: _.inc(pointsUsed),
|
||||
update_time: new Date()
|
||||
});
|
||||
|
||||
const updatedPoints = await db.collection('jjx-user-points').doc(user_id).get();
|
||||
await db.collection('jjx-points-log').add({
|
||||
user_id: user_id,
|
||||
type: 'spend',
|
||||
amount: -pointsUsed,
|
||||
balance_after: updatedPoints.data[0] ? updatedPoints.data[0].balance : 0,
|
||||
reason: '合同审查 — ' + (file_name || ''),
|
||||
biz_type: 'contract_review',
|
||||
biz_id: record_id,
|
||||
_add_time: new Date()
|
||||
});
|
||||
}
|
||||
|
||||
// 保存 HTML 报告到云存储
|
||||
let htmlUrl = '';
|
||||
try {
|
||||
// 如果回复本身就是 HTML(包含 <!DOCTYPE html> 或 <html>)
|
||||
let htmlContent = answer;
|
||||
const htmlMatch = answer.match(/```html\s*([\s\S]*?)```/);
|
||||
if (htmlMatch && htmlMatch[1]) {
|
||||
htmlContent = htmlMatch[1].trim();
|
||||
}
|
||||
|
||||
if (htmlContent.includes('<!DOCTYPE') || htmlContent.includes('<html')) {
|
||||
const cloudPath = 'review-reports/' + Date.now() + '_' + (file_name || 'report').replace(/\.[^.]+$/, '') + '.html';
|
||||
const uploadRes = await uniCloud.uploadFile({
|
||||
cloudPath: cloudPath,
|
||||
fileContent: Buffer.from(htmlContent, 'utf-8')
|
||||
});
|
||||
if (uploadRes.fileID) {
|
||||
const urlRes = await uniCloud.getTempFileURL({ fileList: [uploadRes.fileID] });
|
||||
if (urlRes.fileList && urlRes.fileList[0]) {
|
||||
htmlUrl = urlRes.fileList[0].tempFileURL;
|
||||
}
|
||||
}
|
||||
}
|
||||
} catch (e) {
|
||||
console.log('HTML报告上传失败:', e.message);
|
||||
}
|
||||
|
||||
// 解析 JSON 结构化数据(如果有的话)
|
||||
let contractInfo = {};
|
||||
try {
|
||||
const jsonMatch = answer.match(/```json\s*([\s\S]*?)```/);
|
||||
if (jsonMatch) contractInfo = JSON.parse(jsonMatch[1].trim());
|
||||
} catch (e) {}
|
||||
|
||||
// 更新审查记录为完成
|
||||
await db.collection('jjx-review-records').doc(record_id).update({
|
||||
status: 'completed',
|
||||
result: answer,
|
||||
html_url: htmlUrl,
|
||||
contract_info: contractInfo,
|
||||
points_used: pointsUsed,
|
||||
conversation_id: conversationId || '',
|
||||
update_time: new Date()
|
||||
});
|
||||
|
||||
resolve({ code: 0, msg: '审查完成' });
|
||||
|
||||
} catch (e) {
|
||||
// 更新为失败
|
||||
await db.collection('jjx-review-records').doc(record_id).update({
|
||||
status: 'failed',
|
||||
result: '审查失败: ' + e.message,
|
||||
error: e.message,
|
||||
update_time: new Date()
|
||||
});
|
||||
reject(e);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
// 从回复中提取 HTML 报告下载链接
|
||||
let htmlUrl = '';
|
||||
try {
|
||||
const linkMatch = answer.match(/\[([^\]]*\.html)\]\((https?:\/\/[^\)]+)\)/);
|
||||
if (linkMatch) {
|
||||
htmlUrl = linkMatch[2];
|
||||
} else {
|
||||
const urlMatch = answer.match(/https?:\/\/[^\s\)\]]+\.html[^\s\)\]]*/);
|
||||
if (urlMatch) htmlUrl = urlMatch[0];
|
||||
}
|
||||
} catch (e) {}
|
||||
|
||||
// 解析 JSON 结构化数据
|
||||
let contractInfo = {};
|
||||
try {
|
||||
const jsonMatch = answer.match(/```json\s*([\s\S]*?)```/);
|
||||
if (jsonMatch) contractInfo = JSON.parse(jsonMatch[1].trim());
|
||||
} catch (e) {}
|
||||
|
||||
// 更新审查记录为完成
|
||||
await db.collection('jjx-review-records').doc(record_id).update({
|
||||
status: 'completed',
|
||||
result: answer,
|
||||
html_url: htmlUrl,
|
||||
contract_info: contractInfo,
|
||||
points_used: pointsUsed,
|
||||
conversation_id: conversationId || '',
|
||||
update_time: new Date()
|
||||
|
||||
stream.on('error', async (err) => {
|
||||
await db.collection('jjx-review-records').doc(record_id).update({
|
||||
status: 'failed',
|
||||
result: '审查失败: ' + err.message,
|
||||
error: err.message,
|
||||
update_time: new Date()
|
||||
});
|
||||
reject(err);
|
||||
});
|
||||
|
||||
});
|
||||
|
||||
return { code: 0, msg: '审查完成' };
|
||||
|
||||
} catch (e) {
|
||||
// 审查失败,更新记录
|
||||
await db.collection('jjx-review-records').doc(record_id).update({
|
||||
|
||||
@@ -1,3 +1,4 @@
|
||||
// client/ai/pub/review.js
|
||||
// Markdown 转 HTML
|
||||
function mdToHtml(md) {
|
||||
if (!md) return '';
|
||||
@@ -242,22 +243,7 @@ module.exports = {
|
||||
update_time: new Date()
|
||||
});
|
||||
|
||||
// ===== 触发实际审查(await 确保数据库更新完成)=====
|
||||
try {
|
||||
const doReview = require('./jjxDoReview.js');
|
||||
await doReview.main({
|
||||
data: {
|
||||
record_id: recordId,
|
||||
file_id: fileId,
|
||||
user_id: (userInfo && userInfo.uid) || 'anonymous',
|
||||
file_name: data.file_name || '',
|
||||
file_size: data.file_size || 0
|
||||
}
|
||||
});
|
||||
} catch (e) {
|
||||
console.error('审查任务执行失败:', e);
|
||||
}
|
||||
|
||||
// ===== 立即返回,审查由定时任务触发 =====
|
||||
return {
|
||||
code: 0, msg: 'ok',
|
||||
data: { record_id: recordId, status: 'reviewing' }
|
||||
|
||||
@@ -61,10 +61,11 @@
|
||||
"status": {
|
||||
"bsonType": "string",
|
||||
"title": "审查状态",
|
||||
"description": "uploaded=待审查 reviewing=审查中 completed=已完成 failed=失败",
|
||||
"description": "uploaded=待审查 reviewing=等待处理 processing=审查中 completed=已完成 failed=失败",
|
||||
"enum": [
|
||||
{ "text": "待审查", "value": "uploaded" },
|
||||
{ "text": "审查中", "value": "reviewing" },
|
||||
{ "text": "等待处理", "value": "reviewing" },
|
||||
{ "text": "审查中", "value": "processing" },
|
||||
{ "text": "已完成", "value": "completed" },
|
||||
{ "text": "失败", "value": "failed" }
|
||||
]
|
||||
@@ -77,7 +78,12 @@
|
||||
"points_used": {
|
||||
"bsonType": "number",
|
||||
"title": "消耗积分",
|
||||
"description": "本次审查消耗的积分数量"
|
||||
"description": "本次审查消耗的积分数量(token_count / 100)"
|
||||
},
|
||||
"token_count": {
|
||||
"bsonType": "number",
|
||||
"title": "Token消耗",
|
||||
"description": "扣子返回的 token 总消耗"
|
||||
},
|
||||
"_add_time": {
|
||||
"bsonType": "date",
|
||||
|
||||
Reference in New Issue
Block a user