unique_toolkit 1.4.2__py3-none-any.whl → 1.4.3__py3-none-any.whl

unique_toolkit/agentic/tools/a2a/config.py

@@ -9,6 +9,10 @@ DEFAULT_PARAM_DESCRIPTION_SUB_AGENT_USER_MESSAGE = """
 This is the message that will be sent to the sub-agent.
 """.strip()
 
+DEFAULT_FORMAT_INFORMATION_SUB_AGENT_SYSTEM_MESSAGE = """
+NEVER mention any references from sub-agent answers in your response.
+"""
+
 
 class ResponseDisplayMode(StrEnum):
     HIDDEN = "hidden"
@@ -40,7 +44,9 @@ class SubAgentToolConfig(BaseToolConfig):
     param_description_sub_agent_user_message: str = (
         DEFAULT_PARAM_DESCRIPTION_SUB_AGENT_USER_MESSAGE
     )
-    tool_format_information_for_system_prompt: str = ""
+    tool_format_information_for_system_prompt: str = (
+        DEFAULT_FORMAT_INFORMATION_SUB_AGENT_SYSTEM_MESSAGE
+    )
     tool_description_for_user_prompt: str = ""
     tool_format_information_for_user_prompt: str = ""
 

unique_toolkit/agentic/tools/a2a/postprocessing/postprocessor.py

@@ -182,23 +182,45 @@ def _consolidate_references_in_place(
 
         references = list(sorted(references, key=lambda ref: ref["sequenceNumber"]))
 
+        ref_map = {}
+
         message_new_refs = []
         for reference in references:
             source_id = reference["sourceId"]
 
             if source_id not in existing_refs:
                 message_new_refs.append(reference)
+
                 existing_refs[source_id] = start_index
                 start_index += 1
 
             reference_num = existing_refs[source_id]
-
             seq_num = reference["sequenceNumber"]
-            message["text"] = re.sub(
-                rf"<sup>{seq_num}</sup>",
-                f"<sup>{reference_num}</sup>",
-                message["text"],
-            )
+
+            ref_map[seq_num] = reference_num
+
             reference["sequenceNumber"] = reference_num
 
+        message["text"] = _replace_references_in_text(message["text"], ref_map)
         message["references"] = message_new_refs
+
+
+def _replace_references_in_text_non_overlapping(
+    text: str, ref_map: dict[int, int]
+) -> str:
+    for orig, repl in ref_map.items():
+        text = re.sub(rf"<sup>{orig}</sup>", f"<sup>{repl}</sup>", text)
+    return text
+
+
+def _replace_references_in_text(text: str, ref_map: dict[int, int]) -> str:
+    # 2 phase replacement, since the map keys and values can overlap
+    max_ref = max(ref_map.keys(), default=0) + 1
+    unique_refs = range(max_ref, max_ref + len(ref_map))
+
+    text = _replace_references_in_text_non_overlapping(
+        text, dict(zip(ref_map.keys(), unique_refs))
+    )
+    return _replace_references_in_text_non_overlapping(
+        text, dict(zip(unique_refs, ref_map.values()))
+    )

{unique_toolkit-1.4.2.dist-info → unique_toolkit-1.4.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.4.2
+Version: 1.4.3
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -118,6 +118,9 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), 
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [1.4.3] - 2025-09-30
+- Fix bug with sub-agent post-processing reference numbers.
+
 ## [1.4.2] - 2025-09-30
 - Adding litellm models `litellm:anthropic-claude-sonnet-4-5` and `litellm:anthropic-claude-opus-4-1`
 

{unique_toolkit-1.4.2.dist-info → unique_toolkit-1.4.3.dist-info}/RECORD

@@ -51,7 +51,7 @@ unique_toolkit/agentic/short_term_memory_manager/persistent_short_term_memory_ma
 unique_toolkit/agentic/thinking_manager/thinking_manager.py,sha256=41QWFsdRrbWlQHBfYCFv726UDom4WbcvaRfjCmoUOQI,4183
 unique_toolkit/agentic/tools/__init__.py,sha256=-ToY9-Xiz0K7qCUydH1h1yG6n4h1hQS8sBuSVPNEq2Y,43
 unique_toolkit/agentic/tools/a2a/__init__.py,sha256=NdY0J33b1G4sbx6UWwNS74JVSAeEtu8u_iEXOT64Uq0,187
-unique_toolkit/agentic/tools/a2a/config.py,sha256=exKyR-RyQ3RDJcEAKwfOdyj1flfbBaRhcdn5ROnmNB4,1513
+unique_toolkit/agentic/tools/a2a/config.py,sha256=3Rp5myYQgPwOq-qmM0OaZDFkovXh9a1YJc8TnieN-K0,1711
 unique_toolkit/agentic/tools/a2a/evaluation/__init__.py,sha256=H9YhT22w8EadV9b-6IDqYqKQa41qcA3m6ADzmP7g6Cc,246
 unique_toolkit/agentic/tools/a2a/evaluation/config.py,sha256=o1Xj2H4175C1ALT8-wIfks69Xez3pgY77PFyPBYS4Hs,1692
 unique_toolkit/agentic/tools/a2a/evaluation/evaluator.py,sha256=V3y72yAZ5ynDnzp8V7UxMpnwa1Xyw7gvjPqfgHCeMkU,7660
@@ -60,7 +60,7 @@ unique_toolkit/agentic/tools/a2a/manager.py,sha256=yuuQuBrAcsT3gAWEdxf6EvRnL_iWt
 unique_toolkit/agentic/tools/a2a/memory.py,sha256=4VFBzITCv5E_8YCc4iF4Y6FhzplS2C-FZaZHdeC7DyA,1028
 unique_toolkit/agentic/tools/a2a/postprocessing/__init__.py,sha256=9M5RRSO4gmQVI1gfve6MGEfVktT9WcELQFZ8Sv6xu4Y,160
 unique_toolkit/agentic/tools/a2a/postprocessing/display.py,sha256=rlBWO2M8Lr5Kx-vmvwlV-vEu33BZE4votP-TMr3-3Dw,3366
-unique_toolkit/agentic/tools/a2a/postprocessing/postprocessor.py,sha256=BWXFsv5akPhkNxRQ3JLktrBkoT1_66joTel_uM4BgRY,6963
+unique_toolkit/agentic/tools/a2a/postprocessing/postprocessor.py,sha256=MGOQJ2bcBEv8v8mI5ZOgsv_J8hPFFLQ2RLODXzJ6eVc,7652
 unique_toolkit/agentic/tools/a2a/postprocessing/test/test_display.py,sha256=UVOd5tVTltngVfsdOn6cuyvG7QmRBkfgUj_SruTgmHI,15279
 unique_toolkit/agentic/tools/a2a/schema.py,sha256=T1l5z6trtPE5nhqPzt5tvfRNDhqL_ST1Wj7_lBWJ58g,304
 unique_toolkit/agentic/tools/a2a/service.py,sha256=ySgCZZ_3APPZ8j0sz-7_1CYxfFTvGCSpu0qsUG1YShQ,7916
@@ -137,7 +137,7 @@ unique_toolkit/short_term_memory/schemas.py,sha256=OhfcXyF6ACdwIXW45sKzjtZX_gkcJ
 unique_toolkit/short_term_memory/service.py,sha256=5PeVBu1ZCAfyDb2HLVvlmqSbyzBBuE9sI2o9Aajqjxg,8884
 unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/smart_rules/compile.py,sha256=cxWjb2dxEI2HGsakKdVCkSNi7VK9mr08w5sDcFCQyWI,9553
-unique_toolkit-1.4.2.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-1.4.2.dist-info/METADATA,sha256=aIA-PciV6Tv8mVFR3uJFxGXqFAQ_-mtJ8-1X-he1ehQ,34076
-unique_toolkit-1.4.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-1.4.2.dist-info/RECORD,,
+unique_toolkit-1.4.3.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-1.4.3.dist-info/METADATA,sha256=kqVo0uCpxe7VLrBCbbENvk2O7lCBBA8PxNVb9J-VcQE,34161
+unique_toolkit-1.4.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-1.4.3.dist-info/RECORD,,