from setuptools import setup, find_packages

setup(
    name="sia",
    version="0.1.0",
    packages=find_packages(),
    install_requires=[
        'accelerate>=0.26.0',
        'aiohttp>=3.8.0',
        'bitsandbytes>=0.41.0',
        'dotenv-python>=0.0.1',
        'flash-attn>=2.0.0',
        'huggingface_hub>=0.16.0',
        'lxml>=4.9.0',
        'mistral-common>=1.0.0',
        'mistralai>=0.0.7',
        'openai>=1.0.0',
        'psutil>=5.9.0',
        'python-dotenv>=1.0.0',
        'tiktoken>=0.4.0',
        'torch>=2.0.0',
        'transformers>=4.30.0'
    ],
    entry_points={
        'console_scripts': [
            'sia=sia.__main__:main',
        ],
    },
    classifiers=[
        'Development Status :: 3 - Alpha',
        'Intended Audience :: Developers',
        'Programming Language :: Python :: 3',
        'Programming Language :: Python :: 3.10',
    ],
    python_requires='>=3.10',
)