deepseek-ocr 0.1.1

A simple and efficient Python SDK for DeepSeek-OCR API

DeepSeek-OCR-SDK

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. Common providers include SiliconFlow, DeepSeek Official API, and others. Please check your provider's documentation for the correct base_url.

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")

# Chinese document
text = client.parse("chinese.pdf", mode="free_ocr", chinese_hint=True)

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

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
Chinese documents	FREE_OCR + chinese_hint=True	Best Chinese support
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

Available API Providers:

• SiliconFlow: https://api.siliconflow.cn/v1/chat/completions
• DeepSeek Official: https://api.deepseek.com/v1/chat/completions
• Others: Check your provider's documentation

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/

Performance Benchmarks

See BENCHMARKS.md for detailed performance comparisons with MinerU and Docling.

API Reference

See API_REFERENCE.md for complete API documentation.

Contributing

Contributions are welcome! Please feel free to submit a Pull Request.

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 端点。常见提供商包括硅基流动、DeepSeek 官方 API 等。请查看您的提供商文档以获取正确的 base_url。

架构图

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")

# 中文文档
text = client.parse("chinese.pdf", mode="free_ocr", chinese_hint=True)

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

异步用法

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	避免截断问题
中文文档	FREE_OCR + chinese_hint=True	最佳中文支持
混合内容	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

可用的 API 提供商：

• 硅基流动（SiliconFlow）：https://api.siliconflow.cn/v1/chat/completions
• DeepSeek 官方：https://api.deepseek.com/v1/chat/completions
• 其他：请查看您的提供商文档

编程式配置

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/

性能基准

详细的性能对比（与 MinerU、Docling 等）请参见 BENCHMARKS.md。

API 参考

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

贡献

欢迎贡献！请随时提交 Pull Request。

许可证

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

致谢

• DeepSeek AI 提供的优秀 OCR 模型

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