import botpy from botpy.message import Message, DirectMessage from botpy.types.message import Reference import re import json import threading import asyncio import time import requests import util.unfit_words as uw import os import sys from cores.qqbot.personality import personalities from addons.baidu_aip_judge import BaiduJudge from model.platform.qqchan import QQChan, NakuruGuildMember, NakuruGuildMessage from model.platform.qq import QQ from nakuru import ( CQHTTP, GroupMessage, GroupMemberIncrease, FriendMessage, GuildMessage, Notify ) from nakuru.entities.components import Plain,At,Image from model.provider.provider import Provider from model.command.command import Command from model.command.command_rev_chatgpt import CommandRevChatGPT from model.command.command_rev_edgegpt import CommandRevEdgeGPT from model.command.command_openai_official import CommandOpenAIOfficial from util import general_utils as gu from util.cmd_config import CmdConfig as cc import util.gplugin as gplugin from PIL import Image as PILImage import io import traceback # QQBotClient实例 client = '' # 缓存的会话 session_dict = {} # 最大缓存token（在配置里改 configs/config.yaml） max_tokens = 2000 # 配置信息 config = {} # 统计信息 count = {} # 统计信息 stat_file = '' # 是否独立会话默认值 uniqueSession = False # 日志记录 # logf = open('log.log', 'a+', encoding='utf-8') # 用户发言频率 user_frequency = {} # 时间默认值 frequency_time = 60 # 计数默认值 frequency_count = 2 # 公告（可自定义）： announcement = "" # 机器人私聊模式 direct_message_mode = True # 适配pyinstaller abs_path = os.path.dirname(os.path.realpath(sys.argv[0])) + '/' # 版本 version = '3.0.3' # 语言模型 REV_CHATGPT = 'rev_chatgpt' OPENAI_OFFICIAL = 'openai_official' REV_ERNIE = 'rev_ernie' REV_EDGEGPT = 'rev_edgegpt' provider = None chosen_provider = None # 语言模型对象 # rev_chatgpt = None # rev_edgegpt = None # chatgpt = None llm_instance: dict[str, Provider] = {} llm_command_instance: dict[str, Command] = {} # gpt配置信息 gpt_config = {} # 百度内容审核实例 baidu_judge = None # 回复前缀 reply_prefix = {} # 关键词回复 keywords = {} # QQ频道机器人 qqchannel_bot: QQChan = None PLATFORM_QQCHAN = 'qqchan' qqchan_loop = None # 配置 cc.init_attributes(["qq_forward_threshold"], 200) cc.init_attributes(["qq_welcome"], "欢迎加入本群！\n欢迎给https://github.com/Soulter/QQChannelChatGPT项目一个Star😊~\n输入help查看帮助~\n") cc.init_attributes(["bing_proxy"], "") cc.init_attributes(["qq_pic_mode"], False) cc.init_attributes(["rev_chatgpt_model"], "") cc.init_attributes(["rev_chatgpt_plugin_ids"], []) cc.init_attributes(["rev_chatgpt_PUID"], "") cc.init_attributes(["rev_chatgpt_unverified_plugin_domains"], []) cc.init_attributes(["gocq_host"], "127.0.0.1") cc.init_attributes(["gocq_http_port"], 5700) cc.init_attributes(["gocq_websocket_port"], 6700) cc.init_attributes(["gocq_react_group"], True) cc.init_attributes(["gocq_react_guild"], True) cc.init_attributes(["gocq_react_friend"], True) cc.init_attributes(["gocq_react_group_increase"], True) cc.init_attributes(["gocq_qqchan_admin"], "") cc.init_attributes(["other_admins"], []) cc.init_attributes(["CHATGPT_BASE_URL"], "") # cc.init_attributes(["qq_forward_mode"], False) # QQ机器人 gocq_bot = None PLATFORM_GOCQ = 'gocq' gocq_app = CQHTTP( host=cc.get("gocq_host", "127.0.0.1"), port=cc.get("gocq_websocket_port", 6700), http_port=cc.get("gocq_http_port", 5700), ) admin_qq = "123456" admin_qqchan = None gocq_loop = None nick_qq = None bing_cache_loop = None # 插件 cached_plugins = {} # 全局对象 _global_object = {} # 统计 cnt_total = 0 def new_sub_thread(func, args=()): thread = threading.Thread(target=func, args=args, daemon=True) thread.start() # 写入统计信息 def toggle_count(at: bool, message): global stat_file try: if str(message.guild_id) not in count: count[str(message.guild_id)] = { 'count': 1, 'direct_count': 1, } else: count[str(message.guild_id)]['count'] += 1 if not at: count[str(message.guild_id)]['direct_count'] += 1 stat_file = open(abs_path+"configs/stat", 'w', encoding='utf-8') stat_file.write(json.dumps(count)) stat_file.flush() stat_file.close() except BaseException: pass # 上传统计信息并检查更新 def upload(): global version, session_dict, gocq_bot, qqchannel_bot, cnt_total while True: addr = '' addr_ip = '' session_dict_dump = '{}' try: addr = requests.get('http://myip.ipip.net', timeout=5).text addr_ip = re.findall(r'\d+.\d+.\d+.\d+', addr)[0] json.dumps(session_dict) except BaseException as e: pass try: gocq_cnt = 0 qqchan_cnt = 0 if gocq_bot is not None: gocq_cnt = gocq_bot.get_cnt() if qqchannel_bot is not None: qqchan_cnt = qqchannel_bot.get_cnt() o = {"cnt_total": cnt_total,"admin": admin_qq,"addr": addr, 's': session_dict_dump} o_j = json.dumps(o) res = {"version": version, "count": gocq_cnt+qqchan_cnt, "ip": addr_ip, "others": o_j, "cntqc": qqchan_cnt, "cntgc": gocq_cnt} gu.log(res, gu.LEVEL_DEBUG, tag="Upload", fg = gu.FG_COLORS['yellow'], bg=gu.BG_COLORS['black']) resp = requests.post('https://api.soulter.top/upload', data=json.dumps(res), timeout=5) # print(resp.text) if resp.status_code == 200: ok = resp.json() if ok['status'] == 'ok': cnt_total = 0 if gocq_bot is not None: gocq_cnt = gocq_bot.set_cnt(0) if qqchannel_bot is not None: qqchan_cnt = qqchannel_bot.set_cnt(0) except BaseException as e: gu.log("上传统计信息时出现错误: " + str(e), gu.LEVEL_ERROR, tag="Upload") pass time.sleep(10*60) ''' 初始化机器人 ''' def initBot(cfg, prov): global llm_instance, llm_command_instance global provider, baidu_judge, chosen_provider global reply_prefix, gpt_config, config, uniqueSession, frequency_count, frequency_time, announcement, direct_message_mode, version global reply_prefix, keywords, cached_plugins, _global_object provider = prov config = cfg if 'reply_prefix' in cfg: reply_prefix = cfg['reply_prefix'] # 语言模型提供商 gu.log("--------加载语言模型--------", gu.LEVEL_INFO, fg=gu.FG_COLORS['yellow']) if REV_CHATGPT in prov: gu.log("- 逆向ChatGPT库 -", gu.LEVEL_INFO) if cfg['rev_ChatGPT']['enable']: if 'account' in cfg['rev_ChatGPT']: from model.provider.provider_rev_chatgpt import ProviderRevChatGPT llm_instance[REV_CHATGPT] = ProviderRevChatGPT(cfg['rev_ChatGPT'], base_url=cc.get("CHATGPT_BASE_URL", None)) llm_command_instance[REV_CHATGPT] = CommandRevChatGPT(llm_instance[REV_CHATGPT], _global_object) chosen_provider = REV_CHATGPT else: input("[System-err] 请退出本程序, 然后在配置文件中填写rev_ChatGPT相关配置") if REV_EDGEGPT in prov: gu.log("- New Bing -", gu.LEVEL_INFO) if not os.path.exists('./cookies.json'): input("[System-err] 导入Bing模型时发生错误, 没有找到cookies文件或者cookies文件放置位置错误。windows启动器启动的用户请把cookies.json文件放到和启动器相同的目录下。\n如何获取请看https://github.com/Soulter/QQChannelChatGPT仓库介绍。") else: if cfg['rev_edgegpt']['enable']: try: from model.provider.provider_rev_edgegpt import ProviderRevEdgeGPT llm_instance[REV_EDGEGPT] = ProviderRevEdgeGPT() llm_command_instance[REV_EDGEGPT] = CommandRevEdgeGPT(llm_instance[REV_EDGEGPT], _global_object) chosen_provider = REV_EDGEGPT except BaseException as e: gu.log("加载Bing模型时发生错误, 请检查1. cookies文件是否正确放置 2. 是否设置了代理（梯子）。", gu.LEVEL_ERROR, max_len=60) if OPENAI_OFFICIAL in prov: gu.log("- OpenAI官方 -", gu.LEVEL_INFO) if cfg['openai']['key'] is not None: from model.provider.provider_openai_official import ProviderOpenAIOfficial llm_instance[OPENAI_OFFICIAL] = ProviderOpenAIOfficial(cfg['openai']) llm_command_instance[OPENAI_OFFICIAL] = CommandOpenAIOfficial(llm_instance[OPENAI_OFFICIAL], _global_object) chosen_provider = OPENAI_OFFICIAL gu.log("--------加载个性化配置--------", gu.LEVEL_INFO, fg=gu.FG_COLORS['yellow']) # 得到关键词 if os.path.exists("keyword.json"): with open("keyword.json", 'r', encoding='utf-8') as f: keywords = json.load(f) # 检查provider设置偏好 p = cc.get("chosen_provider", None) if p is not None and p in llm_instance: chosen_provider = p gu.log(f"将使用{chosen_provider}语言模型。", gu.LEVEL_INFO) # 百度内容审核 if 'baidu_aip' in cfg and 'enable' in cfg['baidu_aip'] and cfg['baidu_aip']['enable']: try: baidu_judge = BaiduJudge(cfg['baidu_aip']) gu.log("百度内容审核初始化成功", gu.LEVEL_INFO) except BaseException as e: gu.log("百度内容审核初始化失败", gu.LEVEL_ERROR) threading.Thread(target=upload, daemon=True).start() # 得到私聊模式配置 if 'direct_message_mode' in cfg: direct_message_mode = cfg['direct_message_mode'] gu.log("私聊功能: "+str(direct_message_mode), gu.LEVEL_INFO) # 得到发言频率配置 if 'limit' in cfg: gu.log("发言频率配置: "+str(cfg['limit']), gu.LEVEL_INFO) if 'count' in cfg['limit']: frequency_count = cfg['limit']['count'] if 'time' in cfg['limit']: frequency_time = cfg['limit']['time'] # 得到公告配置 if 'notice' in cfg: if cc.get("qq_welcome", None) != None and cfg['notice'] == '此机器人由Github项目QQChannelChatGPT驱动。': announcement = cc.get("qq_welcome", None) else: announcement = cfg['notice'] gu.log("公告配置: " + announcement, gu.LEVEL_INFO) try: if 'uniqueSessionMode' in cfg and cfg['uniqueSessionMode']: uniqueSession = True else: uniqueSession = False gu.log("独立会话: "+str(uniqueSession), gu.LEVEL_INFO) except BaseException: pass gu.log(f"QQ开放平台AppID: {cfg['qqbot']['appid']} 令牌: {cfg['qqbot']['token']}") if chosen_provider is None: gu.log("检测到没有启动任何一个语言模型。请至少在配置文件中启用一个语言模型。", gu.LEVEL_CRITICAL) global nick_qq nick_qq = cc.get('nick_qq', nick_qq) thread_inst = None gu.log("--------加载插件--------", gu.LEVEL_INFO, fg=gu.FG_COLORS['yellow']) # 加载插件 _command = Command(None, None) ok, err = _command.plugin_reload(cached_plugins) if ok: gu.log("加载插件完成", gu.LEVEL_INFO) else: gu.log(err, gu.LEVEL_ERROR) gu.log("--------加载机器人平台--------", gu.LEVEL_INFO, fg=gu.FG_COLORS['yellow']) global admin_qq, admin_qqchan admin_qq = cc.get('admin_qq', None) admin_qqchan = cc.get('admin_qqchan', None) if admin_qq == None: gu.log("未设置管理者QQ号(管理者才能使用update/plugin等指令)", gu.LEVEL_WARNING) admin_qq = input("请输入管理者QQ号(必须设置): ") gu.log("管理者QQ号设置为: " + admin_qq, gu.LEVEL_INFO, fg=gu.FG_COLORS['yellow']) cc.put('admin_qq', admin_qq) if admin_qqchan == None: gu.log("未设置管理者QQ频道用户号(管理者才能使用update/plugin等指令)", gu.LEVEL_WARNING) admin_qqchan = input("请输入管理者频道用户号(不是QQ号, 可以先回车跳过然后在频道发送指令!myid获取): ") if admin_qqchan == "": gu.log("跳过设置管理者频道用户号", gu.LEVEL_INFO, fg=gu.FG_COLORS['yellow']) else: gu.log("管理者频道用户号设置为: " + admin_qqchan, gu.LEVEL_INFO, fg=gu.FG_COLORS['yellow']) cc.put('admin_qqchan', admin_qqchan) gu.log("管理者QQ: " + admin_qq, gu.LEVEL_INFO) gu.log("管理者频道用户号: " + admin_qqchan, gu.LEVEL_INFO) # GOCQ global gocq_bot if 'gocqbot' in cfg and cfg['gocqbot']['enable']: gu.log("- 启用QQ机器人 -", gu.LEVEL_INFO) global gocq_app, gocq_loop gocq_loop = asyncio.new_event_loop() gocq_bot = QQ(True, cc, gocq_loop) thread_inst = threading.Thread(target=run_gocq_bot, args=(gocq_loop, gocq_bot, gocq_app), daemon=False) thread_inst.start() else: gocq_bot = QQ(False) gu.log("机器人部署教程: https://github.com/Soulter/QQChannelChatGPT/wiki/", gu.LEVEL_INFO, fg=gu.FG_COLORS['yellow']) gu.log("如果有任何问题, 请在 https://github.com/Soulter/QQChannelChatGPT 上提交issue或加群322154837", gu.LEVEL_INFO, fg=gu.FG_COLORS['yellow']) gu.log("请给 https://github.com/Soulter/QQChannelChatGPT 点个star!", gu.LEVEL_INFO, fg=gu.FG_COLORS['yellow']) # QQ频道 if 'qqbot' in cfg and cfg['qqbot']['enable']: gu.log("- 启用QQ频道机器人(旧版) -", gu.LEVEL_INFO) global qqchannel_bot, qqchan_loop qqchannel_bot = QQChan() qqchan_loop = asyncio.new_event_loop() thread_inst = threading.Thread(target=run_qqchan_bot, args=(cfg, qqchan_loop, qqchannel_bot), daemon=False) thread_inst.start() # thread.join() if thread_inst == None: input("[System-Error] 没有启用/成功启用任何机器人，程序退出") exit() thread_inst.join() def run_qqchan_bot(cfg, loop, qqchannel_bot): asyncio.set_event_loop(loop) intents = botpy.Intents(public_guild_messages=True, direct_message=True) global client client = botClient( intents=intents, bot_log=False ) try: qqchannel_bot.run_bot(client, cfg['qqbot']['appid'], cfg['qqbot']['token']) except BaseException as e: gu.log("启动QQ频道机器人时出现错误, 原因如下: " + str(e), gu.LEVEL_CRITICAL, tag="QQ频道") gu.log(r"如果您是初次启动，请修改配置文件（QQChannelChatGPT/config.yaml）详情请看：https://github.com/Soulter/QQChannelChatGPT/wiki。" + str(e), gu.LEVEL_CRITICAL, tag="System") i = input("输入y打开配置文件, 按回车退出程序。") if i == "y": abs_path = os.path.abspath("QQChannelChatGPT/configs/config.yaml") print("配置文件地址：" + abs_path) os.system(f"notepad \"{abs_path}\"") # gu.log("如果你使用了go-cqhttp, 则可以忽略上面的报错。" + str(e), gu.LEVEL_CRITICAL, tag="QQ频道") # input(f"\n[System-Error] 启动QQ频道机器人时出现错误，原因如下：{e}\n可能是没有填写QQBOT appid和token？请在config中完善你的appid和token\n配置教程：https://soulter.top/posts/qpdg.html\n") def run_gocq_bot(loop, gocq_bot, gocq_app): asyncio.set_event_loop(loop) gu.log("正在检查本地GO-CQHTTP连接...端口5700, 6700", tag="QQ") while True: if not gu.port_checker(5700, cc.get("gocq_host", "127.0.0.1")) or not gu.port_checker(6700, cc.get("gocq_host", "127.0.0.1")): gu.log("与GO-CQHTTP通信失败, 请检查GO-CQHTTP是否启动并正确配置。5秒后自动重试。", gu.LEVEL_CRITICAL, tag="QQ") time.sleep(5) else: gu.log("检查完毕，未发现问题。", tag="QQ") break global gocq_client gocq_client = gocqClient() try: gocq_bot.run_bot(gocq_app) except BaseException as e: input("启动QQ机器人出现错误"+str(e)) ''' 检查发言频率 ''' def check_frequency(id) -> bool: ts = int(time.time()) if id in user_frequency: if ts-user_frequency[id]['time'] > frequency_time: user_frequency[id]['time'] = ts user_frequency[id]['count'] = 1 return True else: if user_frequency[id]['count'] >= frequency_count: return False else: user_frequency[id]['count']+=1 return True else: t = {'time':ts,'count':1} user_frequency[id] = t return True ''' 通用回复方法 ''' def send_message(platform, message, res, msg_ref = None, session_id = None): global qqchannel_bot, qqchannel_bot, gocq_loop, session_dict if session_id is not None: if session_id not in session_dict: session_dict[session_id] = { 'cnt': 1, } else: session_dict[session_id]['cnt'] += 1 else: session_dict[session_id]['cnt'] += 1 if platform == PLATFORM_QQCHAN: qqchannel_bot.send_qq_msg(message, res, msg_ref=msg_ref) if platform == PLATFORM_GOCQ: asyncio.run_coroutine_threadsafe(gocq_bot.send_qq_msg(message, res), gocq_loop).result() def oper_msg(message, group: bool=False, msg_ref: Reference = None, platform: str = None): """ 处理消息。 group: 群聊模式, message: 频道是频道的消息对象, QQ是nakuru-gocq的消息对象 """ global provider, session_dict, uniqueSession qq_msg = '' session_id = '' user_id = '' global chosen_provider, reply_prefix, keywords, qqchannel_bot, gocq_bot, gocq_loop, bing_cache_loop, qqchan_loop role = "member" # 角色 hit = False # 是否命中指令 command_result = () # 调用指令返回的结果 global admin_qq, admin_qqchan, cached_plugins, gocq_bot, nick_qq global _global_object, cnt_total cnt_total += 1 with_tag = False # 是否带有昵称 # 将nick_qq(昵称)统一转换为tuple if nick_qq == None: nick_qq = ("ai","!","！") if isinstance(nick_qq, str): nick_qq = (nick_qq,) if isinstance(nick_qq, list): nick_qq = tuple(nick_qq) if platform == PLATFORM_GOCQ or platform == PLATFORM_QQCHAN: _len = 0 for i in message.message: if isinstance(i, Plain): qq_msg += str(i.text).strip() if isinstance(i, At): # @机器人 if message.type == "GuildMessage": if i.qq == message.user_id or i.qq == message.self_tiny_id: with_tag = True if message.type == "FriendMessage": if i.qq == message.self_id: with_tag = True if message.type == "GroupMessage": if i.qq == message.self_id: with_tag = True for i in nick_qq: if i != '' and qq_msg.startswith(i): _len = len(i) with_tag = True break qq_msg = qq_msg[_len:].strip() gu.log(f"收到消息：{qq_msg}", gu.LEVEL_INFO, tag="QQ") user_id = message.user_id if group: # 适配GO-CQHTTP的频道功能 if message.type == "GuildMessage": session_id = message.channel_id else: session_id = message.group_id else: with_tag = True session_id = message.user_id role = "member" if message.type == "GuildMessage": sender_id = str(message.sender.tiny_id) else: sender_id = str(message.sender.user_id) if sender_id == admin_qq or \ sender_id == admin_qqchan or \ sender_id in cc.get("other_admins", []) or \ sender_id == cc.get("gocq_qqchan_admin", ""): # gu.log("检测到管理员身份", gu.LEVEL_INFO, tag="GOCQ") role = "admin" if uniqueSession: # 独立会话时，一个用户一个session session_id = sender_id if platform == PLATFORM_QQCHAN: with_tag = True if qq_msg == "": send_message(platform, message, f"Hi~", msg_ref=msg_ref, session_id=session_id) return if with_tag: # 检查发言频率 if not check_frequency(user_id): send_message(platform, message, f'你的发言超过频率限制(╯▔皿▔)╯。\n管理员设置{frequency_time}秒内只能提问{frequency_count}次。', msg_ref=msg_ref, session_id=session_id) return # logf.write("[GOCQBOT] "+ qq_msg+'\n') # logf.flush() # 关键词回复 for k in keywords: if qq_msg == k: plain_text = "" if 'plain_text' in keywords[k]: plain_text = keywords[k]['plain_text'] else: plain_text = keywords[k] image_url = "" if 'image_url' in keywords[k]: image_url = keywords[k]['image_url'] if image_url != "": res = [Plain(plain_text), Image.fromURL(image_url)] send_message(platform, message, res, msg_ref=msg_ref, session_id=session_id) else: send_message(platform, message, plain_text, msg_ref=msg_ref, session_id=session_id) return # 关键词拦截器 for i in uw.unfit_words_q: matches = re.match(i, qq_msg.strip(), re.I | re.M) if matches: send_message(platform, message, f"你的提问得到的回复未通过【自有关键词拦截】服务, 不予回复。", msg_ref=msg_ref, session_id=session_id) return if baidu_judge != None: check, msg = baidu_judge.judge(qq_msg) if not check: send_message(platform, message, f"你的提问得到的回复未通过【百度AI内容审核】服务, 不予回复。\n\n{msg}", msg_ref=msg_ref, session_id=session_id) return # 检查是否是更换语言模型的请求 temp_switch = "" if qq_msg.startswith('/bing') or qq_msg.startswith('/gpt') or qq_msg.startswith('/revgpt'): target = chosen_provider if qq_msg.startswith('/bing'): target = REV_EDGEGPT elif qq_msg.startswith('/gpt'): target = OPENAI_OFFICIAL elif qq_msg.startswith('/revgpt'): target = REV_CHATGPT l = qq_msg.split(' ') if len(l) > 1 and l[1] != "": # 临时对话模式，先记录下之前的语言模型，回答完毕后再切回 temp_switch = chosen_provider chosen_provider = target qq_msg = l[1] else: # if role != "admin": # send_message(platform, message, "你没有权限更换语言模型。", msg_ref=msg_ref, session_id=session_id) # return chosen_provider = target cc.put("chosen_provider", chosen_provider) send_message(platform, message, f"已切换至【{chosen_provider}】", msg_ref=msg_ref, session_id=session_id) return chatgpt_res = "" if session_id in gocq_bot.waiting and gocq_bot.waiting[session_id] == '': gocq_bot.waiting[session_id] = qq_msg return hit, command_result = llm_command_instance[chosen_provider].check_command( qq_msg, session_id, bing_cache_loop, role, platform, message, cached_plugins, gocq_bot, ) # 指令触发情况 if not hit: if not with_tag: return if chosen_provider == None: send_message(platform, message, f"管理员未启动任何语言模型或者语言模型初始化时失败。", msg_ref=msg_ref, session_id=session_id) return try: if chosen_provider == REV_CHATGPT or chosen_provider == OPENAI_OFFICIAL: if _global_object != None and "web_search" in _global_object and _global_object["web_search"]: chatgpt_res = gplugin.web_search(qq_msg, llm_instance[chosen_provider], session_id) else: chatgpt_res = str(llm_instance[chosen_provider].text_chat(qq_msg, session_id)) elif chosen_provider == REV_EDGEGPT: res, res_code = asyncio.run_coroutine_threadsafe(llm_instance[chosen_provider].text_chat(qq_msg, platform), bing_cache_loop).result() if res_code == 0: # bing不想继续话题，重置会话后重试。 send_message(platform, message, "Bing不想继续话题了, 正在自动重置会话并重试。", msg_ref=msg_ref, session_id=session_id) asyncio.run_coroutine_threadsafe(llm_instance[chosen_provider].forget(), bing_cache_loop).result() res, res_code = asyncio.run_coroutine_threadsafe(llm_instance[chosen_provider].text_chat(qq_msg, platform), bing_cache_loop).result() if res_code == 0: # bing还是不想继续话题，大概率说明提问有问题。 asyncio.run_coroutine_threadsafe(llm_instance[chosen_provider].forget(), bing_cache_loop).result() send_message(platform, message, "Bing仍然不想继续话题, 会话已重置, 请检查您的提问后重试。", msg_ref=msg_ref, session_id=session_id) res = "" chatgpt_res = str(res) if chosen_provider in reply_prefix: chatgpt_res = reply_prefix[chosen_provider] + chatgpt_res except BaseException as e: gu.log(f"调用异常：{traceback.format_exc()}", gu.LEVEL_ERROR, max_len=100000) gu.log("调用语言模型例程时出现异常。原因: "+str(e), gu.LEVEL_ERROR) send_message(platform, message, "调用语言模型例程时出现异常。原因: "+str(e), msg_ref=msg_ref, session_id=session_id) return # 切换回原来的语言模型 if temp_switch != "": chosen_provider = temp_switch # 指令回复 if hit: # 检查指令. command_result是一个元组：(指令调用是否成功, 指令返回的文本结果, 指令类型) if command_result == None: # send_message(platform, message, "指令调用未返回任何信息。", msg_ref=msg_ref, session_id=session_id) return command = command_result[2] if command == "keyword": if os.path.exists("keyword.json"): with open("keyword.json", "r", encoding="utf-8") as f: keywords = json.load(f) else: try: send_message(platform, message, command_result[1], msg_ref=msg_ref, session_id=session_id) except BaseException as e: send_message(platform, message, f"回复消息出错: {str(e)}", msg_ref=msg_ref, session_id=session_id) # 昵称 if command == "nick": nick_qq = cc.get("nick_qq", nick_qq) if command == "update latest r": send_message(platform, message, command_result[1] + "\n\n即将自动重启。", msg_ref=msg_ref, session_id=session_id) py = sys.executable os.execl(py, py, *sys.argv) if not command_result[0]: send_message(platform, message, f"指令调用错误: \n{str(command_result[1])}", msg_ref=msg_ref, session_id=session_id) return # 画图指令 if isinstance(command_result[1], list) and len(command_result) == 3 and command == 'draw': for i in command_result[1]: # i is a link # 保存到本地 pic_res = requests.get(i, stream = True) if pic_res.status_code == 200: image = PILImage.open(io.BytesIO(pic_res.content)) send_message(platform, message, [Image.fromFileSystem(gu.save_temp_img(image))], msg_ref=msg_ref, session_id=session_id) # 其他指令 else: try: send_message(platform, message, command_result[1], msg_ref=msg_ref, session_id=session_id) except BaseException as e: send_message(platform, message, f"回复消息出错: {str(e)}", msg_ref=msg_ref, session_id=session_id) return # 记录日志 # logf.write(f"{reply_prefix} {str(chatgpt_res)}\n") # logf.flush() # 敏感过滤 # 过滤不合适的词 for i in uw.unfit_words: chatgpt_res = re.sub(i, "***", chatgpt_res) # 百度内容审核服务二次审核 if baidu_judge != None: check, msg = baidu_judge.judge(chatgpt_res) if not check: send_message(platform, message, f"你的提问得到的回复【百度内容审核】未通过，不予回复。\n\n{msg}", msg_ref=msg_ref, session_id=session_id) return # 发送信息 try: send_message(platform, message, chatgpt_res, msg_ref=msg_ref, session_id=session_id) except BaseException as e: gu.log("回复消息错误: \n"+str(e), gu.LEVEL_ERROR) # QQ频道机器人 class botClient(botpy.Client): # 收到频道消息 async def on_at_message_create(self, message: Message): toggle_count(at=True, message=message) gu.log(str(message), gu.LEVEL_DEBUG, max_len=9999) # 转换层 nakuru_guild_message = qqchannel_bot.gocq_compatible_receive(message) gu.log(f"转换后: {str(nakuru_guild_message)}", gu.LEVEL_DEBUG, max_len=9999) message_reference = Reference(message_id=message.id, ignore_get_message_error=False) new_sub_thread(oper_msg, (nakuru_guild_message, True, message_reference, PLATFORM_QQCHAN)) # 收到私聊消息 async def on_direct_message_create(self, message: DirectMessage): if direct_message_mode: # 转换层 nakuru_guild_message = qqchannel_bot.gocq_compatible_receive(message) gu.log(f"转换后: {str(nakuru_guild_message)}", gu.LEVEL_DEBUG, max_len=9999) toggle_count(at=False, message=message) new_sub_thread(oper_msg, (nakuru_guild_message, False, None, PLATFORM_QQCHAN)) # QQ机器人 class gocqClient(): # 收到群聊消息 @gocq_app.receiver("GroupMessage") async def _(app: CQHTTP, source: GroupMessage): if cc.get("gocq_react_group", True): if isinstance(source.message[0], Plain): new_sub_thread(oper_msg, (source, True, None, PLATFORM_GOCQ)) if isinstance(source.message[0], At): if source.message[0].qq == source.self_id: new_sub_thread(oper_msg, (source, True, None, PLATFORM_GOCQ)) else: return @gocq_app.receiver("FriendMessage") async def _(app: CQHTTP, source: FriendMessage): if cc.get("gocq_react_friend", True): if isinstance(source.message[0], Plain): new_sub_thread(oper_msg, (source, False, None, PLATFORM_GOCQ)) else: return @gocq_app.receiver("GroupMemberIncrease") async def _(app: CQHTTP, source: GroupMemberIncrease): if cc.get("gocq_react_group_increase", True): global nick_qq, announcement await app.sendGroupMessage(source.group_id, [ Plain(text = announcement), ]) @gocq_app.receiver("Notify") async def _(app: CQHTTP, source: Notify): print(source) if source.sub_type == "poke" and source.target_id == source.self_id: new_sub_thread(oper_msg, (source, False, None, PLATFORM_GOCQ)) @gocq_app.receiver("GuildMessage") async def _(app: CQHTTP, source: GuildMessage): if cc.get("gocq_react_guild", True): if isinstance(source.message[0], Plain): new_sub_thread(oper_msg, (source, True, None, PLATFORM_GOCQ)) if isinstance(source.message[0], At): if source.message[0].qq == source.self_tiny_id: new_sub_thread(oper_msg, (source, True, None, PLATFORM_GOCQ)) else: return