braintrust-sdk-python/py/setup.py at main · braintrustdata/braintrust-sdk-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
import os

import setuptools

dir_name = os.path.abspath(os.path.dirname(__file__))

version_contents = {}
with open(os.path.join(dir_name, "src", "braintrust", "version.py"), encoding="utf-8") as f:
    exec(f.read(), version_contents)

with open(os.path.join(dir_name, "README.md"), "r", encoding="utf-8") as f:
    long_description = f.read()

install_requires = [
    "GitPython",
    "requests",
    "chevron",
    "tqdm",
    "exceptiongroup>=1.2.0",
    "python-dotenv",
    "sseclient-py",
    "python-slugify",
    "typing_extensions>=4.1.0",
    "wrapt",
]

extras_require = {
    "cli": ["boto3", "psycopg2-binary", "uv", "starlette", "uvicorn"],
    "doc": ["pydoc-markdown"],
    "openai-agents": ["openai-agents"],
    "otel": ["opentelemetry-api", "opentelemetry-sdk", "opentelemetry-exporter-otlp-proto-http"],
    # orjson is not compatible with PyPy, so we exclude it for that platform
    "performance": ["orjson; platform_python_implementation != 'PyPy'"],
    "temporal": ["temporalio>=1.19.0; python_version>='3.10'"],
}

extras_require["all"] = sorted({package for packages in extras_require.values() for package in packages})

setuptools.setup(
    name="braintrust",
    version=version_contents["VERSION"],
    author="Braintrust",
    author_email="[email protected]",
    description="SDK for integrating Braintrust",
    long_description=long_description,
    long_description_content_type="text/markdown",
    url="https://www.braintrust.dev",
    project_urls={
        "Source Code": "https://github.com/braintrustdata/braintrust-sdk-python",
        "Bug Tracker": "https://github.com/braintrustdata/braintrust-sdk-python/issues",
    },
    classifiers=[
        "Programming Language :: Python :: 3",
        "Programming Language :: Python :: 3.10",
        "Operating System :: OS Independent",
    ],
    package_dir={"": "src"},
    packages=setuptools.find_packages(where="src"),
    package_data={"braintrust": ["py.typed"]},
    python_requires=">=3.10.0",
    entry_points={
        "console_scripts": ["braintrust = braintrust.cli.__main__:main"],
        "pytest11": ["braintrust = braintrust.wrappers.pytest_plugin.plugin"],
    },
    install_requires=install_requires,
    extras_require=extras_require,
)