Skip to content

Python SDK

安装

bash
pip install openai

基础用法

python
import openai

client = openai.OpenAI(
    api_key="sk-你的API_KEY",
    base_url="https://beesai.cn/v1"
)

response = client.chat.completions.create(
    model="gpt-4o",
    messages=[
        {"role": "system", "content": "你是一个有帮助的AI助手。"},
        {"role": "user", "content": "用一句话解释量子计算。"}
    ]
)

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

流式输出

python
stream = client.chat.completions.create(
    model="gpt-4o",
    messages=[{"role": "user", "content": "写一首关于 AI 的诗"}],
    stream=True
)

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

多轮对话

python
messages = [
    {"role": "system", "content": "你是一个专业的翻译助手。"},
    {"role": "user", "content": "把'Hello World'翻译成中文"},
    {"role": "assistant", "content": "你好,世界"},
    {"role": "user", "content": "再翻译成日文"},
]

response = client.chat.completions.create(model="gpt-4o", messages=messages)
print(response.choices[0].message.content)

环境变量配置(推荐)

python
import os
from openai import OpenAI

client = OpenAI(
    api_key=os.getenv("BEESAI_API_KEY"),
    base_url=os.getenv("BEESAI_BASE_URL", "https://beesai.cn/v1")
)