Yandex.Cloud SDK version constraints workaround

Author: black-roland

custom_components/yandexgpt_conversation/__init__.py

@@ -15,7 +15,8 @@
 from homeassistant.helpers import config_validation as cv
 from homeassistant.helpers import selector
 from homeassistant.helpers.typing import ConfigType
-from yandex_cloud_ml_sdk import AsyncYCloudML
+
+from .sdk_workaround import is_latest_sdk_installed, upgrade_sdk
 
 from .const import (ATTR_FILENAME, ATTR_PROMPT, ATTR_SEED,
                     CONF_ENABLE_SERVER_DATA_LOGGING, CONF_FOLDER_ID,
@@ -73,6 +74,11 @@ async def render_image(call: ServiceCall) -> ServiceResponse:
             LOGGER.error("Error during image generation: %s", str(err), exc_info=True)
             raise HomeAssistantError(f"Image generation failed: {str(err)}") from err
 
+    if not is_latest_sdk_installed():
+        upgrade_success = upgrade_sdk()
+        if not upgrade_success:
+            return False
+
     hass.services.async_register(
         DOMAIN,
         SERVICE_GENERATE_IMAGE,
@@ -99,6 +105,8 @@ async def render_image(call: ServiceCall) -> ServiceResponse:
 
 async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
     """Set up YandexGPT from a config entry."""
+    from yandex_cloud_ml_sdk import AsyncYCloudML
+
     settings = {**entry.data, **entry.options}
 
     entry.runtime_data = AsyncYCloudML(

custom_components/yandexgpt_conversation/manifest.json

@@ -10,6 +10,6 @@
   "iot_class": "cloud_polling",
   "issue_tracker": "https://github.com/black-roland/homeassistant-yandexgpt/issues",
   "loggers": ["yandexgpt_conversation"],
-  "requirements": ["yandex-cloud-ml-sdk==0.13.1", "yandexcloud==0.353.0"],
+  "requirements": [],
   "version": "1.4.2"
 }

custom_components/yandexgpt_conversation/sdk_workaround.py

@@ -0,0 +1,57 @@
+"""Workaround for the Yandex.Cloud SDK dependencies BS 💩."""
+
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at https://mozilla.org/MPL/2.0/.
+
+from homeassistant.util import package as pkg_util
+from .const import LOGGER
+
+# Details:
+# https://github.com/black-roland/homeassistant-yandexgpt/issues/32
+#
+# The history of "we don't give a 💩":
+# https://github.com/yandex-cloud/python-sdk/issues?q=is%3Aissue%20dependencies%20OR%20requirements%20OR%20constraints
+REQUIRED_PACKAGES = [
+    "yandex-cloud-ml-sdk==0.13.1",
+    "yandexcloud==0.353.0"
+]
+
+
+def is_latest_sdk_installed() -> bool:
+    """Check if the required versions of SDKs are installed."""
+    try:
+        for package in REQUIRED_PACKAGES:
+            if not pkg_util.is_installed(package):
+                LOGGER.debug("Package %s is not installed or not in the required version.", package)
+                return False
+
+        LOGGER.debug("All required Yandex Cloud SDK packages are installed in correct versions.")
+        return True
+    except Exception as e:
+        LOGGER.warning("Error checking SDK package installation status: %s", e)
+        return False
+
+
+def upgrade_sdk() -> bool:
+    """Use pkg_util to install the required versions of the SDKs."""
+
+    # Empty args to avoid setting any package constraints
+    _pip_kwargs = {}
+
+    success = True
+    for package in REQUIRED_PACKAGES:
+        try:
+            LOGGER.info("Attempting to install/upgrade Yandex Cloud SDK package: %s", package)
+
+            install_success = pkg_util.install_package(package, **_pip_kwargs)
+            if not install_success:
+                LOGGER.error("Failed to install/upgrade Yandex Cloud SDK package: %s", package)
+                success = False
+                continue
+
+            LOGGER.info("Successfully installed/upgraded Yandex Cloud SDK package: %s", package)
+        except Exception as e:
+            LOGGER.exception("Exception occurred while installing/upgrading package %s: %s", package, e)
+            success = False
+    return success