豆包大模型API调用两种方法

目录

一、环境配置

二、方法一：通过Model_ID调用

三、方法二：通过 Endpoint ID 调用模型

---

一、环境配置

pip install volcengine-python-sdk

二、方法一：通过Model_ID调用

官方教程：快速入门-调型服务--火山方舟大模型服务平台-火山引擎

1. 创建API_key

开通管理页面：账号登录-火山引擎

2. 获取模型ID

        Model_ID详见： 获取模型ID

3. 编写调用代码：

import os
from volcenginesdkarkruntime import Ark

client = Ark(api_key="<ARK_API_KEY>")
completion = client.chat.completions.create(
    # 替换为模型 ID，比如"doubao-1.5-pro-32k-250115"
    model="<Model>",
    messages=[
        {"role": "user", "content": "你好"}
    ]
)

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

ps: 因为是自用，所以这里图方便采用硬编码密钥的形式；如果有需求/安全风险，建议参考官方教程设置ARK_API_KEY环境变量的方式。
 

二编：又去试了一下，如果不建Endpoint创建接入点好像会报错，创建完用Model_ID或者EndpointID都行，不过Model_ID偶尔会报错（但不影响正常运行）。如果都要创建EndpointID的话，Model_ID的意义或者说快速在哪？(⊙_⊙)？

三、方法二：通过 Endpoint ID 调用模型

官方教程：通过 Endpoint ID 调用模型--火山方舟大模型服务平台-火山引擎

1. 获取 Endpoint_ID

创建就按照提示填写信息就行，需要实名认证，有免费额度。

绿框框里就是Endpoint ID。

2. 编写调用代码：

import os
from volcenginesdkarkruntime import Ark

# 从环境变量中读取您的方舟API Key
client = Ark(api_key=os.environ.get("<YOUR_API_KEY>"))
completion = client.chat.completions.create(
    # 将 <Model> 替换为 Endpoint ID
    model="<Model>", 
    messages=[
        {"role": "user", "content": "你好"}
    ]
)

print(completion.choices[0].message)

将 <Model> 替换为 Endpoint ID即可。好处是：

坏处是：看别人的教程走了一大圈弯路hh；基础调用的话方法一就可以满足了，也比较方便快捷，容易上手。