用Python写一个调用chatgpt的程序
时间: 2023-02-27 12:29:10 浏览: 383
您可以使用OpenAI的API来调用ChatGPT。首先,您需要注册OpenAI并获得API密钥。然后,您可以使用Python的requests库发送HTTP请求到OpenAI API。
以下是一个简单的示例代码:
```
import requests
def chat_with_chatgpt(prompt):
api_key = "YOUR_API_KEY"
model = "text-davinci-002"
prompt = (f"{prompt}" + "Agent:")
headers = {
"Content-Type": "application/json",
"Authorization": f"Bearer {api_key}"
}
data = {
"model": model,
"prompt": prompt,
"max_tokens": 100,
"temperature": 0.5,
}
response = requests.post("https://api.openai.com/v1/engines/davinci/jobs", headers=headers, json=data)
if response.status_code != 200:
raise Exception("Failed to get response from API")
return response.json()["choices"][0]["text"]
print(chat_with_chatgpt("你好!"))
```
在代码中,您可以将您的API密钥替换为`YOUR_API_KEY`。然后,您可以调用`chat_with_chatgpt`函数,将您的对话提示作为参数,并得到ChatGPT的回答。
阅读全文