Python SDK

环境要求

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

快速开始

安装

pip install openai

配置客户端

from openai import OpenAI

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

基本调用

response = client.chat.completions.create(
    model="gpt-4o",
    messages=[
        {"role": "system", "content": "你是一个有用的助手"},
        {"role": "user", "content": "你好，请介绍一下自己"}
    ]
)

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

流式调用

stream = client.chat.completions.create(
    model="gpt-4o",
    messages=[
        {"role": "user", "content": "讲一个关于人工智能的笑话"}
    ],
    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": "你好"}
    ]
)

错误处理

from openai import OpenAI
from openai import APIError, RateLimitError

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

try:
    response = client.chat.completions.create(
        model="gpt-4o",
        messages=[{"role": "user", "content": "Hello"}]
    )
except APIError as e:
    print(f"API 错误: {e}")
except RateLimitError:
    print("请求频率超限，请稍后再试")

高级配置

设置超时

from openai import OpenAI

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

设置代理

import os

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

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