完善一下上面的代码
时间: 2023-07-02 22:17:03 浏览: 53
好的,下面是完整的代码:
```python
import openai
import json
# 首先设置 CSDN API key
openai.api_key = "YOUR_API_KEY_HERE"
# 定义对话函数
def talk_to_ai(prompt):
# 设置请求参数
params = {
"prompt": prompt,
"temperature": 0.5,
"max_tokens": 50,
"nft": False,
"stop": "\n"
}
# 发送请求并获取 AI 的回复
response = openai.Completion.create(engine="davinci", prompt=prompt, **params)
message = response.choices[0].text.strip()
# 返回 AI 的回复
return message
# 测试对话函数
while True:
prompt = input("你:")
if prompt.lower() == "bye":
print("ChitGPT: 再见!")
break
message = talk_to_ai(prompt)
print("ChitGPT:", message)
```
其中,`YOUR_API_KEY_HERE` 部分需要替换为你的 CSDN API key。另外,我们使用了 CSDN 的 `davinci` 引擎进行对话,你也可以替换为其他引擎。在 `params` 中,我们设置了一些请求参数,例如温度、最大生成长度等,可以根据需要进行调整。最后,在测试对话函数中,我们使用一个简单的循环来不断获取用户输入并输出 AI 的回复,当用户输入 `bye` 时,程序退出。