from setuptools import setup, find_packages

setup(
    name="LLMPerf",
    version="0.1.0",
    author="Avnish Narayan",
    author_email="avnish@anyscale.com",
    description="A framework for load testing LLM APIs",
    license="Apache-2.0",
    packages=find_packages(),
    python_requires=">=3.8, <3.11",
    install_requires=[
        "pydantic<2.5",
        "ray",
        "pytest>=6.0",
        "seaborn>=0.11",
        "awscli>=1.22",
        "typer>=0.4",
        "litellm>=0.1.738",
        "num2words",
        "transformers",
        "tqdm",
        "boto3",
        "google-cloud-aiplatform"
    ],
    classifiers=[
        "Programming Language :: Python :: 3",
        "License :: OSI Approved :: Apache Software License",
        "Operating System :: OS Independent",
    ],
)