快捷導(dǎo)航

以文檔處理為例講解豆包API調(diào)用教程(Python)

更新時間：2025年01月03日 10:52:35 作者：JungleZhang0719

這篇文章主要介紹了如何在Linux服務(wù)器上配置和使用豆包大模型的API,通過API處理200萬字的文檔,展現(xiàn)了技術(shù)的發(fā)展和應(yīng)用,需要的朋友可以參考下

一、前言：

現(xiàn)在大模型api價格越來越便宜了，豆包大模型分為 lite 和 pro，其中pro 32k 大模型的 api 價格在 0.0008元/千tokens，作為普通人能夠用非常低的價格享受世界上最先進(jìn)的科技，讓我驚嘆于技術(shù)的發(fā)展。

剛好手上有差不多200w字的文檔要處理，我們這里選擇 Doubao-pro-32k 的模型，換算成人民幣差不多兩塊錢（再次震驚）。

備注：這篇文章要求一點點基礎(chǔ)，這里默認(rèn)你在 Linux 服務(wù)器上操作、已經(jīng)有 Conda 環(huán)境、Python 版本大于 3.7、會基礎(chǔ)的 Linux 命令、并且看得懂簡單的 Python代碼。

二、配置環(huán)境

直接在 Windows 上安裝豆包的庫會報錯：

note: This error originates from a subprocess, and is likely not a problem with pip.
  ERROR: Failed building wheel for volcengine-python-sdk
  Running setup.py clean for volcengine-python-sdk
Failed to build volcengine-python-sdk
ERROR: Could not build wheels for volcengine-python-sdk, which is required to install pyproject.toml-based projects

這是因為 Windows 系統(tǒng)有最長路徑限制，解決辦法是自己看官方文檔。我們這里還是以 Ubuntu 為例，在命令行輸入：

pip install 'volcengine-python-sdk[ark]'

從控制臺獲取api key：

復(fù)制以后將 ARK_API_KEY 設(shè)置為全局變量，在命令行輸入：

export ARK_API_KEY="你剛才創(chuàng)建的API-key"

然后我們?nèi)?chuàng)建“模型推理接入點”：登錄火山方舟平臺，點擊左側(cè)導(dǎo)航欄中的【在線推理】，點擊【創(chuàng)建推理接入點】。

然后復(fù)制生成的ENDPOINT_ID

在命令行輸入：

export ENDPOINT_ID="你的ENDPOINT_ID"

這樣就完成了環(huán)境配置

三、調(diào)用 API

執(zhí)行代碼：

from volcenginesdkarkruntime import Ark
import os

client = Ark()

print("----- standard request -----")
completion = client.chat.completions.create(
    model=os.getenv('ENDPOINT_ID'),
    messages = [
        {"role": "system", "content": "你是豆包，是由字節(jié)跳動開發(fā)的 AI 人工智能助手"},
        {"role": "user", "content": "常見的十字花科植物有哪些？"},
    ],
)

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

成功輸出：

----- standard request -----
十字花科是一個較大的科，包含許多常見的蔬菜和花卉。以下是一些常見的十字花科植物：
- **蔬菜**：
    - **白菜類**：包括大白菜、小白菜、菜心等。
    - **甘藍(lán)類**：如結(jié)球甘藍(lán)（卷心菜）、羽衣甘藍(lán)、抱子甘藍(lán)等。
    - **芥菜類**：有莖用芥菜（榨菜）、葉用芥菜（雪里蕻）、根用芥菜（大頭菜）等。
    - **蘿卜類**：常見的有紅蘿卜、白蘿卜、青蘿卜等。
    - **其他**：還有花椰菜（菜花）、西蘭花、油菜、薺菜等。
- **花卉**：
    - **紫羅蘭**：是十字花科紫羅蘭屬的二年生或多年生草本植物，花朵通常為紫色。
    - **桂竹香**：屬于十字花科桂竹香屬，是一種多年生草本植物，花色豐富，有黃色、橙色、紅色等。
    - **諸葛菜**：又名二月蘭，是十字花科諸葛菜屬的一年或二年生草本植物，藍(lán)紫色的花在每年的 4-5 月盛開。
    - **香雪球**：是十字花科香雪球?qū)俚亩嗄晟荼局参?，花朵小而密集，呈白色、淡紫色或紫紅色。

這只是十字花科植物的一小部分，該科還包括許多其他的蔬菜和花卉。十字花科植物的特點是花朵呈十字形排列，果實為角果。這些植物在人們的生活中具有重要的經(jīng)濟和觀賞價值。