unique_toolkit 1.4.1__tar.gz → 1.4.3__tar.gz

{unique_toolkit-1.4.1 → unique_toolkit-1.4.3}/CHANGELOG.md

@@ -5,6 +5,12 @@ 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`
+
 ## [1.4.1] - 2025-09-30
 - Handle sub agent failed assessments better in sub agent evaluator.
 

{unique_toolkit-1.4.1 → unique_toolkit-1.4.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.4.1
+Version: 1.4.3
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -118,6 +118,12 @@ 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`
+
 ## [1.4.1] - 2025-09-30
 - Handle sub agent failed assessments better in sub agent evaluator.
 

{unique_toolkit-1.4.1 → unique_toolkit-1.4.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_toolkit"
-version = "1.4.1"
+version = "1.4.3"
 description = ""
 authors = [
     "Cedric Klinkert <cedric.klinkert@unique.ch>",

{unique_toolkit-1.4.1 → unique_toolkit-1.4.3}/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-1.4.1 → unique_toolkit-1.4.3}/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.1 → unique_toolkit-1.4.3}/unique_toolkit/language_model/infos.py

@@ -37,7 +37,9 @@ class LanguageModelName(StrEnum):
         "litellm:anthropic-claude-3-7-sonnet-thinking"
     )
     ANTHROPIC_CLAUDE_SONNET_4 = "litellm:anthropic-claude-sonnet-4"
+    ANTHROPIC_CLAUDE_SONNET_4_5 = "litellm:anthropic-claude-sonnet-4-5"
     ANTHROPIC_CLAUDE_OPUS_4 = "litellm:anthropic-claude-opus-4"
+    ANTHROPIC_CLAUDE_OPUS_4_1 = "litellm:anthropic-claude-opus-4-1"
     GEMINI_2_0_FLASH = "litellm:gemini-2-0-flash"
     GEMINI_2_5_FLASH = "litellm:gemini-2-5-flash"
     GEMINI_2_5_FLASH_LITE_PREVIEW_0617 = "litellm:gemini-2-5-flash-lite-preview-06-17"
@@ -663,6 +665,26 @@ class LanguageModelInfo(BaseModel):
                     info_cutoff_at=date(2025, 3, 1),
                     published_at=date(2025, 5, 1),
                 )
+            case LanguageModelName.ANTHROPIC_CLAUDE_SONNET_4_5:
+                return cls(
+                    name=model_name,
+                    capabilities=[
+                        ModelCapabilities.FUNCTION_CALLING,
+                        ModelCapabilities.STREAMING,
+                        ModelCapabilities.VISION,
+                        ModelCapabilities.REASONING,
+                    ],
+                    provider=LanguageModelProvider.LITELLM,
+                    version="claude-sonnet-4-5",
+                    encoder_name=EncoderName.O200K_BASE,  # TODO: Update encoder with litellm
+                    token_limits=LanguageModelTokenLimits(
+                        # Input limit is 200_000, we leave 20_000 tokens as buffer due to tokenizer mismatch
+                        token_limit_input=200_000,
+                        token_limit_output=64_000,
+                    ),
+                    info_cutoff_at=date(2025, 7, 1),
+                    published_at=date(2025, 9, 29),
+                )
             case LanguageModelName.ANTHROPIC_CLAUDE_OPUS_4:
                 return cls(
                     name=model_name,
@@ -683,6 +705,25 @@ class LanguageModelInfo(BaseModel):
                     info_cutoff_at=date(2025, 3, 1),
                     published_at=date(2025, 5, 1),
                 )
+            case LanguageModelName.ANTHROPIC_CLAUDE_OPUS_4_1:
+                return cls(
+                    name=model_name,
+                    capabilities=[
+                        ModelCapabilities.FUNCTION_CALLING,
+                        ModelCapabilities.STREAMING,
+                        ModelCapabilities.VISION,
+                        ModelCapabilities.REASONING,
+                    ],
+                    provider=LanguageModelProvider.LITELLM,
+                    version="claude-opus-4",
+                    encoder_name=EncoderName.O200K_BASE,  # TODO: Update encoder with litellm
+                    token_limits=LanguageModelTokenLimits(
+                        token_limit_input=200_000,
+                        token_limit_output=32_000,
+                    ),
+                    info_cutoff_at=date(2025, 3, 1),
+                    published_at=date(2025, 5, 1),
+                )
             case LanguageModelName.GEMINI_2_0_FLASH:
                 return cls(
                     name=model_name,