unique-toolkit 2026.30.0.dev9__tar.gz → 2026.30.0.dev11__tar.gz

{unique_toolkit-2026.30.0.dev9 → unique_toolkit-2026.30.0.dev11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.30.0.dev9
+Version: 2026.30.0.dev11
 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.30.0.dev2,<2026.30.0rc0
+Requires-Dist: unique-sdk>=2026.30.0.dev5,<2026.30.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.30.0.dev9 → unique_toolkit-2026.30.0.dev11}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "2026.30.0.dev9"
+version = "2026.30.0.dev11"
 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.30.0.dev2,<2026.30.0rc0",
+    "unique-sdk>=2026.30.0.dev5,<2026.30.0rc0",
     "jambo>=0.1.2,<0.2",
     "jinja2>=3.1.6,<4",
     "docxtpl>=0.20.1,<0.21",

unique_toolkit-2026.30.0.dev11/unique_toolkit/chat/__init__.py

@@ -0,0 +1,37 @@
+from typing import TYPE_CHECKING, Any
+
+from .cancellation import CancellationEvent as CancellationEvent
+from .cancellation import CancellationWatcher as CancellationWatcher
+from .constants import DOMAIN_NAME as DOMAIN_NAME
+from .schemas import ChatMessage as ChatMessage
+from .schemas import ChatMessageAssessment as ChatMessageAssessment
+from .schemas import ChatMessageAssessmentLabel as ChatMessageAssessmentLabel
+from .schemas import ChatMessageAssessmentStatus as ChatMessageAssessmentStatus
+from .schemas import ChatMessageAssessmentType as ChatMessageAssessmentType
+from .schemas import ChatMessageRole as ChatMessageRole
+from .utils import (
+    convert_chat_history_to_injectable_string as convert_chat_history_to_injectable_string,
+)
+
+if TYPE_CHECKING:
+    from .service import ChatService as ChatService
+
+
+def __getattr__(name: str) -> Any:
+    # Lazy import to break the circular dependency:
+    # chat/__init__ → chat/service → chat/deprecated/service → chat/functions → chat/__init__
+    # ChatService here is the deprecated shim; the canonical path is
+    # unique_toolkit.services.chat_service.ChatService.
+    if name == "ChatService":
+        import warnings
+
+        with warnings.catch_warnings():
+            warnings.simplefilter("ignore", DeprecationWarning)
+            from unique_toolkit.chat.service import ChatService
+
+        # Cache in globals() per PEP 562: subsequent access resolves via
+        # normal attribute lookup, never re-entering __getattr__ (and
+        # never re-opening the non-thread-safe warnings.catch_warnings()).
+        globals()["ChatService"] = ChatService
+        return ChatService
+    raise AttributeError(f"module {__name__!r} has no attribute {name!r}")

{unique_toolkit-2026.30.0.dev9 → unique_toolkit-2026.30.0.dev11}/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,
+    set_stopped_streaming_at: bool = False,
     segment_kind: str | None = None,
 ) -> ChatMessage:
     """Modifies a chat message synchronously.
@@ -84,6 +85,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.
+        set_stopped_streaming_at (bool, optional): Whether to set the stoppedStreamingAt field with the current date time. Use this to signal that model streaming has finished while post-processing (e.g. evaluations) is still running, so clients can unblock user input before completedAt is set. 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:
@@ -108,6 +110,7 @@ def modify_message(
             debug_info=debug_info,
             message_id=message_id,
             set_completed_at=set_completed_at,
+            set_stopped_streaming_at=set_stopped_streaming_at,
             segment_kind=segment_kind,
         )
         message = unique_sdk.Message.modify(**params)
@@ -131,6 +134,7 @@ async def modify_message_async(
     debug_info: dict[str, Any] | None = None,
     message_id: str | None = None,
     set_completed_at: bool = False,
+    set_stopped_streaming_at: bool = False,
     segment_kind: str | None = None,
 ) -> ChatMessage:
     """Modifies a chat message asynchronously.
@@ -149,6 +153,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.
+        set_stopped_streaming_at (bool, optional): Whether to set the stoppedStreamingAt field with the current date time. Use this to signal that model streaming has finished while post-processing (e.g. evaluations) is still running, so clients can unblock user input before completedAt is set. 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:
@@ -173,6 +178,7 @@ async def modify_message_async(
             debug_info=debug_info,
             message_id=message_id,
             set_completed_at=set_completed_at,
+            set_stopped_streaming_at=set_stopped_streaming_at,
             segment_kind=segment_kind,
         )
         message = await unique_sdk.Message.modify_async(**params)
@@ -231,9 +237,11 @@ def _construct_message_modify_params(
     debug_info: dict[str, Any] | None = None,
     message_id: str | None = None,
     set_completed_at: bool = False,
+    set_stopped_streaming_at: bool = False,
     segment_kind: str | None = None,
 ) -> dict[str, Any]:
     completed_at_datetime = None
+    stopped_streaming_at_datetime = None
 
     if message_id:
         # Message ID specified. No need to guess
@@ -249,6 +257,9 @@ def _construct_message_modify_params(
     if set_completed_at:
         completed_at_datetime = _time_utils.get_datetime_now()
 
+    if set_stopped_streaming_at:
+        stopped_streaming_at_datetime = _time_utils.get_datetime_now()
+
     params: dict[str, Any] = {
         "user_id": user_id,
         "company_id": company_id,
@@ -259,6 +270,7 @@ def _construct_message_modify_params(
         "references": map_references(references) if references else [],
         "debugInfo": debug_info,
         "completedAt": completed_at_datetime,
+        "stoppedStreamingAt": stopped_streaming_at_datetime,
     }
     if segment_kind is not None:
         params["segmentKind"] = segment_kind

{unique_toolkit-2026.30.0.dev9 → unique_toolkit-2026.30.0.dev11}/unique_toolkit/services/chat_service.py

@@ -392,6 +392,7 @@ class ChatService(ChatServiceDeprecated):
         debug_info: dict[str, Any] | None = None,
         message_id: str | None = None,
         set_completed_at: bool | None = None,
+        set_stopped_streaming_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.
@@ -403,6 +404,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.
+            set_stopped_streaming_at (Optional[bool]): Whether to set the stoppedStreamingAt field with the current date time. Use this to unblock user input as soon as model streaming ends, ahead of slower post-processing (e.g. evaluations) that still needs to run before completedAt is set. 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:
@@ -427,6 +429,7 @@ class ChatService(ChatServiceDeprecated):
             debug_info=debug_info,
             message_id=message_id,
             set_completed_at=set_completed_at or False,
+            set_stopped_streaming_at=set_stopped_streaming_at or False,
             segment_kind=segment_kind,
         )
 
@@ -438,6 +441,7 @@ class ChatService(ChatServiceDeprecated):
         debug_info: dict[str, Any] | None = None,
         message_id: str | None = None,
         set_completed_at: bool | None = False,
+        set_stopped_streaming_at: bool | None = False,
         segment_kind: str | None = None,
     ) -> ChatMessage:
         """Modifies a message in the chat session asynchronously.
@@ -449,6 +453,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.
+            set_stopped_streaming_at (bool, optional): Whether to set the stoppedStreamingAt field with the current date time. Use this to unblock user input as soon as model streaming ends, ahead of slower post-processing (e.g. evaluations) that still needs to run before completedAt is set. 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:
@@ -472,6 +477,7 @@ class ChatService(ChatServiceDeprecated):
             debug_info=debug_info,
             message_id=message_id,
             set_completed_at=set_completed_at or False,
+            set_stopped_streaming_at=set_stopped_streaming_at or False,
             segment_kind=segment_kind,
         )
 
@@ -655,6 +661,17 @@ class ChatService(ChatServiceDeprecated):
         """Unblocks the next user input"""
         self.modify_assistant_message(set_completed_at=True)
 
+    def signal_streaming_stopped(self) -> None:
+        """Marks model streaming as finished without marking the message fully completed.
+
+        Call this as soon as the model stops emitting tokens, before running any
+        slower post-processing (e.g. evaluations, hallucination checks). Clients
+        that unblock user input on ``stoppedStreamingAt`` (in addition to
+        ``completedAt``) will re-enable input immediately instead of waiting for
+        post-processing to finish.
+        """
+        self.modify_assistant_message(set_stopped_streaming_at=True)
+
     # History Methods
     ############################################################################
 

unique_toolkit-2026.30.0.dev9/unique_toolkit/chat/__init__.py

@@ -1,20 +0,0 @@
-import warnings
-
-from .cancellation import CancellationEvent as CancellationEvent
-from .cancellation import CancellationWatcher as CancellationWatcher
-from .constants import DOMAIN_NAME as DOMAIN_NAME
-from .schemas import ChatMessage as ChatMessage
-from .schemas import ChatMessageAssessment as ChatMessageAssessment
-from .schemas import ChatMessageAssessmentLabel as ChatMessageAssessmentLabel
-from .schemas import ChatMessageAssessmentStatus as ChatMessageAssessmentStatus
-from .schemas import ChatMessageAssessmentType as ChatMessageAssessmentType
-from .schemas import ChatMessageRole as ChatMessageRole
-
-# Import ChatService with deprecation warning suppressed for internal use
-with warnings.catch_warnings():
-    warnings.simplefilter("ignore", DeprecationWarning)
-    from .service import ChatService as ChatService
-
-from .utils import (
-    convert_chat_history_to_injectable_string as convert_chat_history_to_injectable_string,
-)