chore: version update notices

Haidra-Org · Mar 6, 2024 · db0c626 · db0c626
1 parent cb2cff8
commit db0c626
Show file tree

Hide file tree

Showing 4 changed files with 25 additions and 1 deletion.
diff --git a/horde_worker_regen/_version_meta.json b/horde_worker_regen/_version_meta.json
@@ -1,5 +1,5 @@
 {
-    "recommended_version": "4.2.4",
+    "recommended_version": "4.2.6",
     "required_min_version": "4.2.4",
     "required_min_version_update_date": "2024-03-05",
     "required_min_version_info": {
@@ -11,6 +11,9 @@
         },
         "4.2.4": {
             "reason_for_update": "Fixes a bug that causes indefinite worker hangups."
+        },
+        "4.2.6": {
+            "reason_for_update": "Fixes LoRa downloads causing worker instability."
         }
     },
     "beta_version_info": {

diff --git a/horde_worker_regen/load_env_vars.py b/horde_worker_regen/load_env_vars.py
@@ -90,6 +90,8 @@ def load_env_vars() -> None:  # FIXME: there is a dynamic way to do this
             )
             logger.warning(reason_for_update_str)
 
+            os.environ["AIWORKER_NOT_REQUIRED_VERSION"] = "1"
+
         else:
             logger.error(
                 f"Current worker version {horde_worker_regen.__version__} has a required update to "
@@ -103,6 +105,13 @@ def load_env_vars() -> None:  # FIXME: there is a dynamic way to do this
             input("Press Enter to continue...")
             exit(1)
 
+    if not semver.compare(horde_worker_regen.__version__, version_meta.recommended_version) >= 0:
+        logger.warning(
+            f"Current worker version {horde_worker_regen.__version__} is not the recommended version. "
+            f"Please consider updating to {version_meta.recommended_version}.",
+        )
+        os.environ["AIWORKER_NOT_RECOMMENDED_VERSION"] = "1"
+
     if version_meta.beta_version_info:
         current_version_semver = semver.VersionInfo.parse(horde_worker_regen.__version__)
         current_version_simple = (

diff --git a/horde_worker_regen/process_management/process_manager.py b/horde_worker_regen/process_management/process_manager.py
@@ -2870,6 +2870,17 @@ def print_status_method(self) -> None:
 
             logger.success(job_info_message)
 
+            if os.getenv("AIWORKER_NOT_REQUIRED_VERSION"):
+                logger.warning(
+                    "There is a required update available for the AI Worker. "
+                    "`git pull` and `update-runtime` to update.",
+                )
+            elif os.getenv("AIWORKER_NOT_RECOMMENDED_VERSION"):
+                logger.warning(
+                    "There is a recommended update available for the AI Worker. "
+                    "`git pull` and `update-runtime` to update.",
+                )
+
             self._last_status_message_time = time.time()
 
     _bridge_data_loop_interval = 1.0

diff --git a/horde_worker_regen/version_meta.py b/horde_worker_regen/version_meta.py
@@ -15,6 +15,7 @@ class BetaVersionInfo(BaseModel):
 
 
 class VersionMeta(BaseModel):
+    recommended_version: str
     required_min_version: str
     required_min_version_update_date: str
     beta_version_info: dict[str, BetaVersionInfo]