nonebot插件之chatgpt使用詳解

更新時(shí)間：2023年03月14日 08:46:23 作者：萌新源

這篇文章主要為大家介紹了nonebot插件之chatgpt使用詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪

前言

今天我要教大家的是 如何實(shí)現(xiàn)nonebot插件之ChatGpt

注意，本文涉及異步爬蟲，json文件讀寫等知識(shí)點(diǎn)

準(zhǔn)備

1.獲取開發(fā)者key

獲取key的地址：Account API Keys - OpenAI API

如圖所示，我已經(jīng)創(chuàng)建好一個(gè)key了，大家也可以點(diǎn)擊Create new secret key按鈕來創(chuàng)建一個(gè)新的key，注意，千萬不要泄露自己的key哦

2.魔法

在獲取key的過程我們還是需要用到魔法的，且代理必須為國(guó)外的，只要key搞到手，后續(xù)的步驟就不用用到魔法了

開始

1.找接口

之前我原本是想要教大家去對(duì)接OpenAI的官方接口的，但是想到大部分同學(xué)可能不會(huì)“魔法”，如果沒有“魔法”體驗(yàn)感會(huì)大打折扣，所以我們就要借助其他大佬幫助我們完成代理這個(gè)過程

在網(wǎng)上沖浪的時(shí)候，我發(fā)現(xiàn)了這個(gè)寶藏網(wǎng)站 GPT3.5 (cutim.top)

可以看到這個(gè)網(wǎng)站是需要我們提供key的，我這里淺淺解釋一下本次程序的主要思路

整體思路大概就是這樣

到這里，大家應(yīng)該都有key了吧，我們打開剛才的網(wǎng)站，按F12打開開發(fā)者調(diào)試工具

在這里我們可以看到請(qǐng)求的api地址： gpt.cutim.top/question

我們可以看到他是post請(qǐng)求

那么我們打開源分析，不難看出數(shù)據(jù)就是json格式，且有兩個(gè)參數(shù)，一個(gè)key，一個(gè)question

既然我們已經(jīng)找到接口了，那么下一步就是重頭戲——寫代碼了

2.快樂的敲代碼

先上猛料

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

'''
實(shí)現(xiàn)qq群聊調(diào)用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()  # 獲取用戶發(fā)送的鏈接
    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)  # 讀取數(shù)據(jù)
        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)  # 請(qǐng)求數(shù)據(jù)
            response = data.json()
            try:
                # 規(guī)避內(nèi)容獲取失敗
                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:
                # 重置會(huì)話
                person_data[user_id] = ""
                with open(file_name, "w", encoding="UTF-8") as f:
                    json.dump(person_data, f, ensure_ascii=False)
                ai_res = '很抱歉，內(nèi)容獲取失敗，請(qǐng)確認(rèn)您的問題沒有非法成分，如沒有可能是您的會(huì)話歷史已達(dá)到上限，請(qǐng)更換您的提問方式或再試一次'
    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)  # 請(qǐng)求數(shù)據(jù)
            response = data.json()
            ai_res = str(response['content']).strip("\n")
    res = f"[CQ:reply,id={msg_id}]{ai_res}"
    await chatgpt.send(Message(res))

這里我也是直接上全部代碼

接下來我挑一些我認(rèn)為比較有研究?jī)r(jià)值的代碼出來講講

    try:
        with open(file_name, "r", encoding="UTF-8") as f: #①
            person_data = json.load(f)  # 讀取數(shù)據(jù)
        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}"}

比如這一段,在①這個(gè)地方,我讀取了一個(gè)用來儲(chǔ)存用戶會(huì)話的json文件,那么可能有人會(huì)問了,為什么要讀取這樣一個(gè)文件呢?或者說這個(gè)文件有什么作用?

其實(shí)在早期版本沒有會(huì)話文件的時(shí)候,經(jīng)過群友的測(cè)試,我發(fā)現(xiàn)了一個(gè)小問題,那就是對(duì)話不連續(xù),比如說我要跟gpt玩成語接龍,但是會(huì)話是不連續(xù)的呀,于是我就找到了用一個(gè)文件儲(chǔ)存用戶會(huì)話的方法,文件結(jié)構(gòu)大概是這樣

就是把每個(gè)人的會(huì)話數(shù)據(jù)分別儲(chǔ)存起來，這樣對(duì)話就有了連續(xù)性

連續(xù)性問題是解決了，但是又產(chǎn)生了一個(gè)新的問題——會(huì)話太長(zhǎng)，gpt不知道怎么回答或者說是無法正常獲取內(nèi)容，那又怎么辦，于是我想到了下面的方法來解決這個(gè)問題

try:
    # 規(guī)避內(nèi)容獲取失敗
    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:
    # 重置會(huì)話
    person_data[user_id] = ""
    with open(file_name, "w", encoding="UTF-8") as f:
        json.dump(person_data, f, ensure_ascii=False)
    ai_res = '很抱歉，內(nèi)容獲取失敗，請(qǐng)確認(rèn)您的問題沒有非法成分，如沒有可能是您的會(huì)話歷史已達(dá)到上限，請(qǐng)更換您的提問方式或再試一次'

可以看到，我這里寫了個(gè)try語句，當(dāng)獲取內(nèi)容失敗的時(shí)候就把會(huì)話清空，并且返回一個(gè)提示信息給用戶，好讓用戶重新提問

以上就是我認(rèn)為需要講解的代碼部分了，相信大家是可以看懂我的思路的，哦對(duì)，記得替換key

更多關(guān)于nonebot插件chatgpt的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

您可能感興趣的文章: