Merge pull request #1080 from guardrails-ai/feat/version-warnings

Feat/version warnings

Author: zsimjee

guardrails/cli/configure.py

@@ -12,7 +12,7 @@
 from guardrails.cli.hub.console import console
 from guardrails.cli.server.hub_client import AuthenticationError, get_auth
 from guardrails.cli.telemetry import trace_if_enabled
-
+from guardrails.cli.version import version_warnings_if_applicable
 
 DEFAULT_TOKEN = ""
 DEFAULT_ENABLE_METRICS = True
@@ -78,6 +78,7 @@ def configure(
         help="Clear the existing token from the configuration file.",
     ),
 ):
+    version_warnings_if_applicable(console)
     trace_if_enabled("configure")
     existing_token = _get_default_token()
     last4 = existing_token[-4:] if existing_token else ""

guardrails/cli/hub/install.py

@@ -5,7 +5,9 @@
 
 from guardrails.cli.hub.hub import hub_command
 from guardrails.cli.logger import logger
+from guardrails.cli.hub.console import console
 from guardrails.cli.telemetry import trace_if_enabled
+from guardrails.cli.version import version_warnings_if_applicable
 
 
 @hub_command.command()
@@ -40,6 +42,8 @@ def confirm():
                 " local models for local inference?",
             )
 
+        version_warnings_if_applicable(console)
+
         install_multiple(
             package_uris,
             install_local_models=local_models,

guardrails/cli/start.py

@@ -5,6 +5,8 @@
 from guardrails.cli.hub.utils import pip_process
 from guardrails.cli.logger import logger
 from guardrails.cli.telemetry import trace_if_enabled
+from guardrails.cli.version import version_warnings_if_applicable
+from guardrails.cli.hub.console import console
 
 
 def api_is_installed() -> bool:
@@ -39,5 +41,6 @@ def start(
     from guardrails_api.cli.start import start  # type: ignore
 
     logger.info("Starting Guardrails server")
+    version_warnings_if_applicable(console)
     trace_if_enabled("start")
     start(env, config, port)

guardrails/cli/version.py

@@ -0,0 +1,31 @@
+import contextlib
+import requests
+import semver
+from importlib.metadata import version
+from rich.console import Console
+
+
+GUARDRAILS_PACKAGE_NAME = "guardrails-ai"
+
+
+def get_guardrails_version():
+    return version(GUARDRAILS_PACKAGE_NAME)
+
+
+def version_warnings_if_applicable(console: Console):
+    current_version = get_guardrails_version()
+
+    with contextlib.suppress(Exception):
+        res = requests.get(f"https://pypi.org/pypi/{GUARDRAILS_PACKAGE_NAME}/json")
+        version_info = res.json()
+        info = version_info.get("info", {})
+        latest_version = info.get("version")
+
+        is_update_available = semver.compare(latest_version, current_version) > 0
+
+        if is_update_available:
+            console.print(
+                "[yellow]There is a newer version of Guardrails "
+                f"available {latest_version}. Your current version "
+                f"is {current_version}[/yellow]!"
+            )

poetry.lock

@@ -61,6 +61,7 @@ diff-match-patch = "^20230430"
 guardrails-api = ">=0.0.1"
 mlflow = {version = ">=2.0.1", optional = true}
 uvloop = {version = "^0.20.0", optional = true}
+semver = "^3.0.2"
 
 [tool.poetry.extras]
 sql = ["sqlvalidator", "sqlalchemy", "sqlglot"]