unique_toolkit 1.19.3__tar.gz → 1.20.1__tar.gz

{unique_toolkit-1.19.3 → unique_toolkit-1.20.1}/CHANGELOG.md

@@ -6,6 +6,14 @@ 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.20.1] - 2025-10-30
+- Fix typing issues in `LanguageModelFunction`.
+
+## [1.20.0] - 2025-10-30
+- Fix bug where async tasks executed with `SafeTaskExecutor` did not log exceptions.
+- Add option to customize sub agent response display title.
+- Add option to display sub agent responses after the main agent response.
+- Add option to specify postprocessors to run before or after the others in the `PostprocessorManager`.
 
 ## [1.19.3] - 2025-10-29
 - More documentation on advanced rendering
@@ -21,6 +29,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ## [1.19.0] - 2025-10-28
 - Enable additional headers on openai and langchain client
 
+
 ## [1.18.1] - 2025-10-28
 - Fix bug where sub agent references were not properly displayed in the main agent response when the sub agent response was hidden.
 

{unique_toolkit-1.19.3 → unique_toolkit-1.20.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.19.3
+Version: 1.20.1
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -119,6 +119,14 @@ 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.20.1] - 2025-10-30
+- Fix typing issues in `LanguageModelFunction`.
+
+## [1.20.0] - 2025-10-30
+- Fix bug where async tasks executed with `SafeTaskExecutor` did not log exceptions.
+- Add option to customize sub agent response display title.
+- Add option to display sub agent responses after the main agent response.
+- Add option to specify postprocessors to run before or after the others in the `PostprocessorManager`.
 
 ## [1.19.3] - 2025-10-29
 - More documentation on advanced rendering
@@ -134,6 +142,7 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 ## [1.19.0] - 2025-10-28
 - Enable additional headers on openai and langchain client
 
+
 ## [1.18.1] - 2025-10-28
 - Fix bug where sub agent references were not properly displayed in the main agent response when the sub agent response was hidden.
 

{unique_toolkit-1.19.3 → unique_toolkit-1.20.1}/pyproject.toml

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

{unique_toolkit-1.19.3 → unique_toolkit-1.20.1}/unique_toolkit/agentic/postprocessor/postprocessor_manager.py

@@ -85,15 +85,39 @@ class PostprocessorManager:
         self._chat_service = chat_service
         self._postprocessors: list[Postprocessor | ResponsesApiPostprocessor] = []
 
+        # Allow to add postprocessors that should be run before or after the others.
+        self._first_postprocessor: Postprocessor | ResponsesApiPostprocessor | None = (
+            None
+        )
+        self._last_postprocessor: Postprocessor | ResponsesApiPostprocessor | None = (
+            None
+        )
+
     def add_postprocessor(
         self, postprocessor: Postprocessor | ResponsesApiPostprocessor
     ):
         self._postprocessors.append(postprocessor)
 
+    def set_first_postprocessor(
+        self, postprocessor: Postprocessor | ResponsesApiPostprocessor
+    ) -> None:
+        if self._first_postprocessor is not None:
+            raise ValueError("Cannot set first postprocessor if already set.")
+
+        self._first_postprocessor = postprocessor
+
+    def set_last_postprocessor(
+        self, postprocessor: Postprocessor | ResponsesApiPostprocessor
+    ) -> None:
+        if self._last_postprocessor is not None:
+            raise ValueError("Cannot set last postprocessor if already set.")
+
+        self._last_postprocessor = postprocessor
+
     def get_postprocessors(
         self, name: str
     ) -> list[Postprocessor | ResponsesApiPostprocessor]:
-        return self._postprocessors
+        return self._get_all_postprocessors()
 
     async def run_postprocessors(
         self,
@@ -103,14 +127,7 @@ class PostprocessorManager:
             logger=self._logger,
         )
 
-        if isinstance(loop_response, ResponsesLanguageModelStreamResponse):
-            postprocessors = self._postprocessors
-        else:
-            postprocessors = [
-                postprocessor
-                for postprocessor in self._postprocessors
-                if isinstance(postprocessor, Postprocessor)
-            ]
+        postprocessors = self._get_valid_postprocessors_for_loop_response(loop_response)
 
         tasks = [
             task_executor.execute_async(
@@ -122,17 +139,14 @@ class PostprocessorManager:
         ]
         postprocessor_results = await asyncio.gather(*tasks)
 
-        for postprocessor, result in zip(postprocessors, postprocessor_results):
-            if not result.success:
-                self._logger.warning(
-                    "Postprocessor %s failed to run.",
-                    postprocessor.get_name(),
-                    exc_info=result.exception,
-                )
+        successful_postprocessors: list[Postprocessor | ResponsesApiPostprocessor] = []
+        for i in range(len(postprocessors)):
+            if postprocessor_results[i].success:
+                successful_postprocessors.append(postprocessors[i])
 
         modification_results = [
-            postprocessor.apply_postprocessing_to_response(loop_response)  # type: ignore
-            for postprocessor in postprocessors
+            postprocessor.apply_postprocessing_to_response(loop_response)  # type: ignore (checked in `get_valid_postprocessors_for_loop_response`)
+            for postprocessor in successful_postprocessors
         ]
 
         has_been_modified = any(modification_results)
@@ -155,6 +169,44 @@ class PostprocessorManager:
         self,
         text: str,
     ) -> str:
-        for postprocessor in self._postprocessors:
+        for postprocessor in self._get_all_postprocessors():
             text = await postprocessor.remove_from_text(text)
         return text
+
+    def _get_all_postprocessors(
+        self,
+    ) -> list[Postprocessor | ResponsesApiPostprocessor]:
+        return [
+            postprocessor
+            for postprocessor in [
+                self._first_postprocessor,
+                *self._postprocessors,
+                self._last_postprocessor,
+            ]
+            if postprocessor is not None
+        ]
+
+    def _get_valid_postprocessors_for_loop_response(
+        self, loop_response: LanguageModelStreamResponse
+    ) -> list[Postprocessor | ResponsesApiPostprocessor]:
+        all_postprocessors = self._get_all_postprocessors()
+
+        postprocessors: list[Postprocessor | ResponsesApiPostprocessor] = []
+
+        if isinstance(loop_response, ResponsesLanguageModelStreamResponse):
+            """
+            All processore can be executed, since `ResponsesLanguageModelStreamResponse`
+            is a subclass of `LanguageModelStreamResponse`
+            """
+            postprocessors = all_postprocessors
+        else:
+            """
+            Cannot execute Responses API-specific postprocessors
+            """
+            postprocessors = [
+                postprocessor
+                for postprocessor in all_postprocessors
+                if isinstance(postprocessor, Postprocessor)
+            ]
+
+        return postprocessors

{unique_toolkit-1.19.3 → unique_toolkit-1.20.1}/unique_toolkit/agentic/tools/a2a/evaluation/evaluator.py

@@ -52,7 +52,7 @@ def _format_single_assessment_found(name: str, explanation: str) -> str:
     return _SingleAssessmentData(name=name, explanation=explanation).model_dump_json()
 
 
-@failsafe(failure_return_value=False)
+@failsafe(failure_return_value=False, log_exceptions=False)
 def _is_single_assessment_found(value: str) -> bool:
     _ = _SingleAssessmentData.model_validate_json(value)
     return True

{unique_toolkit-1.19.3 → unique_toolkit-1.20.1}/unique_toolkit/agentic/tools/a2a/postprocessing/_display.py

@@ -77,10 +77,17 @@ def _add_line_break(text: str, before: bool = True, after: bool = True) -> str:
     return _wrap_text(text, start_tag, end_tag)
 
 
+def _prepare_title_template(
+    display_title_template: str, display_name_placeholder: str
+) -> str:
+    return display_title_template.replace("{}", "{%s}" % display_name_placeholder)
+
+
 def _get_display_template(
     mode: SubAgentResponseDisplayMode,
     add_quote_border: bool,
     add_block_border: bool,
+    display_title_template: str,
     answer_placeholder: str = "answer",
     assistant_id_placeholder: str = "assistant_id",
     display_name_placeholder: str = "display_name",
@@ -89,7 +96,9 @@ def _get_display_template(
         return ""
 
     assistant_id_placeholder = _wrap_hidden_div("{%s}" % assistant_id_placeholder)
-    display_name_placeholder = _wrap_strong("{%s}" % display_name_placeholder)
+    title_template = _prepare_title_template(
+        display_title_template, display_name_placeholder
+    )
     template = _join_text_blocks(
         assistant_id_placeholder, "{%s}" % answer_placeholder, sep="\n\n"
     )  # Double line break is needed for markdown formatting
@@ -104,20 +113,14 @@ def _get_display_template(
             template = _wrap_with_details_tag(
                 template,
                 "open",
-                display_name_placeholder,
+                title_template,
             )
         case SubAgentResponseDisplayMode.DETAILS_CLOSED:
-            template = _wrap_with_details_tag(
-                template, "closed", display_name_placeholder
-            )
+            template = _wrap_with_details_tag(template, "closed", title_template)
         case SubAgentResponseDisplayMode.PLAIN:
-            display_name_placeholder = _add_line_break(
-                display_name_placeholder, before=False, after=True
-            )
-            template = _join_text_blocks(display_name_placeholder, template)
             # Add a hidden block border to seperate sub agent answers from the rest of the text.
             hidden_block_border = _wrap_hidden_div("sub_agent_answer_block")
-            template = _join_text_blocks(template, hidden_block_border)
+            template = _join_text_blocks(title_template, template, hidden_block_border)
 
     if add_block_border:
         template = _wrap_with_block_border(template)
@@ -130,11 +133,13 @@ def _get_display_removal_re(
     mode: SubAgentResponseDisplayMode,
     add_quote_border: bool,
     add_block_border: bool,
+    display_title_template: str,
 ) -> re.Pattern[str]:
     template = _get_display_template(
         mode=mode,
         add_quote_border=add_quote_border,
         add_block_border=add_block_border,
+        display_title_template=display_title_template,
     )
 
     pattern = template.format(
@@ -147,6 +152,7 @@ def _get_display_removal_re(
 def _build_sub_agent_answer_display(
     display_name: str,
     display_mode: SubAgentResponseDisplayMode,
+    display_title_template: str,
     add_quote_border: bool,
     add_block_border: bool,
     answer: str,
@@ -156,6 +162,7 @@ def _build_sub_agent_answer_display(
         mode=display_mode,
         add_quote_border=add_quote_border,
         add_block_border=add_block_border,
+        display_title_template=display_title_template,
     )
     return template.format(
         display_name=display_name, answer=answer, assistant_id=assistant_id
@@ -168,11 +175,13 @@ def _remove_sub_agent_answer_from_text(
     add_block_border: bool,
     text: str,
     assistant_id: str,
+    display_title_template: str,
 ) -> str:
     pattern = _get_display_removal_re(
         assistant_id=assistant_id,
         mode=display_mode,
         add_quote_border=add_quote_border,
         add_block_border=add_block_border,
+        display_title_template=display_title_template,
     )
     return re.sub(pattern, "", text)

{unique_toolkit-1.19.3 → unique_toolkit-1.20.1}/unique_toolkit/agentic/tools/a2a/postprocessing/config.py

@@ -1,4 +1,5 @@
 from enum import StrEnum
+from typing import Literal
 
 from pydantic import BaseModel, Field
 
@@ -31,3 +32,14 @@ class SubAgentDisplayConfig(BaseModel):
         default=False,
         description="If set, a block border is added around the sub agent response.",
     )
+    display_title_template: str = Field(
+        default="Answer from <strong>{}</strong>",
+        description=(
+            "The template to use for the display title of the sub agent response."
+            "If a placeholder '{}' is present, it will be replaced with the display name of the sub agent."
+        ),
+    )
+    position: Literal["before", "after"] = Field(
+        default="before",
+        description="The position of the sub agent response in the main agent response.",
+    )

{unique_toolkit-1.19.3 → unique_toolkit-1.20.1}/unique_toolkit/agentic/tools/a2a/postprocessing/postprocessor.py

@@ -116,7 +116,9 @@ class SubAgentResponsesPostprocessor(Postprocessor):
             list(displayed_sub_agent_infos.values()), existing_refs
         )
 
-        answers = []
+        answers_displayed_before = []
+        answers_displayed_after = []
+
         for assistant_id in displayed_sub_agent_infos.keys():
             messages = self._assistant_id_to_tool_info[assistant_id]["responses"]
 
@@ -129,17 +131,23 @@ class SubAgentResponsesPostprocessor(Postprocessor):
                 if len(messages) > 1:
                     display_name += f" {sequence_number}"
 
-                answers.append(
-                    _build_sub_agent_answer_display(
-                        display_name=display_name,
-                        assistant_id=assistant_id,
-                        display_mode=display_mode,
-                        answer=message["text"],
-                        add_quote_border=tool_info["display_config"].add_quote_border,
-                        add_block_border=tool_info["display_config"].add_block_border,
-                    )
+                answer = _build_sub_agent_answer_display(
+                    display_name=display_name,
+                    display_mode=display_mode,
+                    add_quote_border=tool_info["display_config"].add_quote_border,
+                    add_block_border=tool_info["display_config"].add_block_border,
+                    answer=message["text"],
+                    assistant_id=assistant_id,
+                    display_title_template=tool_info[
+                        "display_config"
+                    ].display_title_template,
                 )
 
+                if tool_info["display_config"].position == "before":
+                    answers_displayed_before.append(answer)
+                else:
+                    answers_displayed_after.append(answer)
+
                 loop_response.message.references.extend(
                     ContentReference(
                         message_id=self._main_agent_message["id"],
@@ -152,13 +160,20 @@ class SubAgentResponsesPostprocessor(Postprocessor):
                     for ref in message["references"]
                 )
 
-        if len(answers) > 0:
+        if len(answers_displayed_before) > 0:
             loop_response.message.text = (
-                "<br>\n\n".join(answers)
+                "<br>\n\n".join(answers_displayed_before)
                 + "<br>\n\n"
                 + loop_response.message.text.strip()
             )
 
+        if len(answers_displayed_after) > 0:
+            loop_response.message.text = (
+                loop_response.message.text.strip()
+                + "<br>\n\n"
+                + "<br>\n\n".join(answers_displayed_after)
+            )
+
         return True
 
     @override
@@ -172,6 +187,7 @@ class SubAgentResponsesPostprocessor(Postprocessor):
                     assistant_id=assistant_id,
                     add_quote_border=display_config.add_quote_border,
                     add_block_border=display_config.add_block_border,
+                    display_title_template=display_config.display_title_template,
                 )
         return text