#!/bin/bash

# Default values
PORT=9999
NTHREADS=4

# Parse command line arguments
while [[ $# -gt 0 ]]; do
    case $1 in
        --port)
            PORT="$2"
            shift 2
            ;;
        --nthreads)
            NTHREADS="$2"
            shift 2
            ;;
        --help)
            echo "Usage: docker run distributed-llama-worker [OPTIONS]"
            echo ""
            echo "Options:"
            echo "  --port <port>      Worker port (default: 9999)"
            echo "  --nthreads <n>     Number of threads (default: 4)"
            echo ""
            echo "Example:"
            echo "  docker run -p 9999:9999 distributed-llama-worker --port 9999 --nthreads 4"
            exit 0
            ;;
        *)
            echo "Unknown option: $1"
            exit 1
            ;;
    esac
done

# Build the command
CMD="./dllama worker --port $PORT --nthreads $NTHREADS"

echo "Starting worker with command:"
echo "$CMD"
echo ""

exec $CMD