pyproject.toml

[tool.poetry]
name = "ml-dev-bench"
version = "0.1.0"
description = "A framework for evaluating AI agents"
authors = ["MLDevBench"]
readme = "README.md"
packages = [
    { include = "ml_dev_bench" },
    { include = "calipers" },
    { include = "runtime" }
]

[tool.poetry.dependencies]
python = "^3.12"
pydantic = "^2.0"
PyYAML = "^6.0"
python-dotenv = "^1.0.0"
litellm = "^1.52.3"
langchain-community = "^0.3.8"
composio-core = { git = "https://github.com/ml-dev-bench/composio/", branch = "ml_dev_bench_master", subdirectory = "python" }
GitPython = "^3.1.0"
pandas = "*"
Pillow = "*"
hydra-core = "^1.3.2"
google-cloud-aiplatform = ">=1.38"

[tool.poetry.group.aide]
optional = true

[tool.poetry.group.aide.dependencies]
accelerate = "*"
aideml = { git = "https://github.com/ml-dev-bench/aideml/", branch = "ml_dev_bench_compat", rev = "d72c265d1574c56e4326a37a93ac3c521141c665" }
streamlit = ">=1.24.0"
openai = ">=1.0.0"
anthropic = ">=0.5.0"
torch = "==2.2.2"
torchvision = "==0.17.2"
torchaudio = "==2.2.2"
torchtext = "==0.17.2"
torchmetrics = "==1.3.1"
pytorch-lightning = "==2.2.1"
transformers = "<5.0.0"
kornia = "<0.8.0"
timm = "<0.10.0"
fastai = "<3.0.0"
datasets = "*"

[tool.poetry.group.react-agent.dependencies]
accelerate = "*"
langgraph = "^0.2.53"
langgraph-checkpoint-sqlite = "^2.0.1"
langchain-openai = "^0.2.10"
langchain-core = "^0.3.21"
langchain-anthropic = "^0.3.0"
composio-langchain = "^0.6.10"
datasets = "*"

[tool.poetry.group.openhands-agent]
optional = true

[tool.poetry.group.openhands-agent.dependencies]
openhands-ai = { git = "https://github.com/ml-dev-bench/OpenHands", branch = "ml-dev-bench-v0.21.1" }
playwright = "^1.41.0"


[tool.poetry.group.dev.dependencies]
ruff = "0.9.2"
mypy = "1.13.0"
pre-commit = "4.0.1"
build = "*"
jupyter = "1.1.1"
notebook = ">=7.0.0"
pytest = "^7.4.0"
pytest-asyncio = "^0.21.0"
pytest-cov = "^4.1.0"

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

[tool.autopep8]
# autopep8 fights with mypy on line length issue
ignore = [ "E501" ]

[tool.black]
# prevent black (if installed) from changing single quotes to double quotes
skip-string-normalization = true

[tool.mypy]
python_version = "3.12"
warn_return_any = true
warn_unused_configs = true
disallow_untyped_defs = true
check_untyped_defs = true