OpenAI API调用入门

入门·10 分钟·通过可运行代码讲解「OpenAI API调用入门」,覆盖环境准备、核心代码、参数说明、异常处理、工程化改造和练习任务,适合 AI 编程入门。

OpenAI API调用入门

OpenAI API 让你可以在自己的程序中调用大模型能力。学习 API 调用的重点不是只跑通一次,而是理解请求结构、密钥管理、错误处理和结果解析。

准备密钥

不要把 API Key 写进代码。创建 .env 文件:

OPENAI_API_KEY=你的密钥

安装依赖:

pip install openai python-dotenv

最小示例

import os
from dotenv import load_dotenv
from openai import OpenAI
 
load_dotenv()
client = OpenAI(api_key=os.getenv('OPENAI_API_KEY'))
 
response = client.chat.completions.create(
    model='gpt-4o-mini',
    messages=[
        {'role': 'system', 'content': '你是一个耐心的 AI 编程老师'},
        {'role': 'user', 'content': '用三句话解释什么是 RAG'}
    ]
)
print(response.choices[0].message.content)

如果 SDK 版本不同,接口名称可能变化,遇到问题先看官方文档和当前包版本。

messages结构

messages 是对话上下文数组。system 设定模型行为,user 表示用户输入,assistant 表示模型历史回答。多轮对话时,要把必要上下文继续传入。

常用参数

参数 作用
model 选择模型
messages 对话内容
temperature 控制随机性
max_tokens 限制输出长度
stream 是否流式输出

加上异常处理

try:
    response = client.chat.completions.create(
        model='gpt-4o-mini',
        messages=[{'role': 'user', 'content': '你好'}]
    )
    print(response.choices[0].message.content)
except Exception as e:
    print('调用失败:', type(e).__name__, str(e))

真实项目中要区分网络错误、认证错误、限流错误和服务端错误,并记录日志。

小结

API 调用入门要掌握四件事:安全读取密钥、构造 messages、解析返回结果、处理异常。跑通示例只是第一步,封装和日志才是项目可用的关键。