setup.py

import sys
from pathlib import Path
from setuptools import setup, find_packages

if sys.version_info.major != 3:
    raise RuntimeError("This package requires Python 3+")

pkg_name = 'async_openai'
gitrepo = 'GrowthEngineAI/async-openai'

root = Path(__file__).parent
version = root.joinpath('async_openai/version.py').read_text().split('VERSION = ', 1)[-1].strip().replace('-', '').replace("'", '')

requirements = [
    'aiohttpx >= 0.0.12',
    # 'file-io',
    'backoff',
    'tiktoken',
    'lazyops >= 0.2.76', # Pydantic Support
    'pydantic',
    'jinja2',
    'pyyaml',
    # 'pydantic-settings', # remove to allow for v1/v2 support
]

if sys.version_info.minor < 8:
    requirements.append('typing_extensions')

extras = {
    'cache': ['kvdb'], # Adds caching support
    'utils': ['numpy', 'scipy'] # Adds embedding utility support
}

args = {
    'packages': find_packages(include = [f'{pkg_name}', f'{pkg_name}.*',]),
    'install_requires': requirements,
    'include_package_data': True,
    'long_description': root.joinpath('README.md').read_text(encoding='utf-8'),
    'entry_points': {
        "console_scripts": []
    },
    'extras_require': extras,
}

setup(
    name = pkg_name,
    version = version,
    url=f'https://github.com/{gitrepo}',
    license='MIT Style',
    description='Unofficial Async Python client library for the OpenAI API',
    author='Tri Songz',
    author_email='ts@growthengineai.com',
    long_description_content_type="text/markdown",
    classifiers=[
        'Intended Audience :: Developers',
        'License :: OSI Approved :: MIT License',
        'Programming Language :: Python :: 3.7',
        'Topic :: Software Development :: Libraries',
    ],
    **args
)