From 62c8d56e765f609cf505cf54f3d9b15b4ce23401 Mon Sep 17 00:00:00 2001 From: ikechan8370 Date: Tue, 14 Feb 2023 22:54:47 +0800 Subject: [PATCH] Api3 (#149) * feat: api3 WIP * feat: add support for proxy of openai * fix: if no proxy --- apps/chat.js | 13 +++++++ apps/management.js | 12 ++++++- config/index.js | 10 ++++-- utils/message.js | 84 ++++++++++++++++++++++++++++++++++++++++++++++ 4 files changed, 115 insertions(+), 4 deletions(-) create mode 100644 utils/message.js diff --git a/apps/chat.js b/apps/chat.js index 9e7eb96..c1254c4 100644 --- a/apps/chat.js +++ b/apps/chat.js @@ -10,6 +10,7 @@ import { ChatGPTClient, BingAIClient } from '@waylaidwanderer/chatgpt-api' import { getMessageById, tryTimes, upsertMessage } from '../utils/common.js' import { ChatGPTPuppeteer } from '../utils/browser.js' import { KeyvFile } from 'keyv-file' +import {OfficialChatGPTClient} from "../utils/message.js"; // import puppeteer from '../utils/browser.js' // import showdownKatex from 'showdown-katex' const blockWords = Config.blockWords @@ -402,6 +403,18 @@ export class chatgpt extends plugin { invocationId: response.invocationId, conversationSignature: response.conversationSignature } + } else if (use === 'api3') { + // official without cloudflare + let accessToken = await redis.get('CHATGPT:TOKEN') + if (!accessToken) { + throw new Error('未绑定ChatGPT AccessToken，请使用#chatgpt设置token命令绑定token') + } + this.chatGPTApi = new OfficialChatGPTClient({ + accessToken, + apiReverseUrl: Config.api, + timeoutMs: 120000 + }) + return await this.chatGPTApi.sendMessage(prompt, conversation) } else { let completionParams = {} if (Config.model) { diff --git a/apps/management.js b/apps/management.js index ec506db..13ccc49 100644 --- a/apps/management.js +++ b/apps/management.js @@ -43,6 +43,11 @@ export class ChatgptManagement extends plugin { fnc: 'useReversedAPIBasedSolution', permission: 'master' }, + { + reg: '^#chatgpt切换API3$', + fnc: 'useReversedAPIBasedSolution2', + permission: 'master' + }, { reg: '^#chatgpt切换(必应|Bing)$', fnc: 'useReversedBingSolution', @@ -118,7 +123,12 @@ export class ChatgptManagement extends plugin { async useReversedAPIBasedSolution (e) { await redis.set('CHATGPT:USE', 'apiReverse') - await this.reply('已切换到基于第三方Reversed API的解决方案，如果已经对话过建议执行`#结束对话`避免引起404错误') + await this.reply('已切换到基于第三方Reversed CompletionAPI的解决方案，如果已经对话过建议执行`#结束对话`避免引起404错误') + } + + async useReversedAPIBasedSolution2 (e) { + await redis.set('CHATGPT:USE', 'api3') + await this.reply('已切换到基于第三方Reversed ConversastionAPI的解决方案，如果已经对话过建议执行`#结束对话`避免引起404错误') } async useReversedBingSolution (e) { diff --git a/config/index.js b/config/index.js index d938084..3d76e3c 100644 --- a/config/index.js +++ b/config/index.js @@ -1,5 +1,5 @@ // 例如http://127.0.0.1:7890 -const PROXY = '' +const PROXY = 'http://127.0.0.1:7890' const API_KEY = '' export const Config = { @@ -20,8 +20,12 @@ export const Config = { apiKey: API_KEY, // 模型名称，选填。如无特殊需求保持默认即可，会使用chatgpt-api库提供的当前可用的最适合的默认值。保底可用的是 text-davinci-003。当发现新的可用的chatGPT模型会更新这里的值 // 20230211： text-chat-davinci-002-sh-alpha-aoruigiofdj83 中午存活了几分钟 - model: 'text-davinci-002-render', - api: '', + model: '', + // *********************************************************************************************************************************** + // 以下为API3方式的配置 * + // *********************************************************************************************************************************** + // from https://github.com/acheong08/ChatGPT + api: 'https://chat.duti.tech/api/conversation', // *********************************************************************************************************************************** // 以下为API2方式的配置 * // *********************************************************************************************************************************** diff --git a/utils/message.js b/utils/message.js new file mode 100644 index 0000000..aaaf21b --- /dev/null +++ b/utils/message.js @@ -0,0 +1,84 @@ +import { v4 as uuidv4 } from 'uuid' +import { Config } from '../config/index.js' +import HttpsProxyAgent from 'https-proxy-agent' +import _ from 'lodash' +import fetch from 'node-fetch' +export class OfficialChatGPTClient { + constructor (opts = {}) { + const { + accessToken, + apiReverseUrl, + timeoutMs + } = opts + this._accessToken = accessToken + this._apiReverseUrl = apiReverseUrl + this._timeoutMs = timeoutMs + } + + async sendMessage (prompt, opts = {}) { + let { + timeoutMs = this._timeoutMs, + conversationId = uuidv4(), + parentMessageId = uuidv4(), + messageId = uuidv4(), + action = 'next' + } = opts + let abortController = null + if (timeoutMs) { + abortController = new AbortController() + } + const url = this._apiReverseUrl || 'https://chat.openai.com/backend-api/conversation' + const body = { + action, + messages: [ + { + id: messageId, + role: 'user', + content: { + content_type: 'text', + parts: [prompt] + } + } + ], + conversationId, + model: Config.plus ? 'text-davinci-002-render-sha' : 'text-davinci-002-render-sha', + parent_message_id: parentMessageId + } + let option = { + method: 'POST', + body: JSON.stringify(body), + signal: abortController?.signal, + headers: { + accept: 'text/event-stream', + 'x-openai-assistant-app-id': '', + authorization: `Bearer ${this._accessToken}`, + 'content-type': 'application/json', + referer: 'https://chat.openai.com/chat' + }, + referrer: 'https://chat.openai.com/chat' + } + if (Config.proxy) { + option.agent = new HttpsProxyAgent(Config.proxy) + } + const res = await fetch(url, option) + const decoder = new TextDecoder('utf-8') + const bodyBytes = await res.arrayBuffer() + const bodyText = decoder.decode(bodyBytes) + const events = bodyText.split('\n\n').filter(item => !_.isEmpty(item)) + let fullResponse = events[events.length - 2] + fullResponse = _.trimStart(fullResponse, 'data: ') + if (Config.debug) { + logger.debug(fullResponse) + } + fullResponse = JSON.parse(fullResponse) + if (!fullResponse.message) { + throw new Error(fullResponse.detail || 'unkown error') + } + return { + text: fullResponse.message.content.parts[0], + conversationId: fullResponse.conversation_id, + id: fullResponse.message.id, + parentMessageId + } + } +}