unique_toolkit 2026.28.0.dev7__py3-none-any.whl → 2026.28.0.dev9__py3-none-any.whl

unique_toolkit/chat/functions.py

@@ -180,6 +180,7 @@ def map_references(references: list[ContentReference]) -> list[dict[str, Any]]:
     return [
         {
             "name": ref.name,
+            "description": ref.description,
             "url": ref.url,
             "sequenceNumber": ref.sequence_number,
             "sourceId": ref.source_id,
@@ -199,6 +200,7 @@ def map_references_with_original_index(
     return [
         {
             "name": ref.name,
+            "description": ref.description,
             "url": ref.url,
             "sequenceNumber": ref.sequence_number,
             "sourceId": ref.source_id,

unique_toolkit/content/schemas.py

@@ -215,6 +215,17 @@ class ContentReference(BaseModel):
     sequence_number: int
     source: str
     source_id: str
+    description: str | None = Field(
+        default=None,
+        description=(
+            "Optional free-text/structured description carried alongside the "
+            "reference. Used by MCP tool references (UN-22310) to carry an "
+            'enriched ``{"mcp": {connectorName, title, details?, mcpServerId?}}`` '
+            "JSON payload the chat frontend renders as a card. Nullable "
+            "end-to-end (node-chat persists it as a nullable column; empty "
+            "values are dropped before sending)."
+        ),
+    )
     url: str | None = Field(
         default=None,
         description=(
@@ -241,6 +252,8 @@ class ContentReference(BaseModel):
             "source": reference["source"],
             "source_id": reference["sourceId"],
         }
+        if "description" in reference:
+            kwargs["description"] = reference["description"]
         if "originalIndex" in reference:
             kwargs["original_index"] = reference["originalIndex"]
 

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

@@ -2,8 +2,15 @@
 
 Single responsibility: owns every ``unique_sdk.Message.modify_async`` call
 related to a streaming response (``startedStreamingAt``, incremental text
-+ references, ``stoppedStreamingAt``, ``completedAt``), plus the
-``content_chunks`` used to filter references down to what was actually cited.
++ references, and ``stoppedStreamingAt``), plus 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
+``stoppedStreamingAt`` only on rounds that produced answer text; tool-call
+rounds (empty text) leave it null so the message stays in the streaming
+state. ``completedAt`` is intentionally **not** written here — the
+orchestrator marks completion at end-of-turn via ``set_completed_at``.
 
 Attach by calling :meth:`register` once on the owned bus:
 
@@ -162,6 +169,10 @@ class MessagePersistingSubscriber:
         if event.appendices:
             final_text = final_text + "".join(event.appendices)
 
+        # Only mark streaming stopped on the answer round; tool-call rounds have
+        # empty text and must stay "streaming" so the frontend keeps the steps.
+        stopped_streaming_at = now_iso if final_text else None
+
         await unique_sdk.Message.modify_async(
             id=event.message_id,
             chatId=event.chat_id,
@@ -173,6 +184,5 @@ class MessagePersistingSubscriber:
             # Chat completions persist their request as a JSON array; the SDK type is narrower.
             gptRequest=event.gpt_request,  # pyright: ignore[reportArgumentType]
             debugInfo=event.debug_info,
-            stoppedStreamingAt=cast(Any, now_iso),
-            completedAt=cast(Any, now_iso),
+            stoppedStreamingAt=cast(Any, stopped_streaming_at),
         )

{unique_toolkit-2026.28.0.dev7.dist-info → unique_toolkit-2026.28.0.dev9.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.28.0.dev7
+Version: 2026.28.0.dev9
 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.dev10,<2026.28.0rc0
+Requires-Dist: unique-sdk>=2026.28.0.dev11,<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.dev7.dist-info → unique_toolkit-2026.28.0.dev9.dist-info}/RECORD

@@ -248,7 +248,7 @@ unique_toolkit/chat/__init__.py,sha256=N9RubtxWN7UF0DlxP7YgeNTgjjk6p7AwkbGaVvuZC
 unique_toolkit/chat/cancellation.py,sha256=HvGZK-daCMSr0M0AnvMY3EBPH1N_UqQPfrbTjwr8owk,5645
 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=vj0sdeBVm-vsrZOwsHAfCUtpdTwQNgr-aVayvRhUCp4,52360
+unique_toolkit/chat/functions.py,sha256=8PXqo4X2BJhsbHkD9X5Ykqz-iVqbRJ1UlWuCNl1XSX8,52448
 unique_toolkit/chat/rendering.py,sha256=c8YiV9oADRrJQ5A_QBJ4_UFc0NZ-2vVaa7tupoMusso,880
 unique_toolkit/chat/responses_api.py,sha256=QtCOe3TI242N1JChhoFikxUCvGY6xfTvSmaE9Ta8HSM,19811
 unique_toolkit/chat/schemas.py,sha256=l-Epugu27RoajiX5OkFO-BrVNey-wfTmaXiEmNNmG7s,7396
@@ -258,7 +258,7 @@ unique_toolkit/chat/utils.py,sha256=CsD9IER2gA1IOV0fmUOl-N0LmKvA7JP9dGrOqeOY0Gg,
 unique_toolkit/content/__init__.py,sha256=nnQDfnNTEvI8-3dgWvIojoFDYU7edwd8Kloi6lMDG0M,1586
 unique_toolkit/content/constants.py,sha256=1iy4Y67xobl5VTnJB6SxSyuoBWbdLl9244xfVMUZi5o,60
 unique_toolkit/content/functions.py,sha256=avnlEQ1FkVz2NTbXUkafU04jzygqOJtUMQL605rEL4c,34019
-unique_toolkit/content/schemas.py,sha256=00qbe-DlUjqlf-n3QVOotZjpsj0LZNnUoPmdwyMjVHY,10667
+unique_toolkit/content/schemas.py,sha256=AmnlqMpA_JgbgsaZF6Xdy8kt9lcY0YCqlfqpG1tYWeE,11306
 unique_toolkit/content/service.py,sha256=S_sFtG1hJd-zz7XyP-6Gu-B8_Xx-r5v46qZLWdRRyRs,29477
 unique_toolkit/content/smart_rules.py,sha256=9GlHXm63POTi4jy5hW0qUjeqy-JSeL1wIbURM1GKhc4,11481
 unique_toolkit/content/utils.py,sha256=keDkqu_ZzQY0y_g7g_XLOza3p4XsvfqVVDI9ylM91KY,10115
@@ -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=nHh2vQWF6YWurdkijof6UXhTMFzllj_4eSdr8nm9aDI,8016
+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/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
@@ -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.28.0.dev7.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
-unique_toolkit-2026.28.0.dev7.dist-info/METADATA,sha256=nHNAallyeC9qG-si_8OUQL1asr0rDHCppjABATDQfbU,7427
-unique_toolkit-2026.28.0.dev7.dist-info/RECORD,,
+unique_toolkit-2026.28.0.dev9.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
+unique_toolkit-2026.28.0.dev9.dist-info/METADATA,sha256=qXosj8SbRg2-BNbpviYvDe9xj98kVKyr0_KLcJFvYvs,7427
+unique_toolkit-2026.28.0.dev9.dist-info/RECORD,,