from setuptools import setup, find_packages

setup(
    name="nexus-coder",
    version="0.4.0",
    description="Nexus Coder v0.4 - CyberForge edition. MoE 423B/39B + 3M context + CyberGym training.",
    long_description=open("README.md", "r", encoding="utf-8").read() if __import__("os").path.exists("README.md") else "",
    long_description_content_type="text/markdown",
    author="Hieu Louis",
    author_email="mhieuhonda@users.noreply.github.com",
    url="https://github.com/mhieuhonda/NexusCoder",
    license="NAL-1.0 (Attribution Required)",
    packages=find_packages(),
    python_requires="==3.12.13",
    install_requires=[
        "torch>=2.0.0",
        "numpy>=1.24.0",
        "tqdm>=4.65.0",
        "pyyaml>=6.0",
        "datasets>=2.14.0",
        "requests>=2.31.0",
        "cryptography>=41.0.0",
    ],
    extras_require={
        "gpu": ["flash-attn>=2.0.0", "bitsandbytes>=0.41.0", "triton>=2.0.0"],
        "data": ["datasets>=2.14.0", "datasketch>=1.6.0", "langdetect>=1.0.9"],
        "tools": ["ruff>=0.1.0", "black>=23.0.0", "isort>=5.12.0",
                   "sqlparse>=0.4.4", "jsbeautifier>=1.14.0"],
        "crypto": ["cryptography>=41.0.0", "pyjwt>=2.8.0"],
        "database": [
            "sqlalchemy>=2.0.0", "psycopg2-binary>=2.9.0", "pymysql>=1.1.0",
            "redis>=5.0.0", "pymongo>=4.5.0", "elasticsearch>=8.0.0",
            "kafka-python>=2.0.2", "pika>=1.3.0",
        ],
        "web": ["aiohttp>=3.9.0", "websockets>=12.0", "grpcio>=1.59.0",
                 "beautifulsoup4>=4.12.0", "lxml>=4.9.0"],
        "devops": ["paramiko>=3.4.0", "kubernetes>=28.1.0", "docker>=7.0.0"],
        "media": ["Pillow>=10.0.0", "reportlab>=4.0.0", "markdown>=3.5.0"],
        "ml": ["scikit-learn>=1.3.0", "scipy>=1.11.0", "transformers>=4.35.0",
                "accelerate>=0.24.0", "peft>=0.6.0"],
        "distributed": ["deepspeed>=0.12.0", "accelerate>=0.24.0", "flash-attn>=2.0.0"],
    },
    classifiers=[
        "Development Status :: 4 - Beta",
        "License :: Other/Proprietary License",
        "Programming Language :: Python :: 3.12",
        "Programming Language :: Python :: 3.12.13",
        "Topic :: Scientific/Engineering :: Artificial Intelligence",
    ],
)