74 lines
2.6 KiB
JavaScript
74 lines
2.6 KiB
JavaScript
|
/**
|
|||
|
|
* M-DINGTALK · index-stream.js · v1.0
|
|||
|
|
* 开发者：之之（DEV-004）
|
|||
|
|
* 功能：钉钉 Stream 模式机器人主入口
|
|||
|
|
* 修改日期：2026-03-15
|
|||
|
|
*/
|
|||
|
|
require('dotenv').config();
|
|||
|
|
const { DWClient, TOPIC_ROBOT, EventAck } = require('dingtalk-stream');
|
|||
|
|
|
|||
|
|
const client = new DWClient({
|
|||
|
|
clientId: process.env.DINGTALK_APP_KEY,
|
|||
|
|
clientSecret: process.env.DINGTALK_APP_SECRET,
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
client.registerCallbackListener(TOPIC_ROBOT, async (res) => {
|
|||
|
|
console.log("[DEBUG] 收到原始回调:", JSON.stringify(res).substring(0, 100));
|
|||
|
|
try {
|
|||
|
|
const msgData = res.data || res;
|
|||
|
|
const senderNick = msgData.senderNick || msgData.senderId || '未知用户';
|
|||
|
|
const content = msgData.text?.content || msgData.content?.content || '';
|
|||
|
|
const sessionWebhook = msgData.sessionWebhook;
|
|||
|
|
|
|||
|
|
console.log(`[Stream] 收到消息 | 发送者：${senderNick} | 内容：${content.trim()}`);
|
|||
|
|
|
|||
|
|
// 尝试写入 Notion SYSLOG
|
|||
|
|
try {
|
|||
|
|
const { Client } = require('@notionhq/client');
|
|||
|
|
const notion = new Client({ auth: process.env.NOTION_TOKEN });
|
|||
|
|
await notion.pages.create({
|
|||
|
|
parent: { database_id: process.env.NOTION_SYSLOG_DB_ID },
|
|||
|
|
properties: {
|
|||
|
|
'标题': {
|
|||
|
|
title: [{ text: { content: `[钉钉消息] ${senderNick}: ${content.trim().substring(0, 50)}` } }]
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
console.log('[Notion] SYSLOG 写入成功');
|
|||
|
|
} catch (notionErr) {
|
|||
|
|
console.error('[Notion] 写入失败: ', notionErr.message);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// AI 回复
|
|||
|
|
if (process.env.LLM_API_KEY && sessionWebhook) {
|
|||
|
|
const axios = require('axios');
|
|||
|
|
const llmRes = await axios.post(
|
|||
|
|
`${process.env.LLM_BASE_URL || 'https://api.openai.com/v1'}/chat/completions`,
|
|||
|
|
{
|
|||
|
|
model: process.env.LLM_MODEL || 'gpt-4o-mini',
|
|||
|
|
messages: [
|
|||
|
|
{ role: 'system', content: process.env.BOT_SYSTEM_PROMPT || '你是之之秋秋，一个温馨聊天的钉钉机器人。' },
|
|||
|
|
{ role: 'user', content: content.replace(`@${msgData.robotCode || ''}`, '').trim() }
|
|||
|
|
],
|
|||
|
|
max_tokens: 500
|
|||
|
|
},
|
|||
|
|
{
|
|||
|
|
headers: { 'Authorization': `Bearer ${process.env.LLM_API_KEY}` }
|
|||
|
|
}
|
|||
|
|
);
|
|||
|
|
const reply = llmRes.data.choices[0].message.content;
|
|||
|
|
await axios.post(sessionWebhook, {
|
|||
|
|
msgtype: 'text',
|
|||
|
|
text: { content: reply }
|
|||
|
|
});
|
|||
|
|
console.log(`[Stream] 回复已发送, 长度: ${reply.length}`);
|
|||
|
|
}
|
|||
|
|
} catch (err) {
|
|||
|
|
console.error('[Stream] 处理出错: ', err.message);
|
|||
|
|
}
|
|||
|
|
return EventAck.SUCCESS;
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
client.connect();
|
|||
|
|
console.log('M-DINGTALK Phase 9 - Stream 模式已启动');
|