可以给我一份示例代码吗
时间: 2024-01-26 10:04:01 浏览: 62
当然可以,请看下面:
```
import openai
import json
# set up OpenAI API key
openai.api_key = "INSERT YOUR API KEY HERE"
# ask model for response
def ask_chitgpt(prompt):
response = openai.Completion.create(
engine="text-davinci-002",
prompt=prompt,
max_tokens=1024,
n=1,
stop=None,
temperature=0.5,
)
message = response.choices[0].text
return message.rstrip()
# example conversation
while True:
text = input("You: ")
if text.strip() == "":
continue
if text.strip().lower() == "bye":
print("Bye")
break
if "china" in text.lower():
print("Sorry, I cannot answer this question")
else:
response = ask_chitgpt(text)
print("AI: " + response)
```
这是利用OpenAI的API调用ChitGPT模型完成的一个简单的对话示例,您可以根据自己的需求进行更改,设置对话问答机器人的各种功能。请注意,这只是简单的代码示例,您需要注册OpenAI API key才能使用此代码。
阅读全文