mirror of
https://github.com/ikechan8370/chatgpt-plugin.git
synced 2025-12-16 13:27:08 +00:00
Conversation communication (#165)
* init: conversation in api3 (WIP) * feat: beta conversation * fix: fix some problems * fix: add help message
This commit is contained in:
ikechan8370
GitHub
parent
a1e72c7757
commit
a3b14b1a79
11 changed files with 658 additions and 215 deletions
0
REDIS
Normal file
0
REDIS
Normal file
8
Redis.md
Normal file
8
Redis.md
Normal file
|
|
@ -0,0 +1,8 @@
|
|||
## 本插件使用的一些redis键值说明
|
||||
|
||||
### 官方（API3/浏览器模式）
|
||||
* CHATGPT:QQ_CONVERSATION:123456789 qq号为123456789的人当前正在使用哪个对话
|
||||
* CHATGPT:CONVERSATION_LAST_MESSAGE_ID:26001339-9043-435d-8394-c553a3109fdf id为`26001339-9043-435d-8394-c553a3109fdf`的对话，最后一条用户发出的消息的id
|
||||
* CHATGPT:CONVERSATION_LAST_MESSAGE_PROMPT:26001339-9043-435d-8394-c553a3109fdf id为`26001339-9043-435d-8394-c553a3109fdf`的对话，最后一条用户发出的消息的内容（问题）
|
||||
* CHATGPT:CONVERSATION_CREATE_TIME:26001339-9043-435d-8394-c553a3109fdf id为`26001339-9043-435d-8394-c553a3109fdf`的对话创建时间
|
||||
* CHATGPT:CONVERSATION_LENGTH:26001339-9043-435d-8394-c553a3109fdf id为`26001339-9043-435d-8394-c553a3109fdf`的对话当前长度（不计分支）
|
||||
199
apps/chat.js
199
apps/chat.js
|
|
@ -1,32 +1,19 @@
|
|||
import plugin from '../../../lib/plugins/plugin.js'
|
||||
import _ from 'lodash'
|
||||
import { Config } from '../config/index.js'
|
||||
import showdown from 'showdown'
|
||||
import mjAPI from 'mathjax-node'
|
||||
import { uuid } from 'oicq/lib/common.js'
|
||||
import { v4 as uuid } from 'uuid'
|
||||
import delay from 'delay'
|
||||
import { ChatGPTAPI } from 'chatgpt'
|
||||
import { ChatGPTClient, BingAIClient } from '@waylaidwanderer/chatgpt-api'
|
||||
import { getMessageById, makeForwardMsg, tryTimes, upsertMessage, pTimeout } from '../utils/common.js'
|
||||
import { getMessageById, makeForwardMsg, tryTimes, upsertMessage } from '../utils/common.js'
|
||||
import { ChatGPTPuppeteer } from '../utils/browser.js'
|
||||
import { KeyvFile } from 'keyv-file'
|
||||
import { OfficialChatGPTClient } from '../utils/message.js'
|
||||
import fetch from 'node-fetch'
|
||||
// import puppeteer from '../utils/browser.js'
|
||||
// import showdownKatex from 'showdown-katex'
|
||||
import { getConversations, getLatestMessageIdByConversationId } from '../utils/conversation.js'
|
||||
const blockWords = Config.blockWords
|
||||
const converter = new showdown.Converter({
|
||||
extensions: [
|
||||
// showdownKatex({
|
||||
// delimiters: [
|
||||
// { left: '$$', right: '$$', display: false },
|
||||
// { left: '$', right: '$', display: false, inline: true },
|
||||
// { left: '\\(', right: '\\)', display: false },
|
||||
// { left: '\\[', right: '\\]', display: true }
|
||||
// ]
|
||||
// })
|
||||
]
|
||||
})
|
||||
|
||||
/**
|
||||
* 每个对话保留的时长。单个对话内ai是保留上下文的。超时后销毁对话，再次对话创建新的对话。
|
||||
* 单位：秒
|
||||
|
|
@ -42,7 +29,7 @@ mjAPI.config({
|
|||
mjAPI.start()
|
||||
|
||||
export class chatgpt extends plugin {
|
||||
constructor() {
|
||||
constructor () {
|
||||
let toggleMode = Config.toggleMode
|
||||
super({
|
||||
/** 功能名称 */
|
||||
|
|
@ -62,7 +49,7 @@ export class chatgpt extends plugin {
|
|||
},
|
||||
{
|
||||
reg: '#chatgpt对话列表',
|
||||
fnc: 'getConversations',
|
||||
fnc: 'getAllConversations',
|
||||
permission: 'master'
|
||||
},
|
||||
{
|
||||
|
|
@ -95,6 +82,10 @@ export class chatgpt extends plugin {
|
|||
reg: '#OpenAI(剩余)?(余额|额度)',
|
||||
fnc: 'totalAvailable',
|
||||
permission: 'master'
|
||||
},
|
||||
{
|
||||
reg: '^#chatgpt切换对话',
|
||||
fnc: 'attachConversation'
|
||||
}
|
||||
|
||||
]
|
||||
|
|
@ -107,7 +98,7 @@ export class chatgpt extends plugin {
|
|||
* @param e
|
||||
* @returns {Promise<void>}
|
||||
*/
|
||||
async getConversations(e) {
|
||||
async getConversations (e) {
|
||||
let keys = await redis.keys('CHATGPT:CONVERSATIONS:*')
|
||||
if (!keys || keys.length === 0) {
|
||||
await this.reply('当前没有人正在与机器人对话', true)
|
||||
|
|
@ -129,7 +120,7 @@ export class chatgpt extends plugin {
|
|||
* @param e
|
||||
* @returns {Promise<void>}
|
||||
*/
|
||||
async destroyConversations(e) {
|
||||
async destroyConversations (e) {
|
||||
let ats = e.message.filter(m => m.type === 'at')
|
||||
if (ats.length === 0) {
|
||||
let c = await redis.get(`CHATGPT:CONVERSATIONS:${e.sender.user_id}`)
|
||||
|
|
@ -153,7 +144,7 @@ export class chatgpt extends plugin {
|
|||
}
|
||||
}
|
||||
|
||||
async help(e) {
|
||||
async help (e) {
|
||||
let response = 'chatgpt-plugin使用帮助文字版\n' +
|
||||
'@我+聊天内容: 发起对话与AI进行聊天\n' +
|
||||
'#chatgpt对话列表: 查看当前发起的对话\n' +
|
||||
|
|
@ -163,7 +154,7 @@ export class chatgpt extends plugin {
|
|||
await this.reply(response)
|
||||
}
|
||||
|
||||
async switch2Picture(e) {
|
||||
async switch2Picture (e) {
|
||||
let userSetting = await redis.get(`CHATGPT:USER:${e.sender.user_id}`)
|
||||
if (!userSetting) {
|
||||
userSetting = { usePicture: true }
|
||||
|
|
@ -175,7 +166,7 @@ export class chatgpt extends plugin {
|
|||
await this.reply('ChatGPT回复已转换为图片模式')
|
||||
}
|
||||
|
||||
async switch2Text(e) {
|
||||
async switch2Text (e) {
|
||||
let userSetting = await redis.get(`CHATGPT:USER:${e.sender.user_id}`)
|
||||
if (!userSetting) {
|
||||
userSetting = { usePicture: false }
|
||||
|
|
@ -191,7 +182,7 @@ export class chatgpt extends plugin {
|
|||
* #chatgpt
|
||||
* @param e oicq传递的事件参数e
|
||||
*/
|
||||
async chatgpt(e) {
|
||||
async chatgpt (e) {
|
||||
let prompt
|
||||
if (this.toggleMode === 'at') {
|
||||
if (!e.msg || e.msg.startsWith('#')) {
|
||||
|
|
@ -208,7 +199,7 @@ export class chatgpt extends plugin {
|
|||
}
|
||||
}
|
||||
const use = await redis.get('CHATGPT:USE')
|
||||
if (use != 'bing') {
|
||||
if (use !== 'bing') {
|
||||
let randomId = uuid()
|
||||
// 队列队尾插入，开始排队
|
||||
await redis.rPush('CHATGPT:CHAT_QUEUE', [randomId])
|
||||
|
|
@ -240,8 +231,37 @@ export class chatgpt extends plugin {
|
|||
// } catch (e) {
|
||||
// await this.reply('chatgpt初始化出错：' + e.msg, true)
|
||||
// }
|
||||
let previousConversation = await redis.get(`CHATGPT:CONVERSATIONS:${e.sender.user_id}`)
|
||||
let previousConversation
|
||||
let conversation = {}
|
||||
if (use === 'api3') {
|
||||
// api3 支持对话穿插，因此不按照qq号来进行判断了
|
||||
let conversationId = await redis.get(`CHATGPT:QQ_CONVERSATION:${e.sender.user_id}`)
|
||||
if (conversationId) {
|
||||
let lastMessageId = await redis.get(`CHATGPT:CONVERSATION_LAST_MESSAGE_ID:${conversationId}`)
|
||||
if (!lastMessageId) {
|
||||
lastMessageId = await getLatestMessageIdByConversationId(conversationId)
|
||||
}
|
||||
// let lastMessagePrompt = await redis.get(`CHATGPT:CONVERSATION_LAST_MESSAGE_PROMPT:${conversationId}`)
|
||||
// let conversationCreateTime = await redis.get(`CHATGPT:CONVERSATION_CREATE_TIME:${conversationId}`)
|
||||
// let conversationLength = await redis.get(`CHATGPT:CONVERSATION_LENGTH:${conversationId}`)
|
||||
conversation = {
|
||||
conversationId,
|
||||
parentMessageId: lastMessageId
|
||||
}
|
||||
if (Config.debug) {
|
||||
logger.mark({ previousConversation })
|
||||
}
|
||||
} else {
|
||||
let ctime = new Date()
|
||||
previousConversation = {
|
||||
sender: e.sender,
|
||||
ctime,
|
||||
utime: ctime,
|
||||
num: 0
|
||||
}
|
||||
}
|
||||
} else {
|
||||
previousConversation = await redis.get(`CHATGPT:CONVERSATIONS:${e.sender.user_id}`)
|
||||
if (!previousConversation) {
|
||||
let ctime = new Date()
|
||||
previousConversation = {
|
||||
|
|
@ -261,11 +281,14 @@ export class chatgpt extends plugin {
|
|||
conversationSignature: previousConversation.conversationSignature
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
try {
|
||||
if (Config.debug) {
|
||||
logger.mark(conversation)
|
||||
}
|
||||
let chatMessage = await this.sendMessage(prompt, conversation, use)
|
||||
let chatMessage = await this.sendMessage(prompt, conversation, use, e)
|
||||
if (use !== 'api3') {
|
||||
previousConversation.conversation = {
|
||||
conversationId: chatMessage.conversationId
|
||||
}
|
||||
|
|
@ -278,9 +301,10 @@ export class chatgpt extends plugin {
|
|||
previousConversation.conversation.parentMessageId = chatMessage.id
|
||||
}
|
||||
console.log(chatMessage)
|
||||
let response = chatMessage?.text
|
||||
previousConversation.num = previousConversation.num + 1
|
||||
await redis.set(`CHATGPT:CONVERSATIONS:${e.sender.user_id}`, JSON.stringify(previousConversation), CONVERSATION_PRESERVE_TIME > 0 ? { EX: CONVERSATION_PRESERVE_TIME } : {})
|
||||
}
|
||||
let response = chatMessage?.text
|
||||
// 检索是否有屏蔽词
|
||||
const blockWord = blockWords.find(word => response.toLowerCase().includes(word.toLowerCase()))
|
||||
if (blockWord) {
|
||||
|
|
@ -297,14 +321,14 @@ export class chatgpt extends plugin {
|
|||
}
|
||||
if (userSetting.usePicture) {
|
||||
let endTokens = ['.', '。', '……', '!', '！', ']', ')', '）', '】', '?', '？', '~', '"', "'"]
|
||||
let maxTries = 3
|
||||
let maxTries = use === 'api3' ? 3 : 0
|
||||
while (maxTries >= 0 && !endTokens.find(token => response.trimEnd().endsWith(token))) {
|
||||
maxTries--
|
||||
// while (!response.trimEnd().endsWith('.') && !response.trimEnd().endsWith('。') && !response.trimEnd().endsWith('……') &&
|
||||
// !response.trimEnd().endsWith('！') && !response.trimEnd().endsWith('!') && !response.trimEnd().endsWith(']') && !response.trimEnd().endsWith('】')
|
||||
// ) {
|
||||
await this.reply('内容有点多，我正在奋笔疾书，请再等一会', true, { recallMsg: 5 })
|
||||
let responseAppend = await this.sendMessage('Continue', conversation, use)
|
||||
let responseAppend = await this.sendMessage('Continue', conversation, use, e)
|
||||
previousConversation.conversation = {
|
||||
conversationId: responseAppend.conversationId,
|
||||
parentMessageId: responseAppend.id
|
||||
|
|
@ -328,23 +352,22 @@ export class chatgpt extends plugin {
|
|||
// logger.info(response)
|
||||
// markdown转为html
|
||||
// todo部分数学公式可能还有问题
|
||||
let converted = response //converter.makeHtml(response)
|
||||
let converted = response // converter.makeHtml(response)
|
||||
|
||||
/** 最后回复消息 */
|
||||
await e.runtime.render('chatgpt-plugin', use != 'bing' ? 'content/ChatGPT/index' : 'content/Bing/index', { content: converted, prompt, senderName: e.sender.nickname })
|
||||
await e.runtime.render('chatgpt-plugin', use !== 'bing' ? 'content/ChatGPT/index' : 'content/Bing/index', { content: converted, prompt, senderName: e.sender.nickname })
|
||||
} else {
|
||||
let quotemessage = []
|
||||
if (chatMessage?.quote) {
|
||||
chatMessage.quote.forEach(function (item, index) {
|
||||
if (item.trim() != '') {
|
||||
if (item.trim() !== '') {
|
||||
quotemessage.push(item)
|
||||
}
|
||||
})
|
||||
}
|
||||
if (response.length > 1000 ) {
|
||||
if (Config.autoUsePicture && response.length > Config.autoUsePictureThreshold) {
|
||||
// 文字过多时自动切换到图片模式输出
|
||||
let converted = response
|
||||
await e.runtime.render('chatgpt-plugin', use != 'bing' ? 'content/ChatGPT/index' : 'content/Bing/index', { content: converted, prompt, quote: quotemessage, senderName: e.sender.nickname })
|
||||
await e.runtime.render('chatgpt-plugin', use !== 'bing' ? 'content/ChatGPT/index' : 'content/Bing/index', { content: response, prompt, quote: quotemessage, senderName: e.sender.nickname })
|
||||
} else {
|
||||
await this.reply(`${response}`, e.isGroup)
|
||||
if (quotemessage.length > 0) {
|
||||
|
|
@ -370,16 +393,20 @@ export class chatgpt extends plugin {
|
|||
}
|
||||
}
|
||||
|
||||
async sendMessage(prompt, conversation = {}, use) {
|
||||
async sendMessage (prompt, conversation = {}, use, e) {
|
||||
if (!conversation) {
|
||||
conversation = {
|
||||
timeoutMs: Config.defaultTimeoutMs
|
||||
}
|
||||
}
|
||||
// console.log(use)
|
||||
if (use === 'browser') {
|
||||
if (Config.debug) {
|
||||
logger.mark(`using ${use} mode`)
|
||||
}
|
||||
switch (use) {
|
||||
case 'browser': {
|
||||
return await this.chatgptBrowserBased(prompt, conversation)
|
||||
} else if (use === 'apiReverse') {
|
||||
}
|
||||
case 'apiReverse': {
|
||||
const currentDate = new Date().toISOString().split('T')[0]
|
||||
let promptPrefix = `You are ${Config.assistantLabel}, a large language model trained by OpenAI. ${Config.promptPrefixOverride || defaultPropmtPrefix}
|
||||
Current date: ${currentDate}`
|
||||
|
|
@ -420,12 +447,13 @@ export class chatgpt extends plugin {
|
|||
id: response.messageId,
|
||||
parentMessageId: conversation?.parentMessageId
|
||||
}
|
||||
} else if (use === 'bing') {
|
||||
}
|
||||
case 'bing': {
|
||||
let bingToken = await redis.get('CHATGPT:BING_TOKEN')
|
||||
if (!bingToken) {
|
||||
throw new Error('未绑定Bing Cookie，请使用#chatgpt设置必应token命令绑定Bing Cookie')
|
||||
}
|
||||
let cookie = undefined
|
||||
let cookie
|
||||
if (bingToken?.indexOf('=') > -1) {
|
||||
cookie = bingToken
|
||||
}
|
||||
|
|
@ -437,19 +465,6 @@ export class chatgpt extends plugin {
|
|||
let response
|
||||
let reply = ''
|
||||
try {
|
||||
/* bingAIClient中设置了无响应2分钟超时，应该不用单独处理了,后续看情况可以删掉这些代码
|
||||
const responseP = new Promise(
|
||||
async (resolve, reject) => {
|
||||
let bingResponse = await bingAIClient.sendMessage(prompt, conversation || {},(token) => {
|
||||
reply += token
|
||||
})
|
||||
return resolve(bingResponse)
|
||||
})
|
||||
response = await pTimeout(responseP, {
|
||||
milliseconds: Config.bingTimeoutMs,
|
||||
message: reply != '' ? `${reply}\n不行了，我的大脑过载了，处理不过来了!` : '必应的小脑子不好使了，不知道怎么回答！'
|
||||
})
|
||||
*/
|
||||
response = await bingAIClient.sendMessage(prompt, conversation || {}, (token) => {
|
||||
reply += token
|
||||
})
|
||||
|
|
@ -481,7 +496,8 @@ export class chatgpt extends plugin {
|
|||
invocationId: response.invocationId,
|
||||
conversationSignature: response.conversationSignature
|
||||
}
|
||||
} else if (use === 'api3') {
|
||||
}
|
||||
case 'api3': {
|
||||
// official without cloudflare
|
||||
let accessToken = await redis.get('CHATGPT:TOKEN')
|
||||
if (!accessToken) {
|
||||
|
|
@ -492,8 +508,20 @@ export class chatgpt extends plugin {
|
|||
apiReverseUrl: Config.api,
|
||||
timeoutMs: 120000
|
||||
})
|
||||
return await this.chatGPTApi.sendMessage(prompt, conversation)
|
||||
} else {
|
||||
let sendMessageResult = await this.chatGPTApi.sendMessage(prompt, conversation)
|
||||
// 更新最后一条prompt
|
||||
await redis.set(`CHATGPT:CONVERSATION_LAST_MESSAGE_PROMPT:${sendMessageResult.conversationId}`, prompt)
|
||||
// 更新最后一条messageId
|
||||
await redis.set(`CHATGPT:CONVERSATION_LAST_MESSAGE_ID:${sendMessageResult.conversationId}`, sendMessageResult.id)
|
||||
await redis.set(`CHATGPT:QQ_CONVERSATION:${e.sender.user_id}`, sendMessageResult.conversationId)
|
||||
if (!conversation.conversationId) {
|
||||
// 如果是对话的创建者
|
||||
await redis.set(`CHATGPT:CONVERSATION_CREATER_ID:${sendMessageResult.conversationId}`, e.sender.user_id)
|
||||
await redis.set(`CHATGPT:CONVERSATION_CREATER_NICK_NAME:${sendMessageResult.conversationId}`, e.sender.card)
|
||||
}
|
||||
return sendMessageResult
|
||||
}
|
||||
default: {
|
||||
let completionParams = {}
|
||||
if (Config.model) {
|
||||
completionParams.model = Config.model
|
||||
|
|
@ -520,13 +548,14 @@ export class chatgpt extends plugin {
|
|||
return await tryTimes(async () => await this.chatGPTApi.sendMessage(prompt, option), 5)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
async emptyQueue(e) {
|
||||
async emptyQueue (e) {
|
||||
await redis.lTrim('CHATGPT:CHAT_QUEUE', 1, 0)
|
||||
await this.reply('已清空当前等待队列')
|
||||
}
|
||||
|
||||
async removeQueueFirst(e) {
|
||||
async removeQueueFirst (e) {
|
||||
let uid = await redis.lPop('CHATGPT:CHAT_QUEUE', 0)
|
||||
if (!uid) {
|
||||
await this.reply('当前等待队列为空')
|
||||
|
|
@ -535,13 +564,55 @@ export class chatgpt extends plugin {
|
|||
}
|
||||
}
|
||||
|
||||
async totalAvailable(e) {
|
||||
async getAllConversations (e) {
|
||||
const use = await redis.get('CHATGPT:USE')
|
||||
if (use === 'api3') {
|
||||
let conversations = await getConversations(e.sender.user_id)
|
||||
if (Config.debug) {
|
||||
logger.mark('all conversations: ', conversations)
|
||||
}
|
||||
// let conversationsFirst10 = conversations.slice(0, 10)
|
||||
await e.runtime.render('chatgpt-plugin', 'conversation/chatgpt', { conversations })
|
||||
let text = '对话列表\n'
|
||||
text += '对话id | 对话发起者 \n'
|
||||
conversations.forEach(c => {
|
||||
text += c.id + '|' + (c.creater || '未知') + '\n'
|
||||
})
|
||||
text += '您可以通过使用命令#chatgpt切换对话+对话id来加入指定对话'
|
||||
await this.reply(text)
|
||||
} else {
|
||||
return await this.getConversations(e)
|
||||
}
|
||||
}
|
||||
|
||||
async attachConversation (e) {
|
||||
const use = await redis.get('CHATGPT:USE')
|
||||
if (use !== 'api3') {
|
||||
await this.reply('该功能目前仅支持API3模式')
|
||||
} else {
|
||||
let conversationId = _.trimStart(e.msg.trimStart(), '#chatgpt切换对话').trim()
|
||||
if (!conversationId) {
|
||||
await this.reply('无效对话id，请在#chatgpt切换对话后面加上对话id')
|
||||
return false
|
||||
}
|
||||
// todo 验证这个对话是否存在且有效
|
||||
// await getLatestMessageIdByConversationId(conversationId)
|
||||
await redis.set(`CHATGPT:QQ_CONVERSATION:${e.sender.user_id}`, conversationId)
|
||||
await this.reply('切换成功')
|
||||
}
|
||||
}
|
||||
|
||||
async totalAvailable (e) {
|
||||
if (!Config.apiKey) {
|
||||
this.reply('当前未配置OpenAI API key，请在插件配置文件config/index.js中配置。若使用免费的API3则无需关心计费。')
|
||||
return false
|
||||
}
|
||||
// 查询OpenAI Plus剩余试用额度
|
||||
fetch('https://api.openai.com/dashboard/billing/credit_grants', {
|
||||
method: 'GET',
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
'Authorization': 'Bearer ' + Config.apiKey,
|
||||
Authorization: 'Bearer ' + Config.apiKey
|
||||
}
|
||||
})
|
||||
.then(response => response.json())
|
||||
|
|
@ -560,7 +631,7 @@ export class chatgpt extends plugin {
|
|||
* @param prompt 问题
|
||||
* @param conversation 对话
|
||||
*/
|
||||
async chatgptBrowserBased(prompt, conversation) {
|
||||
async chatgptBrowserBased (prompt, conversation) {
|
||||
let option = { markdown: true }
|
||||
if (Config['2captchaToken']) {
|
||||
option.captchaToken = Config['2captchaToken']
|
||||
|
|
|
|||
|
|
@ -34,7 +34,12 @@ let helpData = [
|
|||
{
|
||||
icon: 'list',
|
||||
title: '#chatgpt对话列表',
|
||||
desc: '查询当前哪些人正在与机器人聊天'
|
||||
desc: '查询当前哪些人正在与机器人聊天.目前API3模式下支持切换对话'
|
||||
},
|
||||
{
|
||||
icon: 'switch',
|
||||
title: '#chatgpt切换对话+对话id',
|
||||
desc: '目前仅API3模式下可用，切换到指定的对话中'
|
||||
},
|
||||
{
|
||||
icon: 'destroy',
|
||||
|
|
|
|||
|
|
@ -1,5 +1,5 @@
|
|||
import plugin from '../../../lib/plugins/plugin.js'
|
||||
import {Config} from "../config/index.js";
|
||||
import { Config } from '../config/index.js'
|
||||
import { BingAIClient } from '@waylaidwanderer/chatgpt-api'
|
||||
|
||||
export class ChatgptManagement extends plugin {
|
||||
|
|
@ -99,7 +99,7 @@ export class ChatgptManagement extends plugin {
|
|||
this.finish('saveToken')
|
||||
return
|
||||
}
|
||||
let cookie = undefined
|
||||
let cookie
|
||||
if (token?.indexOf('=') > -1) {
|
||||
cookie = token
|
||||
}
|
||||
|
|
@ -111,7 +111,7 @@ export class ChatgptManagement extends plugin {
|
|||
// 异步就好了，不卡着这个context了
|
||||
bingAIClient.createNewConversation().then(async res => {
|
||||
if (res.clientId) {
|
||||
logger.info("bing token 有效")
|
||||
logger.info('bing token 有效')
|
||||
} else {
|
||||
logger.error('bing token 无效', res)
|
||||
await this.reply(`经检测，Bing Token无效。来自Bing的错误提示：${res.result?.message}`)
|
||||
|
|
@ -141,8 +141,13 @@ export class ChatgptManagement extends plugin {
|
|||
}
|
||||
|
||||
async useOpenAIAPIBasedSolution (e) {
|
||||
let use = await redis.get('CHATGPT:USE')
|
||||
if (use !== 'api') {
|
||||
await redis.set('CHATGPT:USE', 'api')
|
||||
await this.reply('已切换到基于OpenAI API的解决方案，如果已经对话过建议执行`#结束对话`避免引起404错误')
|
||||
} else {
|
||||
await this.reply('当前已经是API模式了')
|
||||
}
|
||||
}
|
||||
|
||||
async useReversedAPIBasedSolution (e) {
|
||||
|
|
@ -151,13 +156,33 @@ export class ChatgptManagement extends plugin {
|
|||
}
|
||||
|
||||
async useReversedAPIBasedSolution2 (e) {
|
||||
let use = await redis.get('CHATGPT:USE')
|
||||
if (use !== 'api3') {
|
||||
await redis.set('CHATGPT:USE', 'api3')
|
||||
await this.reply('已切换到基于第三方Reversed ConversastionAPI的解决方案，如果已经对话过建议执行`#结束对话`避免引起404错误')
|
||||
await this.reply('已切换到基于第三方Reversed Conversastion API(API3)的解决方案')
|
||||
} else {
|
||||
await this.reply('当前已经是API3模式了')
|
||||
}
|
||||
}
|
||||
|
||||
async useReversedBingSolution (e) {
|
||||
let use = await redis.get('CHATGPT:USE')
|
||||
if (use !== 'bing') {
|
||||
await redis.set('CHATGPT:USE', 'bing')
|
||||
// 结束所有人的对话
|
||||
const keys = await redis.keys('CHATGPT:CONVERSATIONS:*')
|
||||
if (keys.length) {
|
||||
const response = await redis.del(keys)
|
||||
if (Config.debug) {
|
||||
console.log('Deleted keys:', response)
|
||||
}
|
||||
} else {
|
||||
console.log('No keys matched the pattern')
|
||||
}
|
||||
await this.reply('已切换到基于微软新必应的解决方案，如果已经对话过务必执行`#结束对话`避免引起404错误')
|
||||
} else {
|
||||
await this.reply('当前已经是必应Bing模式了')
|
||||
}
|
||||
}
|
||||
|
||||
async modeHelp () {
|
||||
|
|
|
|||
|
|
@ -10,6 +10,10 @@ export const Config = {
|
|||
blockWords: ['屏蔽词1', '屏蔽词b'],
|
||||
// 改为true后，全局默认以图片形式回复，并自动发出Continue命令补全回答。长回复可能会有bug。
|
||||
defaultUsePicture: false,
|
||||
// 如果true，字数大于阈值（autoUsePictureThreshold）会自动用图片发送，即使是文本模式。
|
||||
autoUsePicture: true,
|
||||
// 仅当autoUsePicture为true时生效，字数大于该阈值会自动用图片发送，即使是文本模式。
|
||||
autoUsePictureThreshold: 1200,
|
||||
// 每个人发起的对话保留时长。超过这个时长没有进行对话，再进行对话将开启新的对话。单位：秒
|
||||
conversationPreserveTime: 0,
|
||||
// 触发方式 可选值：at 或 prefix 。at模式下只有at机器人才会回复。prefix模式下不需要at，但需要添加前缀#chat
|
||||
|
|
@ -26,6 +30,7 @@ export const Config = {
|
|||
// ***********************************************************************************************************************************
|
||||
// from https://github.com/acheong08/ChatGPT
|
||||
api: 'https://chatgpt.duti.tech/api/conversation',
|
||||
apiBaseUrl: 'https://chatgpt.duti.tech',
|
||||
// ***********************************************************************************************************************************
|
||||
// 以下为API2方式的配置 *
|
||||
// ***********************************************************************************************************************************
|
||||
|
|
@ -60,7 +65,7 @@ export const Config = {
|
|||
'2captchaToken': '',
|
||||
// http或socks5代理
|
||||
proxy: PROXY,
|
||||
debug: false,
|
||||
debug: true,
|
||||
// 各个地方的默认超时时间
|
||||
defaultTimeoutMs: 120000,
|
||||
// bing默认超时时间，bing太慢了有的时候
|
||||
|
|
|
|||
36
resources/conversation/chatgpt.html
Normal file
36
resources/conversation/chatgpt.html
Normal file
|
|
@ -0,0 +1,36 @@
|
|||
<!DOCTYPE html>
|
||||
<html>
|
||||
|
||||
<head>
|
||||
<meta http-equiv="content-type" content="text/html;charset=utf-8" />
|
||||
<link rel="stylesheet" type="text/css" href="{{pluResPath}}conversation/conversation.css" />
|
||||
<link rel="shortcut icon" href="#" />
|
||||
</head>
|
||||
{{@headStyle}}
|
||||
|
||||
<body>
|
||||
<div class="container" id="container">
|
||||
<div class="head_box">
|
||||
<div class="id_text">ChatGPT-Plugin</div>
|
||||
<h2 class="day_text">最近对话列表</h2>
|
||||
<img class="chatgpt_logo" src="{{pluResPath}}img/icon/chatgpt.png"/>
|
||||
</div>
|
||||
<div class="data_box">
|
||||
<div class="list">
|
||||
{{each conversations item}}
|
||||
<div class="item-{{item.status}}">
|
||||
<img class="icon" src="{{pluResPath}}img/icon/chat.png" />
|
||||
<div class="title">
|
||||
<div class="text">{{item.id}}</div>
|
||||
<div class="dec">最近问题：{{item.lastPrompt}}</div>
|
||||
<div class="creater">发起者：{{item.creater}}</div>
|
||||
</div>
|
||||
</div>
|
||||
{{/each}}
|
||||
</div>
|
||||
</div>
|
||||
<div class="logo">Created By Yunzai-Bot and ChatGPT-Plugin</div>
|
||||
</div>
|
||||
</body>
|
||||
|
||||
</html>
|
||||
178
resources/conversation/conversation.css
Normal file
178
resources/conversation/conversation.css
Normal file
|
|
@ -0,0 +1,178 @@
|
|||
@font-face {
|
||||
font-family: "tttgbnumber";
|
||||
src: url("../../../../../resources/font/tttgbnumber.ttf");
|
||||
font-weight: normal;
|
||||
font-style: normal;
|
||||
}
|
||||
* {
|
||||
margin: 0;
|
||||
padding: 0;
|
||||
box-sizing: border-box;
|
||||
user-select: none;
|
||||
}
|
||||
body {
|
||||
font-family: sans-serif;
|
||||
font-size: 16px;
|
||||
width: 630px;
|
||||
color: #1e1f20;
|
||||
transform: scale(1.5);
|
||||
transform-origin: 0 0;
|
||||
}
|
||||
.container {
|
||||
width: 630px;
|
||||
padding: 20px 15px 10px 15px;
|
||||
background-color: #f5f6fb;
|
||||
}
|
||||
.head_box {
|
||||
border-radius: 15px;
|
||||
font-family: tttgbnumber;
|
||||
padding: 10px 20px;
|
||||
position: relative;
|
||||
box-shadow: 0 5px 10px 0 rgb(0 0 0 / 15%);
|
||||
}
|
||||
.head_box .id_text {
|
||||
font-size: 24px;
|
||||
}
|
||||
.head_box .day_text {
|
||||
font-size: 20px;
|
||||
}
|
||||
.head_box .chatgpt_logo {
|
||||
position: absolute;
|
||||
top: 12px;
|
||||
right: 15px;
|
||||
width: 50px;
|
||||
}
|
||||
.base_info {
|
||||
position: relative;
|
||||
padding-left: 10px;
|
||||
}
|
||||
.uid {
|
||||
font-family: tttgbnumber;
|
||||
}
|
||||
|
||||
.data_box {
|
||||
border-radius: 15px;
|
||||
margin-top: 20px;
|
||||
margin-bottom: 15px;
|
||||
padding: 20px 0px 5px 0px;
|
||||
background: #fff;
|
||||
box-shadow: 0 5px 10px 0 rgb(0 0 0 / 15%);
|
||||
position: relative;
|
||||
}
|
||||
.tab_lable {
|
||||
position: absolute;
|
||||
top: -10px;
|
||||
left: -8px;
|
||||
background: #d4b98c;
|
||||
color: #fff;
|
||||
font-size: 14px;
|
||||
padding: 3px 10px;
|
||||
border-radius: 15px 0px 15px 15px;
|
||||
z-index: 20;
|
||||
}
|
||||
.data_line {
|
||||
display: flex;
|
||||
justify-content: space-around;
|
||||
margin-bottom: 14px;
|
||||
}
|
||||
.data_line_item {
|
||||
width: 100px;
|
||||
text-align: center;
|
||||
/*margin: 0 20px;*/
|
||||
}
|
||||
.num {
|
||||
font-family: tttgbnumber;
|
||||
font-size: 24px;
|
||||
}
|
||||
.data_box .lable {
|
||||
font-size: 14px;
|
||||
color: #7f858a;
|
||||
line-height: 1;
|
||||
margin-top: 3px;
|
||||
}
|
||||
|
||||
.list{
|
||||
display: flex;
|
||||
justify-content: flex-start;
|
||||
flex-wrap: wrap;
|
||||
}
|
||||
|
||||
.list .item-normal {
|
||||
width: 575px;
|
||||
display: flex;
|
||||
align-items: center;
|
||||
background: #f1f1f1;
|
||||
padding: 8px 6px 8px 6px;
|
||||
border-radius: 8px;
|
||||
margin: 0 0px 10px 10px;
|
||||
}
|
||||
.list .item-normal .icon{
|
||||
width: 24px;
|
||||
height: 24px;
|
||||
background-repeat: no-repeat;
|
||||
background-size: 100% 100%;
|
||||
position: relative;
|
||||
flex-shrink: 0;
|
||||
}
|
||||
.list .item-normal .title{
|
||||
font-size: 16px;
|
||||
margin-left: 6px;
|
||||
line-height: 20px;
|
||||
}
|
||||
.list .item-normal .title .text{
|
||||
color: #1995A4;
|
||||
white-space: nowrap;
|
||||
}
|
||||
.list .item-normal .title .creater{
|
||||
font-size: 12px;
|
||||
color: #69878B;
|
||||
margin-top: 2px;
|
||||
}
|
||||
.list .item-using .title .dec{
|
||||
font-size: 12px;
|
||||
color: #999;
|
||||
margin-top: 2px;
|
||||
}
|
||||
.list .item-using {
|
||||
width: 575px;
|
||||
display: flex;
|
||||
align-items: center;
|
||||
background: #157985;
|
||||
padding: 8px 6px 8px 6px;
|
||||
border-radius: 8px;
|
||||
margin: 0 0px 10px 10px;
|
||||
}
|
||||
.list .item-using .icon{
|
||||
width: 24px;
|
||||
height: 24px;
|
||||
background-repeat: no-repeat;
|
||||
background-size: 100% 100%;
|
||||
position: relative;
|
||||
flex-shrink: 0;
|
||||
}
|
||||
.list .item-using .title{
|
||||
font-size: 16px;
|
||||
margin-left: 6px;
|
||||
line-height: 20px;
|
||||
}
|
||||
.list .item-using .title .text{
|
||||
color: #CBD4D5;
|
||||
white-space: nowrap;
|
||||
}
|
||||
.list .item-using .title .dec{
|
||||
font-size: 12px;
|
||||
color: #CBD4D5;
|
||||
margin-top: 2px;
|
||||
}
|
||||
.list .item-using .title .creater{
|
||||
font-size: 12px;
|
||||
color: #CBD4D5;
|
||||
margin-top: 2px;
|
||||
}
|
||||
|
||||
.logo {
|
||||
font-size: 14px;
|
||||
font-family: "tttgbnumber";
|
||||
text-align: center;
|
||||
color: #7994a7;
|
||||
}
|
||||
|
|
@ -4,8 +4,6 @@ import StealthPlugin from 'puppeteer-extra-plugin-stealth'
|
|||
import { getOpenAIAuth } from './openai-auth.js'
|
||||
import delay from 'delay'
|
||||
import { v4 as uuidv4 } from 'uuid'
|
||||
import { pTimeout } from './common.js'
|
||||
console.log({ pTimeout })
|
||||
const chatUrl = 'https://chat.openai.com/chat'
|
||||
let puppeteer = {}
|
||||
|
||||
|
|
|
|||
117
utils/conversation.js
Normal file
117
utils/conversation.js
Normal file
|
|
@ -0,0 +1,117 @@
|
|||
import fetch from 'node-fetch'
|
||||
import { Config } from '../config/index.js'
|
||||
|
||||
export async function getConversations (qq = '') {
|
||||
let accessToken = await redis.get('CHATGPT:TOKEN')
|
||||
if (!accessToken) {
|
||||
throw new Error('未绑定ChatGPT AccessToken，请使用#chatgpt设置token命令绑定token')
|
||||
}
|
||||
let response = await fetch(`${Config.apiBaseUrl}/api/conversations?offset=0&limit=20`, {
|
||||
method: 'GET',
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
Authorization: 'Bearer ' + accessToken
|
||||
}
|
||||
})
|
||||
let json = await response.text()
|
||||
if (Config.debug) {
|
||||
logger.mark(json)
|
||||
}
|
||||
let conversations = JSON.parse(json)
|
||||
let result = conversations.items?.sort((a, b) => b.create_time - a.create_time)
|
||||
let map = {}
|
||||
for (let i = 0; i < conversations.items.length; i++) {
|
||||
// 老用户初次更新该功能，这里频繁请求可能会429。由并行改为串行以尽量降低频率。必要时可可能还要等待。
|
||||
let item = conversations.items[i]
|
||||
let cachedConversationLastMessage = await redis.get(`CHATGPT:CONVERSATION_LAST_MESSAGE_PROMPT:${item.id}`)
|
||||
if (!cachedConversationLastMessage) {
|
||||
map[item.id] = cachedConversationLastMessage
|
||||
} else {
|
||||
// 缓存中没有，就去查官方api
|
||||
let conversationDetailResponse = await fetch(`${Config.apiBaseUrl}/api/conversation/${item.id}`, {
|
||||
method: 'GET',
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
Authorization: 'Bearer ' + accessToken
|
||||
}
|
||||
})
|
||||
let conversationDetail = await conversationDetailResponse.text()
|
||||
if (Config.debug) {
|
||||
logger.mark('conversation detail for conversation ' + item.id, conversationDetail)
|
||||
}
|
||||
conversationDetail = JSON.parse(conversationDetail)
|
||||
let messages = Object.values(conversationDetail.mapping)
|
||||
|
||||
messages = messages
|
||||
.filter(message => message.message)
|
||||
.map(messages => messages.message)
|
||||
|
||||
let messagesAssistant = messages.filter(messages => messages.role === 'assistant')
|
||||
.sort((a, b) => b.create_time - a.create_time)
|
||||
let messagesUser = messages.filter(messages => messages.role === 'user')
|
||||
.sort((a, b) => b.create_time - a.create_time)
|
||||
await redis.set(`CHATGPT:CONVERSATION_LENGTH:${item.id}`, messagesUser?.length || 0)
|
||||
let lastMessage = null
|
||||
if (messagesUser.length > 0) {
|
||||
lastMessage = messagesUser[0].content.parts[0]
|
||||
await redis.set(`CHATGPT:CONVERSATION_LAST_MESSAGE_PROMPT:${item.id}`, lastMessage)
|
||||
}
|
||||
if (messagesAssistant.length > 0) {
|
||||
await redis.set(`CHATGPT:CONVERSATION_LAST_MESSAGE_ID:${item.id}`, messagesAssistant[0].id)
|
||||
}
|
||||
await redis.set(`CHATGPT:CONVERSATION_CREATE_TIME:${item.id}`, new Date(conversationDetail.create_time * 1000).toLocaleString())
|
||||
map[item.id] = lastMessage
|
||||
}
|
||||
}
|
||||
let res = []
|
||||
let usingConversationId
|
||||
if (qq) {
|
||||
usingConversationId = await redis.get(`CHATGPT:QQ_CONVERSATION:${qq}`)
|
||||
}
|
||||
let promisesPostProcess = result.map(async conversation => {
|
||||
conversation.lastPrompt = map[conversation.id]
|
||||
conversation.create_time = new Date(conversation.create_time).toLocaleString()
|
||||
// 这里的时间格式还可以。不用管了。conversation.create_time =
|
||||
// title 全是 New chat，不要了
|
||||
delete conversation.title
|
||||
conversation.creater = await redis.get(`CHATGPT:CONVERSATION_CREATER_NICK_NAME:${conversation.id}`)
|
||||
if (qq && conversation.id === usingConversationId) {
|
||||
conversation.status = 'using'
|
||||
} else {
|
||||
conversation.status = 'normal'
|
||||
}
|
||||
if (conversation.lastPrompt?.length > 80) {
|
||||
conversation.lastPrompt = conversation.lastPrompt.slice(0, 80) + '......'
|
||||
}
|
||||
res.push(conversation)
|
||||
})
|
||||
await Promise.all(promisesPostProcess)
|
||||
return res
|
||||
}
|
||||
|
||||
export async function getLatestMessageIdByConversationId (conversationId) {
|
||||
let accessToken = await redis.get('CHATGPT:TOKEN')
|
||||
if (!accessToken) {
|
||||
throw new Error('未绑定ChatGPT AccessToken，请使用#chatgpt设置token命令绑定token')
|
||||
}
|
||||
let conversationDetailResponse = await fetch(`${Config.apiBaseUrl}/api/conversation/${conversationId}`, {
|
||||
method: 'GET',
|
||||
headers: {
|
||||
'Content-Type': 'application/json',
|
||||
Authorization: 'Bearer ' + accessToken
|
||||
}
|
||||
})
|
||||
let conversationDetail = await conversationDetailResponse.text()
|
||||
if (Config.debug) {
|
||||
logger.mark('conversation detail for conversation ' + conversationId, conversationDetail)
|
||||
}
|
||||
conversationDetail = JSON.parse(conversationDetail)
|
||||
let messages = Object.values(conversationDetail.mapping)
|
||||
messages = messages
|
||||
.filter(message => message.message)
|
||||
.map(messages => messages.message)
|
||||
.filter(messages => messages.role === 'assistant')
|
||||
.sort((a, b) => b.create_time - a.create_time)
|
||||
await redis.set(`CHATGPT:CONVERSATION_LAST_MESSAGE_ID:${conversationId}`, messages[0].id)
|
||||
return messages[0].id
|
||||
}
|
||||
0
utils/redis-key.js
Normal file
0
utils/redis-key.js
Normal file
Loading…
Add table
Add a link
Reference in a new issue