pyproject.toml

[build-system]
requires = ["setuptools>=61.0"]
build-backend = "setuptools.build_meta"

[project]
name = "superduperdb"
description = "🔮 Super-power your database with AI 🔮"
readme = "README.md"
license = {file = "LICENSE"}
maintainers = [{name = "Duncan Blythe", email = "opensource@superduperdb.com"}]
keywords = [
    "mongo",
    "mongodb",
    "pymongo",
    "data-science",
    "machine-learning",
    "mlops",
    "vector-database",
    "ai",
]
classifiers = [
    "Development Status :: 3 - Alpha",
    "Environment :: GPU :: NVIDIA CUDA",
    "License :: OSI Approved :: Apache Software License",
    "Operating System :: OS Independent",
    "Programming Language :: Python :: 3.8",
    "Programming Language :: Python :: 3.9",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: Implementation :: CPython",
    "Topic :: Database",
    "Topic :: Scientific/Engineering :: Artificial Intelligence",
    "Topic :: Software Development :: Libraries :: Python Modules",
    "Typing :: Typed"
]
requires-python = ">=3.8"
dynamic = ["version"]
dependencies = [
    "boto3>=1.16",
    "dask[distributed]>=2022.6.0",
    "dill>=0.3.6",
    "fil>=1.1.0",
    "networkx>=2.8.8",
    "requests>=2.22",  # lower bound from openai and boto3
    "tqdm>=4.64.1",
    "typer>=0.7.0",
    "pylance>=0.6.1",
    "readerwriterlock>=1.0.9",
    "flask>=2.2.3",
    "flask-cors>=3.0.10",
    "flask-httpauth>=4.7.0",
    "werkzeug>=2.2.3",
    "pydantic>=1",
    "pymongo>=4.3.3",
    "numpy>=1.24.3",
    "overrides>=7",
    "tenacity>=8.1.0",
    "scikit-learn>=1.1.3",
    "openai>=0.27.6",
    "pillow>=9.4.0",
    "safer>=4.8.0",
    "semver>=3",
    "mongomock>=4.1.2",
    "ibis-framework[sqlite]>=5.1.0",
    "ibis-framework[duckdb]>=5.1.0"
]

[project.optional-dependencies]
apis = [
    "cohere",
    "anthropic",
    ]
torch = [
    "torch>=2.0.0,!=2.0.1",
    "accelerate>=0.20.1",
    "transformers>=4.29.1",
    "torchvision",
]
# TODO lots of these now unused
docs = [
    "furo",
    "ipython",
    "myst_parser",
    "nbsphinx",
    "pandoc",
    "sphinx",
    "sphinx-autodoc-typehints",
    "sphinx_copybutton",
    "sphinxcontrib-mermaid",
    "nbsphinx-link",
]
typing = [
    "mypy>=1.3.0",
    "types-pillow>=9.5.0.4",
    "types-requests>=2.31.0.1",
    "types-tqdm>=4.65.0.1",
    "monkeytype",
    "boto3-stubs>=1.26.154",
]
lint = [
    "black>=23.3",
    "interrogate>=1.5.0",
    "ruff>=0.0.267",
]
tests = [
    "fastapi>=0.103.1",
    "httpx>=0.24.1",
    "impall>=1.2.0",
    "lorem>=0.1.1",
    "pytest>=7.3.1",
    "pytest-cov>=2.12.1",
    "pytest-xdist>=3.3.1",
    "nbval>=0.10.0",
    "tdir>=1.6",
    "vcrpy",
    "pytest-asyncio",
    "urllib3<2",  # see https://github.com/urllib3/urllib3/issues/3053
]
dev = ["superduperdb[apis,docs,lint,tests,typing,torch]"]

[project.urls]
homepage = "https://www.superduperdb.com/"
documentation = "https://docs.superduperdb.com"
source = "https://github.com/SuperDuperDB/superduperdb/"

[tool.setuptools.packages.find]
include = ["superduperdb*"]

[tool.setuptools.dynamic]
version = {attr = "superduperdb.__version__"}

[tool.black]
skip-string-normalization = true
target-version = ["py38"]

[tool.mypy]
ignore_missing_imports = true
no_implicit_optional = true
warn_unused_ignores = true
# disallow_any_generics = true
# disallow_any_unimported = true
# disallow_untyped_calls = true
# disallow_untyped_defs = true
# ignore_missing_imports = false
# warn_return_any = true

[tool.pytest.ini_options]
addopts = "-W ignore"

[tool.interrogate]
# Uncomment these next lines to generate a badge
#
# badge-format = "svg"
# generate-badge = "docs"

color = true
exclude = []
fail-under = 36.1
ignore-magic = true
ignore-nested-classes = false
ignore-nested-functions = true
ignore-private = true
ignore-property-decorators = true
ignore-regex = []
ignore-semiprivate = true
omit-covered-files = true
quiet = false
verbose = 0
whitelist-regex = []

[tool.ruff]
extend-select = [
    "I", # Missing required import (auto-fixable)
]

[tool.ruff.isort]
combine-as-imports = true