Merge pull request #122 from kneelesh48/docker-updates

Docker updates

backend/Dockerfile

@@ -1,16 +1,25 @@
 # Build the virtualenv as a separate step: Only re-execute this step when pyproject.toml changes
 FROM python:3.12-bookworm AS build-venv
-COPY pyproject.toml /build/pyproject.toml
-RUN python -m venv /venv && \
+
+WORKDIR /build
+COPY pyproject.toml .
+COPY README.md .
+
+RUN mkdir -p openui/util && \
+    python -m venv /venv && \
     /venv/bin/pip install --upgrade pip setuptools wheel && \
-    mkdir -p /build/openui/util && touch /build/README.md
-RUN /venv/bin/pip install --disable-pip-version-check /build
+    /venv/bin/pip install --disable-pip-version-check .
 
 # Copy the virtualenv into a distroless image
 FROM python:3.12-slim-bookworm
+
 ENV PATH="/venv/bin:$PATH"
+
 COPY --from=build-venv /venv /venv
 COPY . /app
+
 WORKDIR /app
+
 RUN pip install --no-deps -U /app
-ENTRYPOINT ["python", "-m", "openui"]
+
+ENTRYPOINT ["python", "-m", "openui"]

backend/pyproject.toml

@@ -9,6 +9,8 @@ dependencies = [
     "uvicorn>=0.25.0",
     "fastapi-sso>=0.13.0",
     "boto3>=1.34.67",
+    "Pillow>=8.3.1",
+    "playwright>=1.41.0",
 ]
 name = "openui"
 version = "0.3.1"
@@ -36,8 +38,6 @@ CI = "https://github.com/wandb/openui/actions"
 [project.optional-dependencies]
 test = ["pytest", "pre-commit"]
 eval = [
-    "Pillow>=8.3.1",
-    "playwright>=1.41.0",
     "beautifulsoup4>=4.0.0",
     "mistletoe>=1.0.0",
 ]

docker-compose.yaml

@@ -2,13 +2,15 @@ version: '3.8'
 
 services:
   ollama:
+    container_name: ollama
     image: ollama/ollama:latest
     ports:
       - "11434:11434"
     volumes:
       - ${OLLAMA_VOLUME_PATH:-ollama}:/root/.ollama
 
   backend:
+    container_name: openui
     build:
       context: ./backend
       dockerfile: Dockerfile
@@ -18,7 +20,9 @@ services:
       - OLLAMA_HOST=http://ollama:11434
       # Just remove `=xxx` to have the env variable passed forward
       - OPENAI_API_KEY=xxx
+      - GROQ_API_KEY=xxx
     depends_on:
       - ollama
+
 volumes:
-  ollama:
+  ollama: