Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions .github/workflows/goose.yml
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,7 @@ permissions:
issues: write

env:
PROVIDER_API_KEY: ${{ secrets.GROQ_API_KEY }}
PROVIDER_API_KEY: ${{ secrets.GOOGLE_API_KEY }}
PR_NUMBER: ${{ github.event.pull_request.number }}
GH_TOKEN: ${{ github.token }}

Expand Down Expand Up @@ -47,8 +47,8 @@ jobs:
run: |
mkdir -p ~/.config/goose
cat > ~/.config/goose/config.yaml <<'EOF'
GOOSE_PROVIDER: groq
GOOSE_MODEL: llama-3.3-70b-versatile
GOOSE_PROVIDER: google
GOOSE_MODEL: gemini-2.0-flash-exp
keyring: false
EOF

Expand All @@ -69,7 +69,7 @@ jobs:

- name: Run Goose and filter output
env:
GROQ_API_KEY: ${{ secrets.GROQ_API_KEY }}
GOOGLE_API_KEY: ${{ secrets.GOOGLE_API_KEY }}
run: |
goose run --instructions instructions.txt \
| sed -E 's/\x1B\[[0-9;]*[mK]//g' \
Expand Down
69 changes: 69 additions & 0 deletions Dockerfile
Original file line number Diff line number Diff line change
@@ -0,0 +1,69 @@
# syntax=docker/dockerfile:1
# BuildKit optimized Dockerfile with advanced caching and multi-arch support

ARG TARGETPLATFORM
ARG BUILDPLATFORM
ARG TARGETOS
ARG TARGETARCH

# Stage 1: Dependencies stage with cache mounts
FROM --platform=$BUILDPLATFORM python:3.11-slim AS dependencies

WORKDIR /app

# Use cache mount for apt packages
RUN --mount=type=cache,target=/var/cache/apt \
--mount=type=cache,target=/var/lib/apt \
apt-get update && apt-get install -y \
gcc \
python3-dev

# Use cache mount for pip packages
COPY requirements.txt .
RUN --mount=type=cache,target=/root/.cache/pip \
pip install --user -r requirements.txt

# Stage 2: Model training stage
FROM dependencies AS trainer

COPY train.py .

# Train the model with cache mount for any downloads
RUN --mount=type=cache,target=/tmp \
python train.py

# Stage 3: Final production stage
FROM --platform=$TARGETPLATFORM python:3.11-slim AS production

# Display build info for debugging
RUN echo "Building for platform: $TARGETPLATFORM, architecture: $TARGETARCH"

# Create non-root user
RUN useradd --create-home --shell /bin/bash mluser

WORKDIR /app

# Copy Python packages from dependencies stage
COPY --from=dependencies /root/.local /home/mluser/.local

# Copy application files
COPY app.py .
COPY requirements.txt .

# Copy trained models from trainer stage
COPY --from=trainer /app/*.pkl ./

# Set proper ownership
RUN chown -R mluser:mluser /app
USER mluser

# Update PATH for user-installed packages
ENV PATH=/home/mluser/.local/bin:$PATH

# Health check with improved reliability
HEALTHCHECK --interval=30s --timeout=10s --start-period=15s --retries=3 \
CMD python -c "import urllib.request; urllib.request.urlopen('http://localhost:7860', timeout=5)" || exit 1

EXPOSE 7860

CMD ["python", "app.py"]