OpenAI的API调用之初探，python调用GPT-API（交互式，支持多轮对话）

准备工作

1. 关于如何开通gpt账号、API功能、获取API的key，请大家自行百度，当大家看到如下界面，代表第一步的准备工作完成：
2. 关于python的环境安装以及OpenAI 第三方库的安装： pip list

需求场景

有两个场景，单轮对话和多轮对话。

1.prompt的输入：
能够支持从文件路径下直接读取已经准备好的prompt（通常是比较复杂的，有准备的），同时能够支持实时对话输入。

2.计算输入prompt的token，根据每个版本的gpt接口token限制给出可用的model列表

3.GPT返回的对话结果能够保存文件到本地

单轮对话的实现

1.引入需要的包

javascript代码解读复制代码import os
import tiktoken
import inquirer
import datetime
from openai import OpenAI
from common.openapi_invoke import openapi_invoke, openapi_choice_model

2.获取基础参数

ini代码解读复制代码client = OpenAI()
#计算token的方法
encoding = tiktoken.get_encoding("cl100k_base")

#取api-key，为了防止泄露，大家可以配置在环境变量中
client.api_key = os.getenv("OPENAI_API_KEY")

3.输入输出的声明

ini代码解读复制代码# 获取input目录下的所有txt文件
input_dir = "input"
txt_files = [os.path.join(input_dir, f) for f in os.listdir(input_dir) if
             os.path.isfile(os.path.join(input_dir, f)) and f.endswith('.txt')]

# 定义结果输出目录，使用时间戳
timestamp = datetime.datetime.now().strftime("%Y-%m-%d_%H-%M-%S")
output_dir = "output"
output_file = f"{timestamp}.txt"

4.prompt方式选择

ini代码解读复制代码prompt_type = input("\n请选择：\n1.选择已有prompt\n2.直接进行对话\n")
if prompt_type == "1":
    prompt = prompt_read()
else:
    prompt = input("\nprompt:\n")

5.api的调用

ini代码解读复制代码selected_model = openapi_choice_model(prompt)
dialogue = openapi_invoke(prompt, selected_model)

6.获取api的对话返回并存储

python代码解读复制代码with open(os.path.join(output_dir, output_file), 'w', encoding='utf-8') as