SMOLTRACE/pyproject.toml at main · Mandark-droid/SMOLTRACE · 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
[build-system]
requires = ["setuptools>=61.0"]
build-backend = "setuptools.build_meta"

[tool.setuptools]
packages = ["smoltrace", "smoltrace.exporters"]

[project]
name = "smoltrace"
version = "0.0.14"
authors = [
    {name = "Kshitij Thakkar", email = "kshitijthakkar@rocketmail.com"}
]
description = "A comprehensive benchmarking and evaluation framework for smolagents."
readme = "README.md"
requires-python = ">=3.10"
classifiers = [
    "Programming Language :: Python :: 3",
    "License :: OSI Approved :: GNU Affero General Public License v3",
    "Operating System :: OS Independent",
]
dependencies = [
    "smolagents>=1.0.0",
    "datasets",
    "huggingface_hub",
    "opentelemetry-sdk",
    "genai-otel-instrument[openinference]>=0.1.22",
    "duckduckgo-search",
    "ddgs"
]

[project.optional-dependencies]
gpu = [
    "genai-otel-instrument[gpu]>=0.1.22",
    "nvidia-ml-py",
    "transformers>=4.36.0",
    "torch>=2.0.0",
    "accelerate>=0.20.0"
]
opensearch = [
    "opensearch-py>=2.4.0",
]
dev = [
    # Testing
    "pytest>=7.4.0",
    "pytest-cov>=4.1.0",
    "pytest-mock>=3.11.1",
    "pytest-asyncio>=0.21.1",

    # Code quality
    "black>=23.7.0",
    "isort>=5.12.0",
    "pylint>=2.17.5",
    "ruff>=0.0.285",
    "mypy>=1.4.1",
    "pre-commit>=3.5.0",

    # Build and release
    "build>=0.10.0",
    "twine>=4.0.2",

    # Security
    "safety>=2.3.5",
    "bandit>=1.7.5",
]

[project.urls]
"Homepage" = "https://github.com/Mandark-droid/SMOLTRACE"
"Bug Tracker" = "https://github.com/Mandark-droid/SMOLTRACE/issues"
"Documentation" = "https://github.com/Mandark-droid/SMOLTRACE#readme"
"Changelog" = "https://github.com/Mandark-droid/SMOLTRACE/blob/main/CHANGELOG.md"


[project.scripts]
smoltrace-eval = "smoltrace.cli:main"
smoltrace-cleanup = "smoltrace.cleanup:main"
smoltrace-copy-datasets = "smoltrace.copy_datasets:main"

[tool.black]
line-length = 100

[tool.isort]
profile = "black"
line_length = 100

[tool.ruff]
line-length = 100