Feat/poetry to uv (#29)

Update from poetry to uv. Fix minor dtype bug for sentence-transformers

Author: rbs333

.github/workflows/release.yml

@@ -15,31 +15,17 @@ jobs:
       - name: Check out repository
         uses: actions/checkout@v4
 
-      - name: Install Python
-        uses: actions/setup-python@v5
+      - name: Install uv
+        uses: astral-sh/setup-uv@v7
         with:
           python-version: ${{ env.PYTHON_VERSION }}
+          enable-cache: true
 
-      - name: Install Poetry
-        uses: snok/install-poetry@v1
-        with:
-          version: latest
-          virtualenvs-create: true
-          virtualenvs-in-project: true
-
-      - name: Load cached venv
-        id: cached-poetry-dependencies
-        uses: actions/cache@v3
-        with:
-          path: .venv
-          key: venv-${{ runner.os }}-${{ steps.setup-python.outputs.python-version }}-${{ hashFiles('**/poetry.lock') }}
-
-      - name: Install dependencies
-        if: steps.cached-poetry-dependencies.outputs.cache-hit != 'true'
-        run: poetry install --only=main
+      - name: Sync dependencies
+        run: uv sync --no-dev-dependencies
 
       - name: Build package
-        run: poetry build
+        run: uv build
 
       - name: Upload build
         uses: actions/upload-artifact@v4
@@ -55,28 +41,11 @@ jobs:
       - name: Check out repository
         uses: actions/checkout@v4
 
-      - name: Install Python
-        uses: actions/setup-python@v5
+      - name: Install uv
+        uses: astral-sh/setup-uv@v7
         with:
           python-version: ${{ env.PYTHON_VERSION }}
-
-      - name: Install Poetry
-        uses: snok/install-poetry@v1
-        with:
-          version: latest
-          virtualenvs-create: true
-          virtualenvs-in-project: true
-
-      - name: Load cached venv
-        id: cached-poetry-dependencies
-        uses: actions/cache@v3
-        with:
-          path: .venv
-          key: venv-${{ runner.os }}-${{ steps.setup-python.outputs.python-version }}-${{ hashFiles('**/poetry.lock') }}
-
-      - name: Install dependencies
-        if: steps.cached-poetry-dependencies.outputs.cache-hit != 'true'
-        run: poetry install --only=main
+          enable-cache: true
 
       - name: Download build artifacts
         uses: actions/download-artifact@v4
@@ -86,5 +55,5 @@ jobs:
 
       - name: Publish to PyPI
         env:
-          POETRY_PYPI_TOKEN_PYPI: ${{ secrets.PYPI }}
-        run: poetry publish
+          UV_PUBLISH_TOKEN: ${{ secrets.PYPI }}
+        run: uv publish --token $UV_PUBLISH_TOKEN

.github/workflows/test.yml

@@ -12,7 +12,7 @@ jobs:
 
     services:
       redis:
-        image: redis/redis-stack:latest
+        image: ${{ matrix.redis-image }}
         ports:
           - 6379:6379
         options: >-
@@ -24,34 +24,26 @@ jobs:
     strategy:
       matrix:
         python-version: ["3.12"]
+        redis-version: ["latest", "8.4.0"]
+        include:
+          - redis-version: "latest"
+            redis-image: redis/redis-stack:latest
+          - redis-version: "8.4.0"
+            redis-image: redis:8.4.0
 
     steps:
-    - uses: actions/checkout@v4
-
-    - name: Set up Python ${{ matrix.python-version }}
-      uses: actions/setup-python@v4
-      with:
-        python-version: ${{ matrix.python-version }}
-
-    - name: Install Poetry
-      uses: snok/install-poetry@v1
-      with:
-        version: latest
-        virtualenvs-create: true
-        virtualenvs-in-project: true
-
-    - name: Load cached venv
-      id: cached-poetry-dependencies
-      uses: actions/cache@v3
-      with:
-        path: .venv
-        key: venv-${{ runner.os }}-${{ steps.setup-python.outputs.python-version }}-${{ hashFiles('**/poetry.lock') }}
-
-    - name: Install dependencies
-      if: steps.cached-poetry-dependencies.outputs.cache-hit != 'true'
-      run: poetry install --all-extras
-
-    - name: Run tests
-      run: poetry run test
-      env:
-        REDIS_URL: redis://localhost:6379/0
+      - uses: actions/checkout@v4
+
+      - name: Install uv
+        uses: astral-sh/setup-uv@v7
+        with:
+          python-version: ${{ matrix.python-version }}
+          enable-cache: true
+
+      - name: Sync dependencies
+        run: uv sync
+
+      - name: Run tests
+        run: uv run test
+        env:
+          REDIS_URL: redis://localhost:6379/0

.vscode/settings.json

@@ -1,6 +1,6 @@
 {
     "python.testing.pytestArgs": [
-        "tests"
+        "."
     ],
     "python.testing.unittestEnabled": false,
     "python.testing.pytestEnabled": true

CLAUDE.md

@@ -1,7 +1,7 @@
 .PHONY: install format lint test clean redis-start redis-stop check-types check
 
 install:
-	poetry install --all-extras
+	uv sync
 
 redis-start:
 	docker run -d --name redis-stack -p 6379:6379 -p 8001:8001 redis/redis-stack:latest
@@ -10,16 +10,16 @@ redis-stop:
 	docker stop redis-stack
 
 format:
-	poetry run format
-	poetry run sort-imports
+	uv run format
+	uv run sort-imports
 
 check-types:
-	poetry run check-mypy
+	uv run check-mypy
 
 lint: format check-types
-	
+
 test:
-	poetry run test
+	uv run test
 
 check: lint test
 

Makefile

@@ -449,7 +449,7 @@ optimizer = CacheThresholdOptimizer(cache, test_data, eval_metric="precision") #
 optimizer = CacheThresholdOptimizer(cache, test_data, eval_metric="recall")    # Recall
 ```
 
-For a complete example, see [docs/examples/threshold_optimization/00_threshold_optimization.ipynb](https://github.com/redis-applied-ai/redis-retrieval-optimizer/blob/1be3e2683a301a6bebe298ec8d126b7429701bc9/docs/examples/threshold_optimization/00_threshold_optimization.ipynb).
+For complete an example, see [docs/examples/threshold_optimization/00_threshold_optimization.ipynb](https://github.com/redis-applied-ai/redis-retrieval-optimizer/blob/1be3e2683a301a6bebe298ec8d126b7429701bc9/docs/examples/threshold_optimization/00_threshold_optimization.ipynb).
 
 ## 🔧 Custom Processors and Search Methods