Python調(diào)用openai API實現(xiàn)多輪對話

如何實現(xiàn)多輪對話？

gpt-3.5-turbo 模型調(diào)用方法 openai.ChatCompletion.create 里傳入的 message 是一個列表，列表里每個元素是字典，包含了角色和內(nèi)容，我們只需將每輪對話都存儲起來，然后每次提問都帶上之前的問題和回答即可。

最終效果

實現(xiàn)代碼

import openai
import json
import os

os.environ["HTTP_PROXY"] = "http://127.0.0.1:20171"
os.environ["HTTPS_PROXY"] = "http://127.0.0.1:20171"

# 獲取 api
def get_api_key():
    # 可以自己根據(jù)自己實際情況實現(xiàn)
    # 以我為例子，我是存在一個 openai_key 文件里，json 格式
    '''
    {"api": "你的 api keys"}
    '''
    openai_key_file = './envs/openai_key'
    with open(openai_key_file, 'r', encoding='utf-8') as f:
        openai_key = json.loads(f.read())
    return openai_key['api']
openai.api_key = get_api_key() 


class ChatGPT:
    def __init__(self, user):
        self.user = user
        self.messages = [{"role": "system", "content": "一個有10年P(guān)ython開發(fā)經(jīng)驗的資深算法工程師"}]
        self.filename="./user_messages.json"

    def ask_gpt(self):
        # q = "用python實現(xiàn)：提示手動輸入3個不同的3位數(shù)區(qū)間，輸入結(jié)束后計算這3個區(qū)間的交集，并輸出結(jié)果區(qū)間"
        rsp = openai.ChatCompletion.create(
          model="gpt-3.5-turbo",
          messages=self.messages
        )
        return rsp.get("choices")[0]["message"]["content"]

    def writeTojson(self):
        try:
            # 判斷文件是否存在
            if not os.path.exists(self.filename):
                with open(self.filename, "w") as f:
                    # 創(chuàng)建文件
                    pass
            # 讀取
            with open(self.filename, 'r', encoding='utf-8') as f:
                content = f.read()
                msgs = json.loads(content) if len(content) > 0 else {}
            # 追加
            msgs.update({self.user : self.messages})
            # 寫入
            with open(self.filename, 'w', encoding='utf-8') as f:
                json.dump(msgs, f)
        except Exception as e:
            print(f"錯誤代碼：{e}")           

def main():
    user = input("請輸入用戶名稱: ")
    chat = ChatGPT(user)
    
    # 循環(huán)
    while 1:
        # 限制對話次數(shù)
        if len(chat.messages) >= 11:
            print("******************************")
            print("*********強制重置對話**********")
            print("******************************")
            # 寫入之前信息
            chat.writeTojson()
            user = input("請輸入用戶名稱: ")
            chat = ChatGPT(user)
            
        # 提問
        q = input(f"【{chat.user}】")
        
        # 邏輯判斷
        if q == "0":
            print("*********退出程序**********")
            # 寫入之前信息
            chat.writeTojson()
            break
        elif q == "1":
            print("**************************")
            print("*********重置對話**********")
            print("**************************")
            # 寫入之前信息
            chat.writeTojson()
            user = input("請輸入用戶名稱: ")
            chat = ChatGPT(user)
            continue
            
        # 提問-回答-記錄
        chat.messages.append({"role": "user", "content": q})
        answer = chat.ask_gpt()
        print(f"【ChatGPT】{answer}")
        chat.messages.append({"role": "assistant", "content": answer})

if __name__ == '__main__':
    main()

代碼解析：

ChatGPT 類，包含三個函數(shù)：

• __init__初始化函數(shù)，初始化了三個個實例變量，user、messages、filename（當前用戶、消息列表、存儲記錄的文件路徑）。

• ask_gpt函數(shù)，將當前用戶所有歷史消息+最新提問發(fā)送給 gpt-3.5-turbo ，并返回響應(yīng)結(jié)果。

• writeTojson函數(shù)，結(jié)束/重置用戶時記錄當前用戶之前的訪問數(shù)據(jù)。

• main函數(shù)，程序入口函數(shù)，用戶輸入用戶名后進入與 ChatGPT 的循環(huán)對話中，輸入 0 退出程序，輸入 1 重置用戶，退出和重置都會將當前用戶之前訪問數(shù)據(jù)記錄搭配 json 文件中。

• 由于 gpt-3.5-turbo 單次請求最大 token 數(shù)為：4096，所以代碼里限制了下對話次數(shù)。文章來源地址http://www.zghlxwxcb.cn/news/detail-437728.html

更多拓展

• 你可以寫個函數(shù)，從 json 文件讀取歷史用戶訪問記錄，然后每次訪問可以選用戶。
• 你可以寫個 web 服務(wù)，使用 session 或者數(shù)據(jù)庫支持多用戶同時登錄，同時訪問。
• 你可以基于之前分享的釘釘機器人項目，將 gpt-3.5-turbo 接入釘釘機器人。

到了這里，關(guān)于【課件】Python調(diào)用OpenAI API實現(xiàn)ChatGPT多輪對話的文章就介紹完了。如果您還想了解更多內(nèi)容，請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！