unique_toolkit 2026.30.0.dev5__py3-none-any.whl → 2026.30.0.dev7__py3-none-any.whl

unique_toolkit/chat/cancellation.py

@@ -57,6 +57,16 @@ class CancellationWatcher:
     def on_cancellation(self) -> TypedEventBus[CancellationEvent]:
         return self._bus
 
+    @property
+    def assistant_message_id(self) -> str:
+        return self._assistant_message_id
+
+    def set_assistant_message_id(self, message_id: str) -> None:
+        """Retarget polling to the active assistant segment."""
+        if self._cancelled:
+            return
+        self._assistant_message_id = message_id
+
     async def check_cancellation_async(self) -> bool:
         """Poll the DB once.  Returns ``True`` if the message was cancelled.
 

unique_toolkit/experimental/integrations/openai/streaming/event_routing/subscribers/message_persister.py

@@ -1,9 +1,12 @@
 """Subscriber that persists assistant messages in response to stream events.
 
-Single responsibility: owns every ``unique_sdk.Message.modify_async`` call
-related to a streaming response (``startedStreamingAt``, incremental text
-+ references, and ``stoppedStreamingAt``), plus the ``content_chunks`` used
-to filter references down to what was actually cited.
+Single responsibility: owns every SDK write related to a streaming
+response. Stream boundaries go through ``unique_sdk.Message.modify_async``
+(``startedStreamingAt`` on start, final text + ``stoppedStreamingAt`` on
+end); incremental text deltas go through
+``unique_sdk.Message.create_event_async`` on the hot path. It also owns
+the ``content_chunks`` used to filter references down to what was actually
+cited.
 
 The same message is streamed into once per agent round, and this subscriber
 only sees per-request boundaries (never end-of-turn). It therefore stamps
@@ -66,7 +69,10 @@ def _now_utc_iso() -> str:
 
 
 class MessagePersistingSubscriber:
-    """Translates text lifecycle events into ``unique_sdk.Message.modify_async`` calls.
+    """Translates text lifecycle events into ``unique_sdk.Message`` SDK writes.
+
+    Stream start/end are persisted via ``Message.modify_async``; incremental
+    :class:`TextUpdate` deltas are persisted via ``Message.create_event_async``.
 
     Holds the retrieved chunks for the currently active stream (keyed by
     ``message_id``) so reference filtering on :class:`TextUpdate` and
@@ -139,8 +145,8 @@ class MessagePersistingSubscriber:
         # written again in :meth:`on_ended`, so a dropped delta degrades
         # to a slightly coarser UI update rather than data loss.
         try:
-            await unique_sdk.Message.modify_async(
-                id=event.message_id,
+            await unique_sdk.Message.create_event_async(
+                messageId=event.message_id,
                 chatId=event.chat_id,
                 user_id=self._settings.context.auth.user_id.get_secret_value(),
                 company_id=self._settings.context.auth.company_id.get_secret_value(),

unique_toolkit/services/chat_service.py

@@ -200,6 +200,11 @@ class ChatService(ChatServiceDeprecated):
         """Cancellation watcher for this chat session."""
         return self._cancellation_watcher
 
+    def _set_assistant_message_id(self, message_id: str) -> None:
+        """Point active assistant operations and cancellation polling at *message_id*."""
+        self._assistant_message_id = message_id
+        self._cancellation_watcher.set_assistant_message_id(message_id)
+
     @property
     def elicitation(self) -> ElicitationService:
         """Get the ElicitationService for this chat session."""
@@ -515,8 +520,7 @@ class ChatService(ChatServiceDeprecated):
             response_turn_id=response_turn_id,
             segment_index=segment_index,
         )
-        # Update the assistant message id
-        self._assistant_message_id = chat_message.id or "unknown"
+        self._set_assistant_message_id(chat_message.id or "unknown")
         return chat_message
 
     async def create_assistant_message_async(
@@ -564,8 +568,7 @@ class ChatService(ChatServiceDeprecated):
             response_turn_id=response_turn_id,
             segment_index=segment_index,
         )
-        # Update the assistant message id
-        self._assistant_message_id = chat_message.id or "unknown"
+        self._set_assistant_message_id(chat_message.id or "unknown")
         return chat_message
 
     def create_user_message(

{unique_toolkit-2026.30.0.dev5.dist-info → unique_toolkit-2026.30.0.dev7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.30.0.dev5
+Version: 2026.30.0.dev7
 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>

{unique_toolkit-2026.30.0.dev5.dist-info → unique_toolkit-2026.30.0.dev7.dist-info}/RECORD

@@ -245,7 +245,7 @@ unique_toolkit/app/unique_settings.py,sha256=-bYvxwCw_DvhSViHXgc-pYoNboxhHB8VG2L
 unique_toolkit/app/verification.py,sha256=GxFFwcJMy25fCA_Xe89wKW7bgqOu8PAs5y8QpHF0GSc,3861
 unique_toolkit/app/webhook.py,sha256=evpxLDnvz7P3D7-TV1XBkl8WcoldXpjOkw3tvFRxHr0,2368
 unique_toolkit/chat/__init__.py,sha256=N9RubtxWN7UF0DlxP7YgeNTgjjk6p7AwkbGaVvuZCv0,938
-unique_toolkit/chat/cancellation.py,sha256=HvGZK-daCMSr0M0AnvMY3EBPH1N_UqQPfrbTjwr8owk,5645
+unique_toolkit/chat/cancellation.py,sha256=k5HMjpeBmofIiInwq5TjOR_5k91oq1bWG00lJXZSoJ0,5970
 unique_toolkit/chat/constants.py,sha256=05kq6zjqUVB2d6_P7s-90nbljpB3ryxwCI-CAz0r2O4,83
 unique_toolkit/chat/deprecated/service.py,sha256=8_XoopUWHT396Z_5gwMmsI6v4NGVgQirSJCYUW4Gx8c,9339
 unique_toolkit/chat/functions.py,sha256=h-8MEhDpBbTRZc3A9JoDGMm-so3BWDdV6gv6NpXYIOg,54678
@@ -336,7 +336,7 @@ unique_toolkit/experimental/integrations/openai/streaming/event_routing/response
 unique_toolkit/experimental/integrations/openai/streaming/event_routing/responses/text_delta_event_handler.py,sha256=JB-7W89K4pT6K8A9Z_f_Pow5p-EIGwLoLUSb9BI3qW4,3634
 unique_toolkit/experimental/integrations/openai/streaming/event_routing/responses/tool_call_event_handler.py,sha256=iELT-H3-u6eG3D3X9Qz0QfTlDj-IfHBAq7bWYR7Dd5U,3198
 unique_toolkit/experimental/integrations/openai/streaming/event_routing/subscribers/__init__.py,sha256=4wpUt9kzfPzncO9tlKcvOQUzQG5e6kVp5AKZI5grxOk,276
-unique_toolkit/experimental/integrations/openai/streaming/event_routing/subscribers/message_persister.py,sha256=cqwDI6PLLiocaBUBL6VPLqgucH_Y_wKTdN3HdadFq6k,8635
+unique_toolkit/experimental/integrations/openai/streaming/event_routing/subscribers/message_persister.py,sha256=JVA0lcF3_WCYEUgWqX53gO5Lj7zR1iFCCuNXuWDPGbo,8930
 unique_toolkit/experimental/integrations/openai/streaming/event_routing/subscribers/progress_log_persister.py,sha256=VCYGZY3WSpI7mvVs2gO89-M0JenakW01BGfBfFkPDrw,3499
 unique_toolkit/experimental/resources/__init__.py,sha256=KdMPaaneY0twZ-JBfwImRjXZaWT68NVuOIeqdNXFh_Y,3377
 unique_toolkit/experimental/resources/content_folder/__init__.py,sha256=c5MUj1xIP10L7tcM4-hG2m1_yUMVfKr_jHAncUXVNvo,2492
@@ -395,7 +395,7 @@ unique_toolkit/monitoring/middleware.py,sha256=JNh5v_INAkexuLtXhCG9n5n2t6Pnf8ZdP
 unique_toolkit/monitoring/registry.py,sha256=8r9jqaQoRHZveGRZe_GQvBzvQpmUV4cpw8Ej1ssDIdE,5056
 unique_toolkit/protocols/support.py,sha256=x5edzbBmKAUuvm3oLFh52gkgQsqR5LU7M0t8RHK0DZk,5161
 unique_toolkit/services/__init__.py,sha256=b9Wl7Wvz1JmZDbDexIZzBoXZ0wscXuczGAH8LoL-Qak,572
-unique_toolkit/services/chat_service.py,sha256=Y-Eoxelp02eLVm3gv6_aqeTLtCsAJ6s23XuQRsade1A,97493
+unique_toolkit/services/chat_service.py,sha256=MAueuSfATFJH8aSeNxbnBI_Ne1UgYUIk3i5F1f2ALFQ,97692
 unique_toolkit/services/factory.py,sha256=7gNeanyHkgX23Iz7RMCQTCkE9gI5asCI11nXxM0jjx4,9832
 unique_toolkit/services/knowledge_base.py,sha256=a9itNRxIrGDIuXkLiTLBsA-m2ncy8F8DwCLCXtWm_GY,43446
 unique_toolkit/short_term_memory/__init__.py,sha256=2mI3AUrffgH7Yt-xS57EGqnHf7jnn6xquoKEhJqk3Wg,185
@@ -406,6 +406,6 @@ unique_toolkit/short_term_memory/service.py,sha256=2JMBEBn6GoyK-a0xvdcWpPft-3C74
 unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/smart_rules/compile.py,sha256=vRe0aMJWuVIK-VsqomgxQImzke7MDmwgSxxJfYCa7zw,1386
 unique_toolkit/test_utilities/events.py,sha256=FLrhMh0A_mQutAUjoD0FdMcVQ4Gq31r35OVGzRjvLro,6760
-unique_toolkit-2026.30.0.dev5.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
-unique_toolkit-2026.30.0.dev5.dist-info/METADATA,sha256=4UB6dPpbvtoQdDASGwf3Ld_iqYFiLTy98BnyqmU3wOg,7426
-unique_toolkit-2026.30.0.dev5.dist-info/RECORD,,
+unique_toolkit-2026.30.0.dev7.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
+unique_toolkit-2026.30.0.dev7.dist-info/METADATA,sha256=tf1qkimmD41c7yoIbUFNmKEgYkiJx-zSyOA1trTxzgQ,7426
+unique_toolkit-2026.30.0.dev7.dist-info/RECORD,,