unique_toolkit 1.19.3__py3-none-any.whl → 1.20.0__py3-none-any.whl

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/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/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/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/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
 

unique_toolkit/agentic/tools/a2a/postprocessing/test/test_display.py

@@ -379,7 +379,10 @@ def test_get_display_template__returns_empty__when_hidden_mode() -> None:
 
     # Act
     result = _get_display_template(
-        mode=mode, add_quote_border=False, add_block_border=False
+        mode=mode,
+        add_quote_border=False,
+        add_block_border=False,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -403,7 +406,10 @@ def test_get_display_template__includes_placeholders__for_all_modes() -> None:
     for mode in modes:
         # Act
         result = _get_display_template(
-            mode=mode, add_quote_border=False, add_block_border=False
+            mode=mode,
+            add_quote_border=False,
+            add_block_border=False,
+            display_title_template="Answer from <strong>{}</strong>",
         )
 
         # Assert
@@ -424,7 +430,10 @@ def test_get_display_template__wraps_assistant_id__as_hidden_div() -> None:
 
     # Act
     result = _get_display_template(
-        mode=mode, add_quote_border=False, add_block_border=False
+        mode=mode,
+        add_quote_border=False,
+        add_block_border=False,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -444,7 +453,10 @@ def test_get_display_template__wraps_display_name__as_strong() -> None:
 
     # Act
     result = _get_display_template(
-        mode=mode, add_quote_border=False, add_block_border=False
+        mode=mode,
+        add_quote_border=False,
+        add_block_border=False,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -465,7 +477,10 @@ def test_get_display_template__adds_details_open__when_details_open_mode() -> No
 
     # Act
     result = _get_display_template(
-        mode=mode, add_quote_border=False, add_block_border=False
+        mode=mode,
+        add_quote_border=False,
+        add_block_border=False,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -487,7 +502,10 @@ def test_get_display_template__adds_details_closed__when_details_closed_mode() -
 
     # Act
     result = _get_display_template(
-        mode=mode, add_quote_border=False, add_block_border=False
+        mode=mode,
+        add_quote_border=False,
+        add_block_border=False,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -510,7 +528,10 @@ def test_get_display_template__adds_line_break_after_name__in_plain_mode() -> No
 
     # Act
     result = _get_display_template(
-        mode=mode, add_quote_border=False, add_block_border=False
+        mode=mode,
+        add_quote_border=False,
+        add_block_border=False,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -530,7 +551,10 @@ def test_get_display_template__adds_quote_border__when_flag_true() -> None:
 
     # Act
     result = _get_display_template(
-        mode=mode, add_quote_border=True, add_block_border=False
+        mode=mode,
+        add_quote_border=True,
+        add_block_border=False,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -550,7 +574,10 @@ def test_get_display_template__adds_block_border__when_flag_true() -> None:
 
     # Act
     result = _get_display_template(
-        mode=mode, add_quote_border=False, add_block_border=True
+        mode=mode,
+        add_quote_border=False,
+        add_block_border=True,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -570,7 +597,10 @@ def test_get_display_template__adds_both_borders__when_both_flags_true() -> None
 
     # Act
     result = _get_display_template(
-        mode=mode, add_quote_border=True, add_block_border=True
+        mode=mode,
+        add_quote_border=True,
+        add_block_border=True,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -601,6 +631,7 @@ def test_get_display_removal_re__returns_pattern__for_plain_mode() -> None:
         mode=mode,
         add_quote_border=False,
         add_block_border=False,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -629,6 +660,7 @@ def test_get_display_removal_re__returns_pattern__for_details_modes() -> None:
             mode=mode,
             add_quote_border=False,
             add_block_border=False,
+            display_title_template="Answer from <strong>{}</strong>",
         )
 
         # Assert
@@ -653,6 +685,7 @@ def test_get_display_removal_re__includes_assistant_id__in_pattern() -> None:
         mode=mode,
         add_quote_border=False,
         add_block_border=False,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -676,6 +709,7 @@ def test_get_display_removal_re__has_capture_groups__for_answer_and_name() -> No
         mode=mode,
         add_quote_border=False,
         add_block_border=False,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -707,6 +741,7 @@ def test_build_sub_agent_answer_display__creates_html__for_plain_mode() -> None:
         add_block_border=False,
         answer=answer,
         assistant_id=assistant_id,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -738,6 +773,7 @@ def test_build_sub_agent_answer_display__creates_details__for_details_open() ->
         add_block_border=False,
         answer=answer,
         assistant_id=assistant_id,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -769,6 +805,7 @@ def test_build_sub_agent_answer_display__creates_details__for_details_closed() -
         add_block_border=False,
         answer=answer,
         assistant_id=assistant_id,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -800,6 +837,7 @@ def test_build_sub_agent_answer_display__returns_empty__for_hidden_mode() -> Non
         add_block_border=False,
         answer=answer,
         assistant_id=assistant_id,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -830,6 +868,7 @@ def test_remove_sub_agent_answer__removes_plain_display__from_text() -> None:
         add_block_border=False,
         answer=answer,
         assistant_id=assistant_id,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     text_with_display = f"Before content\n{display}\nAfter content"
@@ -841,6 +880,7 @@ def test_remove_sub_agent_answer__removes_plain_display__from_text() -> None:
         add_block_border=False,
         text=text_with_display,
         assistant_id=assistant_id,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -871,6 +911,7 @@ def test_remove_sub_agent_answer__removes_details_open__from_text() -> None:
         add_block_border=False,
         answer=answer,
         assistant_id=assistant_id,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     text_with_display = f"Start\n{display}\nEnd"
@@ -882,6 +923,7 @@ def test_remove_sub_agent_answer__removes_details_open__from_text() -> None:
         add_block_border=False,
         text=text_with_display,
         assistant_id=assistant_id,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -913,6 +955,7 @@ def test_remove_sub_agent_answer__removes_details_closed__from_text() -> None:
         add_block_border=False,
         answer=answer,
         assistant_id=assistant_id,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     text_with_display = f"Beginning\n{display}\nEnding"
@@ -924,6 +967,7 @@ def test_remove_sub_agent_answer__removes_details_closed__from_text() -> None:
         add_block_border=False,
         text=text_with_display,
         assistant_id=assistant_id,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -955,6 +999,7 @@ def test_remove_sub_agent_answer__removes_with_quote_border__from_text() -> None
         add_block_border=False,
         answer=answer,
         assistant_id=assistant_id,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     text_with_display = f"Before\n{display}\nAfter"
@@ -966,6 +1011,7 @@ def test_remove_sub_agent_answer__removes_with_quote_border__from_text() -> None
         add_block_border=False,
         text=text_with_display,
         assistant_id=assistant_id,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -997,6 +1043,7 @@ def test_remove_sub_agent_answer__removes_with_block_border__from_text() -> None
         add_block_border=True,
         answer=answer,
         assistant_id=assistant_id,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     text_with_display = f"Start\n{display}\nFinish"
@@ -1008,6 +1055,7 @@ def test_remove_sub_agent_answer__removes_with_block_border__from_text() -> None
         add_block_border=True,
         text=text_with_display,
         assistant_id=assistant_id,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -1039,6 +1087,7 @@ def test_remove_sub_agent_answer__removes_with_both_borders__from_text() -> None
         add_block_border=True,
         answer=answer,
         assistant_id=assistant_id,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     text_with_display = f"Prefix\n{display}\nSuffix"
@@ -1050,6 +1099,7 @@ def test_remove_sub_agent_answer__removes_with_both_borders__from_text() -> None
         add_block_border=True,
         text=text_with_display,
         assistant_id=assistant_id,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -1082,6 +1132,7 @@ def test_remove_sub_agent_answer__preserves_other_content__with_multiple_display
         add_block_border=False,
         answer="Answer from agent 1",
         assistant_id=assistant_id_1,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     display_2 = _build_sub_agent_answer_display(
@@ -1091,6 +1142,7 @@ def test_remove_sub_agent_answer__preserves_other_content__with_multiple_display
         add_block_border=False,
         answer="Answer from agent 2",
         assistant_id=assistant_id_2,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     text_with_displays = f"Start\n{display_1}\nMiddle\n{display_2}\nEnd"
@@ -1102,6 +1154,7 @@ def test_remove_sub_agent_answer__preserves_other_content__with_multiple_display
         add_block_border=False,
         text=text_with_displays,
         assistant_id=assistant_id_1,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -1135,6 +1188,7 @@ def test_remove_sub_agent_answer__handles_multiline_answer__with_dotall_flag() -
         add_block_border=False,
         answer=answer,
         assistant_id=assistant_id,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     text_with_display = f"Before\n{display}\nAfter"
@@ -1146,6 +1200,7 @@ def test_remove_sub_agent_answer__handles_multiline_answer__with_dotall_flag() -
         add_block_border=False,
         text=text_with_display,
         assistant_id=assistant_id,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -1179,6 +1234,7 @@ def test_remove_sub_agent_answer__handles_special_regex_chars__in_answer() -> No
         add_block_border=False,
         answer=answer,
         assistant_id=assistant_id,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     text_with_display = f"Start\n{display}\nEnd"
@@ -1190,6 +1246,7 @@ def test_remove_sub_agent_answer__handles_special_regex_chars__in_answer() -> No
         add_block_border=False,
         text=text_with_display,
         assistant_id=assistant_id,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -1220,6 +1277,7 @@ def test_remove_sub_agent_answer__handles_empty_answer__successfully() -> None:
         add_block_border=False,
         answer=answer,
         assistant_id=assistant_id,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     text_with_display = f"Beginning\n{display}\nEnding"
@@ -1231,6 +1289,7 @@ def test_remove_sub_agent_answer__handles_empty_answer__successfully() -> None:
         add_block_border=False,
         text=text_with_display,
         assistant_id=assistant_id,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert
@@ -1259,6 +1318,7 @@ def test_remove_sub_agent_answer__no_op_when_assistant_not_found() -> None:
         add_block_border=False,
         answer="Present answer",
         assistant_id=assistant_id_present,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     text_with_display = f"Start\n{display}\nEnd"
@@ -1271,6 +1331,7 @@ def test_remove_sub_agent_answer__no_op_when_assistant_not_found() -> None:
         add_block_border=False,
         text=text_with_display,
         assistant_id=assistant_id_absent,
+        display_title_template="Answer from <strong>{}</strong>",
     )
 
     # Assert

unique_toolkit/agentic/tools/utils/execution/execution.py

@@ -16,7 +16,7 @@ P = ParamSpec("P")
 R = TypeVar("R")
 
 
-logger = logging.getLogger(__name__)
+_logger = logging.getLogger(__name__)
 
 
 class Result(Generic[R]):
@@ -87,7 +87,7 @@ class SafeTaskExecutor:
         self._ignored_exceptions = tuple(ignored_exceptions)
         self._log_exceptions = log_exceptions
         self._log_exc_info = log_exc_info
-        self._logger = logger
+        self._logger = logger or _logger
 
     def execute(
         self, f: Callable[P, R], *args: P.args, **kwargs: P.kwargs
@@ -98,7 +98,9 @@ class SafeTaskExecutor:
             if isinstance(e, self._ignored_exceptions):
                 raise e
             if self._log_exceptions:
-                logger.error(f"Error in {f.__name__}: {e}", exc_info=self._log_exc_info)
+                self._logger.error(
+                    f"Error in {f.__name__}: {e}", exc_info=self._log_exc_info
+                )
             return Result(False, exception=e)
 
     async def execute_async(
@@ -110,7 +112,9 @@ class SafeTaskExecutor:
             if isinstance(e, self._ignored_exceptions):
                 raise e
             if self._log_exceptions:
-                logger.error(f"Error in {f.__name__}: {e}", exc_info=self._log_exc_info)
+                self._logger.error(
+                    f"Error in {f.__name__}: {e}", exc_info=self._log_exc_info
+                )
             return Result(False, exception=e)
 
 

{unique_toolkit-1.19.3.dist-info → unique_toolkit-1.20.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.19.3
+Version: 1.20.0
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -119,6 +119,11 @@ 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.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 +139,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.dist-info → unique_toolkit-1.20.0.dist-info}/RECORD

@@ -47,7 +47,7 @@ unique_toolkit/agentic/history_manager/history_construction_with_contents.py,sha
 unique_toolkit/agentic/history_manager/history_manager.py,sha256=6V4D5-YJFCsl_WthGVGw0Eq4WobXFgkFiX2YUtdjOZw,9275
 unique_toolkit/agentic/history_manager/loop_token_reducer.py,sha256=4XUX2-yVBnaYthV8p0zj2scVBUdK_3IhxBgoNlrytyQ,18498
 unique_toolkit/agentic/history_manager/utils.py,sha256=VIn_UmcR3jHtpux0qp5lQQzczgAm8XYSeQiPo87jC3A,3143
-unique_toolkit/agentic/postprocessor/postprocessor_manager.py,sha256=Z6rMQjhD0x6uC4p1cdxbUVv3jO-31hZTyNE1SiGYIu8,5680
+unique_toolkit/agentic/postprocessor/postprocessor_manager.py,sha256=s6HFhA61TE05aAay15NFTWI1JvdSlxmGpEVfpBbGFyM,7684
 unique_toolkit/agentic/reference_manager/reference_manager.py,sha256=x51CT0D8HHu2LzgXdHGy0leOYpjnsxVbPZ2nc28G9mA,4005
 unique_toolkit/agentic/responses_api/__init__.py,sha256=9WTO-ef7fGE9Y1QtZJFm8Q_jkwK8Srtl-HWvpAD2Wxs,668
 unique_toolkit/agentic/responses_api/postprocessors/code_display.py,sha256=qbrxXL_AQ3ufBOW2TuNgml7d8u6qY_WGriS5jyZlJlE,1902
@@ -61,16 +61,16 @@ unique_toolkit/agentic/tools/a2a/config.py,sha256=6diTTSiS2prY294LfYozB-db2wmJ6j
 unique_toolkit/agentic/tools/a2a/evaluation/__init__.py,sha256=_cR8uBwLbG7lyXoRskTpItzacgs4n23e2LeqClrytuc,354
 unique_toolkit/agentic/tools/a2a/evaluation/_utils.py,sha256=GtcPAMWkwGwJ--hBxn35ow9jN0VKYx8h2qMUXR8DCho,1877
 unique_toolkit/agentic/tools/a2a/evaluation/config.py,sha256=Ra5rzJArS3r8C7RTmzKB8cLEYh4w-u3pe_XLgul3GOA,2355
-unique_toolkit/agentic/tools/a2a/evaluation/evaluator.py,sha256=-XDqkC2fLDFMkl8KZIefb1fl9dvlwug-uzcWmhil0vk,9144
+unique_toolkit/agentic/tools/a2a/evaluation/evaluator.py,sha256=CMSoJK2yLf2vvqm7r0n_fAX_Upma8xKjWEqVGcgu4qM,9166
 unique_toolkit/agentic/tools/a2a/evaluation/summarization_user_message.j2,sha256=acP1YqD_sCy6DT0V2EIfhQTmaUKeqpeWNJ7RGgceo8I,271
 unique_toolkit/agentic/tools/a2a/manager.py,sha256=FkO9jY7o8Td0t-HBkkatmxwhJGSJXmYkFYKFhPdbpMo,1674
 unique_toolkit/agentic/tools/a2a/postprocessing/__init__.py,sha256=R90CSecxJrKH7TbwiMYPyTwsXUUmouL8fbEUlL4ee9Q,362
-unique_toolkit/agentic/tools/a2a/postprocessing/_display.py,sha256=rdWk-6M6V27v7G3dqaaDj1vXcgsFvDrHswFuSKQfd2I,5186
+unique_toolkit/agentic/tools/a2a/postprocessing/_display.py,sha256=iSHgcHuz3ryBZajgSf_uqYI2DpC801fxLdU1Ym4qGCw,5452
 unique_toolkit/agentic/tools/a2a/postprocessing/_utils.py,sha256=JsWwylR2Ao_L0wk1UlhqeN2fTxPnrbhoi1klYHVBnLk,750
-unique_toolkit/agentic/tools/a2a/postprocessing/config.py,sha256=hqo3vQZX63yXoLT7wN13vf0rC7qKiozKKfdkicYCQno,1061
-unique_toolkit/agentic/tools/a2a/postprocessing/postprocessor.py,sha256=MMy3NeN4koHWmamaXvsEnrRbsh7XBaWhD0PRcr1i7C4,11931
+unique_toolkit/agentic/tools/a2a/postprocessing/config.py,sha256=XhOeGu2qi_amz4VY2qF1ZGYQ1frYeVmVtAWYaKskgfY,1582
+unique_toolkit/agentic/tools/a2a/postprocessing/postprocessor.py,sha256=VkQDKEEYhNu0ztbXHizbKLeXIADEdFQT4IzBGYY8svg,12623
 unique_toolkit/agentic/tools/a2a/postprocessing/test/test_consolidate_references.py,sha256=HvisnbhPUwuR9uELGunxLjNHvCXheoD5eW52L0tvUXk,26789
-unique_toolkit/agentic/tools/a2a/postprocessing/test/test_display.py,sha256=Hn2GOsVhpCkpFV9Asvy_IyiYvCd88rxDzi0E_zt2kWc,37553
+unique_toolkit/agentic/tools/a2a/postprocessing/test/test_display.py,sha256=ISuLZ2ZDT5IPyk0U5UBubsN6jlprNuVvBiAVm9R9wOg,40445
 unique_toolkit/agentic/tools/a2a/postprocessing/test/test_postprocessor_reference_functions.py,sha256=GxSkkY-Xgd61Bk8sIRfEtkT9hqL1VgPLWrq-6XoB0rA,11360
 unique_toolkit/agentic/tools/a2a/prompts.py,sha256=0ILHL_RAcT04gFm2d470j4Gho7PoJXdCJy-bkZgf_wk,2401
 unique_toolkit/agentic/tools/a2a/tool/__init__.py,sha256=JIJKZBTLTA39OWhxoUd6uairxmqINur1Ex6iXDk9ef8,197
@@ -99,7 +99,7 @@ unique_toolkit/agentic/tools/tool_manager.py,sha256=DtxJobe_7QKFe6CjnMhCP-mnKO6M
 unique_toolkit/agentic/tools/tool_progress_reporter.py,sha256=ixud9VoHey1vlU1t86cW0-WTvyTwMxNSWBon8I11SUk,7955
 unique_toolkit/agentic/tools/utils/__init__.py,sha256=s75sjY5nrJchjLGs3MwSIqhDW08fFXIaX7eRQjFIA4s,346
 unique_toolkit/agentic/tools/utils/execution/__init__.py,sha256=OHiKpqBnfhBiEQagKVWJsZlHv8smPp5OI4dFIexzibw,37
-unique_toolkit/agentic/tools/utils/execution/execution.py,sha256=vjG2Y6awsGNtlvyQAGCTthQ5thWHYnn-vzZXaYLb3QE,7922
+unique_toolkit/agentic/tools/utils/execution/execution.py,sha256=ocPGGfUwa851207HNTLYiBJ1pNzJp4VhMZ49OPP33gU,8022
 unique_toolkit/agentic/tools/utils/source_handling/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/agentic/tools/utils/source_handling/schema.py,sha256=iHBKuks6tUy8tvian4Pd0B6_-8__SehVVNcxIUAUjEA,882
 unique_toolkit/agentic/tools/utils/source_handling/source_formatting.py,sha256=uZ0QXqrPWgId3ZA67dvjHQ6xrW491LK1xxx_sVJmFHg,9160
@@ -166,7 +166,7 @@ unique_toolkit/short_term_memory/service.py,sha256=5PeVBu1ZCAfyDb2HLVvlmqSbyzBBu
 unique_toolkit/smart_rules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 unique_toolkit/smart_rules/compile.py,sha256=Ozhh70qCn2yOzRWr9d8WmJeTo7AQurwd3tStgBMPFLA,1246
 unique_toolkit/test_utilities/events.py,sha256=_mwV2bs5iLjxS1ynDCjaIq-gjjKhXYCK-iy3dRfvO3g,6410
-unique_toolkit-1.19.3.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_toolkit-1.19.3.dist-info/METADATA,sha256=kcF3i4FpSU1Dnp8OTmkjqWtIw6uyIwMZukexyeyvjcU,39142
-unique_toolkit-1.19.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_toolkit-1.19.3.dist-info/RECORD,,
+unique_toolkit-1.20.0.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_toolkit-1.20.0.dist-info/METADATA,sha256=clAfPiA8yIUHHPqoIS3JcmZJR_SJGj53uH0UTN_RVwQ,39492
+unique_toolkit-1.20.0.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_toolkit-1.20.0.dist-info/RECORD,,