graph-agent/pyproject.toml at master · hugegraph/graph-agent · 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
[tool.poetry]
name = "chat2graph"
version = "0.1.0"
description = "Chat2Graph: A Graph Native Agentic System."
authors = ["Florian Fan <fanzhidongyzby@gmail.com>"]
packages = [
    { include = "app" }
]

[tool.poetry.dependencies]
python = ">=3.10,<3.12"
uv = "^0.8.0"
openai = "^1.86.0"
litellm = {extras = ["proxy"], version = "^1.72.6"} # pip install "aiohttp>=3.12.13" # litellm has conflict with dbgpt which requires aiohttp = "3.8.4"
pydantic = "^2.11.7"
typing-inspect = ">=0.9.0"
pytest-asyncio = "^1.0.0"
chromadb = "^1.0.12"
neo4j = "^5.28.1"
sqlalchemy-utils = "^0.41.2"
matplotlib = "^3.10.3"
memfuse = "^0.3.3"
networkx = "^3.4.2"
mcp = "^1.9.3"
pyfiglet = "^1.0.3"
playwright = "^1.53.0"
python-magic = "^0.4.27"
google-generativeai = "^0.8.5"
google-genai = "^1.36.0"
datasets = "<=3.6.0"  # required by benchmark of huggingface

[[tool.poetry.source]]
name = "PyPI"
priority = "primary"

[[tool.poetry.source]]
name = "aliyun"
priority = "supplemental"
url = "https://mirrors.aliyun.com/pypi/simple/"

[build-system]
requires = ["poetry-core>=1.0.0"]
build-backend = "poetry.core.masonry.api"

[tool.poetry.group.dev.dependencies]
pytest = "^8.4.0"
ruff = "^0.11.13"
mypy = "^1.18.1"

[tool.poetry.group.service.dependencies]
flask = "3.1.1"
flask-sqlalchemy = "3.1.1"
flask-cors = "6.0.1"

[tool.poetry.group.test.dependencies]
pytest = "8.4.0"
pytest-cov = "6.2.1"
pytest-mock = "^3.14.1"

[tool.poetry.group.db-gpt.dependencies]
# dbgpt = { version = "0.7.3", extras = ["agent", "simple_framework", "framework"] }
# dbgpt-ext = { version = "0.7.3", extras = ["rag", "graph_rag", "storage_chromadb"] }

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

[tool.ruff.lint]
select = [
    "E",  # pycodestyle error
    "F",  # pyflakes
    "I",  # isort
    "B",  # flake8-bugbear
    "C4", # flake8-comprehensions
    "UP", # pyupgrade
    "EXE",
]
ignore = [
    "UP006",    # use List not list
    "UP035",
    "UP007",
    "UP045",
]

[tool.ruff.lint.isort]
combine-as-imports = true
force-sort-within-sections = true
known-first-party = ["app"]

[tool.ruff.format]
quote-style = "double"
indent-style = "space"
skip-magic-trailing-comma = false
line-ending = "auto"

[tool.pytest.ini_options]
testpaths = ["test"]
python_files = ["test_*.py"]
addopts = "-v"
asyncio_mode = "auto"  # Enable asyncio mode
markers = [
    "asyncio: mark test as async"
]