pyproject.toml

[project]
name = "datacommons-agent-toolkit"
version = "0.0.1"
description = 'Data Commons tools using the Model Context Protocol (MCP).'
readme = "README.md"
requires-python = ">=3.11, <3.14"
license = "MIT"
license-files = ["LICENSE"]
keywords = []
authors = []
classifiers = [
  "Development Status :: 3 - Alpha",
  "Intended Audience :: Developers",
  "Programming Language :: Python :: 3",
]

dependencies = [
  "datacommons-mcp",
  "ruff>=0.12.5",
  "pytest>=8.4.1",
  "pytest-asyncio>=1.1.0",
  "pre-commit>=4.3.0",
]

[project.optional-dependencies]
test = [
  "google-adk",
  "google-adk[eval]",
  "pytest>=8.4.1",
  "pytest-asyncio>=1.1.0",
  "requests-mock",
  "rouge-score",
]

[tool.uv.sources]
datacommons-mcp = { workspace = true }

[[tool.uv.index]]
url = "https://pypi.org/simple"

[tool.uv.workspace]
members = ["packages/*"]

[tool.ruff]
# TODO(clincoln8): Implement lint checks for ipynb's (including pre-push hook).
exclude = [
    "**/*.ipynb",  # Ignores all notebooks
]

[tool.ruff.lint]
# Visit https://docs.astral.sh/ruff/rules/ for detailed rule descriptions.
extend-select = [
  "ERA",  # removes commented-out code from Python files.
  "FAST", # linting FastAPI specifics
  "YTT",  # Flake8 checks for misuse of `sys.version` or `sys.version_info`
  "ANN",  # Flake8 Type Annotation Checks
  "S",    # Automated security testing with bandit and flake8.
  "BLE",  # A flake8 extension that checks for blind except: statements
  "FBT",  # A flake8 plugin to detect boolean traps.
  "B",    # flake8-bugbear finds likely bugs and design problems.
  "A",    # flake8-builtins detects shadowed python builtins
  "COM",  # flake8 lint for trailing commas.
  "C4",   # flake8 plugin to help write better list/set/dict comprehensions
  "ISC",  # flake8 plugin to encourage correct string literal concatenation
  "LOG",  # flake8 plugin checks for issues with standard library logging module
  "G",    # flake8 plugin to encourage better formatting of built-in logging strings
  "PIE",  # flake8 misc lints
  "PT",   # flake8 plugin checks common pitfalls with pytest-based tests
  "RET",  # encourages good return practices
  "SIM",  # flake8-simplify checks common simplification patterns
  "TID",  # flake8-tidy-imports encourages tidier imports
  "TD",   # flake8-todos lints TODOs throughout the project for consistency
  "TC",   # flake8 plugin for managing type-checking imports & forward references
  "ARG",  # flake8-unused-arguments warns on unused function arguments
  "I",    # isort sorts Python imports
  "N",    # check PEP-8 naming conventions
  "PERF", # Pylint extension with performance anti-patterns
  "E",    # pycodestyle errors
  "W",    # pycodestyle warnings
  "F",    # pyflakes analyzes and detects various errors
  "PL",   # pylint
  "Q",    # defaults to double quotes
  "UP", # Encourage syntax supported by newer python versions
  # "CPY", # check for copyright headers # uncomment once graduated from preview
]
extend-ignore = [
  "TD003",  # missing-todo-link: Missing issue link for this TODO
  "PLC",    # pylint conventions
  "PLR",    # pylint refactor
  "COM812", # missing-trailing-comma may conflict with formatter
  "E501",   # line-too-long
  "BLE001"  # Allow catching blind exceptions (`Exception`)
]

[tool.ruff.lint.per-file-ignores]
"packages/*/tests/**/*.py" = [
    "S101",   # Allow using `assert` in pytests
    "ANN001", # Ignore missing type annotations in test files
    "ANN201", # Ignore missing return types in public test functions
    "ANN202", # Ignore missing return types in private teest functions
]
"packages/*/evals/**/test_*.py" = [
    "S101",   # Allow using `assert` in pytests
    "ANN001", # Ignore missing type annotations in test files
    "ANN201", # Ignore missing return types in public test functions
    "ANN202", # Ignore missing return types in private test functions,
]