OpenAI website
https://openai.com/
Sample code used in the demo
pip install openai
import openai
import json
import datetime
d = datetime.datetime.now()
f = '%Y%m%d%H%M%S'
strdatetime = d.strftime(f)
file_name = "history_" + strdatetime + ".json"
model_engine = "gpt-3.5-turbo"
openai.api_key = "******************************************" # Please replace with a secret key.
messages = []
open(file_name, "w")
while True:
user_input = input("\nuser: ")
if user_input == "quit":
break
user_message = {"role": "user", "content": user_input}
messages.append(user_message)
with open(file_name, "a") as f:
f.write(json.dumps(user_message, ensure_ascii=False) + "\n")
print("ChatGPT: ")
response = openai.ChatCompletion.create(
model=model_engine,
messages=messages,
)
chat_response = response["choices"][0]["message"]["content"].strip()
chat_message = {"role": "assistant", "content": chat_response}
messages.append(chat_message)
with open(file_name, "a") as f:
f.write(json.dumps(chat_message, ensure_ascii=False) + "\n")
print(chat_response)