pyproject.toml

[project]
name = "vellum-ai"

[tool.poetry]
name = "vellum-ai"
version = "1.14.7"
description = ""
readme = "README.md"
authors = []
keywords = []
license = "MIT"
classifiers = [
    "Intended Audience :: Developers",
    "Programming Language :: Python",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.8",
    "Programming Language :: Python :: 3.9",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
    "Operating System :: OS Independent",
    "Operating System :: POSIX",
    "Operating System :: MacOS",
    "Operating System :: POSIX :: Linux",
    "Operating System :: Microsoft :: Windows",
    "Topic :: Software Development :: Libraries :: Python Modules",
    "Typing :: Typed",
    "License :: OSI Approved :: MIT License"
]
urls = { Repository = "https://github.com/vellum-ai/vellum-python-sdks" }
packages = [
    { include = "vellum", from = "src" },
    { include = "vellum_cli", from = "ee" },
    { include = "vellum_ee", from = "ee" },
]

[tool.poetry.dependencies]
python = "^3.9"
Jinja2 = ">=3.1.0,<4.0.0"
click = ">=8.1.7,<9.0.0"
docker = ">=7.1.0,<8.0.0"
httpx = ">=0.21.2"
jsonschema = ">=4.0.0"
mypy = { version = ">=1.13.0", optional = true}
orderly-set = "^5.2.2"
publication = "0.0.3"
pydantic = ">= 1.9.2"
pydantic-core = ">=2.18.2"
pydash = ">=7.0.0,<8.0.0"
python-dateutil = ">=2.8.0,<3.0.0"
python-dotenv = ">=1.0.0,<2.0.0"
pytz = ">=2022.0,<2026.0"
pyyaml = ">=6.0.0,<7.0.0"
requests = ">=2.31.0,<3.0.0"
tomli = ">=2.0.0,<3.0.0"
typing_extensions = ">= 4.0.0"
zuban = { version = ">=0.4.2", optional = true}

[tool.poetry.group.dev.dependencies]
mypy = "==1.13.0"
pytest = "^7.4.0"
pytest-asyncio = "^0.23.5"
python-dateutil = "^2.9.0"
types-python-dateutil = "^2.9.0.20240316"
black = "24.8.0"
deepdiff = "8.0.1"
flake8 = "7.1.1"
flake8-breakpoint = "1.1.0"
flake8-isort = "6.1.1"
flake8-print = "5.0.0"
flake8-pyproject = "1.2.3"
flake8-unused-arguments = "^0.0.13"
ipdb = "0.13.13"
ipython = "8.12.3"
isort = "5.13.2"
pre-commit = "3.5.0"
pytest-cov = "^4.0.0"
pytest-mock = "3.10.0"
pytest-timeout = "2.3.1"
pytest-xdist = "^3.8.0"
requests-mock = "1.11.0"
ruff = "==0.11.5"
tomli_w = "1.1.0"
ty = ">=0.0.1a29"
types-PyYAML = "6.0.12.20240808"
types-pytz = "2024.1.0.20240417"
types-requests = "2.32.0.20240712"

[tool.pytest.ini_options]
testpaths = [ "tests" ]
asyncio_mode = "auto"
filterwarnings = [
    "ignore::pytest.PytestCollectionWarning"
]

[tool.mypy]
plugins = [
    "pydantic.mypy",
    "vellum.plugins.vellum_mypy",
]
check_untyped_defs = true
exclude = ["venv", "examples/workflows", "ee/codegen/python_file_merging/tests/fixtures", "ee/codegen_integration/fixtures/simple_scheduled_trigger"]  #TODO: Uncomment trigger test once we fix mypy issue with graph: https://linear.app/vellum/issue/APO-2127/fix-mypy-issue-of-graph-types
overrides = [
    { module = "deepdiff.*", ignore_missing_imports = true },
    { module = "docker.*", ignore_missing_imports = true },
    { module = "jsonschema.*", ignore_missing_imports = true },
    { module = "setuptools.*", ignore_missing_imports = true },
]
[tool.ruff]
line-length = 120

[tool.ruff.lint]
select = [
    "E",  # pycodestyle errors
    "F",  # pyflakes
    "I",  # isort
]
ignore = [
    "E402",  # Module level import not at top of file
    "E501",  # Line too long
    "E711",  # Comparison to `None` should be `cond is not None`
    "E712",  # Avoid equality comparisons to `True`; use `if ...:` checks
    "E721",  # Use `is` and `is not` for type comparisons, or `isinstance()` for insinstance checks
    "E722",  # Do not use bare `except`
    "E731",  # Do not assign a `lambda` expression, use a `def`
    "F821",  # Undefined name
    "F841"   # Local variable ... is assigned to but never used
]

[tool.ruff.lint.isort]
section-order = ["future", "standard-library", "third-party", "first-party"]

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

[tool.poetry.extras]
mypy=["mypy"]
zuban=["zuban"]

[tool.poetry.scripts]
vellum = "vellum_cli:main"
generate_node_definitions = "vellum_ee.scripts.generate_node_definitions:main"

[tool.isort]
skip = ["src/vellum/client", "tests/client", ".venv", "src/vellum/__init__.py", "examples/workflows", "src/vellum/client/raw_client.py"]
multi_line_output = 3
force_grid_wrap = 0
line_length = 120
combine_as_imports = true
force_sort_within_sections = true
include_trailing_comma = true
use_parentheses = true
known_testing = "pytest"
known_typing = ["typing", "typing_extensions"]
known_vellum = ["vellum", "vellum_cli", "vellum_ee"]
default_section = "THIRDPARTY"
sections = ["FUTURE", "TESTING", "STDLIB", "TYPING", "THIRDPARTY", "VELLUM", "FIRSTPARTY", "LOCALFOLDER"]
no_lines_before = ["STDLIB", "TYPING"]
sort_order = "native"

[tool.black]
line-length = 120
target-version = [
    "py38",
]
include = ".*\\.py$"
exclude = "(.git|venv|src/vellum/client|tests/client|scripts|examples/workflows|ee/codegen/python_file_merging/tests/fixtures)"

[tool.flake8]
max-line-length = 120
exclude = ".git,.venv,ee/codegen/node_modules,ee/codegen_integration/fixtures,scripts,src/vellum/__init__.py,src/vellum/client,src/vellum/core,src/vellum/environment.py,src/vellum/errors,src/vellum/raw_client.py,src/vellum/resources,src/vellum/types,tests/client,examples/workflows,ee/codegen/python_file_merging/tests/fixtures"
ignore = "E203, W503, E704, U100, U101"

[tool.coverage.run]
source = ["src/vellum", "ee/vellum_cli", "ee/vellum_ee"]
omit = [
    "src/vellum/client/*",
    "src/vellum/core/*",
    "src/vellum/environment.py",
    "src/vellum/errors/*",
    "src/vellum/resources/*",
    "src/vellum/types/*",
    "tests/*",
    "examples/*",
    "scripts/*",
    "*/tests/*",
    "*/__pycache__/*",
]

[tool.coverage.report]
fail_under = 25
show_missing = true
skip_covered = false

[tool.coverage.html]
directory = "htmlcov"