unique_toolkit 2026.30.0.dev6__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/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.dev6.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.dev6
+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.dev6.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
@@ -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.dev6.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
-unique_toolkit-2026.30.0.dev6.dist-info/METADATA,sha256=mEqu4--XWmzlsENu8qaH8Hcg_TyeP6Rfc0fblZI0cpQ,7426
-unique_toolkit-2026.30.0.dev6.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,,