本文介绍了如何使用 Python 调用 OpenAI API 进行文本生成、代码生成和图像生成。内容包括获取 API 密钥、安装官方包、以及针对 GPT-3、Codex 和 DALL-E 模型的具体代码示例。通过设置环境变量和配置参数，开发者可以快速构建 AI 应用原型，并包含错误处理与最佳实践建议。
OpenAI 提供了强大的 AI 模型接口，开发者可以通过 API 快速集成自然语言处理、代码生成及图像创作能力。本文将使用 Python 演示如何接入 OpenAI API，实现文本生成、代码生成和图像生成的基本功能。
在使用 API 之前，需要登录 OpenAI 账户并生成 API Key。
export OPENAI_API_KEY="sk-xxxxxxxxxxxxxxxxxxxxxxxx"
通过 pip 安装官方 OpenAI 包：
pip install openai
文本生成适用于情感分析、自动对话、摘要等场景。主要使用 Completion 端点。
以下示例演示如何使用 GPT-3 模型进行情感判断：
import os
import openai
openai.api_key = os.getenv("OPENAI_API_KEY")
prompt = """
Decide whether a Mike's sentiment is positive, neutral, or negative.
Mike: I don't like homework!
Sentiment:
"""
response = openai.Completion.create(
model="text-davinci-003",
prompt=prompt,
max_tokens=100,
temperature=0
)
print(response.choices[0].text.strip())
text-davinci-003。{
"choices": [
{
"finish_reason": "stop",
"index": 0,
"logprobs": null,
"text": "Negative"
}
],
...
}
利用 Codex 模型，可以将自然语言注释转化为可执行代码，或优化现有代码逻辑。
生成 Python 代码以创建上海温度数组：
import os
import openai
openai.api_key = os.getenv("OPENAI_API_KEY")
response = openai.Completion.create(
model="code-davinci-002",
prompt="\nCreate an array of weather temperatures for Shanghai\n",
temperature=0,
max_tokens=256,
top_p=1,
frequency_penalty=0,
presence_penalty=0
)
print(response.choices[0].text.strip())
import numpy as np
def create_temperatures(n):
"""
Create an array of weather temperatures for Shanghai
"""
temperatures = np.random.uniform(low=14.0, high=20.0, size=n)
return temperatures
使用 DALL-E 模型根据文本描述生成图像。细节越丰富，生成结果越符合预期。
import openai
response = openai.Image.create(
prompt="A fluffy white cat with blue eyes sitting in a basket of flowers, looking up adorably at the camera",
n=1,
size="1024x1024"
)
image_url = response['data'][0]['url']
print(image_url)
在实际开发中，网络波动或配额限制可能导致请求失败，建议添加异常处理机制。
try:
response = openai.Completion.create(...)
except openai.error.Timeout:
print("请求超时")
except openai.error.RateLimitError:
print("达到速率限制")
except Exception as e:
print(f"发生错误：{e}")
此外，注意控制 max_tokens 以节省成本，并在生产环境中严格保护 API Key，不要提交到公共代码仓库。
本文介绍了基于 Python 调用 OpenAI API 的核心流程，涵盖了文本、代码和图像三种生成模式。通过合理配置参数和处理异常，开发者可以高效构建 AI 应用原型。更多高级用法请参考官方文档。
微信公众号「极客日志」，在微信中扫描左侧二维码关注。展示文案：极客日志 zeeklog
生成新的随机RSA私钥和公钥pem证书。 在线工具，RSA密钥对生成器在线工具，online
基于 Mermaid.js 实时预览流程图、时序图等图表，支持源码编辑与即时渲染。 在线工具，Mermaid 预览与可视化编辑在线工具，online
解析常见 curl 参数并生成 fetch、axios、PHP curl 或 Python requests 示例代码。 在线工具，curl 转代码在线工具，online
将字符串编码和解码为其 Base64 格式表示形式即可。 在线工具，Base64 字符串编码/解码在线工具，online
将字符串、文件或图像转换为其 Base64 表示形式。 在线工具，Base64 文件转换器在线工具，online
将 Markdown（GFM）转为 HTML 片段，浏览器内 marked 解析；与 HTML转Markdown 互为补充。 在线工具，Markdown转HTML在线工具，online