支持的模型

零一万物 API支持的模型和上下文长度:

零一万物模型

平台	模式标识符	备注
yi	yi-34b-chat-0205
yi	yi-34b-chat-200k
yi	yi-vl-plus

from openai import OpenAI

client = OpenAI(
    api_key = "自己的API key",
    base_url = "https://api.fe8.cn/v1"
)

chat_completion = client.chat.completions.create(
    messages=[
        {
            "role": "user",
            "content": "你是谁，用中文回答",
        }
    ],
    model="yi-34b-chat-0205",
)
print(chat_completion.choices[0].message.content)