使用python接入chatgpt api的小demo
本demo 使用的是3.5版本的
要求:需要在自己电脑安装openai。
pip install openai
没有逐字输出功能,需要的自己去开发。如果回复内容较多较慢,gpt会自动分开部分回复出来。
以下是源码。
import openai
openai.api_key = "密钥"
def ask_gpt(message):
completion = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=message
)
return completion.choices[0].message
if __name__ == '__main__':
message = []
gpt_msg = []
try:
while True:
user_input = input("我: ").strip()
if user_input == "exit" or user_input == "退出":
exit()
content = {"role": "user", "content": ""}
content["content"] = user_input
message.append(content)
gpt_res = ask_gpt(message)
# print(gpt_res)
gpt_msg = {"content": "", "role": "assistant"}
gpt_msg["role"] = gpt_res["role"]
gpt_msg["content"] = gpt_res["content"]
print("" + gpt_res["content"])
message.append(gpt_msg)
except Exception as e:
print(e)
使用python接入chatgpt api的小demo
本demo 使用的是3.5版本的
要求:需要在自己电脑安装openai。
pip install openai
没有逐字输出功能,需要的自己去开发。如果回复内容较多较慢,gpt会自动分开部分回复出来。
以下是源码。
import openai
openai.api_key = "密钥"
def ask_gpt(message):
completion = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=message
)
return completion.choices[0].message
if __name__ == '__main__':
message = []
gpt_msg = []
try:
while True:
user_input = input("我: ").strip()
if user_input == "exit" or user_input == "退出":
exit()
content = {"role": "user", "content": ""}
content["content"] = user_input
message.append(content)
gpt_res = ask_gpt(message)
# print(gpt_res)
gpt_msg = {"content": "", "role": "assistant"}
gpt_msg["role"] = gpt_res["role"]
gpt_msg["content"] = gpt_res["content"]
print("" + gpt_res["content"])
message.append(gpt_msg)
except Exception as e:
print(e)