Changes in docker image building.

Author: plmick

Dockerfile

@@ -1,30 +1,30 @@
 FROM python:3.13.7-slim-trixie
 
 ARG version=prod
+ARG GIT_REF=main
 
 LABEL authors="RadLab"
 LABEL version=$version
 
-# If used build.sh ENV is overwrited by default or set value
-ENV branch=main
-
 ENV LLM_PROXY_API_MINIMUM=1
 
-RUN apt-get update && apt-get install -y supervisor htop curl htop jq git vim cron && rm -rf /var/lib/apt/lists/*
+RUN apt-get update && apt-get upgrade -y && apt-get install -y supervisor htop curl htop jq git vim cron && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /srv/
 
-RUN git clone -b ${branch} https://github.com/radlab-dev-group/llm-router.git && \
-    cd /srv/llm-router  && \
+
+RUN git clone https://github.com/radlab-dev-group/llm-router.git && \
+    cd /srv/llm-router && \
+    git checkout ${GIT_REF} && \
     cat .git/HEAD > .version && git log -1 | head -1 >> .version && \
     rm -rf .git .gitignore
 
 WORKDIR /srv/llm-router
 
-RUN pip3 install -r requirements.txt
+RUN pip3 install --upgrade pip && pip3 install -r requirements.txt
 
 RUN pip3 install git+https://github.com/radlab-dev-group/ml-utils.git
 
-RUN chmod +x run-rest-api.sh
+RUN chmod +x run-rest-api.sh && chmod +x entrypoint.sh
 
-CMD ["/srv/llm-router/run-rest-api.sh"]
+ENTRYPOINT ["/srv/llm-router/entrypoint.sh"]

entrypoint.sh

@@ -0,0 +1,9 @@
+#!/bin/bash
+
+if [ "$1" = "SLEEP" ]; then
+  echo "Debug mode"
+  sleep infinity
+else
+  echo "Starting application..."
+  exec ./run-rest-api.sh
+fi

requirements.txt

@@ -1,4 +1,5 @@
 flask
 gunicorn
+redis
 
 # gevent