GPT-4 初体验
Last updated
# call.py
import os
import openai
openai.api_key = os.getenv("OPENAI_API_KEY") # 在环境变量中配置 OPENAI_API_KEY=key
# 将 key 替换成在 openai 中获取到的实际值
def send_message(message):
response = openai.ChatCompletion.create(
model="gpt-4",
messages=[
{"role": "system", "content": "You are a helpful assistant."},
*previous_messages,
{"role": "user", "content": message},
]
)
assistant_response = response.choices[0].message["content"]
previous_messages.extend([
{"role": "user", "content": message},
{"role": "assistant", "content": assistant_response},
])
return assistant_response
previous_messages = []
while True:
print("User: \n---------------")
user_input = ""
for line in iter(input, ''): # 为解决多行输入的问题,读者需要在确认输入时
# 多按一次 Enter
user_input += line + " "
assistant_response = send_message(user_input)
assistant_response_lines = assistant_response.split('\n')
print("---------------\nGPT-4: \n")
for line in assistant_response_lines:
print(line)
print("---------------\n\n")