Allow overriding the version of bugbug used while building the HTTP service worker image (#1345)

* Prioritize the TAG environment variable over the bugbug package version

* Allow overriding the version of bugbug used while building the HTTP service worker image

bugbug/utils.py

@@ -202,10 +202,12 @@ def get_last_modified(url):
 
 
 def download_model(model_name):
-    try:
-        version = f"v{get_bugbug_version()}"
-    except DistributionNotFound:
-        version = os.getenv("TAG", "latest")
+    version = os.getenv("TAG")
+    if not version:
+        try:
+            version = f"v{get_bugbug_version()}"
+        except DistributionNotFound:
+            version = "latest"
 
     path = f"{model_name}model"
     url = f"https://community-tc.services.mozilla.com/api/index/v1/task/project.relman.bugbug.train_{model_name}.{version}/artifacts/public/{path}.zst"

http_service/Dockerfile.bg_worker

@@ -16,6 +16,9 @@ WORKDIR /code/
 ARG CHECK_MODELS
 ENV CHECK_MODELS="${CHECK_MODELS}"
 
+ARG TAG
+ENV TAG="${TAG}"
+
 RUN bash /code/http_service/ensure_models.sh
 
 CMD bugbug-http-worker high default low