openai/openai-python
Publicmirrored fromhttps://github.com/openai/openai-pythonAvailable
setup.py
51lines · 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 | setup( |
| 13 | name="openai", |
| 14 | description="Python client library for the OpenAI API", |
| 15 | version=version_contents["VERSION"], |
| 16 | install_requires=[ |
| 17 | "requests>=2.20", # to get the patch for CVE-2018-18074 |
| 18 | "tqdm", # Needed for progress bars |
| 19 | "pandas>=1.2.3", # Needed for CLI fine-tuning data preparation tool |
| 20 | "pandas-stubs>=1.1.0.11", # Needed for type hints for mypy |
| 21 | "openpyxl>=3.0.7", # Needed for CLI fine-tuning data preparation tool xlsx format |
| 22 | "numpy", |
| 23 | "typing_extensions", # Needed for type hints for mypy |
| 24 | ], |
| 25 | extras_require={ |
| 26 | "dev": ["black~=21.6b0", "pytest==6.*"], |
| 27 | "wandb": ["wandb"], |
| 28 | "embeddings": [ |
| 29 | "scikit-learn>=1.0.2", # Needed for embedding utils, versions >= 1.1 require python 3.8 |
| 30 | "tenacity>=8.0.1", |
| 31 | "matplotlib", |
| 32 | "sklearn", |
| 33 | "plotly", |
| 34 | ], |
| 35 | }, |
| 36 | python_requires=">=3.7.1", |
| 37 | entry_points={ |
| 38 | "console_scripts": [ |
| 39 | "openai=openai._openai_scripts:main", |
| 40 | ], |
| 41 | }, |
| 42 | packages=find_packages(exclude=["tests", "tests.*"]), |
| 43 | package_data={ |
| 44 | "openai": [ |
| 45 | "py.typed", |
| 46 | ] |
| 47 | }, |
| 48 | author="OpenAI", |
| 49 | author_email="support@openai.com", |
| 50 | url="https://github.com/openai/openai-python", |
| 51 | ) |