Switch to UV from PDM

.github/workflows/llm_checks.yml

@@ -14,37 +14,35 @@ jobs:
         with:
           ref: ${{ github.event.inputs.BRANCH }}
 
+      - name: Install uv
+        run: astral-sh/setup-uv@v3
+        with:
+          enable-cache: true
+          cache-local-path: "./llm-service/uv.lock"
+
       - name: Set up Python
         uses: pdm-project/setup-pdm@v3
         with:
           python-version: '3.10'
           cache: false
 
-      - name: Cache pdm cache folder
-        uses: actions/cache@v4
-        id: python_cache
-        with:
-          path: ./llm-service/.venv
-          key: pdm-venv-${{ hashFiles('./llm-service/pdm.lock') }}
-
       - name: Install Python dependencies
-        if: steps.python_cache.outputs.cache-hit != 'true'
         run: |
-          pdm install
+          uv python install
         working-directory: llm-service
 
       - name: Run ruff
         run: |
-          pdm run ruff check app
+          uv run ruff check app
         working-directory: llm-service
 
       - name: Run mypy
         run: |
-          pdm run mypy app
+          uv run mypy app
         working-directory: llm-service
 
       - name: Test with pytest
         run: |
-          pdm run pytest -sxvvra
+          uv run pytest -sxvvra
         working-directory: llm-service
 

README.md

@@ -89,7 +89,8 @@ the Node service locally, you can do so by following these steps:
 - Install Python (via [pyenv](https://github.com/pyenv/pyenv), probably) (directly via brew, if you must)
 - `cd llm-service`
 - Create a virtual environment (`python -m venv venv; source venv/bin/activate`)
-- Install dependencies (`python -m pip install -r requirements.txt`)
+- Install UV (`python -m pip install uv`)
+- Install dependencies (`uv pip install -r pyproject.toml`)
 - `fastapi dev`
   - ends up running on port 8000
 

llm-service/Dockerfile

@@ -1,9 +1,9 @@
 FROM docker-private.infra.cloudera.com/cloudera_base/hardened/cloudera-python:3.10
-RUN pip install pdm
+RUN pip install uv
 COPY ./pyproject.toml /app/
-COPY ./pdm.lock /app/
+COPY ./uv.lock /app/
 WORKDIR /app
-RUN pdm install
+RUN uv install
 
 COPY ./app /app
 WORKDIR /app

llm-service/pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "llm-service"
 version = "0.1.0"
-description = "Default template for PDM package"
+description = "Default template for UV package"
 authors = [
     {name = "Conrado Silva Miranda", email = "[email protected]"},
 ]

local-dev.sh

@@ -65,11 +65,11 @@ docker run --name qdrant_dev --rm -d -p 6333:6333 -p 6334:6334 -v $(pwd)/databas
 cd llm-service
 python3.10 -m venv venv
 source venv/bin/activate
-python -m pip install pdm
-pdm install
-pdm run pytest -sxvvra
+python -m pip install uv
+uv pip install -r pyproject.toml
+uv run pytest -sxvvra
 
-pdm run fastapi dev &
+uv run fastapi dev &
 
 # start up the jarva
 cd ../backend

scripts/refresh_project.sh

@@ -46,8 +46,9 @@ cd ui/express
 npm install
 
 cd ../../llm-service
-pip install pdm
-pdm install -v
+pip install uv
+uv venv
+uv pip install -r pyproject.toml
 
 cd ..
 mkdir -p artifacts

scripts/startup_app.sh

@@ -61,7 +61,7 @@ qdrant/qdrant &
 
 # start Python backend
 cd llm-service
-pdm run fastapi run --host 127.0.0.1 --port 8081 &
+uv run fastapi run --host 127.0.0.1 --port 8081 &
 
 # start up the jarva
 cd ..