unique_toolkit 1.30.0__tar.gz → 1.31.1__tar.gz

{unique_toolkit-1.30.0 → unique_toolkit-1.31.1}/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.31.1] - 2025-11-27
+- Various fixes to sub agent answers.
+
+## [1.31.0] - 2025-11-20
+- Adding model `litellm:anthropic-claude-opus-4-5` to `language_model/info.py`
+
 ## [1.30.0] - 2025-11-26
 - Add option to only display parts of sub agent responses.
 

{unique_toolkit-1.30.0 → unique_toolkit-1.31.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.30.0
+Version: 1.31.1
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -121,6 +121,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.31.1] - 2025-11-27
+- Various fixes to sub agent answers.
+
+## [1.31.0] - 2025-11-20
+- Adding model `litellm:anthropic-claude-opus-4-5` to `language_model/info.py`
+
 ## [1.30.0] - 2025-11-26
 - Add option to only display parts of sub agent responses.
 

{unique_toolkit-1.30.0 → unique_toolkit-1.31.1}/pyproject.toml

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

{unique_toolkit-1.30.0 → unique_toolkit-1.31.1}/unique_toolkit/agentic/tools/a2a/postprocessing/_display_utils.py

@@ -1,6 +1,7 @@
 import re
 from typing import Literal, NamedTuple
 
+from unique_toolkit._common.utils.jinja.render import render_template
 from unique_toolkit.agentic.tools.a2a.postprocessing.config import (
     SubAgentDisplayConfig,
     SubAgentResponseDisplayMode,
@@ -84,6 +85,12 @@ def _prepare_title_template(
     return display_title_template.replace("{}", "{%s}" % display_name_placeholder)
 
 
+def _clean_linebreaks(text: str) -> str:
+    text = text.strip()
+    text = re.sub(r"^(<br>)*|(<br>)*$", "", text)
+    return text
+
+
 def _get_display_template(
     mode: SubAgentResponseDisplayMode,
     add_quote_border: bool,
@@ -126,7 +133,7 @@ def _get_display_template(
     if add_block_border:
         template = _wrap_with_block_border(template)
 
-    return template.strip()
+    return _clean_linebreaks(template)
 
 
 def _get_display_removal_re(
@@ -167,8 +174,7 @@ def get_sub_agent_answer_parts(
 
     substrings = []
     for config in display_config.answer_substrings_config:
-        match = re.search(config.regexp, answer)
-        if match is not None:
+        for match in config.regexp.finditer(answer):
             text = match.group(0)
             substrings.append(
                 SubAgentAnswerPart(
@@ -183,7 +189,7 @@ def get_sub_agent_answer_parts(
 def get_sub_agent_answer_display(
     display_name: str,
     display_config: SubAgentDisplayConfig,
-    answer: str | list[str],
+    answer: str | list[SubAgentAnswerPart],
     assistant_id: str,
 ) -> str:
     template = _get_display_template(
@@ -194,7 +200,12 @@ def get_sub_agent_answer_display(
     )
 
     if isinstance(answer, list):
-        answer = display_config.answer_substrings_separator.join(answer)
+        answer = render_template(
+            display_config.answer_substrings_jinja_template,
+            {
+                "substrings": [answer.formatted_text for answer in answer],
+            },
+        )
 
     return template.format(
         display_name=display_name, answer=answer, assistant_id=assistant_id

{unique_toolkit-1.30.0 → unique_toolkit-1.31.1}/unique_toolkit/agentic/tools/a2a/postprocessing/config.py

@@ -1,3 +1,4 @@
+import re
 from enum import StrEnum
 from typing import Literal
 
@@ -16,7 +17,7 @@ class SubAgentResponseDisplayMode(StrEnum):
 class SubAgentAnswerSubstringConfig(BaseModel):
     model_config = get_configuration_dict()
 
-    regexp: str = Field(
+    regexp: re.Pattern[str] = Field(
         description="The regular expression to use to extract the substring. The first capture group will always be used.",
     )
     display_template: str = Field(
@@ -25,6 +26,13 @@ class SubAgentAnswerSubstringConfig(BaseModel):
     )
 
 
+_ANSWER_SUBSTRINGS_JINJA_TEMPLATE = """
+{% for substring in substrings %}
+{{ substring }}
+{% endfor %}
+""".strip()
+
+
 class SubAgentDisplayConfig(BaseModel):
     model_config = get_configuration_dict()
 
@@ -64,7 +72,7 @@ class SubAgentDisplayConfig(BaseModel):
         default=[],
         description="If set, only parts of the answer matching the provided regular expressions will be displayed.",
     )
-    answer_substrings_separator: str = Field(
-        default="\n",
-        description="The separator to use between the substrings.",
+    answer_substrings_jinja_template: str = Field(
+        default=_ANSWER_SUBSTRINGS_JINJA_TEMPLATE,
+        description="The template to use in order to format the different answer substrings, if any.",
     )

{unique_toolkit-1.30.0 → unique_toolkit-1.31.1}/unique_toolkit/agentic/tools/a2a/postprocessing/display.py

@@ -7,6 +7,7 @@ import unique_sdk
 from pydantic import BaseModel, Field
 
 from unique_toolkit._common.pydantic_helpers import get_configuration_dict
+from unique_toolkit._common.utils.jinja.render import render_template
 from unique_toolkit.agentic.postprocessor.postprocessor_manager import Postprocessor
 from unique_toolkit.agentic.tools.a2a.postprocessing._display_utils import (
     get_sub_agent_answer_display,
@@ -38,20 +39,22 @@ class SubAgentDisplaySpec(NamedTuple):
     display_config: SubAgentDisplayConfig
 
 
+_ANSWERS_JINJA_TEMPLATE = """
+{% for answer in answers %}
+{{ answer }}
+{% endfor %}
+""".strip()
+
+
 class SubAgentResponsesPostprocessorConfig(BaseModel):
     model_config = get_configuration_dict()
 
     sleep_time_before_update: float = Field(
-        default=1, description="Time to sleep before updating the main agent message."
+        default=0, description="Time to sleep before updating the main agent message."
     )
-
-    remove_duplicate_answers: bool = Field(
-        default=False,
-        description="If set, duplicate answers will only be displayed once. If sub agent is configured to display only substrings, this will remove duplicate substrings across different responses.",
-    )
-    answer_separator: str = Field(
-        default="",
-        description="The separator to use between the different sub agent answers.",
+    answers_jinja_template: str = Field(
+        default=_ANSWERS_JINJA_TEMPLATE,
+        description="The template to use to display the sub agent answers.",
     )
 
 
@@ -106,8 +109,6 @@ class SubAgentResponsesDisplayPostprocessor(Postprocessor):
         answers_displayed_before = []
         answers_displayed_after = []
 
-        all_answers_displayed = set()
-
         for assistant_id, responses in displayed_sub_agent_responses.items():
             for response in responses:
                 message = response.message
@@ -130,34 +131,20 @@ class SubAgentResponsesDisplayPostprocessor(Postprocessor):
                         assistant_id,
                         response.sequence_number,
                     )
-
-                message_text = message["text"] or ""
+                    continue
 
                 answer_parts = get_sub_agent_answer_parts(
-                    answer=message_text,
+                    answer=message["text"],
                     display_config=tool_info.display_config,
                 )
 
                 if len(answer_parts) == 0:
                     continue
 
-                answer_display_texts = []
-                if self._config.remove_duplicate_answers:
-                    for answer_part in answer_parts:
-                        if answer_part.matching_text in all_answers_displayed:
-                            continue
-
-                        all_answers_displayed.add(answer_part.matching_text)
-                        answer_display_texts.append(answer_part.formatted_text)
-                else:
-                    answer_display_texts = [
-                        answer_part.formatted_text for answer_part in answer_parts
-                    ]
-
                 answer = get_sub_agent_answer_display(
                     display_name=display_name,
                     display_config=tool_info.display_config,
-                    answer=answer_display_texts,
+                    answer=answer_parts,
                     assistant_id=assistant_id,
                 )
 
@@ -170,7 +157,7 @@ class SubAgentResponsesDisplayPostprocessor(Postprocessor):
             text=loop_response.message.text,
             answers_before=answers_displayed_before,
             answers_after=answers_displayed_after,
-            sep=self._config.answer_separator,
+            template=self._config.answers_jinja_template,
         )
 
         return True
@@ -211,12 +198,12 @@ def _get_final_answer_display(
     text: str,
     answers_before: list[str],
     answers_after: list[str],
-    sep: str = "<br>\n\n",
+    template: str = _ANSWERS_JINJA_TEMPLATE,
 ) -> str:
     if len(answers_before) > 0:
-        text = sep.join(answers_before) + sep + text
+        text = render_template(template, {"answers": answers_before}) + text
 
     if len(answers_after) > 0:
-        text = text + sep + sep.join(answers_after)
+        text = text + render_template(template, {"answers": answers_after})
 
     return text.strip()

{unique_toolkit-1.30.0 → unique_toolkit-1.31.1}/unique_toolkit/agentic/tools/a2a/postprocessing/test/test_display_utils.py

@@ -1492,9 +1492,7 @@ def test_get_sub_agent_answer_parts__returns_empty_list__when_no_matches() -> No
 
 
 @pytest.mark.ai
-def test_get_sub_agent_answer_parts__extracts_first_match_only__for_each_regexp() -> (
-    None
-):
+def test_get_sub_agent_answer_parts__extracts_all_matches__for_each_regexp() -> None:
     """
     Purpose: Verify only first match per regexp is extracted.
     Why this matters: Function uses re.search which finds first occurrence.
@@ -1513,7 +1511,7 @@ def test_get_sub_agent_answer_parts__extracts_first_match_only__for_each_regexp(
     result = get_sub_agent_answer_parts(answer=answer, display_config=config)
 
     # Assert
-    assert len(result) == 1
+    assert len(result) == 2
     assert result[0].matching_text == "42"
 
 
@@ -1661,10 +1659,12 @@ def test_get_sub_agent_answer_parts__preserves_order__of_configs_not_matches() -
     result = get_sub_agent_answer_parts(answer=answer, display_config=config)
 
     # Assert
-    assert len(result) == 2
+    assert len(result) == 4
     # Results follow config order, not text order
     assert result[0].matching_text == "first"
-    assert result[1].matching_text == "123"
+    assert result[1].matching_text == "then"
+    assert result[2].matching_text == "abc"
+    assert result[3].matching_text == "123"
 
 
 @pytest.mark.ai

{unique_toolkit-1.30.0 → unique_toolkit-1.31.1}/unique_toolkit/language_model/infos.py

@@ -47,6 +47,7 @@ class LanguageModelName(StrEnum):
     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"
+    ANTHROPIC_CLAUDE_OPUS_4_5 = "litellm:anthropic-claude-opus-4-5"
     GEMINI_2_0_FLASH = "litellm:gemini-2-0-flash"
     GEMINI_2_5_FLASH = "litellm:gemini-2-5-flash"
     GEMINI_2_5_FLASH_LITE = "litellm:gemini-2-5-flash-lite"
@@ -946,7 +947,7 @@ class LanguageModelInfo(BaseModel):
                         ModelCapabilities.REASONING,
                     ],
                     provider=LanguageModelProvider.LITELLM,
-                    version="claude-opus-4",
+                    version="claude-opus-4-1",
                     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
@@ -956,6 +957,26 @@ class LanguageModelInfo(BaseModel):
                     info_cutoff_at=date(2025, 3, 1),
                     published_at=date(2025, 5, 1),
                 )
+            case LanguageModelName.ANTHROPIC_CLAUDE_OPUS_4_5:
+                return cls(
+                    name=model_name,
+                    capabilities=[
+                        ModelCapabilities.FUNCTION_CALLING,
+                        ModelCapabilities.STREAMING,
+                        ModelCapabilities.VISION,
+                        ModelCapabilities.REASONING,
+                    ],
+                    provider=LanguageModelProvider.LITELLM,
+                    version="claude-opus-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=180_000,
+                        token_limit_output=64_000,
+                    ),
+                    info_cutoff_at=date(2025, 8, 1),
+                    published_at=date(2025, 11, 13),
+                )
             case LanguageModelName.GEMINI_2_0_FLASH:
                 return cls(
                     name=model_name,