明树Git Lab

Commit 28506bce authored by zfp1's avatar zfp1

update

parent 6d7b69a1
...@@ -2,47 +2,47 @@ const express = require('express'); ...@@ -2,47 +2,47 @@ const express = require('express');
const crypto = require('crypto'); const crypto = require('crypto');
const { OpenAI } = require('openai'); const { OpenAI } = require('openai');
const xml2js = require('xml2js'); const xml2js = require('xml2js');
const app = express(); const app = express();
app.use(express.text({ type: 'text/xml' })); app.use(express.text({ type: 'text/xml' }));
// 配置参数 [5]() // 配置参数 [5]()
const openai = new OpenAI({ const openai = new OpenAI({
baseURL: 'https://api.deepseek.com/v1', baseURL: 'https://api.deepseek.com/v1',
apiKey: 'sk-14f0cf16bad245169cd2563e0f9b678e' apiKey: 'sk-14f0cf16bad245169cd2563e0f9b678e'
}); });
// 微信服务号验证 // 微信服务号验证
app.get('/wechat', (req, res) => { app.get('/wechat', (req, res) => {
const { signature, timestamp, nonce, echostr } = req.query; const { signature, timestamp, nonce, echostr } = req.query;
const token = 'deepseektest'; const token = 'deepseektest';
const sorted = [token, timestamp, nonce].sort().join(''); const sorted = [token, timestamp, nonce].sort().join('');
const sha1 = crypto.createHash('sha1').update(sorted).digest('hex'); const sha1 = crypto.createHash('sha1').update(sorted).digest('hex');
sha1 === signature ? res.send(echostr) : res.status(403).send(' 验证失败'); sha1 === signature ? res.send(echostr) : res.status(403).send(' 验证失败');
}); });
// 消息处理流程 // 消息处理流程
app.post('/wechat', async (req, res) => { app.post('/wechat', async (req, res) => {
try { try {
// XML解析 // XML解析
const result = await xml2js.parseStringPromise(req.body); const result = await xml2js.parseStringPromise(req.body);
const message = result.xml.Content[0]; const message = result.xml.Content[0];
const fromUser = result.xml.FromUserName[0]; const fromUser = result.xml.FromUserName[0];
// API调用 [5]() // API调用 [5]()
const completion = await openai.chat.completions.create({ const completion = await openai.chat.completions.create({
model: "deepseek-chat", model: "deepseek-chat",
messages: [ messages: [
{ role: "system", content: "你是一个微信服务号智能助手, 限制你的字符输出要低于2048个字节。" }, { { role: "system", content: "你是一个微信服务号智能助手,限制你的字符输出要低于2048个字节。" },{
role: "user", role: "user",
content: message content: message
}], }],
stream: false stream: false
}); });
let fullContent = completion.choices[0].message.content|| '';
console.log("completion:::", completion);
const xmlEntities = { const xmlEntities = {
'&': '&', '&': '&',
'<': '&lt;', '<': '&lt;',
...@@ -60,25 +60,25 @@ app.post('/wechat', async (req, res) => { ...@@ -60,25 +60,25 @@ app.post('/wechat', async (req, res) => {
return escaped; return escaped;
} }
fullContent = escapeWechatContent(fullContent); let fullContent = escapeWechatContent(completion.choices[0].message.content);
// 响应构造 // 响应构造
const responseXML = ` const responseXML = `
<xml> <xml>
<ToUserName><![CDATA[${fromUser}]]></ToUserName> <ToUserName><![CDATA[${fromUser}]]></ToUserName>
<FromUserName><![CDATA[${result.xml.ToUserName[0]}]]></FromUserName> <FromUserName><![CDATA[${result.xml.ToUserName[0]}]]></FromUserName>
<CreateTime>${Math.floor(Date.now() / 1000)}</CreateTime> <CreateTime>${Math.floor(Date.now()/1000)}</CreateTime>
<MsgType><![CDATA[text]]></MsgType> <MsgType><![CDATA[text]]></MsgType>
<Content><![CDATA[${fullContent}]]></Content> <Content><![CDATA[${fullContent}]]></Content>
</xml> </xml>
`; `;
res.set('Content-Type', 'text/xml'); res.set('Content-Type', 'text/xml');
console.log(responseXML) console.log(responseXML)
res.send(responseXML); res.send(responseXML);
} catch (error) { } catch (error) {
console.error(error); console.error(error);
res.status(500).send(' 服务异常'); res.status(500).send(' 服务异常');
} }
}); });
app.listen(3001, () => console.log(' 服务运行在3001端口')); app.listen(3001, () => console.log(' 服务运行在3001端口'));
\ No newline at end of file \ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment