Fix download routing for BASE_URL (#852)

* base url handling for download * no bundles in docker image * version = "1.12.1"
h2oai · Sep 12, 2024 · 7952d0f · 7952d0f
1 parent f864b93
commit 7952d0f
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 1 deletion.
diff --git a/.dockerignore b/.dockerignore
@@ -2,6 +2,7 @@
 .github
 .mypy_cache
 .neptune
+bundles
 wandb
 .pytest_cache
 data

diff --git a/llm_studio/app_utils/utils.py b/llm_studio/app_utils/utils.py
@@ -2143,6 +2143,10 @@ def get_download_link(q: Q, artifact_path):
  # behind a reverse proxy or being accessed by a public IP in a public
  # cloud.
 
+ # add the BASE_URL to the path
+ if "H2O_CLOUD_ENVIRONMENT" in os.environ:
+ url_path = f"{os.environ['H2O_WAVE_BASE_URL']}/{url_path}"
+
  return url_path
 
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -1,6 +1,6 @@
 [project]
 name = "h2o-llmstudio"
-version = "1.13.0-dev"
+version = "1.12.1"
 readme = "README.md"
 license = {file = "LICENSE"}