Bootstrap

豆包API调用新手教程_Windows系统

一、环境配置

python需要的环境不多说,需单独配置的库如下:

# 加载env环境文件
pip install python-dotenv

# 环境SDK的安装
pip install volcengine-python-sdk

二、获取API密钥

登录火山方舟主页 https://www.volcengine.com/product/ark

1. 创建API Key

2. 创建接入点

将API key和上述截图中框出来的接入点名称复制单独保存,准备工作就算做好了。

三、API调用

在pycharm中创建新的项目,其中env文件中输入以下数据并保存为api_key.env(也可以取其他名字,调用时代码里记得改成相应的名字,能保证读取到就行了)

VOLC_ACCESSKEY = your Access Key
VOLC_SECRETKEY = your Secret Access Key
ENDPOINT_ID = your ENDPOINT_ID

your Access Key 即为接入点名称(创建接入点截图中框出来标记的那个)

your Secret Access Key 即为API key

ENDPOINT_ID 与your Access Key相同

调用API

import os
from volcenginesdkarkruntime import Ark
import dotenv

# 读取env文件 加载环境变量
dotenv.load_dotenv("api_key.env")

# 初始化 Ark 客户端
client = Ark()
client = Ark(api_key="your Secret Access Key", region="cn-beijing")  #此处your Secret Access Key为前面保存的API Key

# 从环境变量中获取模型 ID
model_id = os.getenv("ENDPOINT_ID")

# 欢迎语
Welcome_Text = "您好,我是豆包,您的大模型对话助手,请问有什么可以帮到您?(输入 'exit' 退出对话)"
print(Welcome_Text)

# 进入多轮对话的循环
while True:
    # 从终端获取用户输入
    user_input = input("User:\r\n")

    # 检查用户是否想退出
    if user_input.lower() in ["exit", "quit"]:
        print("AI:感谢您的使用,再见!")
        break

    # 创建流式对话请求
    stream = client.chat.completions.create(
        model=model_id,
        messages=[
            {"role": "system", "content": "你是豆包,是由字节跳动开发的 AI 人工智能助手"},
            {"role": "user", "content": user_input},  # 使用终端输入的内容
        ],
        stream=True
    )

    print("AI:")
    # 初始化一个空字符串来存储所有文本
    full_text = ""

    # 逐块读取流式输出并将结果打印
    for chunk in stream:
        if not chunk.choices:
            continue
        # 获取文本内容
        text = chunk.choices[0].delta.content

        # 输出文本到控制台
        print(text, end="")

        # 将文本累积到 full_text
        full_text += text
    print("\r\n")

;