Python SDK

环境要求

• Python 3.8 或更高版本
• 安装 openai 包：pip install openai

快速开始

安装

pip install openai

配置客户端

from openai import OpenAI

client = OpenAI(
    base_url="https://ai-tokenhub.com/api/v1",
    api_key="$YOUR_API_KEY"
)

基本调用

response = client.chat.completions.create(
    model="gpt-4o",
    messages=[
        {"role": "system", "content": "You are a helpful assistant"},
        {"role": "user", "content": "Hello, please introduce yourself"}
    ]
)

print(response.choices[0].message.content)

流式调用

stream = client.chat.completions.create(
    model="gpt-4o",
    messages=[
        {"role": "user", "content": "Tell a joke about artificial intelligence"}
    ],
    stream=True
)

for chunk in stream:
    if chunk.choices[0].delta.content:
        print(chunk.choices[0].delta.content, end="", flush=True)

使用 Claude 模型

response = client.chat.completions.create(
    model="claude-3-5-sonnet-20241022",
    messages=[
        {"role": "user", "content": "Hello"}
    ]
)

错误处理

from openai import OpenAI
from openai import APIError, RateLimitError

client = OpenAI(
    base_url="https://ai-tokenhub.com/api/v1",
    api_key="$YOUR_API_KEY"
)

try:
    response = client.chat.completions.create(
        model="gpt-4o",
        messages=[{"role": "user", "content": "Hello"}]
    )
except APIError as e:
    print(f"API error: {e}")
except RateLimitError:
    print("Rate limit exceeded, please try again later")

高级配置

设置超时

from openai import OpenAI

client = OpenAI(
    base_url="https://ai-tokenhub.com/api/v1",
    api_key="$YOUR_API_KEY",
    timeout=60.0
)

设置代理

import os

os.environ["HTTPS_PROXY"] = "http://proxy.example.com:8080"

client = OpenAI(
    base_url="https://ai-tokenhub.com/api/v1",
    api_key="$YOUR_API_KEY"
)