-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathpyproject.toml
70 lines (60 loc) · 2.07 KB
/
pyproject.toml
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
[tool.poetry]
name = "dibench"
version = "0.1.0"
description = "Evaluating LLM-based Code Intelligence on Dependency Inference in Repository Scope"
authors = ["DI-Bench <[email protected]>"]
license = "MIT"
readme = "README.md"
packages = [{include = "dibench"}]
[tool.poetry.dependencies]
python = "^3.10"
fire = "^0.6.0"
pygithub = "^2.4.0"
alive-progress = "^3.1.5"
tabulate = "^0.9.0"
toml = "^0.10.2"
poetry = "^1.8.3"
tree-sitter = "0.21.3"
tree_sitter_languages = ">=1.10.2"
datasets = "^3.0.1"
lxml = "^5.3.0"
rich = "^13.9.4"
tenacity = "^9.0.0"
pytest = { version = "^8.3.3", optional = true }
pre-commit = { version = "^3.8.0", optional = true }
docker = { version = "^7.1.0", optional = true }
anthropic = { version = "^0.34.2", optional = true }
openai = { version = "^1.50.0", optional = true }
transformers = { version = "^4.45.1", optional = true }
google-generativeai = { version = "^0.8.2", optional = true }
stop-sequencer = { version = "^1.2.3", optional = true }
vllm = { version = "^0.6.2", optional = true }
tree-sitter-c-sharp = { version = "^0.23.1", optional = true }
tree-sitter-rust = { version = "^0.23.2", optional = true }
tree-sitter-python = { version = "^0.23.4", optional = true }
tree-sitter-typescript = { version = "^0.23.2", optional = true }
[tool.poetry.group.dev.dependencies]
pytest = "^8.3.3"
pre-commit = "^3.8.0"
[tool.poetry.group.eval.dependencies]
docker = "^7.1.0"
[tool.poetry.group.llm.dependencies]
anthropic = "^0.34.2"
openai = "^1.50.0"
transformers = "^4.45.1"
google-generativeai = "^0.8.2"
stop-sequencer = "^1.2.3"
vllm = "^0.6.2"
[tool.poetry.group.pattern.dependencies]
tree-sitter-c-sharp = "^0.23.1"
tree-sitter-rust = "^0.23.2"
tree-sitter-python = "^0.23.4"
tree-sitter-typescript = "^0.23.2"
[tool.poetry.extras]
dev = ["pytest", "pre-commit"]
eval = ["docker"]
llm = ["anthropic", "openai", "transformers", "google-generativeai", "stop-sequencer", "vllm"]
pattern = ["tree-sitter-c-sharp", "tree-sitter-rust", "tree-sitter-python", "tree-sitter-typescript"]
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"