添加bing第三方绘图

This commit is contained in:
zyc404 2023-10-12 16:31:20 +08:00
parent 8dcfb54e04
commit fed2fd8dd4
3 changed files with 41 additions and 10 deletions

View file

@ -1665,6 +1665,29 @@ export class chatgpt extends plugin {
if (Config.debug) { if (Config.debug) {
logger.mark(`开始生成内容:${response.details.imageTag}`) logger.mark(`开始生成内容:${response.details.imageTag}`)
} }
if (Config.bingDrawApi) {
// 调用第三方API进行绘图
const drawOption = {
method: 'POST',
body: JSON.stringify({
prompt: response.details.imageTag,
width: 512,
height: 512
})
}
const drawData = await fetch(Config.bingDrawApi, drawOption)
if (drawData.ok) {
let draw = await cacheres.json()
if (draw.images) {
for(let image of draw.images) {
this.reply(segment.image(`base64://${image}`), true)
}
}
} else {
await e.reply('绘图失败:第三方绘图服务器错误')
}
} else {
let client = new BingDrawClient({ let client = new BingDrawClient({
baseUrl: Config.sydneyReverseProxy, baseUrl: Config.sydneyReverseProxy,
userToken: bingToken userToken: bingToken
@ -1677,6 +1700,7 @@ export class chatgpt extends plugin {
await e.reply('绘图失败:' + err) await e.reply('绘图失败:' + err)
} }
} }
}
// 如果token曾经有异常则清除异常 // 如果token曾经有异常则清除异常
let Tokens = JSON.parse((await redis.get('CHATGPT:BING_TOKENS')) || '[]') let Tokens = JSON.parse((await redis.get('CHATGPT:BING_TOKENS')) || '[]')

View file

@ -636,6 +636,12 @@
"label": "必应验证码pass服务", "label": "必应验证码pass服务",
"placeholder": "必应出验证码会自动用该服务绕过", "placeholder": "必应出验证码会自动用该服务绕过",
"data": "bingCaptchaOneShotUrl" "data": "bingCaptchaOneShotUrl"
},
{
"type": "url",
"label": "第三方绘图",
"placeholder": "Stable Diffusion API地址",
"data": "bingDrawApi"
} }
] ]
}, },

View file

@ -96,6 +96,7 @@ const defaultConfig = {
maxNumUserMessagesInConversation: 20, maxNumUserMessagesInConversation: 20,
sydneyApologyIgnored: true, sydneyApologyIgnored: true,
enforceMaster: false, enforceMaster: false,
bingDrawApi: '',
serverPort: 3321, serverPort: 3321,
serverHost: '', serverHost: '',
viewHost: '', viewHost: '',