Skip to main content

A simple and efficient Python SDK for DeepSeek-OCR API

Project description

DeepSeek-OCR-SDK

PyPI version License: MIT Python 3.10+ Code style: black

English | 中文


English

Overview

DeepSeek-OCR-SDK is a simple and efficient Python SDK for the DeepSeek OCR API. It provides a clean, production-ready interface for converting documents (PDF, images) to Markdown text with high accuracy and performance.

Key Features

  • Simple API: Clean and intuitive interface, minimal learning curve
  • Three OCR Modes:
    • FREE_OCR: Fast mode for 80% of use cases (3.95-10.95s)
    • GROUNDING: Advanced mode for complex tables (5.18-8.31s)
    • OCR_IMAGE: Detailed word-level extraction (19-26s)
  • Intelligent Fallback: Automatically switches modes for better quality
  • Batch Processing: Process multiple documents efficiently with progress tracking
  • Async & Sync: Full support for both asynchronous and synchronous workflows
  • Type Hints: 100% type coverage for better IDE support

Installation

Using pip

pip install deepseek-ocr

Using uv (recommended)

uv add deepseek-ocr

Install from source

# Clone repository
git clone https://github.com/BukeLy/DeepSeek-OCR-SDK
cd DeepSeek-OCR-SDK

# Install with uv
uv sync

# Or install with pip
pip install -e .

Quick Start

from deepseek_ocr import DeepSeekOCR

# Initialize client (choose your API provider)
client = DeepSeekOCR(
    api_key="your_api_key",
    base_url="https://api.siliconflow.cn/v1/chat/completions"  # or your provider's endpoint
)

# Parse document
text = client.parse("document.pdf")
print(text)

Note: This SDK supports any OpenAI-compatible API endpoint that provides the DeepSeek-OCR model. Currently known provider: SiliconFlow (api.siliconflow.cn). DeepSeek's official API does not support the DeepSeek-OCR model.

Architecture

flowchart TD
    A[User Input: PDF/Image] --> B[DeepSeekOCR Client]
    B --> C{Select Mode}
    C -->|FREE_OCR| D[Fast Processing]
    C -->|GROUNDING| E[Complex Table Processing]
    C -->|OCR_IMAGE| F[Detailed Extraction]

    D --> G[PDF to Base64]
    E --> G
    F --> G

    G --> H[Build Prompt]
    H --> I[API Request]
    I --> J{Response OK?}

    J -->|Yes| K[Extract Text]
    J -->|No| L[Retry/Error]
    L --> I

    K --> M{Check Output Length}
    M -->|< 500 chars & fallback enabled| N[Switch to GROUNDING]
    M -->|OK| O[Clean Output]
    N --> G

    O --> P[Post-process]
    P --> Q[Return Markdown]

Usage Examples

Basic Usage

from deepseek_ocr import DeepSeekOCR

client = DeepSeekOCR(
    api_key="your_api_key",
    base_url="https://api.siliconflow.cn/v1/chat/completions"  # or your provider's endpoint
)

# Simple document
text = client.parse("invoice.pdf", mode="free_ocr")

# Complex table
text = client.parse("statement.pdf", mode="grounding")

# Custom DPI
text = client.parse("document.pdf", dpi=300)

Multi-Page PDF Processing

⚠️ Breaking Change in v0.2.0: PDF processing now handles all pages by default.

from deepseek_ocr import DeepSeekOCR

client = DeepSeekOCR(
    api_key="your_api_key",
    base_url="https://api.siliconflow.cn/v1/chat/completions"
)

# Process all pages (new default behavior)
text = client.parse("multi_page.pdf")
# Returns: Page 1 content\n\n---\n\nPage 2 content\n\n---\n\nPage 3 content

# Process only the first page (old behavior)
text = client.parse("multi_page.pdf", pages=1)

# Process specific pages (e.g., pages 1, 3, and 5)
text = client.parse("multi_page.pdf", pages=[1, 3, 5])

# Process a range of pages
text = client.parse("multi_page.pdf", pages=list(range(1, 6)))  # Pages 1-5

Note: Processing multiple pages will increase API usage and costs proportionally. Each page is processed independently with intelligent per-page fallback.

Async Usage

import asyncio
from deepseek_ocr import DeepSeekOCR

async def main():
    client = DeepSeekOCR(
        api_key="your_api_key",
        base_url="https://api.siliconflow.cn/v1/chat/completions"  # or your provider's endpoint
    )
    text = await client.parse_async("document.pdf")
    print(text)

asyncio.run(main())

Batch Processing

import asyncio
from pathlib import Path
from deepseek_ocr import DeepSeekOCR, BatchProcessor

async def batch_example():
    client = DeepSeekOCR(
        api_key="your_api_key",
        base_url="https://api.siliconflow.cn/v1/chat/completions"  # or your provider's endpoint
    )
    processor = BatchProcessor(client, max_concurrent=5)

    files = list(Path("docs").glob("*.pdf"))
    summary = await processor.process_batch(
        files,
        mode="free_ocr",
        show_progress=True
    )

    summary.print_summary()

asyncio.run(batch_example())

Mode Selection Guide

Document Type Recommended Mode Reason
Simple text (invoice, letter) FREE_OCR Fastest, 80% accuracy
Complex tables (≥20 rows) GROUNDING Better structure preservation
Simple tables (<10 rows) FREE_OCR Avoids truncation issues
Mixed content GROUNDING Handles complexity well

Configuration

Environment Variables

export DS_OCR_API_KEY="your_api_key"
export DS_OCR_BASE_URL="https://api.siliconflow.cn/v1/chat/completions"  # REQUIRED: Set to your provider's endpoint
export DS_OCR_MODEL="deepseek-ai/DeepSeek-OCR"
export DS_OCR_TIMEOUT=60
export DS_OCR_MAX_TOKENS=4000
export DS_OCR_DPI=200
export DS_OCR_FALLBACK_ENABLED=true
export DS_OCR_FALLBACK_MODE="grounding"
export DS_OCR_MIN_OUTPUT_THRESHOLD=500
export DS_OCR_PAGE_SEPARATOR="\n\n---\n\n"  # Separator between pages in multi-page PDFs

Available API Providers:

  • SiliconFlow: https://api.siliconflow.cn/v1/chat/completions (Verified ✅)
  • Others: Contact third-party API providers for DeepSeek-OCR support

Note: DeepSeek's official API (api.deepseek.com) does not support the DeepSeek-OCR model.

Programmatic Configuration

from deepseek_ocr import DeepSeekOCR, OCRConfig

# Method 1: Direct initialization
client = DeepSeekOCR(
    api_key="your_api_key",
    base_url="https://api.siliconflow.cn/v1/chat/completions",  # or your provider's endpoint
    timeout=120,
    dpi=300
)

# Method 2: Using config object (requires DS_OCR_BASE_URL environment variable)
config = OCRConfig.from_env(api_key="your_api_key", dpi=300)
client = DeepSeekOCR(api_key=config.api_key, base_url=config.base_url)

DPI Recommendations

  • 150 DPI: May cause hallucinations, not recommended
  • 200 DPI: ⭐ Optimal balance (recommended)
  • 300 DPI: Larger file size, minimal quality improvement

Error Handling

from deepseek_ocr import DeepSeekOCR, APIError, FileProcessingError

client = DeepSeekOCR(
    api_key="your_api_key",
    base_url="https://api.siliconflow.cn/v1/chat/completions"  # or your provider's endpoint
)

try:
    text = client.parse("document.pdf")
except FileProcessingError as e:
    print(f"File error: {e}")
except APIError as e:
    print(f"API error: {e.status_code} - {e}")
except Exception as e:
    print(f"Unexpected error: {e}")

Development

Setup Development Environment

# Clone repository
git clone https://github.com/BukeLy/DeepSeek-OCR-SDK
cd DeepSeek-OCR-SDK

# Install dependencies with uv
uv sync --all-extras

# Activate virtual environment
source .venv/bin/activate  # Linux/Mac
# or
.venv\\Scripts\\activate  # Windows

Run Tests

uv run pytest

Code Quality

# Format code
uv run black deepseek_ocr/
uv run isort deepseek_ocr/

# Type checking
uv run mypy deepseek_ocr/

# Linting
uv run flake8 deepseek_ocr/

API Reference

See API_REFERENCE.md for complete API documentation.

License

This project is licensed under the MIT License - see the LICENSE file for details.

Acknowledgments

  • DeepSeek AI for the excellent OCR model

Disclaimer: This is an unofficial, third-party SDK and is not affiliated with DeepSeek AI or any API service provider. Users are responsible for choosing their own API provider and complying with the provider's terms of service.


中文

简介

DeepSeek-OCR-SDK 是一个简单高效的 Python SDK,用于调用 DeepSeek OCR API。它提供了简洁、生产级的接口,可以高精度、高性能地将文档(PDF、图片)转换为 Markdown 文本。

核心特性

  • 简单易用:API 简洁直观,学习成本低
  • 三种 OCR 模式
    • FREE_OCR:快速模式,适用于 80% 的场景(3.95-10.95秒)
    • GROUNDING:高级模式,适用于复杂表格(5.18-8.31秒)
    • OCR_IMAGE:详细模式,提供词级别提取(19-26秒)
  • 智能回退:自动切换模式以获得更好的质量
  • 批量处理:高效处理多个文档,带进度跟踪
  • 异步 & 同步:完整支持异步和同步工作流
  • 类型提示:100% 类型覆盖,更好的 IDE 支持

安装

使用 pip

pip install deepseek-ocr

使用 uv(推荐)

uv add deepseek-ocr

从源码安装

# 克隆仓库
git clone https://github.com/BukeLy/DeepSeek-OCR-SDK
cd DeepSeek-OCR-SDK

# 使用 uv 安装
uv sync

# 或使用 pip 安装
pip install -e .

快速开始

from deepseek_ocr import DeepSeekOCR

# 初始化客户端(选择您的 API 提供商)
client = DeepSeekOCR(
    api_key="your_api_key",
    base_url="https://api.siliconflow.cn/v1/chat/completions"  # 或您的提供商端点
)

# 解析文档
text = client.parse("document.pdf")
print(text)

注意:本 SDK 支持任何提供 DeepSeek-OCR 模型的 OpenAI 兼容 API 端点。目前已知的提供商:硅基流动 (api.siliconflow.cn)。DeepSeek 官方 API 不支持 DeepSeek-OCR 模型。

架构图

flowchart TD
    A[用户输入: PDF/图片] --> B[DeepSeekOCR 客户端]
    B --> C{选择模式}
    C -->|FREE_OCR| D[快速处理]
    C -->|GROUNDING| E[复杂表格处理]
    C -->|OCR_IMAGE| F[详细提取]

    D --> G[PDF 转 Base64]
    E --> G
    F --> G

    G --> H[构建提示词]
    H --> I[API 请求]
    I --> J{响应正常?}

    J -->|是| K[提取文本]
    J -->|否| L[重试/错误]
    L --> I

    K --> M{检查输出长度}
    M -->|< 500 字符 & 启用回退| N[切换到 GROUNDING]
    M -->|正常| O[清理输出]
    N --> G

    O --> P[后处理]
    P --> Q[返回 Markdown]

使用示例

基础用法

from deepseek_ocr import DeepSeekOCR

client = DeepSeekOCR(
    api_key="your_api_key",
    base_url="https://api.siliconflow.cn/v1/chat/completions"  # 或您的提供商端点
)

# 简单文档
text = client.parse("invoice.pdf", mode="free_ocr")

# 复杂表格
text = client.parse("statement.pdf", mode="grounding")

# 自定义 DPI
text = client.parse("document.pdf", dpi=300)

多页 PDF 处理

⚠️ v0.2.0 破坏性变更:PDF 处理现在默认处理所有页面

from deepseek_ocr import DeepSeekOCR

client = DeepSeekOCR(
    api_key="your_api_key",
    base_url="https://api.siliconflow.cn/v1/chat/completions"
)

# 处理所有页面(新的默认行为)
text = client.parse("multi_page.pdf")
# 返回: 第1页内容\n\n---\n\n第2页内容\n\n---\n\n第3页内容

# 只处理第一页(旧的行为)
text = client.parse("multi_page.pdf", pages=1)

# 处理特定页面(例如第 1、3、5 页)
text = client.parse("multi_page.pdf", pages=[1, 3, 5])

# 处理一个范围的页面
text = client.parse("multi_page.pdf", pages=list(range(1, 6)))  # 第 1-5 页

注意:处理多个页面将按比例增加 API 使用量和费用。每个页面都独立处理,并带有智能的逐页回退机制。

异步用法

import asyncio
from deepseek_ocr import DeepSeekOCR

async def main():
    client = DeepSeekOCR(
        api_key="your_api_key",
        base_url="https://api.siliconflow.cn/v1/chat/completions"  # 或您的提供商端点
    )
    text = await client.parse_async("document.pdf")
    print(text)

asyncio.run(main())

批量处理

import asyncio
from pathlib import Path
from deepseek_ocr import DeepSeekOCR, BatchProcessor

async def batch_example():
    client = DeepSeekOCR(
        api_key="your_api_key",
        base_url="https://api.siliconflow.cn/v1/chat/completions"  # 或您的提供商端点
    )
    processor = BatchProcessor(client, max_concurrent=5)

    files = list(Path("docs").glob("*.pdf"))
    summary = await processor.process_batch(
        files,
        mode="free_ocr",
        show_progress=True
    )

    summary.print_summary()

asyncio.run(batch_example())

模式选择指南

文档类型 推荐模式 原因
简单文本(发票、信件) FREE_OCR 最快,80% 准确率
复杂表格(≥20 行) GROUNDING 更好的结构保留
简单表格(<10 行) FREE_OCR 避免截断问题
混合内容 GROUNDING 处理复杂性好

配置

环境变量

export DS_OCR_API_KEY="your_api_key"
export DS_OCR_BASE_URL="https://api.siliconflow.cn/v1/chat/completions"  # 必填:设置为您的提供商端点
export DS_OCR_MODEL="deepseek-ai/DeepSeek-OCR"
export DS_OCR_TIMEOUT=60
export DS_OCR_MAX_TOKENS=4000
export DS_OCR_DPI=200
export DS_OCR_FALLBACK_ENABLED=true
export DS_OCR_FALLBACK_MODE="grounding"
export DS_OCR_MIN_OUTPUT_THRESHOLD=500
export DS_OCR_PAGE_SEPARATOR="\n\n---\n\n"  # Separator between pages in multi-page PDFs

可用的 API 提供商

  • 硅基流动(SiliconFlow)https://api.siliconflow.cn/v1/chat/completions (已验证 ✅)
  • 其他:联系第三方 API 提供商以获取 DeepSeek-OCR 支持

注意:DeepSeek 官方 API (api.deepseek.com) 不支持 DeepSeek-OCR 模型。

编程式配置

from deepseek_ocr import DeepSeekOCR, OCRConfig

# 方法 1:直接初始化
client = DeepSeekOCR(
    api_key="your_api_key",
    base_url="https://api.siliconflow.cn/v1/chat/completions",  # 或您的提供商端点
    timeout=120,
    dpi=300
)

# 方法 2:使用配置对象(需要设置 DS_OCR_BASE_URL 环境变量)
config = OCRConfig.from_env(api_key="your_api_key", dpi=300)
client = DeepSeekOCR(api_key=config.api_key, base_url=config.base_url)

DPI 推荐

  • 150 DPI:可能产生幻觉,不推荐
  • 200 DPI:⭐ 最佳平衡(推荐)
  • 300 DPI:文件更大,质量提升不明显

错误处理

from deepseek_ocr import DeepSeekOCR, APIError, FileProcessingError

client = DeepSeekOCR(
    api_key="your_api_key",
    base_url="https://api.siliconflow.cn/v1/chat/completions"  # 或您的提供商端点
)

try:
    text = client.parse("document.pdf")
except FileProcessingError as e:
    print(f"文件错误: {e}")
except APIError as e:
    print(f"API 错误: {e.status_code} - {e}")
except Exception as e:
    print(f"未预期的错误: {e}")

开发

设置开发环境

# 克隆仓库
git clone https://github.com/BukeLy/DeepSeek-OCR-SDK
cd DeepSeek-OCR-SDK

# 使用 uv 安装依赖
uv sync --all-extras

# 激活虚拟环境
source .venv/bin/activate  # Linux/Mac
# 或
.venv\\Scripts\\activate  # Windows

运行测试

uv run pytest

代码质量

# 格式化代码
uv run black deepseek_ocr/
uv run isort deepseek_ocr/

# 类型检查
uv run mypy deepseek_ocr/

# 代码检查
uv run flake8 deepseek_ocr/

API 参考

完整的 API 文档请参见 API_REFERENCE.md

许可证

本项目采用 MIT 许可证 - 详情请见 LICENSE 文件。

致谢

  • DeepSeek AI 提供的优秀 OCR 模型

免责声明:这是一个非官方的第三方 SDK,与 DeepSeek AI 或任何 API 服务提供商无关联。用户需自行选择 API 提供商并遵守提供商的服务条款。

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

deepseek_ocr-0.2.0.tar.gz (25.8 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

deepseek_ocr-0.2.0-py3-none-any.whl (19.5 kB view details)

Uploaded Python 3

File details

Details for the file deepseek_ocr-0.2.0.tar.gz.

File metadata

  • Download URL: deepseek_ocr-0.2.0.tar.gz
  • Upload date:
  • Size: 25.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.10.19

File hashes

Hashes for deepseek_ocr-0.2.0.tar.gz
Algorithm Hash digest
SHA256 65a2cc9fe86965b3b5cc3b081b71eb2aaabee3c6874d465001d6ab8250fa6495
MD5 c043bd944fce997f25cd702a1b453958
BLAKE2b-256 97c0d790ae0040752fe0f7d90ea7290bf21869c4ed8f9ff32e01faf35c8a9a86

See more details on using hashes here.

File details

Details for the file deepseek_ocr-0.2.0-py3-none-any.whl.

File metadata

  • Download URL: deepseek_ocr-0.2.0-py3-none-any.whl
  • Upload date:
  • Size: 19.5 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/6.2.0 CPython/3.10.19

File hashes

Hashes for deepseek_ocr-0.2.0-py3-none-any.whl
Algorithm Hash digest
SHA256 9e645ca7a93e75fdba9572439bee714cdec24857766ce4aa346c086f38733ed5
MD5 838e772fa7bc1711bdb2531819e2d65e
BLAKE2b-256 0118301e932dc50be65e158e48bfe92656a892d0b864c073bf5b89bd982daa56

See more details on using hashes here.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page