unique_toolkit 2026.30.0.dev4__py3-none-any.whl → 2026.30.0.dev6__py3-none-any.whl

unique_toolkit/agentic/tools/mcp/tool_wrapper.py

@@ -89,13 +89,14 @@ class MCPToolWrapper(Tool[MCPToolConfig]):
     def tool_description_for_system_prompt(self) -> str:
         """Return tool description for system prompt"""
         # Not using jinja here to keep it simple and not import new packages.
-        description = (
-            f"**MCP Server**: {self._mcp_server.name}\n"
-            f"**Tool Name**: {self.name}\n"
-            f"{self._mcp_tool.system_prompt}"
-        )
-
-        return description
+        lines = [
+            f"**MCP Server**: {self._mcp_server.name}",
+            f"**Tool Name**: {self.name}",
+        ]
+        if self._mcp_tool.system_prompt:
+            lines.append(self._mcp_tool.system_prompt)
+
+        return "\n".join(lines)
 
     def tool_description_for_user_prompt(self) -> str:
         return self._mcp_tool.user_prompt or ""

unique_toolkit/app/__init__.py

@@ -40,6 +40,9 @@ from .schemas import (
 from .schemas import (
     EventUserMessage as EventUserMessage,
 )
+from .schemas import (
+    McpReferenceMapping as McpReferenceMapping,
+)
 from .schemas import (
     McpServer as McpServer,
 )

unique_toolkit/app/schemas.py

@@ -89,6 +89,45 @@ class BaseEvent(BaseModel, Generic[FilterOptionsT]):
 ###
 
 
+class McpReferenceMapping(BaseModel):
+    """Optional per-tool rule for destructuring a list-returning MCP result into
+    individual citation references. A Unique-specific override; when unset the
+    generic citation heuristic is used. Paths are dotted (dict keys and numeric
+    list indices), e.g. ``"issues"`` or ``"fields.summary"``.
+    """
+
+    model_config = model_config
+
+    list_path: Optional[str] = Field(
+        default=None,
+        description="Dotted path to the array of records, e.g. 'issues' or 'result.items'.",
+    )
+    title_path: Optional[str] = Field(
+        default=None,
+        description="Dotted path to each record's title, e.g. 'key' or 'fields.summary'.",
+    )
+    title_template: Optional[str] = Field(
+        default=None,
+        description="Optional title template with {dotted.path} tokens, e.g. '{key} — {fields.summary}'. Takes precedence over title_path.",
+    )
+    details_path: Optional[str] = Field(
+        default=None,
+        description="Optional dotted path to a details line for each record.",
+    )
+    title_from_text: Optional[bool] = Field(
+        default=None,
+        description="Title from the leading text line instead of a field/template: for a single plain-text document (e.g. a fetched Markdown doc) or each plain-text item of a list.",
+    )
+    title_text_path: Optional[str] = Field(
+        default=None,
+        description="When titling a list of objects from text, the dotted path to the text field within each item (e.g. 'content'); leave unset when each list item is itself a plain string.",
+    )
+    title_max_chars: Optional[int] = Field(
+        default=None,
+        description="Max length of the title derived via title_from_text (default 120).",
+    )
+
+
 class McpTool(BaseModel):
     model_config = model_config
 
@@ -117,6 +156,10 @@ class McpTool(BaseModel):
         default=None,
         description="An optional tool format information. This is a Unique specific field.",
     )
+    reference_mapping: Optional[McpReferenceMapping] = Field(
+        default=None,
+        description="Optional rule for destructuring a list-returning result into individual citation references. This is a Unique specific field.",
+    )
     is_connected: bool = Field(
         description="Whether the tool is connected to the MCP server. This is a Unique specific field.",
     )

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-2026.30.0.dev4.dist-info → unique_toolkit-2026.30.0.dev6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.30.0.dev4
+Version: 2026.30.0.dev6
 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.dev0,<2026.30.0rc0
+Requires-Dist: unique-sdk>=2026.30.0.dev1,<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.dev4.dist-info → unique_toolkit-2026.30.0.dev6.dist-info}/RECORD

@@ -201,7 +201,7 @@ unique_toolkit/agentic/tools/factory.py,sha256=YK9uh2EqotQFb9stsXs-A8lsNaxklTCd_
 unique_toolkit/agentic/tools/mcp/__init__.py,sha256=RLF_p-LDRC7GhiB3fdCi4u3bh6V9PY_w26fg61BLyco,122
 unique_toolkit/agentic/tools/mcp/manager.py,sha256=EfWbfNhB8Lnx2EqBgUXhK7klFLaBv2eezvYq8J0TXyo,3808
 unique_toolkit/agentic/tools/mcp/models.py,sha256=DaBVzwJ4dqtgzm8Mv0-NOGBU04zyVe-RgWntBhkpmnE,742
-unique_toolkit/agentic/tools/mcp/tool_wrapper.py,sha256=ehxs3JW-QwWWp5GcI94OnR6wA1AhcMMmvkg3BFx0cgk,15497
+unique_toolkit/agentic/tools/mcp/tool_wrapper.py,sha256=6rrJ83dSzyiteOPXl5ZTca8hSPkH8aOGNjI41IcRp78,15544
 unique_toolkit/agentic/tools/names.py,sha256=MDF0ONLrxeC-iz--EtyrHAox_RN6zasxo5MSbXQYGyw,412
 unique_toolkit/agentic/tools/openai_builtin/__init__.py,sha256=Q6d9mhsLRVF3erT2yMxLld62D3FTZ8nbovI4vLv4Nso,701
 unique_toolkit/agentic/tools/openai_builtin/base.py,sha256=Ob5n1S6HduFbI5RsjIklasRjoKsgCwh840KAb6p2IaI,1858
@@ -230,7 +230,7 @@ unique_toolkit/agentic_table/__init__.py,sha256=bTMTBUomZysXhMtP9Rnzk6hVa2lZ7Xmr
 unique_toolkit/agentic_table/schemas.py,sha256=IhVVSWZyDVIeyVoiP-FWp26MWwXv8oL4KFPTzeQquCc,12568
 unique_toolkit/agentic_table/service.py,sha256=26hblR33Vp8FWWMgtaQCJ0IQT8hmGzGCQy86BDi2wOo,19670
 unique_toolkit/agentic_table/synthetic_chat_event.py,sha256=MylvYuhx5CVt2DbPetzV0sR8GlkL218DmNFi7A6RYVs,2373
-unique_toolkit/app/__init__.py,sha256=VnyvvCugDYZzEK93vVpSL1zHGduuev6yOvLJOGzpmMc,1568
+unique_toolkit/app/__init__.py,sha256=qear5ErW3smLDATxtnUsgkX3f177zsn65ixVW8T0JVE,1641
 unique_toolkit/app/chat_event_filter_options_settings.py,sha256=W0Ydio1UeED5_xqaTzbcGzK2tH4EpA1WHUzdF-iWpqI,364
 unique_toolkit/app/dev_util.py,sha256=cHDuC6jtaptdVISbtMmOCqRTTWBPeCoIUQvUQvcmVyY,5517
 unique_toolkit/app/fast_api_factory.py,sha256=ffoi6h_PD2_DaAuKZNd_DNS53RA0VXJKzGhsDta5SvM,5801
@@ -240,7 +240,7 @@ unique_toolkit/app/init_logging.py,sha256=Lv0FuEiUmxvtV2yovjD0gWNIvLdsi2xdMFeEiN
 unique_toolkit/app/init_sdk.py,sha256=5_oDoETr6akwYyBCb0ivTdMNu3SVgPSkrXcDS6ELyY8,2269
 unique_toolkit/app/performance/async_tasks.py,sha256=8Xv5kEmmFDJO7YyyBxv4e7bcNgq-3z_E2_4z1Yn9EqU,1970
 unique_toolkit/app/performance/async_wrapper.py,sha256=yVVcRDkcdyfjsxro-N29SBvi-7773wnfDplef6-y8xw,1077
-unique_toolkit/app/schemas.py,sha256=NWaj7cpNiXaxU6W-U-bw911cdpXjqsbSUGeVoN4Q3Js,16534
+unique_toolkit/app/schemas.py,sha256=_jHuLDZO9YSP0xf-1AxNVorWYg2NgWfNZ9yXpGt2DHE,18518
 unique_toolkit/app/unique_settings.py,sha256=-bYvxwCw_DvhSViHXgc-pYoNboxhHB8VG2L9YNPiDHU,26861
 unique_toolkit/app/verification.py,sha256=GxFFwcJMy25fCA_Xe89wKW7bgqOu8PAs5y8QpHF0GSc,3861
 unique_toolkit/app/webhook.py,sha256=evpxLDnvz7P3D7-TV1XBkl8WcoldXpjOkw3tvFRxHr0,2368
@@ -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
@@ -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.dev4.dist-info/WHEEL,sha256=eh7sammvW2TypMMMGKgsM83HyA_3qQ5Lgg3ynoecH3M,79
-unique_toolkit-2026.30.0.dev4.dist-info/METADATA,sha256=4iN85BIBq_wHZenyR9KfKIvdQ9JF699LsQ_PVG3HMFY,7426
-unique_toolkit-2026.30.0.dev4.dist-info/RECORD,,
+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,,