GraphRAG/pyproject.toml at main · MvdB/GraphRAG · 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
[build-system]
requires = ["setuptools>=68", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "graphrag"
version = "0.1.0"
description = "Graph-based RAG system with PostgreSQL/AGE and MCP server"
readme = "README.md"
requires-python = ">=3.10"
license = { text = "MIT" }
dependencies = [
    "pydantic>=2.6",
    "pydantic-settings>=2.2",
    "asyncpg>=0.29",
]

[project.optional-dependencies]
watcher = [
    "watchdog>=4.0",
    "sentence-transformers>=3.0,<4.0",
    "transformers>=4.40,<4.49",
    "torch>=2.3",
    "mistune>=3.0",
]
mcp = [
    "mcp[cli]>=1.2",
    "sentence-transformers>=3.0,<4.0",
    "transformers>=4.40,<4.49",
    "torch>=2.3",
    "uvicorn[standard]>=0.29",
    "starlette>=0.37",
]
enricher = [
    "openai>=1.0",
]
dev = [
    "pytest>=8.0",
    "pytest-asyncio>=0.23",
    "pytest-mock>=3.12",
    "ruff>=0.4",
    "mypy>=1.9",
    "graphrag[watcher,mcp]",
]

[tool.setuptools.packages.find]
where = ["src"]

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

[tool.ruff]
line-length = 100
target-version = "py311"

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

[tool.mypy]
python_version = "3.11"
strict = true
ignore_missing_imports = true