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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.28.0.dev15
+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.dev15,<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.dev15 → unique_toolkit-2026.28.0.dev17}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "2026.28.0.dev15"
+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.dev15,<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.dev15 → 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`.

{unique_toolkit-2026.28.0.dev15 → unique_toolkit-2026.28.0.dev17}/unique_toolkit/chat/functions.py

@@ -66,6 +66,7 @@ def modify_message(
     debug_info: dict[str, Any] | None = None,
     message_id: str | None = None,
     set_completed_at: bool = False,
+    segment_kind: str | None = None,
 ) -> ChatMessage:
     """Modifies a chat message synchronously.
 
@@ -83,6 +84,7 @@ def modify_message(
         references (list[ContentReference]): list of ContentReference objects. Defaults to None.
         debug_info (dict[str, Any]], optional): Debug information. Defaults to None.
         set_completed_at (bool, optional): Whether to set the completedAt field with the current date time. Defaults to False.
+        segment_kind (str, optional): The segment kind (e.g. "PROCESS", "PREFACE", "ELICITATION", "ANSWER") to relabel the segment to in place. Defaults to None (kind unchanged).
 
     Returns:
         ChatMessage: The modified message.
@@ -106,6 +108,7 @@ def modify_message(
             debug_info=debug_info,
             message_id=message_id,
             set_completed_at=set_completed_at,
+            segment_kind=segment_kind,
         )
         message = unique_sdk.Message.modify(**params)
         return ChatMessage(**message)
@@ -128,6 +131,7 @@ async def modify_message_async(
     debug_info: dict[str, Any] | None = None,
     message_id: str | None = None,
     set_completed_at: bool = False,
+    segment_kind: str | None = None,
 ) -> ChatMessage:
     """Modifies a chat message asynchronously.
 
@@ -145,6 +149,7 @@ async def modify_message_async(
         references (list[ContentReference]): list of ContentReference objects. Defaults to None.
         debug_info (dict[str, Any]], optional): Debug information. Defaults to None.
         set_completed_at (bool, optional): Whether to set the completedAt field with the current date time. Defaults to False.
+        segment_kind (str, optional): The segment kind (e.g. "PROCESS", "PREFACE", "ELICITATION", "ANSWER") to relabel the segment to in place. Defaults to None (kind unchanged).
 
     Returns:
         ChatMessage: The modified message.
@@ -168,6 +173,7 @@ async def modify_message_async(
             debug_info=debug_info,
             message_id=message_id,
             set_completed_at=set_completed_at,
+            segment_kind=segment_kind,
         )
         message = await unique_sdk.Message.modify_async(**params)
         return ChatMessage(**message)
@@ -225,6 +231,7 @@ def _construct_message_modify_params(
     debug_info: dict[str, Any] | None = None,
     message_id: str | None = None,
     set_completed_at: bool = False,
+    segment_kind: str | None = None,
 ) -> dict[str, Any]:
     completed_at_datetime = None
 
@@ -242,7 +249,7 @@ def _construct_message_modify_params(
     if set_completed_at:
         completed_at_datetime = _time_utils.get_datetime_now()
 
-    params = {
+    params: dict[str, Any] = {
         "user_id": user_id,
         "company_id": company_id,
         "id": message_id,
@@ -253,6 +260,8 @@ def _construct_message_modify_params(
         "debugInfo": debug_info,
         "completedAt": completed_at_datetime,
     }
+    if segment_kind is not None:
+        params["segmentKind"] = segment_kind
     return params
 
 

{unique_toolkit-2026.28.0.dev15 → unique_toolkit-2026.28.0.dev17}/unique_toolkit/services/chat_service.py

@@ -387,6 +387,7 @@ class ChatService(ChatServiceDeprecated):
         debug_info: dict[str, Any] | None = None,
         message_id: str | None = None,
         set_completed_at: bool | None = None,
+        segment_kind: str | None = None,
     ) -> ChatMessage:
         """Modifies a message in the chat session synchronously if parameter is not specified the corresponding field will remain as is.
 
@@ -397,6 +398,7 @@ class ChatService(ChatServiceDeprecated):
             debug_info (dict[str, Any]]]): Debug information. Defaults to {}.
             message_id (Optional[str]): The message ID. Defaults to None.
             set_completed_at (Optional[bool]): Whether to set the completedAt field with the current date time. Defaults to False.
+            segment_kind (str, optional): The segment kind (e.g. "PROCESS", "PREFACE", "ELICITATION", "ANSWER") to relabel the segment to in place, e.g. relabeling M0 from PROCESS to PREFACE once content is set. Defaults to None (kind unchanged).
 
         Returns:
             ChatMessage: The modified message.
@@ -420,6 +422,7 @@ class ChatService(ChatServiceDeprecated):
             debug_info=debug_info,
             message_id=message_id,
             set_completed_at=set_completed_at or False,
+            segment_kind=segment_kind,
         )
 
     async def modify_assistant_message_async(
@@ -430,6 +433,7 @@ class ChatService(ChatServiceDeprecated):
         debug_info: dict[str, Any] | None = None,
         message_id: str | None = None,
         set_completed_at: bool | None = False,
+        segment_kind: str | None = None,
     ) -> ChatMessage:
         """Modifies a message in the chat session asynchronously.
 
@@ -440,6 +444,7 @@ class ChatService(ChatServiceDeprecated):
             references (list[ContentReference]): list of ContentReference objects. Defaults to None.
             debug_info (dict[str, Any]], optional): Debug information. Defaults to None.
             set_completed_at (bool, optional): Whether to set the completedAt field with the current date time. Defaults to False.
+            segment_kind (str, optional): The segment kind (e.g. "PROCESS", "PREFACE", "ELICITATION", "ANSWER") to relabel the segment to in place, e.g. relabeling M0 from PROCESS to PREFACE once content is set. Defaults to None (kind unchanged).
 
         Returns:
             ChatMessage: The modified message.
@@ -462,6 +467,7 @@ class ChatService(ChatServiceDeprecated):
             debug_info=debug_info,
             message_id=message_id,
             set_completed_at=set_completed_at or False,
+            segment_kind=segment_kind,
         )
 
     def create_assistant_message(