DocAtlas-Harness/pyproject.toml at main · microsoft/DocAtlas-Harness · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
[build-system]
requires = ["setuptools>=83.0.0", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "docatlas"
version = "0.3.0"
description = "DocAtlas — a document-understanding harness with portable Agent Skills."
readme = "README.md"
license = "MIT"
authors = [{ name = "Microsoft Research" }]
keywords = ["agents", "document-understanding", "multimodal", "pdf", "retrieval"]
requires-python = ">=3.11,<3.14"
classifiers = [
    "Development Status :: 4 - Beta",
    "Intended Audience :: Developers",
    "Intended Audience :: Science/Research",
    "Operating System :: OS Independent",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Programming Language :: Python :: 3.13",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
]
dependencies = [
    # ── LLM / runtime core ──
    "openai>=1.40",
    "azure-identity>=1.15",
    "python-dotenv>=1.0",
    "pyyaml>=6.0",
    "tqdm>=4.66",

    # ── PDF I/O (Read skill, build-md, build-tree shared helpers) ──
    "pypdf>=6.21.1",
    "pymupdf>=1.23",
    "pillow>=13.0.0",
    "pycryptodome>=3.20",   # optional crypto backend for encrypted PDFs

    # ── PDF → markdown / figures (build-md, Docling backend) ──
    # Heron layout + TableFormer; OCR is opt-in. Uses a GPU automatically
    # if torch sees one. Model weights are cached from HuggingFace on first use.
    # These Docling versions support Pillow 13 throughout the dependency closure.
    "docling>=2.142.1,<3",
    "docling-core>=2.104.0,<3",

    # ── Tree construction (build-tree wraps PageIndex) ──
    # PageIndex is vendored under docatlas/_vendor/pageindex; see its
    # UPSTREAM.md for provenance. No separate PyPI install is needed.
    # We list pageindex's only non-trivial dep (tiktoken) here so it lands
    # in our uv.lock.
    "tiktoken>=0.7",

    # ── Security floors for the provider/document dependency closure ──
    # These already-transitive packages are explicit so built wheel metadata
    # also rejects vulnerable versions; uv-only constraints would not ship.
    "anyio>=4.14.2",
    "certifi>=2026.10.19",
    "fsspec>=2026.6.0",
    "pyjwt>=2.15.1",
    "urllib3>=2.8.0",
]

[project.optional-dependencies]
gpu = [                        # explicit Torch floor for GPU deployments
    "torch>=2.13.0",
]
dev = [
    "pytest>=8.0",
    "pytest-asyncio>=0.23",
]

[project.scripts]
docatlas = "docatlas.__main__:main"
harness = "docatlas.__main__:main"

[project.urls]
Homepage = "https://github.com/microsoft/DocAtlas-Harness"
Repository = "https://github.com/microsoft/DocAtlas-Harness"
Issues = "https://github.com/microsoft/DocAtlas-Harness/issues"

[tool.setuptools.packages.find]
where = ["."]
include = ["docatlas*"]

[tool.setuptools.package-data]
"docatlas.profiles" = ["*.yaml"]
"docatlas.skills" = ["README.md"]
"docatlas.skills._common" = ["README.md"]
"docatlas.skills.note" = ["SKILL.md", "tool.json"]
"docatlas.skills.read" = ["SKILL.md", "tool.json"]
"docatlas.skills.review" = ["SKILL.md", "tool.json"]
"docatlas.skills.search" = ["SKILL.md", "tool.json"]
"docatlas._vendor.pageindex" = ["config.yaml", "LICENSE", "UPSTREAM.md"]

[tool.pytest.ini_options]
testpaths = ["tests"]

[tool.ruff]
extend-exclude = ["docatlas/_vendor"]
line-length = 100
target-version = "py310"

[tool.ruff.lint]
select = ["E", "F", "I", "UP"]
ignore = ["E501"]

[tool.mypy]
check_untyped_defs = true
exclude = "^docatlas/_vendor/"
ignore_missing_imports = true
python_version = "3.10"

[[tool.mypy.overrides]]
module = "docatlas._vendor.*"
ignore_errors = true

[tool.coverage.run]
omit = ["docatlas/_vendor/*"]

[tool.coverage.report]
exclude_also = ["if TYPE_CHECKING:", "if __name__ == .__main__.:"]