编程学习网 > 编程语言 > Python > python调用api接口教程(Python 调用 ChatGPT-3.5-API)
2023
05-27

python调用api接口教程(Python 调用 ChatGPT-3.5-API)

手把手教你使用Python调用 ChatGPT-3.5-API。Hands teach you to use Python to call ChatGPT-3.5-API

这是 手把手教你使用Python调用 ChatGPT-3.5-API系列第二篇文章,也是第二个项目:ChatAPI ,项目功能介绍如下图所示:


先跑起来,再理解
关键代码解析
常见 Web 状态码含义
更多拓展
先跑起来,再理解
首先你需要有一个 openai 账号,如何注册我就不多说了,网上教程很多,而且很详细,如果有问题可以加我微信:pythonbrief,添加通过后请直接描述你的问题+问题截图。
访问下面页面,登录 openai 账号后,创建一个 api keys。
# api keys 创建页面
https://platform.openai.com/account/api-keys
接下来很简单了,
1> 下载项目代码:
git clone https://github.com/XksA-me/ChatGPT-3.5-API.git
# 如果慢使用下面指令(加了个国内源)
git clone https://ghproxy.com/https://github.com/XksA-me/ChatGPT-3.5-API.git     
目录结构:
├── ChatAPI      # 实现 ChatGPT API 服务,Paas应用
├── ChatCMD      # 实现控制台 ChatGPT 多轮对话应用
├── LICENSE  
├── README.md    # 项目总说明
├── blog         # 各个应用实现、使用教程
├── envs         # 存放 key 和各种配置
└── static       # 存放文档静态数据
2> 修改配置文件:

配置文件在 envs/configs,文件内容如下:

{"api": "你的 api key", "jwt_secret_key": "你的 JWT_SECRET_KEY"}
将 api 设置为你的 openai api key
jwt_secret_key用于指定签名 JWT(JSON Web Token)的密钥,可以随便设置,复杂点为好(后面涉及相关代码会再介绍)
3> 进入项目目录,安装依赖:

cd ChatGPT-3.5-API/ChatAPI
# ChatAPI 项目目录介绍
'''
.
├── README.md            # 项目说明文件
├── chatapi.py           # web 服务主程序
├── chatapi_test.ipynb   # 测试代码
├── chatgpt.py           # 请求 chatgpt 代码
├── file                 # 项目文件
│   ├── chatgpt.db       # sqlite 数据库
│   └── user.csv         # user 表测试数据
├── processing_data.py   # sqlite 数据库操作代码
├── ../envs             # 上级目录下的 envs 文件夹,存放 key
└── requirements.txt     # 项目依赖
'''
安装依赖:

# 注意现在是在:ChatGPT-3.5-API/ChatAPI 目录下
pip install requirements.txt
4> 运行代码:

初始化数据库和数据 processing_data.py
processing_data.py  包含了初始化代码,直接运行会自动在当前文件夹中的 file 文件夹下生成一个 chatgpt.db 数据库文件,包含两个表:users 和 tokens,默认初始化数据存在:./file/user.csv,可以根据需要修改~
# 初始化数据
python processing_data.py
服务端启动服务
如果是本地测试 项目运行后服务访问地址:
- 生成token : http://127.0.0.1:5000/token
- 调用api : http://127.0.0.1:5000/api
如果使用的服务器,将 chatapi.py 代码中 127.0.0.1 改成服务器 IP 地址即可。

python chatapi.py

'''
* Serving Flask app 'chatapi'
 * Debug mode: on
WARNING: This is a development server. Do not use it in a production deployment. Use a production WSGI server instead.
 * Running on all addresses (0.0.0.0)
 * Running on http://127.0.0.1:5000
 * Running on http://192.168.31.225:5000
Press CTRL+C to quit
 * Restarting with stat
 * Debugger is active!
 * Debugger PIN: 929-327-786
'''
客户端调用

获取 Token 代码
# 如果你在自己服务运行,请将 127.0.0.1 改成你自己服务器地址
import requests

# 1. 获取Token
def get_token(username, password):
    url = 'http://127.0.0.1:5000/token'
    data = {
        'u': username,
        'p': password
    }
    response = requests.post(url, json=data)
    print(response.text)
    if response.status_code in {200, 201}:
        return response.json()['message']['token']
    # 其他情况在 服务 里会处理
    return response.json()


发送 API 请求代码
# 2. 发送API请求
import requests
def call_api(token, q):
    url = 'http://127.0.0.1:5000/api'
    headers = {
        'Authorization': 'Bearer ' + token
    }
    response = requests.get(url, headers=headers, params={"q": q})
    if response.status_code == 200 and response.json()['statu']:
        return response.json()['message']
    else:
        # 其他情况在 服务 里会处理
        return response.json()

服务端日志
'''
127.0.0.1 - - [11/Mar/2023 14:42:51] "POST /token HTTP/1.1" 200 -
127.0.0.1 - - [11/Mar/2023 15:06:14] "GET /api?q=你好,python写一个合并多个+excel+xls文件的案例,每个excel表头一样 HTTP/1.1" 200 
'''
关键代码解析
使用 Flask-JWT-Extended 实现token生成和有效期管理
from flask import Flask, jsonify, request
from flask_jwt_extended import (
    JWTManager, jwt_required, create_access_token, get_jwt_identity
)
from datetime import datetime, timedelta
from chatgpt import *
from processing_data import (
    get_user, insert_token, get_token_info, update_token, day_diff, get_config
)


app = Flask(__name__)
app.config['JWT_SECRET_KEY'] = get_config()['jwt_secret_key']  # JWT_SECRET_KEY 随机只有自己知道就行,可在 ../envs/configs 里修改
app.config['JWT_ACCESS_TOKEN_EXPIRES'] = timedelta(days=30)  # Token有效期为30天
jwt = JWTManager(app)
这段代码使用了 Flask 框架和 Flask-JWT-Extended 扩展库,实现了对 RESTful API 访问的身份认证和授权控制。

其中JWT_ACCESS_TOKEN_EXPIRES为生成 token 的有效期,上面设置的是 30 天。

api 请求接口
# 设置单日用户最大请求次数
Request_Count = 10

# API接口路由
@app.route('/api', methods=['GET'])
@jwt_required()
def api():
    # 获取当前Token对应user
    user = get_jwt_identity()
    q = request.args.get('q')
    # 检查 token 对应 user 是否存在
    token_info = get_token_info(user)
    if not token_info:
        return jsonify({'statu': False, 'message': 'Token does not exist. Please contact the administrator to apply.'}), 401
    # 获取用户访问次数
    count = token_info[2]
    
    # 检查访问次数是否需要重置(每日需要重置下次数)
    # 上次请求日期与当前日期相差大于等于一天,且count > 0
    if day_diff(datetime.now().date(), token_info[4]) and count > 0:
        # 重置访问次数
        count = 0

    # 检查Token访问次数是否达到限制
    if count >= Request_Count:
        return jsonify({'statu': False, 'message': 'Token access limit reached.'}), 429
    try:
        messages = [{"role": "user", "content": q}]
        rsp = ask_gpt(messages)
    except Exception as e:
        return jsonify({'statu': False, 'message': f'An error occurred in the chatgpt request, {e}'}), 500
    # 增加Token访问次数
    count+= 1
    last_use_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
    update_token(username=token_info[0], count=count, last_use_time=last_use_time)
    # 返回API响应
    return jsonify({'statu': True, 'message': rsp}), 200

jwt_required() 装饰器用于保护需要授权才能访问的接口;
get_jwt_identity() 函数用于获取当前 Token 的身份标识;
jsonify() 函数用于将 Python 数据结构转换为 JSON 格式;


基本说明:
方法:GET
路由:/api
认证:JWT
功能:根据用户提供的问题返回机器人回答
请求参数:
q: 用户提问内容
返回结果:
statu: 返回结果状态
message: 返回信息内容
状态码:
200: 请求成功
401: Token不存在或已失效
429: Token请求次数超出限制
500: 服务器内部错误

请求流程:

用户发起 API 请求,包含 Token 和查询参数 q。
服务端获取 Token 对应的用户信息。
如果 Token 不存在,返回错误提示信息,状态码 401。
获取当前用户访问次数 count。
判断访问次数是否需要重置,如果上次请求日期与当前日期相差大于等于一天,且 count > 0,重置访问次数。
判断 Token 访问次数是否达到限制 Request_Count。
如果 Token 访问次数达到限制,返回错误提示信息,状态码 429。
发送用户查询请求给 GPT 模型。
捕获 GPT 模型请求异常,返回错误提示信息,状态码 500。
增加用户访问次数 count,更新 Token 最后使用时间 last_use_time。
返回 API 响应,状态码 200。
token 请求
# Token生成路由
@app.route('/token', methods=['POST'])
def token():
    # 获取用户名和密码
    username = request.json.get('u', None)
    password = request.json.get('p', None)
    user = get_user(username)
    # 检查用户名和密码是否正确
    if not user or password != user[2]:
        return jsonify({'statu': False, 'message': 'Invalid username or password'}), 401
    try:
        # 每个用户只能有一个token
        token_info = get_token_info(username)
        if token_info:
            return jsonify({'statu': True, 'message': {'username': username, 'token': token_info[1], 'expiration_time': token_info[3]}}), 200
        # 生成Token
        token = create_access_token(identity=username)
        # 记录到数据表
        expiration_time = (datetime.now() + timedelta(days=30)).strftime("%Y-%m-%d %H:%M:%S")
        insert_token(username, token, expiration_time)
    except Exception as e:
        return jsonify({'statu': False, 'message': f'Error in create token, Exception: {e}'}), 500
    # 返回Token
    return jsonify({'statu': True, 'message': {'username': username, 'token': token, 'expiration_time': expiration_time}}), 201

create_access_token() 函数用于生成 Token;


基本说明:
方法:POST
路由:/token
功能:用于生成 Token
请求参数:
u(string):用户名,必填。
p(string):密码,必填。
返回结果:
status(bool):请求状态,成功为 true,失败为 false。
message(object):响应信息,成功时包含以下字段:
username(string):用户名。
token(string):生成的 Token。

expiration_time(string):Token 过期时间,格式为 YYYY-MM-DD HH:MM:SS。
状态码:
200:服务器成功处理了请求,并返回了JSON格式的响应;
201:服务器成功创建了新资源,并返回了JSON格式的响应;
401:表示请求要求身份验证,客户端需要提供有效的用户名和密码;
500:服务器内部错误,无法完成请求的处理。

请求流程:

客户端向服务器发送一个POST请求,请求的URL是'/token';
服务器接收到请求后,获取请求体中的用户名和密码,并调用get_user函数查找数据库中是否存在该用户;
如果用户不存在或密码错误,服务器返回一个包含状态码401和错误信息的JSON响应;
如果用户存在且密码正确,服务器调用get_token_info函数查找该用户是否已经有一个token;
如果已经有一个token,服务器返回一个包含状态码200和该token信息的JSON响应;
如果没有token,服务器调用create_access_token函数生成一个新的token,并调用insert_token函数将该token信息插入到数据库中;
如果生成和插入token过程中发生异常,服务器返回一个包含状态码500和异常信息的JSON响应;
服务器返回一个包含状态码201和新生成token信息的JSON响应。
注: 以上两个接口请求流程是 ChatGPT 根据代码描写的。

常见 Web 状态码含义
在Web访问中,当客户端向服务器发起请求时,服务器会向客户端返回一个状态码,用于告诉客户端请求的处理情况。以下是一些常见的状态码及其含义:

200 OK:请求成功,服务器已成功处理了请求。
201 Created:请求成功并且服务器创建了新的资源。
204 No Content:请求成功,但是响应中没有实体的主体部分。
301 Moved Permanently:请求的资源已经永久移动到新的URL。
302 Found:请求的资源临时从不同的URL响应请求,但是请求者应继续使用原始URL访问资源。
304 Not Modified:请求的资源未被修改,服务器返回此状态码时,不会返回资源的主体部分。
400 Bad Request:请求无效,服务器无法识别请求的格式,常见的原因包括请求中的参数有误或者缺失。
401 Unauthorized:请求未被授权,需要客户端进行身份验证。
403 Forbidden:服务器已经理解请求,但是拒绝执行它,通常是因为客户端没有足够的权限。
404 Not Found:请求的资源不存在,服务器无法找到请求的资源。
429 Too Many Requests:这意味着客户端发送的请求过多,超过了服务器可以处理的限制。
500 Internal Server Error:服务器遇到了意外的情况,无法完成请求。
更多拓展
也许你会有疑问,为什么只有 api 和 token 接口,没有 register 注册接口,每次给新用户还得自己手动操作,太蠢了?

本应用是一个 Paas 层的 API 接口,所以没有提供 register 这种应用层面接口,下一步会出一个 ChatWeb 项目,这个项目会有用户登录、注册界面,这时候和 ChatAPI 结合起来,就是一个较为完整的 Web 应用了。(描述和认知不一定准确,如果你有不同看法,欢迎留言指正、交流)

以上就是python调用api接口教程(Python 调用 ChatGPT-3.5-API)的详细内容,想要了解更多Python教程欢迎持续关注编程学习网。

扫码二维码 获取免费视频学习资料

Python编程学习

查 看2022高级编程视频教程免费获取