setup.py

#!/usr/bin/env python

"""Setup script for the package."""

import logging
import os
import sys
from pathlib import Path

import setuptools

PACKAGE_NAME = "vecs"
MINIMUM_PYTHON_VERSION = (3, 8, 0, "", 0)


def check_python_version():
    """Exit when the Python version is too low."""
    if sys.version_info < MINIMUM_PYTHON_VERSION:
        sys.exit(
            "At least Python {0}.{1}.{2} is required.".format(
                *MINIMUM_PYTHON_VERSION[:3]
            )
        )


def read_package_variable(key, filename="__init__.py"):
    """Read the value of a variable from the package without importing."""
    module_path = os.path.join("src", PACKAGE_NAME, filename)
    with open(module_path) as module:
        for line in module:
            parts = line.strip().split(" ", 2)
            if parts[:-1] == [key, "="]:
                return parts[-1].strip("'").strip('"')
    logging.warning("'%s' not found in '%s'", key, module_path)
    raise KeyError(key)


check_python_version()

long_description = (Path(__file__).parent / "README.md").read_text()

REQUIRES = [
    "pgvector==0.1.*",
    "sqlalchemy==2.*",
    "psycopg2-binary==2.9.*",
    "flupy==1.*",
    "deprecated==1.2.*",
]


setuptools.setup(
    name=read_package_variable("__project__"),
    version=read_package_variable("__version__"),
    description="pgvector client",
    long_description=long_description,
    long_description_content_type="text/markdown",
    url="https://github.com/supabase/vecs",
    author="Oliver Rice",
    packages=setuptools.find_packages("src", exclude="tests"),
    package_dir={"": "src"},
    package_data={"": ["py.typed"]},
    tests_require=["pytest"],
    license="MIT",
    classifiers=[
        "Development Status :: 4 - Beta",
        "Natural Language :: English",
        "Operating System :: OS Independent",
        "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",
    ],
    install_requires=REQUIRES,
    extras_require={
        "dev": ["pytest", "parse", "numpy", "pytest-cov"],
        "docs": [
            "mkdocs",
            "pygments",
            "pymdown-extensions",
            "pymarkdown",
            "mike",
        ],
        "text_embedding": ["sentence-transformers==2.*"],
    },
)