openai/openai-python
Publicmirrored from https://github.com/openai/openai-pythonAvailable
setup.py
56lines · modecode
| 1 | import os |
| 2 | |
| 3 | from setuptools import find_packages, setup |
| 4 | |
| 5 | version_contents = {} |
| 6 | version_path = os.path.join( |
| 7 | os.path.abspath(os.path.dirname(__file__)), "openai/version.py" |
| 8 | ) |
| 9 | with open(version_path, "rt") as f: |
| 10 | exec(f.read(), version_contents) |
| 11 | |
| 12 | with open("README.md", "r") as fh: |
| 13 | long_description = fh.read() |
| 14 | |
| 15 | setup( |
| 16 | name="openai", |
| 17 | description="Python client library for the OpenAI API", |
| 18 | long_description=long_description, |
| 19 | long_description_content_type="text/markdown", |
| 20 | version=version_contents["VERSION"], |
| 21 | install_requires=[ |
| 22 | "requests>=2.20", # to get the patch for CVE-2018-18074 |
| 23 | "tqdm", # Needed for progress bars |
| 24 | "pandas>=1.2.3", # Needed for CLI fine-tuning data preparation tool |
| 25 | "pandas-stubs>=1.1.0.11", # Needed for type hints for mypy |
| 26 | "openpyxl>=3.0.7", # Needed for CLI fine-tuning data preparation tool xlsx format |
| 27 | "numpy", |
| 28 | 'typing_extensions;python_version<"3.8"', # Needed for type hints for mypy |
| 29 | ], |
| 30 | extras_require={ |
| 31 | "dev": ["black~=21.6b0", "pytest==6.*"], |
| 32 | "wandb": ["wandb"], |
| 33 | "embeddings": [ |
| 34 | "scikit-learn>=1.0.2", # Needed for embedding utils, versions >= 1.1 require python 3.8 |
| 35 | "tenacity>=8.0.1", |
| 36 | "matplotlib", |
| 37 | "sklearn", |
| 38 | "plotly", |
| 39 | ], |
| 40 | }, |
| 41 | python_requires=">=3.7.1", |
| 42 | entry_points={ |
| 43 | "console_scripts": [ |
| 44 | "openai=openai._openai_scripts:main", |
| 45 | ], |
| 46 | }, |
| 47 | packages=find_packages(exclude=["tests", "tests.*"]), |
| 48 | package_data={ |
| 49 | "openai": [ |
| 50 | "py.typed", |
| 51 | ] |
| 52 | }, |
| 53 | author="OpenAI", |
| 54 | author_email="support@openai.com", |
| 55 | url="https://github.com/openai/openai-python", |
| 56 | ) |