unique-toolkit 2026.28.0.dev16__tar.gz → 2026.28.0.dev17__tar.gz

{unique_toolkit-2026.28.0.dev16 → unique_toolkit-2026.28.0.dev17}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.28.0.dev16
+Version: 2026.28.0.dev17
 Summary: 
 Author: Cedric Klinkert, Martin Fadler, Sadique Sheik, Fabian Schläpfer, Pascal Hauri, Ejiroghene Oki-Peter, Rami Azouz, Ricardo Visini, Sanzio Monti
 Author-email: Cedric Klinkert <cedric.klinkert@unique.ch>, Martin Fadler <martin.fadler@unique.ch>, Sadique Sheik <sadique@unique.ch>, Fabian Schläpfer <fabian@unique.ch>, Pascal Hauri <pascal@unique.ch>, Ejiroghene Oki-Peter <oki-peter.ejiroghene@unique.ai>, Rami Azouz <rami.ext@unique.ch>, Ricardo Visini <ricardo.visini@unique.ch>, Sanzio Monti <sanzio.monti@unique.ai>
@@ -16,7 +16,7 @@ Requires-Dist: sseclient>=0.0.27,<0.0.28
 Requires-Dist: openai>=1.105.0,<3
 Requires-Dist: platformdirs>=4.0.0,<5
 Requires-Dist: pillow>=12.1.1
-Requires-Dist: unique-sdk>=2026.28.0.dev18,<2026.28.0rc0
+Requires-Dist: unique-sdk>=2026.28.0.dev19,<2026.28.0rc0
 Requires-Dist: jambo>=0.1.2,<0.2
 Requires-Dist: jinja2>=3.1.6,<4
 Requires-Dist: docxtpl>=0.20.1,<0.21

{unique_toolkit-2026.28.0.dev16 → unique_toolkit-2026.28.0.dev17}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "2026.28.0.dev16"
+version = "2026.28.0.dev17"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12"
@@ -28,7 +28,7 @@ dependencies = [
     "openai>=1.105.0,<3",
     "platformdirs>=4.0.0,<5",
     "pillow>=12.1.1",
-    "unique-sdk>=2026.28.0.dev18,<2026.28.0rc0",
+    "unique-sdk>=2026.28.0.dev19,<2026.28.0rc0",
     "jambo>=0.1.2,<0.2",
     "jinja2>=3.1.6,<4",
     "docxtpl>=0.20.1,<0.21",

{unique_toolkit-2026.28.0.dev16 → unique_toolkit-2026.28.0.dev17}/unique_toolkit/app/unique_settings.py

@@ -17,6 +17,7 @@ from pydantic import (
 )
 from pydantic_settings import BaseSettings, SettingsConfigDict
 from typing_extensions import Protocol, deprecated
+from unique_sdk.api_resources._llm_models import LLMModels
 
 from unique_toolkit._common.config_checker import register_config
 from unique_toolkit.app.chat_event_filter_options_settings import (
@@ -321,6 +322,14 @@ class ApiType(StrEnum):
     LEGACY = "legacy"
 
 
+class UniqueApiConnectionError(Exception):
+    """Raised when the API connection check fails."""
+
+    def __init__(self, base_url: str, message: str) -> None:
+        self.base_url = base_url
+        super().__init__(f"{message} (base URL: {base_url})")
+
+
 @register_config()
 class UniqueApi(BaseSettings):
     base_url: str = Field(
@@ -337,6 +346,8 @@ class UniqueApi(BaseSettings):
 
     api_type: ApiType = Field(default=ApiType.LEGACY)
 
+    _probe_check_failed: bool = PrivateAttr(default=False)
+
     version: str = Field(
         default="2023-12-06",
         validation_alias=AliasChoices(
@@ -403,6 +414,8 @@ class UniqueApi(BaseSettings):
         return parsed, base_path
 
     def sdk_url(self) -> str:
+        if self._probe_check_failed:
+            return self.base_url
         parsed, base_path = self.base_path()
         return urlunparse(parsed._replace(path=base_path, query=None, fragment=None))
 
@@ -411,6 +424,15 @@ class UniqueApi(BaseSettings):
         path = base_path + "/openai-proxy"
         return urlunparse(parsed._replace(path=path, query=None, fragment=None))
 
+    async def check_connection(self, user_id: str, company_id: str) -> bool:
+        try:
+            model_list = await LLMModels.get_models_async(user_id, company_id)
+            self._probe_check_failed = False
+            return len(model_list["models"]) > 0
+        except Exception as exc:
+            self._probe_check_failed = True
+            raise UniqueApiConnectionError(self.base_url, str(exc)) from exc
+
 
 # EventFilterOptions
 # Settings only as only obtained from the environment.
@@ -612,7 +634,6 @@ class UniqueSettings:
             logger.warning(
                 f"Environment file '{filename}' not found. Falling back to environment variables only."
             )
-            # Fall back to environment variables only
             return cls.from_env()
 
     def init_sdk(self) -> None:
@@ -643,6 +664,29 @@ class UniqueSettings:
         settings.init_sdk()
         return settings
 
+    async def check_connection(self) -> bool:
+        """Verify the API is reachable by listing available models via the SDK.
+
+        Returns:
+            True if the API is reachable and at least one model is available.
+            False if the API is reachable but no models are configured.
+
+        Raises:
+            UniqueApiConnectionError: If the SDK call fails due to a network or auth error.
+        """
+        # Reset before init_sdk so each retry always probes the canonical computed
+        # sdk_url(), not a stale base_url left over from a previous failure.
+        self.api._probe_check_failed = False
+        self.init_sdk()
+        try:
+            return await self.api.check_connection(
+                self.authcontext.get_confidential_user_id(),
+                self.authcontext.get_confidential_company_id(),
+            )
+        finally:
+            # Re-sync api_base with sdk_url() which may have just fallen back.
+            self.init_sdk()
+
     @classmethod
     def from_chat_event(cls, event: ChatEvent) -> UniqueSettings:
         """Build a :class:`UniqueSettings` from a :class:`ChatEvent`.