A simple and efficient Python SDK for DeepSeek-OCR API
Project description
DeepSeek-OCR-SDK
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
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
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
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
65a2cc9fe86965b3b5cc3b081b71eb2aaabee3c6874d465001d6ab8250fa6495
|
|
| MD5 |
c043bd944fce997f25cd702a1b453958
|
|
| BLAKE2b-256 |
97c0d790ae0040752fe0f7d90ea7290bf21869c4ed8f9ff32e01faf35c8a9a86
|
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
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
9e645ca7a93e75fdba9572439bee714cdec24857766ce4aa346c086f38733ed5
|
|
| MD5 |
838e772fa7bc1711bdb2531819e2d65e
|
|
| BLAKE2b-256 |
0118301e932dc50be65e158e48bfe92656a892d0b864c073bf5b89bd982daa56
|