Python 调用豆包 API 使用指南 | 极客日志

PythonAI

Python 调用豆包 API 使用指南

如何使用 Python 调用字节跳动豆包 API，涵盖基础环境准备、库安装、单轮对话实现、多轮对话、流式响应处理以及错误重试机制。通过示例代码展示了代码生成解释和内容摘要生成的实际应用，并提供了速率限制、输入验证及密钥管理等最佳实践建议，帮助开发者将 AI 能力集成至应用中。

忘忧发布于 2026/3/30更新于 2026/4/131 浏览

Python 调用豆包 API 使用指南

豆包是字节跳动推出的 AI 对话助手，提供了强大的自然语言处理能力。本文将详细介绍如何使用 Python 调用豆包 API，实现智能对话功能。

准备工作

在开始之前，您需要：

注册豆包开发者账号
获取 API 密钥（通常在开发者控制台中创建应用后获得）
安装必要的 Python 库

安装所需库

pip install requests
pip install python-dotenv

基本 API 调用

首先，让我们实现一个简单的豆包 API 调用：

import requests
import json

def call_doubao_api(api_key, prompt, model="doubao-pro"):
    """ 调用豆包 API 的基本函数
    参数:
        api_key: 您的豆包 API 密钥
        prompt: 输入的提示文本
        model: 使用的模型版本，默认为'doubao-pro'
    返回:
        API 的响应内容
    """
    url = "https://api.doubao.com/v1/chat/completions"
    headers = {"Content-Type": "application/json", "Authorization": f"Bearer {api_key}"}
    data = {"model": model, "messages": [{"role": "user", "content": prompt}], "temperature": 0.7}
    response = requests.post(url, headers=headers, json=data)
    if response.status_code == 200:
        return response.json()
    else:
        raise Exception(f"API 调用失败，状态码：{response.status_code}, 错误：{response.text}")

# 使用示例
api_key = "your_api_key_here"  # 替换为您的实际 API 密钥
response = call_doubao_api(api_key, "Python 是什么？")
print(response[][][][])

极客日志微信公众号二维码

更多推荐文章

相关免费在线工具

RSA密钥对生成器
生成新的随机RSA私钥和公钥pem证书。在线工具，RSA密钥对生成器在线工具，online
Mermaid 预览与可视化编辑
基于 Mermaid.js 实时预览流程图、时序图等图表，支持源码编辑与即时渲染。在线工具，Mermaid 预览与可视化编辑在线工具，online
curl 转代码
解析常见 curl 参数并生成 fetch、axios、PHP curl 或 Python requests 示例代码。在线工具，curl 转代码在线工具，online
Base64 字符串编码/解码
将字符串编码和解码为其 Base64 格式表示形式即可。在线工具，Base64 字符串编码/解码在线工具，online
Base64 文件转换器
将字符串、文件或图像转换为其 Base64 表示形式。在线工具，Base64 文件转换器在线工具，online
Markdown 转 HTML
将 Markdown（GFM）转为 HTML 片段，浏览器内 marked 解析；与 HTML 转 Markdown 互为补充。在线工具，Markdown 转 HTML在线工具，online

def multi_turn_conversation(api_key):
    conversation_history = []
    while True:
        user_input = input("你：")
        if user_input.lower() in ["退出", "exit", "quit"]:
            break
        conversation_history.append({"role": "user", "content": user_input})
        response = call_doubao_api(
            api_key=api_key,
            prompt=conversation_history,
            model="doubao-pro"
        )
        assistant_reply = response["choices"][0]["message"]["content"]
        conversation_history.append({"role": "assistant", "content": assistant_reply})
        print(f"豆包：{assistant_reply}")

# 使用示例
# multi_turn_conversation("your_api_key_here")

def stream_doubao_response(api_key, prompt):
    url = "https://api.doubao.com/v1/chat/completions"
    headers = {"Content-Type": "application/json", "Authorization": f"Bearer {api_key}"}
    data = {"model": "doubao-pro", "messages": [{"role": "user", "content": prompt}], "stream": True}
    with requests.post(url, headers=headers, json=data, stream=True) as response:
        for line in response.iter_lines():
            if line:
                decoded_line = line.decode('utf-8')
                if decoded_line.startswith("data:"):
                    json_data = decoded_line[5:].strip()
                    if json_data != "[DONE]":
                        try:
                            chunk = json.loads(json_data)
                            content = chunk.get("choices", [{}])[0].get("delta", {}).get("content", "")
                            print(content, end="", flush=True)
                        except json.JSONDecodeError:
                            pass
        print()

# 使用示例
# stream_doubao_response("your_api_key_here", "请详细解释 Python 的生成器")

import time
from requests.exceptions import RequestException

def robust_doubao_call(api_key, prompt, max_retries=3):
    retries = 0
    last_error = None
    while retries < max_retries:
        try:
            response = call_doubao_api(api_key, prompt)
            return response
        except RequestException as e:
            last_error = e
            retries += 1
            if retries < max_retries:
                time.sleep(2 ** retries)  # 指数退避
        except Exception as e:
            last_error = e
            break
    raise Exception(f"API 调用失败，重试{max_retries}次后仍不成功。最后错误：{str(last_error)}")

def generate_code_explanation(api_key, code_snippet):
    prompt = f"""
    请解释以下 Python 代码的功能和工作原理：
    {code_snippet}
    请按照以下格式回答：
    1. 代码功能概述
    2. 关键代码段解析
    3. 可能的改进建议
    """
    response = robust_doubao_call(api_key, prompt)
    return response["choices"][0]["message"]["content"]

def generate_summary(api_key, text, length="short"):
    length_map = {"short": "50 字以内", "medium": "100-150 字", "long": "200-300 字"}
    prompt = f"""
    请为以下文本生成一个{length_map.get(length, "适中")}的摘要：
    {text}
    摘要要求：
    - 保留核心信息
    - 语言简洁明了
    - 保持客观中立
    """
    response = robust_doubao_call(api_key, prompt)
    return response["choices"][0]["message"]["content"]

from dotenv import load_dotenv
import os

load_dotenv()
api_key = os.getenv("DOUBAO_API_KEY")