Dify 允许创建 AI 应用,并提供二次开发的能力。这里我将演示创建一个法律问答助手的 AI 应用,称作“知法”。在本篇教程中,我将指导你为“知法”接入企业微信。
企业微信的管理员权限
一个 Dify 的帐号
一个 Laf 云的帐号
(可选)一个 OpenAI 的 API Key。如果没有,可以使用 Dify 免费提供的 200 次调用机会用于测试。
(可选)在电脑上新建一个 env.txt 的文件,将下面内容复制到 env.txt 中。在接下来的教程中,我们会一步步把相关的信息填入这个文件。需要保存信息的步骤会高亮显示。
WXWORK_TOKEN=""
WXWORK_AESKEY=""
WXWORK_CORPID=""
WXWORK_AGENTID=""
WXWORK_CORPSECRET=""
DIFY_APPTOKEN=""
这一章节将会介绍如何创建一个法律知识的数据集,并将数据集和应用关联起来。
随时查看文档中关于搭建数据集的更多操作:【数据集管理】
为了让“知法”了解到更多的上下文,我们需要创建一个法律知识的数据库。
随时查看文档中关于创建应用的更多操作 【创建应用】
DIFY_APPTOKEN
。请注意不要把密钥泄漏给任何人,以免造成财产损失。.png)
WXWORK_CORPID
.png)
![](https://cdn.jsdelivr.us/gh/langgenius/dify-docs@main/zh_CN/.gitbook/assets/image (23).png" alt="" width="375">
.png)
.png)
.png)
在 API 接收消息页面,点一下两个【随机获取】按钮,它会自动生成一个 Token 和 EncodingAESKey,我们分别记为 WXWORK_TOKEN 和 WXWORK_AESKEY。注意,不要关掉这个页面,Laf 侧配置完毕后我们再来填写 URL。
.png)
.png)
@wecom/crypto
, xml2js
两个依赖。添加好后,你的依赖列表应该像下面一样。.png)
.png)
POST
, GET
,点击确定。.png)
在创建好云函数中,删除默认的代码,并将文末“附录”中的代码全部粘贴到这里。
.png)
.png)
现在把 URL 粘贴到企业微信后台【设置 API 接收】的页面中刚刚留白的地方,然后点击保存。
.png)
点击日志,应当能看到这样一条报错 'not allow to access from your ip
'
.png)
点击查看这条日志详情,记录日志中给出的 Laf 云 IP。
.png)
回到企业微信的管理后台,点击刚刚创建的应用,为应用配置可行 IP。
.png)
在这里把刚刚的日志中记录的 IP 填入即可。
.png)
.png)
这篇深度参考以下文章,感谢原作者的辛勤付出。https://forum.laf.run/d/556/3
import cloud from '@lafjs/cloud'
import { decrypt, getSignature } from '@wecom/crypto'
import xml2js from 'xml2js'
function genConversationKey(userName) {
return `${process.env.WXWORK_AGENTID}:${userName}`
}
function genWxAppAccessTokenKey() {
return `${process.env.WXWORK_AGENTID}:access-token`
}
async function getToken() {
console.log('[getToken] called')
const cache = cloud.shared.get(genWxAppAccessTokenKey())
if%20(cache && cache.expires >= Date.now()) return cache.token
const res = await cloud.fetch({
url: 'https://qyapi.weixin.qq.com/cgi-bin/gettoken',
method: 'GET',
params: {
corpid: process.env.WXWORK_CORPID,
corpsecret: process.env.WXWORK_CORPSECRET,
}
})
const token = res.data.access_token
cloud.shared.set(genWxAppAccessTokenKey(), { token, expires: Date.now() + res.data.expires_in * 1000 })
return token
}
async function sendWxMessage(message, user) {
console.log('[sendWxMessage] called', user, message)
const res = await cloud.fetch({
url: 'https://qyapi.weixin.qq.com/cgi-bin/message/send',
method: 'POST',
params: {
access_token: await getToken()
},
data: {
"touser": user,
"msgtype": "text",
"agentid": process.env.WXWORK_AGENTID,
"text": {
"content": message
},
"safe": 0,
"enable_id_trans": 0,
"enable_duplicate_check": 0,
"duplicate_check_interval": 1800
},
})
console.log('[sendWxMessage] received', res.data)
}
async function sendDifyMessage(message, userName, onMessage) {
console.log('[sendDifyMessage] called', message, userName)
const conversationId = cloud.shared.get(genConversationKey(userName)) || null
let newConversationId = ''
let responseText = ''
try {
const response = await cloud.fetch({
url: 'https://api.dify.ai/v1/chat-messages',
method: 'POST',
headers: {
'Authorization': `Bearer ${process.env.DIFY_APPTOKEN}`
},
data: {
inputs: {},
response_mode: "streaming",
query: message,
user: userName,
conversation_id: conversationId
},
responseType: "stream"
})
let firstHalfMessage = ''
response.data.on('data',%20(data) => {
let message = data.toString()
try {
if%20(firstHalfMessage) {
message += firstHalfMessage
firstHalfMessage = ''
}
// 检查是不是sse协议
if%20(!message.startsWith('data: ')) return
const parsedChunk: Record<string, any> = JSON.parse(message.substring(6))
if%20(!newConversationId) {
newConversationId = parsedChunk.conversation_id
cloud.shared.set(genConversationKey(userName), newConversationId)
}
const { answer } = parsedChunk
responseText += answer
// 伪流式响应
if%20(answer.endsWith('\n\n') ||%20(responseText.length > 120 && /[?。;!]$/.test(responseText))) {
onMessage(responseText.replace('\n\n', ''))
console.log('[sendDifyMessage] received', responseText, newConversationId)
responseText = ''
}
} catch%20(e) {
firstHalfMessage = message
console.error('[sendDifyMessage] error', message)
}
})
// stream结束时把剩下的消息全部发出去
response.data.on('end',%20() => {
onMessage(responseText.replace('\n\n', ''))
})
} catch%20(e) {
console.error("[sendDifyMessage] error", e)
}
}
async function asyncSendMessage(xml) {
console.log('[asyncSendMessage] called', xml)
if%20(xml.MsgType[0] !== 'text') return
const message = xml.Content[0]
const userName = xml.FromUserName[0]
if%20(message === '/new') {
// 重置conversation id
cloud.shared.set(genConversationKey(userName), null)
sendWxMessage('新建成功,开始新的对话吧~~', userName)
return
}
sendWxMessage('AI思考中, 请耐心等待~~', userName)
try {
sendDifyMessage(message, userName,%20(message) => {
sendWxMessage(message, userName)
})
}
catch%20(e) {
console.error('[sendDifyMessage] error', e)
sendWxMessage('接口请求失败,请联系管理员查看错误信息', userName)
}
}
export default async function%20(ctx: FunctionContext) {
const { query } = ctx
const { msg_signature, timestamp, nonce, echostr } = query
const token = process.env.WXWORK_TOKEN
const key = process.env.WXWORK_AESKEY
console.log('[main] called', ctx.method, ctx.request.url)
// 签名验证专用
if%20(ctx.method === 'GET') {
const signature = getSignature(token, timestamp, nonce, echostr)
if%20(signature !== msg_signature) {
return { message: '签名验证失败', code: 401 }
}
const { message } = decrypt(key, echostr)
return message
}
const payload = ctx.body.xml
const encrypt = payload.encrypt[0]
const signature = getSignature(token, timestamp, nonce, encrypt)
if%20(signature !== msg_signature) {
return { message: '签名验证失败', code: 401 }
}
const { message } = decrypt(key, encrypt)
const {
xml
} = await xml2js.parseStringPromise(message)
// 由于GPT API耗时较久,这里提前返回,防止企业微信超时重试,后续再手动调用发消息接口
ctx.response.sendStatus(200)
await asyncSendMessage(xml)
return { message: true, code: 0 }
}
手机扫一扫
移动阅读更方便
你可能感兴趣的文章