Adding setup.py

on_startup.sh

@@ -19,6 +19,8 @@ cd ~/app
 git clone https://github.com/ray-project/llmperf.git
 cd llmperf
 git checkout afd137a
+mv ~/app/setup.py .
+pip install -e .
 
 # Add dark theme
 mkdir -p ~/.jupyter/lab/user-settings/@jupyterlab/apputils-extension/ && \

setup.py

@@ -0,0 +1,31 @@
+from setuptools import setup, find_packages
+
+setup(
+    name="LLMPerf",
+    version="0.1.0",
+    author="Avnish Narayan",
+    author_email="[email protected]",
+    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",
+    ],
+)