from setuptools import setup, find_packages

setup(
    name="sia",
    version="0.1.0",
    packages=find_packages(),
    entry_points={
        'console_scripts': [
            'sia=sia.__main__:main',
        ],
    },
    install_requires=[
        'torch>=2.0.0',
        'accelerate>=0.26.0',
        'aiohttp>=3.8.0',
        'bitsandbytes>=0.41.0',
        'dotenv-python>=0.0.1',
        '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',
        'transformers>=4.30.0',
        'xml_schema_validator @ file:///root/sia/lib/xml_schema_validator',
    ],
    python_requires='>=3.10',
)