ChatGPT生成文件的下载方案：从API调用到本地存储的完整实现

1次阅读

没有评论

共计 1423 个字符，预计需要花费 4 分钟才能阅读完成。

在实际开发中，使用 ChatGPT API 生成文件后，下载环节往往会遇到几个棘手的问题：

大文件处理 ：当生成的内容较大时（比如长文档或高分辨率图片），直接下载可能导致内存溢出或超时
格式转换 ：API 返回的数据可能需要转换为特定格式（如 Markdown 转 PDF）
断点续传 ：网络不稳定时如何恢复中断的下载
内容解析 ：正确处理不同 MIME 类型（如 application/json、image/png 等）的响应

开发者通常有几种主流实现方式可选：

直接 API 下载
优点：实现简单，适合小文件
缺点：内存占用高，无进度反馈
流式处理
优点：内存友好，支持实时处理
缺点：代码复杂度稍高
分块下载
优点：支持断点续传，适合大文件
缺点：需要处理分块逻辑

以下是一个完整的文件下载实现，包含错误处理和类型判断：

import requests
import os
from mimetypes import guess_extension

def download_file(api_url, save_path, api_key):
    headers = {'Authorization': f'Bearer {api_key}',
        'Accept': '*/*'
    }

    try:
        with requests.get(api_url, headers=headers, stream=True) as r:
            r.raise_for_status()

            # 从 Content-Type 获取文件扩展名
            content_type = r.headers.get('Content-Type', '')
            ext = guess_extension(content_type.split(';')[0].strip()) or '.bin'

            final_path = f'{save_path}{ext}'

            with open(final_path, 'wb') as f:
                for chunk in r.iter_content(chunk_size=8192):
                    if chunk:  # 过滤 keep-alive chunks
                        f.write(chunk)

            return final_path
    except requests.exceptions.RequestException as e:
        print(f'下载失败: {e}')
        if os.path.exists(final_path):
            os.remove(final_path)
        return None

对于生产环境，建议考虑以下优化措施：