Merge pull request #9 from EetuaLaine/main

Add pyproject.toml, uv installation steps, new requirements.txt file.

Author: gabriben

README.md

@@ -9,6 +9,15 @@ pip install -e .
 wraval generate
 ```
 
+## Install with uv
+
+```bash
+uv venv
+source .venv/bin/activate
+uv pip install -e .
+wraval generate
+```
+
 ## Step by step
 
 ### 1. Start by generating evaluation data for each of the writing assistant tasks (a.k.a. tones)
@@ -99,7 +108,7 @@ You can use the [CloudFormation yaml](src/cloudformation.yml) to start a Sagemak
 - [ ] merge generate_all_datasets and generate_specific_datasets
 - [ ] batch processing for Bedrock
 - [ ] batch processing for Sagemaker endpoint
-- [ ] uv?
+- [x] uv
 - [x] from main.py to setup.py
 - [ ] better sagemaker inference output parsing
 - [x] add a model_router.py

pyproject.toml

@@ -0,0 +1,45 @@
+[build-system]
+requires = ["setuptools>=61", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "wraval"
+version = "0.1.0"
+description = """WRAVAL helps in evaluating LLMs for writing assistant tasks like summarization,
+professional tone, witty tone, etc."""
+readme = "README.md"
+requires-python = ">=3.9"
+authors = [{ name = "Gabriel Benedict", email = "[email protected]" }]
+
+# Dependencies – see note below for loading from requirements.txt
+dependencies = [
+    "tqdm==4.66.4",
+    "pandas==2.2.3",
+    "beautifulsoup4==4.12.3",
+    "boto3==1.34.143",
+    "plotly==5.24.1",
+    "transformers==4.48.1",
+    "datasets==3.2.0",
+    "evaluate==0.4.3",
+    "dynaconf==3.2.7",
+    "torch~=2.6.0",
+    "botocore~=1.34.162",
+    "sagemaker",
+    "numpy",
+    "requests"
+]
+
+[project.scripts]
+wraval = "wraval.main:main"
+
+[tool.setuptools]
+package-dir = {"" = "src"}
+include-package-data = true
+
+[tool.setuptools.packages.find]
+where = ["src"]
+
+[tool.setuptools.data-files]
+# This copies the config file into the installation (non-package dir)
+"config" = ["config/settings.toml"]
+

requirements.txt

@@ -1,13 +1,293 @@
-tqdm==4.66.4
-numpy==2.2.2
-pandas==2.2.3
+# This file was autogenerated by uv via the following command:
+#    uv pip compile pyproject.toml -o requirements.txt
+aiohappyeyeballs==2.6.1
+    # via aiohttp
+aiohttp==3.11.18
+    # via
+    #   datasets
+    #   fsspec
+aiosignal==1.3.2
+    # via aiohttp
+annotated-types==0.7.0
+    # via pydantic
+antlr4-python3-runtime==4.9.3
+    # via omegaconf
+anyio==4.9.0
+    # via starlette
+attrs==23.2.0
+    # via
+    #   aiohttp
+    #   jsonschema
+    #   referencing
+    #   sagemaker
 beautifulsoup4==4.12.3
+    # via wraval (pyproject.toml)
 boto3==1.34.143
-plotly==5.24.1
-transformers==4.48.1
+    # via
+    #   wraval (pyproject.toml)
+    #   sagemaker
+    #   sagemaker-core
+botocore==1.34.162
+    # via
+    #   wraval (pyproject.toml)
+    #   boto3
+    #   s3transfer
+certifi==2025.4.26
+    # via requests
+charset-normalizer==3.4.2
+    # via requests
+click==8.1.8
+    # via uvicorn
+cloudpickle==2.2.1
+    # via sagemaker
 datasets==3.2.0
-evaluate==0.4.3
+    # via
+    #   wraval (pyproject.toml)
+    #   evaluate
+dill==0.3.8
+    # via
+    #   datasets
+    #   evaluate
+    #   multiprocess
+    #   pathos
+docker==7.1.0
+    # via sagemaker
 dynaconf==3.2.7
-torch~=2.6.0
-requests~=2.32.3
-botocore~=1.34.162
+    # via wraval (pyproject.toml)
+evaluate==0.4.3
+    # via wraval (pyproject.toml)
+fastapi==0.115.12
+    # via sagemaker
+filelock==3.18.0
+    # via
+    #   datasets
+    #   huggingface-hub
+    #   torch
+    #   transformers
+frozenlist==1.6.0
+    # via
+    #   aiohttp
+    #   aiosignal
+fsspec==2024.9.0
+    # via
+    #   datasets
+    #   evaluate
+    #   huggingface-hub
+    #   torch
+google-pasta==0.2.0
+    # via sagemaker
+h11==0.16.0
+    # via uvicorn
+hf-xet==1.1.0
+    # via huggingface-hub
+huggingface-hub==0.31.1
+    # via
+    #   datasets
+    #   evaluate
+    #   tokenizers
+    #   transformers
+idna==3.10
+    # via
+    #   anyio
+    #   requests
+    #   yarl
+importlib-metadata==6.11.0
+    # via
+    #   sagemaker
+    #   sagemaker-core
+jinja2==3.1.6
+    # via torch
+jmespath==1.0.1
+    # via
+    #   boto3
+    #   botocore
+jsonschema==4.23.0
+    # via
+    #   sagemaker
+    #   sagemaker-core
+jsonschema-specifications==2025.4.1
+    # via jsonschema
+markdown-it-py==3.0.0
+    # via rich
+markupsafe==3.0.2
+    # via jinja2
+mdurl==0.1.2
+    # via markdown-it-py
+mock==4.0.3
+    # via sagemaker-core
+mpmath==1.3.0
+    # via sympy
+multidict==6.4.3
+    # via
+    #   aiohttp
+    #   yarl
+multiprocess==0.70.16
+    # via
+    #   datasets
+    #   evaluate
+    #   pathos
+networkx==3.4.2
+    # via torch
+numpy==1.26.4
+    # via
+    #   wraval (pyproject.toml)
+    #   datasets
+    #   evaluate
+    #   pandas
+    #   sagemaker
+    #   transformers
+omegaconf==2.2.3
+    # via sagemaker
+packaging==25.0
+    # via
+    #   datasets
+    #   evaluate
+    #   huggingface-hub
+    #   plotly
+    #   sagemaker
+    #   transformers
+pandas==2.2.3
+    # via
+    #   wraval (pyproject.toml)
+    #   datasets
+    #   evaluate
+    #   sagemaker
+pathos==0.3.2
+    # via sagemaker
+platformdirs==4.3.8
+    # via
+    #   sagemaker
+    #   sagemaker-core
+plotly==5.24.1
+    # via wraval (pyproject.toml)
+pox==0.3.6
+    # via pathos
+ppft==1.7.7
+    # via pathos
+propcache==0.3.1
+    # via
+    #   aiohttp
+    #   yarl
+protobuf==4.25.7
+    # via sagemaker
+psutil==7.0.0
+    # via sagemaker
+pyarrow==20.0.0
+    # via datasets
+pydantic==2.11.4
+    # via
+    #   fastapi
+    #   sagemaker-core
+pydantic-core==2.33.2
+    # via pydantic
+pygments==2.19.1
+    # via rich
+python-dateutil==2.9.0.post0
+    # via
+    #   botocore
+    #   pandas
+pytz==2025.2
+    # via pandas
+pyyaml==6.0.2
+    # via
+    #   datasets
+    #   huggingface-hub
+    #   omegaconf
+    #   sagemaker
+    #   sagemaker-core
+    #   transformers
+referencing==0.36.2
+    # via
+    #   jsonschema
+    #   jsonschema-specifications
+regex==2024.11.6
+    # via transformers
+requests==2.32.3
+    # via
+    #   wraval (pyproject.toml)
+    #   datasets
+    #   docker
+    #   evaluate
+    #   huggingface-hub
+    #   sagemaker
+    #   transformers
+rich==13.9.4
+    # via sagemaker-core
+rpds-py==0.24.0
+    # via
+    #   jsonschema
+    #   referencing
+s3transfer==0.10.4
+    # via boto3
+safetensors==0.5.3
+    # via transformers
+sagemaker==2.236.0
+    # via wraval (pyproject.toml)
+sagemaker-core==1.0.16
+    # via sagemaker
+schema==0.7.7
+    # via sagemaker
+setuptools==80.3.1
+    # via torch
+six==1.17.0
+    # via
+    #   google-pasta
+    #   python-dateutil
+smdebug-rulesconfig==1.0.1
+    # via sagemaker
+sniffio==1.3.1
+    # via anyio
+soupsieve==2.7
+    # via beautifulsoup4
+starlette==0.46.2
+    # via fastapi
+sympy==1.13.1
+    # via torch
+tblib==3.1.0
+    # via sagemaker
+tenacity==9.1.2
+    # via plotly
+tokenizers==0.21.1
+    # via transformers
+torch==2.6.0
+    # via wraval (pyproject.toml)
+tqdm==4.66.4
+    # via
+    #   wraval (pyproject.toml)
+    #   datasets
+    #   evaluate
+    #   huggingface-hub
+    #   sagemaker
+    #   transformers
+transformers==4.48.1
+    # via wraval (pyproject.toml)
+typing-extensions==4.13.2
+    # via
+    #   anyio
+    #   fastapi
+    #   huggingface-hub
+    #   pydantic
+    #   pydantic-core
+    #   referencing
+    #   torch
+    #   typing-inspection
+typing-inspection==0.4.0
+    # via pydantic
+tzdata==2025.2
+    # via pandas
+urllib3==2.4.0
+    # via
+    #   botocore
+    #   docker
+    #   requests
+    #   sagemaker
+uvicorn==0.34.2
+    # via sagemaker
+xxhash==3.5.0
+    # via
+    #   datasets
+    #   evaluate
+yarl==1.20.0
+    # via aiohttp
+zipp==3.21.0
+    # via importlib-metadata