用python接入chat gpt
时间: 2023-07-17 13:11:39 浏览: 125
用Chat GPT写论文的正确打开方式
要使用 Python 接入 ChatGPT,你可以使用 OpenAI 的 `openai.ChatCompletion.create()` 方法。以下是一个基本的示例代码:
```python
import openai
# 设置 OpenAI API 密钥
openai.api_key = 'YOUR_API_KEY'
# 定义对话历史
history = [
{"role": "system", "content": "You are a helpful assistant."},
{"role": "user", "content": "Who won the world series in 2020?"},
{"role": "assistant", "content": "The Los Angeles Dodgers won the World Series in 2020."},
{"role": "user", "content": "Where was it played?"}
]
# 发送 API 请求
response = openai.ChatCompletion.create(
model="gpt-3.5-turbo",
messages=history,
max_tokens=100,
n=1,
stop=None
)
# 提取回复消息
reply = response['choices'][0]['message']['content']
print("Assistant:", reply)
```
在代码中,你需要将 `YOUR_API_KEY` 替换为你的 OpenAI API 密钥。然后,你可以定义一个对话历史列表 `history`,其中包含对话的角色和内容。向 API 发送请求时,你需要指定模型名称(此示例使用了 `gpt-3.5-turbo`),并设置其他参数,如 `max_tokens`、`n` 和 `stop`。最后,你可以提取 API 响应中的回复消息,并进行打印或其他处理。
请注意,使用 ChatGPT 需要访问 OpenAI 的 GPT-3.5 Turbo 模型,可能需要付费。如果你还没有 OpenAI API 密钥或访问权限,请访问 OpenAI 官方网站了解更多信息。
阅读全文