mirror of
https://github.com/AstrBotDevs/AstrBot
synced 2026-07-20 02:55:08 +08:00
9f91b0c92b
2. 支持临时使用其他语言模型回复(如 /bing hello) #79 perf: 😊1. 优化代码结构，降低耦合度 2. 启动前检查依赖库安装情况 fix: 🤔修复bing模型死锁（正忙）的问题
53 lines
1.8 KiB
Python
53 lines
1.8 KiB
Python
from model.provider.provider import Provider
|
||
from EdgeGPT import Chatbot, ConversationStyle
|
||
import json
|
||
|
||
class ProviderRevEdgeGPT(Provider):
|
||
def __init__(self):
|
||
self.busy = False
|
||
self.wait_stack = []
|
||
with open('./cookies.json', 'r') as f:
|
||
cookies = json.load(f)
|
||
self.bot = Chatbot(cookies=cookies)
|
||
|
||
def is_busy(self):
|
||
return self.busy
|
||
|
||
async def forget(self):
|
||
try:
|
||
await self.bot.reset()
|
||
return True
|
||
except BaseException:
|
||
return False
|
||
|
||
async def text_chat(self, prompt):
|
||
if self.busy:
|
||
return
|
||
self.busy = True
|
||
resp = 'err'
|
||
err_count = 0
|
||
retry_count = 5
|
||
|
||
while err_count < retry_count:
|
||
try:
|
||
resp = await self.bot.ask(prompt=prompt, conversation_style=ConversationStyle.creative)
|
||
resp = resp['item']['messages'][len(resp['item']['messages'])-1]['text']
|
||
if 'I\'m sorry but I prefer not to continue this conversation. I\'m still learning so I appreciate your understanding and patience.' in resp:
|
||
self.busy = False
|
||
return '', 0
|
||
if resp == prompt:
|
||
resp += '\n\n如果你没有让我复述你的话，那代表我可能不想和你继续这个话题了，请输入reset重置会话😶'
|
||
break
|
||
except BaseException as e:
|
||
print(e.with_traceback)
|
||
err_count += 1
|
||
if err_count >= retry_count:
|
||
self.busy = False
|
||
raise e
|
||
print("[RevEdgeGPT] 请求出现了一些问题, 正在重试。次数"+str(err_count))
|
||
self.busy = False
|
||
|
||
print("[RevEdgeGPT] "+str(resp))
|
||
return resp, 1
|
||
|