Merge pull request #57 from fieldsoftheworld/migrate-to-uv

Migrate to uv

Author: asmithml

.github/workflows/ci.yml

@@ -13,22 +13,30 @@ jobs:
       matrix:
         os:
           - ubuntu-latest
-          # - macos-latest # ignore for now
-          # - windows-latest # ignore for now
     steps:
     - uses: actions/checkout@v4
 
-    - name: Set up Pixi
-      uses: prefix-dev/setup-pixi@v0.8.3
+    - name: Set up Python
+      uses: actions/setup-python@v5
       with:
-        environments: test lint
-        cache: true
+        python-version: '3.12'
 
-    - name: Run linting and format checks
-      run: pixi run -e lint lint
+    - name: Install UV
+      uses: astral-sh/setup-uv@v5
+      with:
+        enable-cache: true
 
-    - name: Run tests with coverage
-      run: pixi run -e test test
+    - name: Install dependencies
+      run: uv sync --group dev
 
     - name: Check code formatting
-      run: pixi run -e lint check
+      run: uv run ruff format --check .
+
+    - name: Run linting checks
+      run: uv run ruff check .
+
+    # - name: Type check
+    #   run: uv run mypy server/app
+
+    - name: Run tests with coverage
+      run: cd server && uv run pytest -v --cov=app --cov-report=xml --cov-report=term-missing

README.md

@@ -4,17 +4,18 @@ This project provides a FastAPI-based implementation of the Fields of the World
 
 ## Installation
 
-1. Install [Pixi](https://pixi.sh/):
+1. Install [UV](https://docs.astral.sh/uv/):
    ```bash
-   curl -fsSL https://pixi.sh/install.sh | sh  # macOS/Linux
-   # or: brew install pixi
+   curl -LsSf https://astral.sh/uv/install.sh | sh  # macOS/Linux/WSL
+   # or: brew install uv  # macOS
+   # or: pip install uv   # Any platform
    ```
 
 2. Clone and setup:
    ```bash
    git clone https://github.com/fieldsoftheworld/ftw-inference-api
    cd ftw-inference-api
-   pixi install
+   uv sync --group dev
    ```
 
 ## Deployment
@@ -33,9 +34,9 @@ curl -L https://raw.githubusercontent.com/fieldsoftheworld/ftw-inference-api/mai
 ```
 
 This script will:
-- Install Pixi package manager
+- Install UV package manager
 - Clone the repository and checkout the specified branch
-- Install dependencies using Pixi production environment
+- Install production dependencies using UV
 - Enable GPU support in configuration
 - Configure a systemd service for automatic startup
 - Set up log rotation
@@ -71,22 +72,22 @@ sudo journalctl -u ftw-inference-api --since today  # Today's logs
 2. **Start services**:
    ```bash
    # Terminal 1: Start DynamoDB Local
-   pixi run dynamodb-local
+   docker run -p 8001:8000 amazon/dynamodb-local:latest -jar DynamoDBLocal.jar -sharedDb -inMemory
 
    # Terminal 2: Export AWS credentials (local dev only) and start server
    export AWS_ACCESS_KEY_ID="test" AWS_SECRET_ACCESS_KEY="test"
-   pixi run start
+   cd server && uv run python run.py --debug
    ```
 
 ### Alternative: Direct Server Start
 
 ```bash
-pixi run start  # Development server with debug mode and auto reload
+cd server && uv run python run.py --debug  # Development server with debug mode and auto reload
 ```
 
-Or run directly with options:
+Or run with custom options:
 ```bash
-pixi run python server/run.py --host 127.0.0.1 --port 8080 --debug
+cd server && uv run python run.py --host 127.0.0.1 --port 8080 --debug
 ```
 
 **Command-line options:**
@@ -102,7 +103,7 @@ The server loads configuration from `server/config/base.toml` by default. Settin
 You can specify a custom configuration file using the `--config` command-line option:
 
 ```bash
-python run.py --config /path/to/custom_config.toml
+cd server && uv run python run.py --config /path/to/custom_config.toml
 ```
 
 ## API Endpoints
@@ -164,20 +165,20 @@ server/
 Uses [Ruff](https://docs.astral.sh/ruff/) for linting/formatting and pre-commit hooks for quality checks.
 
 ```bash
-pixi run lint     # Run all pre-commit hooks
-pixi run format   # Format code
-pixi run check    # Check without fixing
+uv run ruff check .           # Check code without fixing
+uv run ruff format .          # Auto-format code
+uv run mypy server/app        # Type check
 ```
 
 Setup pre-commit:
 ```bash
-pixi run pre-commit install
+uv run pre-commit install
 ```
 
 ### Running Tests
 
 ```bash
-pixi run test  # All tests with coverage
+cd server && uv run pytest -v --cov=app --cov-report=xml --cov-report=term-missing
 ```
 
 ## License

deploy.sh

@@ -1,5 +1,5 @@
 #!/bin/bash
-# Deployment script for Ubuntu Deep Learning AMI instances with Pixi and systemd service
+# Deployment script for Ubuntu Deep Learning AMI instances with UV and systemd service
 
 set -e
 
@@ -31,17 +31,17 @@ sudo apt-get update
 sudo apt-get upgrade -y
 
 echo "Installing system dependencies..."
-sudo apt-get install -y  curl
+sudo apt-get install -y curl python3.12 python3.12-venv
 
-echo "Installing Pixi..."
-if command -v pixi &> /dev/null; then
-    echo "Pixi already installed, skipping..."
+echo "Installing UV..."
+if command -v uv &> /dev/null; then
+    echo "UV already installed, skipping..."
 else
-    curl -fsSL https://pixi.sh/install.sh | sh
-    # Add pixi to PATH for current session
-    export PATH="$HOME/.pixi/bin:$PATH"
+    curl -LsSf https://astral.sh/uv/install.sh | sh
+    # Add uv to PATH for current session
+    export PATH="$HOME/.local/bin:$PATH"
     # Add to bashrc for future sessions
-    echo 'export PATH="$HOME/.pixi/bin:$PATH"' >> ~/.bashrc
+    echo 'export PATH="$HOME/.local/bin:$PATH"' >> ~/.bashrc
 fi
 
 echo "Cloning repository..."
@@ -57,10 +57,10 @@ else
     cd ftw-inference-api
 fi
 
-echo "Installing Pixi environment..."
-# Ensure pixi is in PATH for this command
-export PATH="$HOME/.pixi/bin:$PATH"
-pixi install
+echo "Installing Python dependencies with UV..."
+# Ensure uv is in PATH for this command
+export PATH="$HOME/.local/bin:$PATH"
+uv sync
 
 echo "Creating data directories..."
 mkdir -p server/data/uploads server/data/results server/logs
@@ -91,16 +91,8 @@ APP_DIR="$(pwd)"
 USER="$(whoami)"
 HOME_DIR="$(eval echo ~$USER)"
 
-# Conditionally build the ExecStart command based on the GPU configuration.
-# We check the final destination of the production.env file.
-BASE_EXEC_START="${HOME_DIR}/.pixi/bin/pixi run --environment production python run.py --host 0.0.0.0 --port 8000"
-EXEC_START_CMD=$BASE_EXEC_START
-
-if grep -q "PROCESSING__GPU=null" /etc/ftw-inference-api/production.env; then
-    echo "GPU is disabled. Applying CONDA_OVERRIDE_GLIBC to the service."
-    # Prepend the environment variable directly to the command.
-    EXEC_START_CMD="CONDA_OVERRIDE_GLIBC=2.17 ${BASE_EXEC_START}"
-fi
+# Build the ExecStart command using UV
+BASE_EXEC_START="${HOME_DIR}/.local/bin/uv run python run.py --host 0.0.0.0 --port 8000"
 
 sudo tee /etc/systemd/system/ftw-inference-api.service > /dev/null <<EOF
 [Unit]
@@ -113,10 +105,9 @@ User=${USER}
 Group=${USER}
 WorkingDirectory=${APP_DIR}/server
 # Set a reliable PATH for the service environment.
-Environment="PATH=${HOME_DIR}/.pixi/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin"
+Environment="PATH=${HOME_DIR}/.local/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin"
 EnvironmentFile=/etc/ftw-inference-api/production.env
-# Use /bin/sh -c to correctly process the command string with the conditional variable.
-ExecStart=/bin/sh -c "${EXEC_START_CMD}"
+ExecStart=${BASE_EXEC_START}
 Restart=always
 RestartSec=10
 StandardOutput=journal