-
Notifications
You must be signed in to change notification settings - Fork 33
/
Copy pathpyproject.toml
70 lines (62 loc) · 1.92 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 = "continuous-eval"
version = "0.3.14post2"
description = "Open-Source Evaluation for GenAI Applications."
authors = ["Yi Zhang <[email protected]>", "Pasquale Antonante <[email protected]>"]
readme = "README.md"
license = "Apache-2.0"
packages = [{include = "continuous_eval"}]
[tool.poetry.dependencies]
python = ">=3.10,<3.13"
python-dotenv = "^1.0.0"
nltk = "^3.9.1"
rouge = "^1.0.1"
openai = "^1.58.1"
tqdm = "^4.66.1"
requests = "^2.31.0"
posthog = "^3.7.4"
pyyaml = "^6.0.1"
appdirs = "^1.4.4"
munkres = "^1.1.4"
thefuzz = "^0.22.1"
tenacity = ">=8.1.0,<9.0.0"
json-repair = "^0.31.0"
sqlglot = "^26.0.0"
tiktoken = "^0.8.0"
scikit-learn = "^1.6.0"
jinja2 = "^3.1.4"
# Optional model providers
boto3 = {version = "^1.35.77", optional = true}
azure-ai-inference = {version = "^1.0.0b6", optional = true}
anthropic = {version = "^0.40.0", optional = true}
cohere = {version = "^5.13.3", optional = true}
google-generativeai = {version = "^0.8.3", optional = true}
# Semantic
torch = {version = "^2.1.1", optional = true, python = "^3.11"}
transformers = {version = "^4.47.1", optional = true}
sentence-transformers = {version = "^3.3.1", optional = true, python = "^3.11"}
pandas = {version = "^2.2.3", optional = true}
sentencepiece = {version = "^0.2.0", optional = true}
[tool.poetry.extras]
semantic = ["torch", "transformers", "sentencepiece", "sentence-transformers", "pandas"]
bedrock = ["boto3"]
azure = ["azure-ai-inference"]
anthropic = ["anthropic"]
cohere = ["cohere"]
google = ["google-generativeai"]
[tool.poetry.group.dev.dependencies]
pytest = "^7.4.3"
pre-commit = "^4.0.1"
ruff = "^0.7.4"
codeflash = ">=0.9.0"
[tool.ruff]
line-length = 80
[tool.codeflash]
module-root = "continuous_eval"
tests-root = "tests"
test-framework = "pytest"
ignore-paths = []
formatter-cmds = ["ruff check --exit-zero --fix $file", "ruff format $file"]
[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"