基于python实现openai可结合上下文的问答，含html在线版

本文使用 OpenAI GPT（Generative Pre-Training）聊天机器人模型，实现可自动回复提问的聊天功能。

代码解释

首先，我们导入相关的库，例如 openai，Path，time 等。

接下来，为了使模型可以正常工作，我们需要设置 openai 的 api_key ，以及一些初始变量，如 text, turns, last_result，用来记录聊天记录。

之后，我们定义了一个函数 chatgpt，目的是为了接收用户输入的问题，并返回 GPT 模型生成的回答。函数中，除了指定使用 davinci-003 模型外，我们还设置了 temperature、max_tokens、frequency_penalty、presence_penalty 等参数，用来控制结果的随机性和字数，以达到最佳的回答效果。

最后，在 if __name__ == '__main__': 下，我们初始化两个列表，用来存放用户输入的问题和 GPT 模型自动生成的回答，然后在 while 循环中，接收用户输入的问题，并调用 chatgpt 函数，最后将问题和回答分别存储到对应的列表中，最终将内容保存到文件中。

代码使用说明

使用该代码，你需要先申请 OpenAI 的api_key，并将其输入到代码中，然后运行该程序，
输入你的问题，即可获得 GPT 模型的回答；
若输入exit则直接退出当前对话；
程序结束时，会将问答的内容记录到文件中，以便下次查看。

ini配置文件

在目录下创建config.ini文件，内容如下

[openai]ai_account_key = sk-AsqirFnBSHKvalmEe1AnT3BlbkFJe2rX0xxxxxxxxxxx

对话模式代码

点击查看代码

import openai
from pathlib import Path
import time
import configparserANSI_COLOR_GREEN = "\x1b[32m"
ANSI_COLOR_RESET = "\x1b[0m"# 从ini文件中读取api_key
config = configparser.ConfigParser()
config.read('config.ini')
openai.api_key = config['openai']['ai_account_key']text = ""  # 设置一个字符串变量
turns = []  # 设置一个列表变量，turn指对话时的话轮
last_result = ""def chatgpt(question):global textglobal turnsglobal last_resultprompt = text + "\nHuman: " + questiontry:response = openai.Completion.create(model="text-davinci-003",  # 这里我们使用的是davinci-003的模型，准确度更高。prompt=prompt,  # 你输入的问题temperature=0.9,  # 控制结果的随机性，如果希望结果更有创意可以尝试 0.9，或者希望有固定结果可以尝试 0.0max_tokens=2048,  # 这里限制的是回答的长度，你可以可以限制字数，如:写一个300字作文等。top_p=1,# [控制字符的重复度] -2.0 ~ 2.0 之间的数字，正值会根据新 tokens 在文本中的现有频率对其进行惩罚，从而降低模型逐字重复同一行的可能性frequency_penalty=0,# [控制主题的重复度] -2.0 ~ 2.0 之间的数字，正值会根据到目前为止是否出现在文本中来惩罚新 tokens，从而增加模型谈论新主题的可能性presence_penalty=0)result = response["choices"][0]["text"].strip()last_result = resultturns += [question] + [result]  # 只有这样迭代才能连续提问理解上下文if len(turns) <= 10:  # 为了防止超过字数限制程序会爆掉，所以提交的话轮语境为10次。text = " ".join(turns)else:text = " ".join(turns[-10:])return resultexcept Exception as exc:  # 捕获异常后打印出来print(exc)if __name__ == '__main__':# 将问题和回复记录下来，待结束后保存到文件中question_list = []answer_list = []while True:question = input(ANSI_COLOR_GREEN +"\n请输入问题，若输入exit退出\n" + ANSI_COLOR_RESET)question_list.append(question)if question == "exit":breakanswer = chatgpt(question)answer_list.append(answer)print("AI: " + answer)# 保存到文件中timestamp = time.strftime("%Y%m%d-%H%M-%S", time.localtime())file_name = 'output/chat ' + timestamp + '.md'f = Path(file_name)f.parent.mkdir(parents=True, exist_ok=True)with open(file_name, "w", encoding="utf-8") as f:for q, a in zip(question_list, answer_list):f.write(f"question: {q}\nanswer: {a}\n\n")print(ANSI_COLOR_GREEN + "对话内容已保存到文件中: " + file_name + ANSI_COLOR_RESET)

单问答模式代码

点击查看代码

import openai
from pathlib import Path
import time
import configparserANSI_COLOR_GREEN    = "\x1b[32m"
ANSI_COLOR_RESET    = "\x1b[0m"def get_ai_answer(prompt, save=True):# 去除字符串前后的空白符prompt = prompt.strip()# 发起请求if len(prompt) != 0:print(f'已发起请求，问题描述{len(prompt)}个长度，请稍等...')# 从ini文件中读取api_keyconfig = configparser.ConfigParser()config.read('config.ini')openai.api_key = config['openai']['ai_account_key']# Get my answerresponse = openai.Completion.create(prompt=prompt,model="text-davinci-003",temperature=0.9,max_tokens=2048,  #返回结果的长度top_p=1,frequency_penalty=0.0,presence_penalty=0.0)# Print my answer# print(response)answer = response["choices"][0]["text"].strip()print(answer)# 将内容写到以时间戳为名的md文件if save:timestamp = time.strftime("%Y%m%d-%H%M-%S", time.localtime())file_name = 'output/' + timestamp + '.md'f = Path(file_name)f.parent.mkdir(parents=True, exist_ok=True)text = f'# Q\n{prompt}\n# A\n{answer}\n'f.write_text(text, encoding='utf-8')print(ANSI_COLOR_GREEN  +"对话内容已保存到文件中: " + file_name + ANSI_COLOR_RESET)return answerif __name__ == '__main__':prompt = '''
你今年几岁了'''get_ai_answer(prompt)

gitee在线版

此外，我用html写了一个可直接对话的openai gpt3在线版，用该页面需要提前自备openai的apikey。

项目源码https://gitee.com/x223222981/chat-gpt.js