from flask import Flask, request, render_template, jsonify
from revChatGPT.V1 import Chatbot
import jsonapp = Flask(__name__)
chatbot = Chatbot(config={"access_token": "你的access_token 访问https://chat.openai.com/api/auth/session 获取"
})@app.route("/chat",methods=["POST"])
def chat():if request.get_data() is None:return_dict['return_code'] = '5004'return_dict['return_info'] = '请求参数为空'return json.dumps(return_dict, ensure_ascii=False)# 获取传入的参数prompt = get_Data.get('prompt')response = chatbot_ask(prompt)return jsonify({"message": "请求成功","data": response,"status": "Success",})def chatbot_ask(prompt):response = ''for data in chatbot.ask(prompt):response = data["message"]return responseif __name__ == "__main__": app.run(host="0.0.0.0",port="8000")
简单写了一个基于flask 的chatgpt 请求接口