nonebot插件之chatgpt使用详解

from nonebot import on_keyword
from nonebot.typing import T_State
from nonebot.adapters.onebot.v11 import GroupMessageEvent, Bot, Message
import httpx
import json

'''
实现qq群聊调用chatgpt
write by 萌新源 at 2023/3/5
'''
chatgpt = on_keyword({"#gpt"})


@chatgpt.handle()
async def yy(bot: Bot, event: GroupMessageEvent, state: T_State):
    get_msg = str(event.message).strip()  # 获取用户发送的链接
    user_question = get_msg.strip("#gpt")
    msg_id = event.message_id  # 获取消息id
    user_id = event.user_id
    file_name = "chatgpt.json"
    try:
        with open(file_name, "r", encoding="UTF-8") as f:
            person_data = json.load(f)  # 读取数据
        try:
            user_data = person_data[str(user_id)]
        except KeyError:
            user_data = ""
        form_data = {'key': '填写你自己的key',
                     'question': user_data + f"----{user_question}"}
        async with httpx.AsyncClient() as client:
            headers = {
                'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/63.0.3239.132 Safari/537.36 QIHU 360SE'
            }
            url = f"http://gpt.cutim.top/question"
            data = await client.post(url=url, headers=headers, json=form_data, timeout=None)  # 请求数据
            response = data.json()
            try:
                # 规避内容获取失败
                res_ai = str(response['content'])
                try:
                    user_data = person_data[str(user_id)] + f"----{user_question}"
                    question = user_data + f'\n{res_ai}'
                    person_data[str(user_id)] = question
                except KeyError:
                    person_data[str(user_id)] = f'----{user_question}'
                with open(file_name, "w", encoding="UTF-8") as f:
                    json.dump(person_data, f, ensure_ascii=False)
                ai_res = str(response['content']).strip("\n")
            except KeyError:
                # 重置会话
                person_data[user_id] = ""
                with open(file_name, "w", encoding="UTF-8") as f:
                    json.dump(person_data, f, ensure_ascii=False)
                ai_res = '很抱歉，内容获取失败，请确认您的问题没有非法成分，如没有可能是您的会话历史已达到上限，请更换您的提问方式或再试一次'
    except FileNotFoundError:
        with open(file_name, "w", encoding="UTF-8") as f:
            json.dump({user_id: f'----{user_question}'}, f, ensure_ascii=False)
        form_data = {'key': '填写你自己的key', 'question': user_question}
        async with httpx.AsyncClient() as client:
            headers = {
                'User-Agent': 'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/63.0.3239.132 Safari/537.36 QIHU 360SE'
            }
            url = f"http://gpt.cutim.top/question"
            data = await client.post(url=url, headers=headers, json=form_data, timeout=None)  # 请求数据
            response = data.json()
            ai_res = str(response['content']).strip("\n")
    res = f"[CQ:reply,id={msg_id}]{ai_res}"
    await chatgpt.send(Message(res))

    try:
        with open(file_name, "r", encoding="UTF-8") as f: #①
            person_data = json.load(f)  # 读取数据
        try:
            user_data = person_data[str(user_id)]
        except KeyError:
            user_data = ""
        form_data = {'key': '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',
                     'question': user_data + f"----{user_question}"}

try:
    # 规避内容获取失败
    res_ai = str(response['content'])
    try:
        user_data = person_data[str(user_id)] + f"----{user_question}"
        question = user_data + f'\n{res_ai}'
        person_data[str(user_id)] = question
    except KeyError:
        person_data[str(user_id)] = f'----{user_question}'
    with open(file_name, "w", encoding="UTF-8") as f:
        json.dump(person_data, f, ensure_ascii=False)
    ai_res = str(response['content']).strip("\n")
except KeyError:
    # 重置会话
    person_data[user_id] = ""
    with open(file_name, "w", encoding="UTF-8") as f:
        json.dump(person_data, f, ensure_ascii=False)
    ai_res = '很抱歉，内容获取失败，请确认您的问题没有非法成分，如没有可能是您的会话历史已达到上限，请更换您的提问方式或再试一次'