python调用openai API接口
时间: 2024-11-29 16:17:48 浏览: 4
Python 调用 OpenAI API 接口通常需要通过 `requests` 库来发送 HTTP 请求,因为 OpenAI 的 API 并不是 Python 内置的一部分。以下是基本步骤:
1. **安装依赖**:首先确保已经安装了 `requests` 和可能需要的其他库(如 `python-dotenv` 来处理环境变量)。
```bash
pip install requests python-dotenv
```
2. **设置API密钥**:你需要从 OpenAI 官网获取 API 密钥并将其保存在一个安全的地方,通常会使用 `.env` 文件来管理环境变量。
3. **编写代码**:在 Python 中,你可以使用下面的代码模板来调用 API:
```python
import os
from dotenv import load_dotenv
import requests
# 加载环境变量
load_dotenv() # 如果你有.env文件
OPENAI_API_KEY = os.environ['OPENAI_API_KEY']
# 示例:发送一个文本生成请求
def call_openai_api(prompt):
url = "https://api.openai.com/v1/engines/davinci-codex/completions"
headers = {
'Content-Type': 'application/json',
'Authorization': f"Bearer {OPENAI_API_KEY}",
}
data = {
'prompt': prompt,
'max_tokens': 50, # 指定生成的字符数
'temperature': 0.7, # 温度值控制生成的随机性
'n': 1, # 只返回一个响应
}
response = requests.post(url, json=data, headers=headers)
response.raise_for_status() # 确保请求成功
return response.json()
# 使用示例
prompt = "你好,我是来自"
response = call_openai_api(prompt)
print(response['choices'][0]['text'])
```
请确保替换 `OPENAI_API_KEY` 为你实际的 API 密钥,并根据 OpenAI 文档调整 API 路径和参数。
阅读全文