安装
pip install openai
基本使用
from openai import OpenAI
client = OpenAI(
api_key="你的_API_KEY",
base_url="https://api.westtree.top/v1"
)
# 对话
response = client.chat.completions.create(
model="claude-3-5-sonnet",
messages=[{"role": "user", "content": "解释量子计算"}],
max_tokens=500
)
print(response.choices[0].message.content)
# 流式
stream = client.chat.completions.create(
model="gpt-4o",
messages=[{"role": "user", "content": "写一首诗"}],
stream=True
)
for chunk in stream:
print(chunk.choices[0].delta.content or "", end="")
错误处理
from openai import OpenAI, APIError, RateLimitError
try:
response = client.chat.completions.create(...)
except RateLimitError:
print("触发限流 - 退避重试")
except APIError as e:
print(f"API 错误: {e}")