olivebot/ai_module/nlp_gpt.py

from revChatGPT.V1 import Chatbot
from core.content_db import Content_Db
from utils import config_util as cfg
import time 

count = 0
def question(cont):
    global count
    try:
        chatbot = Chatbot(config={
            "access_token": cfg.key_gpt_access_token,
            "paid": False,
            "collect_analytics": True,
            "model": "gpt-4",
            "conversation_id":cfg.key_gpt_conversation_id
            },conversation_id=cfg.key_gpt_conversation_id,
            parent_id=None)

        prompt = cont
        response = ""
        for data in chatbot.ask(prompt):
            response = data["message"]
        count = 0
        return response
    except Exception as e:
        count += 1
        if count < 3:
            time.sleep(15)
            return question(cont)
        return 'gpt当前繁忙，请稍后重试' + e