glaip-sdk 0.1.3__py3-none-any.whl → 0.6.19__py3-none-any.whl

glaip_sdk/cli/update_notifier.py

@@ -8,7 +8,7 @@ from __future__ import annotations
 
 import importlib
 import logging
-import os
+import sys
 from collections.abc import Callable, Iterable, Iterator
 from contextlib import contextmanager
 from typing import Any, Literal
@@ -26,7 +26,9 @@ from glaip_sdk.branding import (
     WARNING_STYLE,
 )
 from glaip_sdk.cli.commands.update import update_command
-from glaip_sdk.cli.utils import command_hint, format_command_hint
+from glaip_sdk.cli.constants import UPDATE_CHECK_ENABLED
+from glaip_sdk.cli.hints import format_command_hint
+from glaip_sdk.cli.utils import command_hint
 from glaip_sdk.rich_components import AIPPanel
 
 FetchLatestVersion = Callable[[], str | None]
@@ -72,7 +74,11 @@ def _fetch_latest_version(package_name: str) -> str | None:
 
 def _should_check_for_updates() -> bool:
     """Return False when update checks are explicitly disabled."""
-    return os.getenv("AIP_NO_UPDATE_CHECK") is None
+    # Check module attribute first (for test overrides), then fall back to imported constant
+    module = sys.modules.get(__name__)
+    if module and hasattr(module, "UPDATE_CHECK_ENABLED"):
+        return getattr(module, "UPDATE_CHECK_ENABLED")
+    return UPDATE_CHECK_ENABLED
 
 
 def _build_update_panel(
@@ -244,6 +250,7 @@ def _suppress_library_logging(
 def _refresh_installed_version(console: Console, ctx: Any) -> None:
     """Reload runtime metadata after an in-process upgrade."""
     new_version: str | None = None
+    branding_module: Any | None = None
 
     try:
         version_module = importlib.reload(importlib.import_module("glaip_sdk._version"))
@@ -252,16 +259,18 @@ def _refresh_installed_version(console: Console, ctx: Any) -> None:
         _LOGGER.debug("Failed to reload glaip_sdk._version: %s", exc, exc_info=True)
 
     try:
-        branding_module = importlib.import_module("glaip_sdk.branding")
+        branding_module = importlib.reload(importlib.import_module("glaip_sdk.branding"))
         if new_version:
             branding_module.SDK_VERSION = new_version
     except Exception as exc:  # pragma: no cover - defensive guard
         _LOGGER.debug("Failed to update branding metadata: %s", exc, exc_info=True)
+        branding_module = None
 
     session = _get_slash_session(ctx)
     if session and hasattr(session, "refresh_branding"):
         try:
-            session.refresh_branding(new_version)
+            branding_cls = getattr(branding_module, "AIPBranding", None) if branding_module else None
+            session.refresh_branding(new_version, branding_cls=branding_cls)
             return
         except Exception as exc:  # pragma: no cover - defensive guard
             _LOGGER.debug("Failed to refresh active slash session: %s", exc, exc_info=True)