mirror of
https://github.com/zhayujie/bot-on-anything.git
synced 2026-02-22 08:27:58 +08:00
62 lines
2.3 KiB
Python
62 lines
2.3 KiB
Python
# encoding:utf-8
|
|
import asyncio
|
|
from model.model import Model
|
|
from config import model_conf_val
|
|
from common import log
|
|
from EdgeGPT import Chatbot, ConversationStyle
|
|
|
|
user_session = dict()
|
|
# newBing对话模型逆向网页gitAPI
|
|
|
|
class BingModel(Model):
|
|
|
|
style = ConversationStyle.creative
|
|
bot: Chatbot = None
|
|
|
|
def __init__(self):
|
|
try:
|
|
self.bot = Chatbot(cookies=model_conf_val("bing", "cookies"))
|
|
except Exception as e:
|
|
log.exception(e)
|
|
|
|
def reply(self, query: str, context=None) -> tuple[str, dict]:
|
|
bot = user_session.get(context['from_user_id'], None)
|
|
if (bot == None):
|
|
bot = self.bot
|
|
log.info("[NewBing] query={}".format(query))
|
|
task = bot.ask(query, conversation_style=self.style)
|
|
answer = asyncio.run(task)
|
|
|
|
# 最新一条回复
|
|
reply = answer["item"]["messages"][-1]
|
|
reply_text = reply["text"]
|
|
reference = ""
|
|
if "sourceAttributions" in reply:
|
|
for i, attribution in enumerate(reply["sourceAttributions"]):
|
|
display_name = attribution["providerDisplayName"]
|
|
url = attribution["seeMoreUrl"]
|
|
reference += f"{i+1}、[{display_name}]({url})\n\n"
|
|
|
|
if len(reference) > 0:
|
|
reference = "***\n"+reference
|
|
throttling = answer["item"]["throttling"]
|
|
throttling_str = ""
|
|
|
|
if throttling["numUserMessagesInConversation"] == throttling["maxNumUserMessagesInConversation"]:
|
|
self.reset_chat(context['from_user_id'])
|
|
throttling_str = "(对话轮次已达上限，本次聊天已结束，将开启新的对话)"
|
|
else:
|
|
throttling_str = f"对话轮次: {throttling['numUserMessagesInConversation']}/{throttling['maxNumUserMessagesInConversation']}\n"
|
|
|
|
response = f"{reply_text}\n{reference}\n***\n{throttling_str}"
|
|
log.info("[NewBing] reply={}", response)
|
|
user_session[context['from_user_id']] = bot
|
|
return response
|
|
else:
|
|
self.reset_chat(context['from_user_id'])
|
|
log.warn("[NewBing] reply={}", answer)
|
|
return "对话被接口拒绝，已开启新的一轮对话。"
|
|
|
|
def reset_chat(self, from_user_id):
|
|
asyncio.run(user_session.get(from_user_id, None).reset())
|