DataMind/pyproject.toml at main · OpenDCAI/DataMind · 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
122
123
124
125
126
127
128
[build-system]
requires = ["setuptools>=68", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "datamind"
description = "Composable inference-time data framework for LLM agents — KB / Graph / DB / Skills / Memory / Hooks behind one tool registry"
requires-python = ">=3.11"
readme = {file = "README.md", content-type = "text/markdown"}
license = {text = "Apache-2.0"}
authors = [
    {name = "Hao Liang", email = "hao.liang@stu.pku.edu.cn"},
    {name = "Wentao Zhang", email = "wentao.zhang@pku.edu.cn"},
]
keywords = [
    "AI", "agent", "RAG", "GraphRAG", "Text2SQL",
    "knowledge-graph", "memory", "MCP", "data-agent", "claude",
]
classifiers = [
    "Development Status :: 4 - Beta",
    "Intended Audience :: Developers",
    "Intended Audience :: Science/Research",
    "License :: OSI Approved :: Apache Software License",
    "Operating System :: OS Independent",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Programming Language :: Python :: 3 :: Only",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
    "Topic :: Software Development :: Libraries :: Python Modules",
]
dynamic = ["version"]
dependencies = [
    # Model client — we drive the agent loop ourselves via the Anthropic
    # Messages API (/v1/messages). This keeps us decoupled from the `claude`
    # CLI, which on some installs is a vendor-rebranded binary that ignores
    # ANTHROPIC_API_KEY.
    "anthropic>=0.40",
    "httpx[http2]>=0.27",

    # Config / validation / logging primitives
    "pydantic>=2.6",
    "pydantic-settings>=2.2",

    # KB
    "chromadb>=1.0",
    "rank-bm25>=0.2",

    # Graph
    "networkx>=3.0",

    # DB abstraction (dialect providers pluggable)
    "sqlalchemy>=2.0",

    # SQL parsing for hooks (DestructiveSqlHook)
    "sqlglot>=22",

    # Server + CLI
    "fastapi>=0.110",
    "uvicorn[standard]>=0.27",
    "typer>=0.12",
    "rich>=13",

    # Async primitives
    "anyio>=4.2",
]

[project.urls]
Homepage      = "https://github.com/OpenDCAI/DataMind"
Documentation = "https://haolpku.github.io/DataMind-Doc/"
Repository    = "https://github.com/OpenDCAI/DataMind"
Issues        = "https://github.com/OpenDCAI/DataMind/issues"

[project.optional-dependencies]
# Legacy v0.1 stack — kept so the old main.py / server.py / modules/* keep
# working side-by-side with the new datamind/ package.
legacy = [
    "llama-index>=0.11",
]

# Dialect / provider extras — installed on demand, no hard requirement
mysql = ["pymysql>=1.1", "cryptography>=42"]
postgres = ["psycopg[binary]>=3.1"]
voyage = ["voyageai>=0.2"]
huggingface = ["sentence-transformers>=2.5"]
dev = [
    "pytest>=8.0",
    "pytest-asyncio>=0.23",
    "build>=1.0",
    "twine>=4.0",
]

[project.scripts]
datamind = "datamind.cli:app"

[tool.setuptools.dynamic]
version = {attr = "datamind.__version__"}

# Explicit include + exclude. `data*` was a foot-gun: it matches `datamind`
# itself and silently produced empty wheels. We keep the include narrow
# (only the `datamind` package and its sub-packages) and exclude only the
# in-package test directory.
[tool.setuptools.packages.find]
where = ["."]
include = ["datamind", "datamind.*"]
exclude = ["datamind.tests", "datamind.tests.*"]

# Bundle non-Python resources so `pip install datamind` ships a working
# server (browser UI under datamind/static/) and a default skill catalog
# (datamind/skills/<name>/SKILL.md). Repo-side static/ and .claude/skills/
# remain authoritative for development; server.py / config.py prefer them
# when running from a checkout.
[tool.setuptools.package-data]
datamind = [
    "static/*.html",
    "static/*.css",
    "static/*.js",
    "skills/*/SKILL.md",
    "skills/*/*.md",
    "skills/*/*.py",
    "skills/*/*.yaml",
    "skills/*/*.yml",
    "skills/*/*.json",
]

[tool.pytest.ini_options]
asyncio_mode = "auto"
testpaths = ["datamind/tests"]