unique_toolkit 1.17.3__tar.gz → 1.18.1__tar.gz

{unique_toolkit-1.17.3 → unique_toolkit-1.18.1}/CHANGELOG.md

@@ -5,6 +5,14 @@ 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.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.
+
+## [1.18.0] - 2025-10-27
+- Temporary fix to rendering of sub agent responses.
+- Add config option `stop_condition` to `SubAgentToolConfig`
+- Add config option `tool_choices` to `SubAgentToolConfig`
+- Make sub agent evaluation use the name of the sub agent name if only one assessment is valid
 
 ## [1.17.3] - 2025-10-27
 - Update Hallucination check citation regex parsing pattern

{unique_toolkit-1.17.3 → unique_toolkit-1.18.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_toolkit
-Version: 1.17.3
+Version: 1.18.1
 Summary: 
 License: Proprietary
 Author: Cedric Klinkert
@@ -118,6 +118,14 @@ 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.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.
+
+## [1.18.0] - 2025-10-27
+- Temporary fix to rendering of sub agent responses.
+- Add config option `stop_condition` to `SubAgentToolConfig`
+- Add config option `tool_choices` to `SubAgentToolConfig`
+- Make sub agent evaluation use the name of the sub agent name if only one assessment is valid
 
 ## [1.17.3] - 2025-10-27
 - Update Hallucination check citation regex parsing pattern

{unique_toolkit-1.17.3 → unique_toolkit-1.18.1}/pyproject.toml

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

{unique_toolkit-1.17.3 → unique_toolkit-1.18.1}/unique_toolkit/agentic/tools/a2a/evaluation/evaluator.py

@@ -3,6 +3,7 @@ from typing import override
 
 import unique_sdk
 from jinja2 import Template
+from pydantic import BaseModel
 from typing_extensions import TypedDict
 
 from unique_toolkit.agentic.evaluation.evaluation_manager import Evaluation
@@ -21,6 +22,7 @@ from unique_toolkit.agentic.tools.a2a.evaluation.config import (
     SubAgentEvaluationServiceConfig,
 )
 from unique_toolkit.agentic.tools.a2a.tool import SubAgentTool
+from unique_toolkit.agentic.tools.utils import failsafe
 from unique_toolkit.chat.schemas import (
     ChatMessageAssessmentLabel,
     ChatMessageAssessmentStatus,
@@ -38,7 +40,27 @@ class _SubAgentToolInfo(TypedDict):
     display_name: str
 
 
-NO_ASSESSMENTS_FOUND = "NO_ASSESSMENTS_FOUND"
+_NO_ASSESSMENTS_FOUND = "NO_ASSESSMENTS_FOUND"
+
+
+class _SingleAssessmentData(BaseModel):
+    name: str
+    explanation: str
+
+
+def _format_single_assessment_found(name: str, explanation: str) -> str:
+    return _SingleAssessmentData(name=name, explanation=explanation).model_dump_json()
+
+
+@failsafe(failure_return_value=False)
+def _is_single_assessment_found(value: str) -> bool:
+    _ = _SingleAssessmentData.model_validate_json(value)
+    return True
+
+
+def _parse_single_assessment_found(value: str) -> tuple[str, str]:
+    data = _SingleAssessmentData.model_validate_json(value)
+    return data.name, data.explanation
 
 
 class SubAgentEvaluationService(Evaluation):
@@ -99,15 +121,36 @@ class SubAgentEvaluationService(Evaluation):
 
                 sub_agents_display_data.append(data)
 
+        # No valid assessments found
         if len(sub_agents_display_data) == 0:
             logger.warning("No valid sub agent assessments found")
 
             return EvaluationMetricResult(
                 name=self.get_name(),
-                value=NO_ASSESSMENTS_FOUND,
+                # This is a trick to be able to indicate to `evaluation_metric_to_assessment`
+                # that no valid assessments were found
+                value=_NO_ASSESSMENTS_FOUND,
                 reason="No sub agents assessments found",
             )
 
+        # Only one valid assessment found, no need to perform summarization
+        if (
+            len(sub_agents_display_data) == 1
+            and len(sub_agents_display_data[0]["assessments"]) == 1
+        ):
+            assessment = sub_agents_display_data[0]["assessments"][0]
+            explanation = assessment["explanation"] or ""
+            name = sub_agents_display_data[0]["name"]
+            label = assessment["label"]
+
+            return EvaluationMetricResult(
+                name=self.get_name(),
+                value=label,
+                # This is a trick to be able to pass the display name to the UI in `evaluation_metric_to_assessment`
+                reason=_format_single_assessment_found(name, explanation),
+                is_positive=label == ChatMessageAssessmentLabel.GREEN,
+            )
+
         reason = await self._get_reason(sub_agents_display_data)
 
         return EvaluationMetricResult(
@@ -121,7 +164,7 @@ class SubAgentEvaluationService(Evaluation):
     async def evaluation_metric_to_assessment(
         self, evaluation_result: EvaluationMetricResult
     ) -> EvaluationAssessmentMessage:
-        if evaluation_result.value == NO_ASSESSMENTS_FOUND:
+        if evaluation_result.value == _NO_ASSESSMENTS_FOUND:
             return EvaluationAssessmentMessage(
                 status=ChatMessageAssessmentStatus.DONE,
                 explanation="No valid sub agents assessments found to consolidate.",
@@ -130,6 +173,16 @@ class SubAgentEvaluationService(Evaluation):
                 type=self.get_assessment_type(),
             )
 
+        if _is_single_assessment_found(evaluation_result.reason):
+            name, reason = _parse_single_assessment_found(evaluation_result.reason)
+            return EvaluationAssessmentMessage(
+                status=ChatMessageAssessmentStatus.DONE,
+                explanation=reason,
+                title=name,
+                label=evaluation_result.value,  # type: ignore
+                type=self.get_assessment_type(),
+            )
+
         return EvaluationAssessmentMessage(
             status=ChatMessageAssessmentStatus.DONE,
             explanation=evaluation_result.reason,
@@ -182,12 +235,6 @@ class SubAgentEvaluationService(Evaluation):
         )
 
     async def _get_reason(self, sub_agents_display_data: list[dict]) -> str:
-        if (
-            len(sub_agents_display_data) == 1
-            and len(sub_agents_display_data[0]["assessments"]) == 1
-        ):
-            return sub_agents_display_data[0]["assessments"][0]["explanation"] or ""
-
         messages = (
             MessagesBuilder()
             .system_message_append(self._config.summarization_system_message)

{unique_toolkit-1.17.3 → unique_toolkit-1.18.1}/unique_toolkit/agentic/tools/a2a/postprocessing/_display.py

@@ -94,13 +94,17 @@ def _get_display_template(
         assistant_id_placeholder, "{%s}" % answer_placeholder, sep="\n\n"
     )  # Double line break is needed for markdown formatting
 
+    template = _add_line_break(template, before=True, after=False)
+
     if add_quote_border:
         template = _wrap_with_quote_border(template)
 
     match mode:
         case SubAgentResponseDisplayMode.DETAILS_OPEN:
             template = _wrap_with_details_tag(
-                template, "open", display_name_placeholder
+                template,
+                "open",
+                display_name_placeholder,
             )
         case SubAgentResponseDisplayMode.DETAILS_CLOSED:
             template = _wrap_with_details_tag(

{unique_toolkit-1.17.3 → unique_toolkit-1.18.1}/unique_toolkit/agentic/tools/a2a/postprocessing/postprocessor.py

@@ -1,9 +1,12 @@
+import asyncio
 import logging
 import re
 from typing import TypedDict, override
 
 import unique_sdk
+from pydantic import BaseModel, Field
 
+from unique_toolkit._common.pydantic_helpers import get_configuration_dict
 from unique_toolkit.agentic.postprocessor.postprocessor_manager import Postprocessor
 from unique_toolkit.agentic.tools.a2a.postprocessing._display import (
     _build_sub_agent_answer_display,
@@ -37,12 +40,21 @@ class _SubAgentToolInfo(TypedDict):
     responses: dict[int, _SubAgentMessageInfo]
 
 
+class SubAgentResponsesPostprocessorConfig(BaseModel):
+    model_config = get_configuration_dict()
+
+    sleep_time_before_update: float = Field(
+        default=0.5, description="Time to sleep before updating the main agent message."
+    )
+
+
 class SubAgentResponsesPostprocessor(Postprocessor):
     def __init__(
         self,
         user_id: str,
         company_id: str,
         main_agent_chat_id: str,
+        config: SubAgentResponsesPostprocessorConfig | None = None,
     ) -> None:
         super().__init__(name=self.__class__.__name__)
 
@@ -52,6 +64,7 @@ class SubAgentResponsesPostprocessor(Postprocessor):
 
         self._assistant_id_to_tool_info: dict[str, _SubAgentToolInfo] = {}
         self._main_agent_message: SpaceMessage | None = None
+        self._config = config or SubAgentResponsesPostprocessorConfig()
 
     @override
     async def run(self, loop_response: LanguageModelStreamResponse) -> None:
@@ -60,6 +73,9 @@ class SubAgentResponsesPostprocessor(Postprocessor):
             company_id=self._company_id,
             chat_id=self._main_agent_chat_id,
         )
+        await asyncio.sleep(
+            self._config.sleep_time_before_update
+        )  # Frontend rendering issues
 
     @override
     def apply_postprocessing_to_response(
@@ -79,17 +95,29 @@ class SubAgentResponsesPostprocessor(Postprocessor):
                 "Main agent message is not set, the `run` method must be called first"
             )
 
+        _add_sub_agent_references_in_place(
+            loop_response=loop_response,
+            sub_agent_infos=list(self._assistant_id_to_tool_info.values()),
+        )
+
         existing_refs = {
             ref.source_id: ref.sequence_number
             for ref in loop_response.message.references
         }
 
+        displayed_sub_agent_infos = dict(
+            (assistant_id, sub_agent_info)
+            for assistant_id, sub_agent_info in self._assistant_id_to_tool_info.items()
+            if sub_agent_info["display_config"].mode
+            != SubAgentResponseDisplayMode.HIDDEN
+        )
+
         _consolidate_references_in_place(
-            list(self._assistant_id_to_tool_info.values()), existing_refs, loop_response
+            list(displayed_sub_agent_infos.values()), existing_refs
         )
 
         answers = []
-        for assistant_id in self._assistant_id_to_tool_info.keys():
+        for assistant_id in displayed_sub_agent_infos.keys():
             messages = self._assistant_id_to_tool_info[assistant_id]["responses"]
 
             for sequence_number in sorted(messages):
@@ -124,9 +152,12 @@ class SubAgentResponsesPostprocessor(Postprocessor):
                     for ref in message["references"]
                 )
 
-        loop_response.message.text = (
-            "\n\n".join(answers) + "<br>\n\n" + loop_response.message.text.strip()
-        )
+        if len(answers) > 0:
+            loop_response.message.text = (
+                "<br>\n\n".join(answers)
+                + "<br>\n\n"
+                + loop_response.message.text.strip()
+            )
 
         return True
 
@@ -147,13 +178,6 @@ class SubAgentResponsesPostprocessor(Postprocessor):
     def register_sub_agent_tool(
         self, tool: SubAgentTool, display_config: SubAgentDisplayConfig
     ) -> None:
-        if display_config.mode == SubAgentResponseDisplayMode.HIDDEN:
-            logger.info(
-                "Sub agent tool %s has display mode `hidden`, responses will be ignored.",
-                tool.config.assistant_id,
-            )
-            return
-
         if tool.config.assistant_id not in self._assistant_id_to_tool_info:
             tool.subscribe(self)
             self._assistant_id_to_tool_info[tool.config.assistant_id] = (
@@ -200,10 +224,73 @@ class SubAgentResponsesPostprocessor(Postprocessor):
         }
 
 
+def _add_sub_agent_references_in_place(
+    loop_response: LanguageModelStreamResponse,
+    sub_agent_infos: list[_SubAgentToolInfo],
+) -> None:
+    message_text = loop_response.message.text
+
+    existing_refs = {
+        ref.source_id: ref.sequence_number
+        for ref in loop_response.message.references or []
+    }
+    start_index = max(existing_refs.values(), default=0) + 1
+
+    for sub_agent_info in sub_agent_infos:
+        sub_agent_name = sub_agent_info["name"]
+        for sequence_number in sorted(sub_agent_info["responses"]):
+            sub_agent_response = sub_agent_info["responses"][sequence_number]
+
+            for reference in sorted(
+                sub_agent_response["references"], key=lambda r: r["sequenceNumber"]
+            ):
+                ref_num = reference["sequenceNumber"]
+                source_id = reference["sourceId"]
+
+                reference_str = SubAgentTool.get_sub_agent_reference_format(
+                    name=sub_agent_name,
+                    sequence_number=sequence_number,
+                    reference_number=ref_num,
+                )
+
+                if reference_str not in message_text:
+                    # Reference not used
+                    continue
+
+                if source_id in existing_refs:
+                    new_ref_num = existing_refs[source_id]
+
+                else:
+                    new_ref_num = start_index
+                    existing_refs[source_id] = new_ref_num
+                    start_index += 1
+
+                    loop_response.message.references.append(
+                        ContentReference(
+                            name=reference["name"],
+                            url=reference["url"] or "",
+                            sequence_number=new_ref_num,
+                            source_id=source_id,
+                            source=reference["source"],
+                        )
+                    )
+
+                message_text = re.sub(
+                    rf"\s*{re.escape(reference_str)}",
+                    f" <sup>{new_ref_num}</sup>",
+                    message_text,
+                )
+
+    loop_response.message.text = message_text
+    # Remove spaces between consecutive references
+    loop_response.message.text = re.sub(
+        r"</sup>\s*<sup>", "</sup><sup>", loop_response.message.text
+    )
+
+
 def _consolidate_references_in_place(
     messages: list[_SubAgentToolInfo],
     existing_refs: dict[str, int],
-    loop_response: LanguageModelStreamResponse,
 ) -> None:
     start_index = max(existing_refs.values(), default=0) + 1
 
@@ -237,29 +324,5 @@ def _consolidate_references_in_place(
                 ref_map[reference["sequenceNumber"]] = reference_num
                 reference["sequenceNumber"] = reference_num
 
-            loop_response.message.text = (
-                _replace_sub_agent_references_in_main_agent_message(
-                    loop_response.message.text,
-                    assistant_tool_info["name"],
-                    sequence_number,
-                    ref_map,
-                )
-            )
             message["text"] = _replace_references_in_text(message["text"], ref_map)
             message["references"] = message_new_refs
-
-
-def _replace_sub_agent_references_in_main_agent_message(
-    message: str, sub_agent_name: str, sequence_number: int, ref_map: dict[int, int]
-) -> str:
-    for old_seq_num, new_seq_num in ref_map.items():
-        reference = SubAgentTool.get_sub_agent_reference_format(
-            name=sub_agent_name,
-            sequence_number=sequence_number,
-            reference_number=old_seq_num,
-        )
-        message = re.sub(rf"\s*{reference}", f" <sup>{new_seq_num}</sup>", message)
-
-    # Remove spaces between consecutive references
-    message = re.sub(r"</sup>\s*<sup>", "</sup><sup>", message)
-    return message

{unique_toolkit-1.17.3 → unique_toolkit-1.18.1}/unique_toolkit/agentic/tools/a2a/postprocessing/test/test_consolidate_references.py

@@ -22,9 +22,8 @@ class TestConsolidateReferencesInPlace:
         """Test with empty messages list."""
         messages = []
         existing_refs = {}
-        loop_response = self._create_mock_loop_response()
 
-        _consolidate_references_in_place(messages, existing_refs, loop_response)
+        _consolidate_references_in_place(messages, existing_refs)
 
         assert existing_refs == {}
 
@@ -53,9 +52,8 @@ class TestConsolidateReferencesInPlace:
             }
         ]
         existing_refs = {}
-        loop_response = self._create_mock_loop_response()
 
-        _consolidate_references_in_place(messages, existing_refs, loop_response)  # type: ignore
+        _consolidate_references_in_place(messages, existing_refs)  # type: ignore
 
         # Should start from index 1 since existing_refs is empty
         assert existing_refs == {"source1": 1}
@@ -90,9 +88,8 @@ class TestConsolidateReferencesInPlace:
             }
         ]
         existing_refs = {"existing_source": 5}
-        loop_response = self._create_mock_loop_response()
 
-        _consolidate_references_in_place(messages, existing_refs, loop_response)  # type: ignore
+        _consolidate_references_in_place(messages, existing_refs)  # type: ignore
 
         # Should start from 6 (max existing + 1)
         assert existing_refs == {"existing_source": 5, "new_source": 6}
@@ -141,8 +138,7 @@ class TestConsolidateReferencesInPlace:
         ]
         existing_refs = {}
 
-        loop_response = self._create_mock_loop_response()
-        _consolidate_references_in_place(messages, existing_refs, loop_response)  # type: ignore
+        _consolidate_references_in_place(messages, existing_refs)  # type: ignore
 
         # Both should map to the same consolidated reference number
         assert existing_refs == {"shared_source": 1}
@@ -187,8 +183,7 @@ class TestConsolidateReferencesInPlace:
         ]
         existing_refs = {}
 
-        loop_response = self._create_mock_loop_response()
-        _consolidate_references_in_place(messages, existing_refs, loop_response)  # type: ignore
+        _consolidate_references_in_place(messages, existing_refs)  # type: ignore
 
         assert existing_refs == {"source1": 1, "source2": 2}
         # Both references should be in the message since they're unique
@@ -240,8 +235,7 @@ class TestConsolidateReferencesInPlace:
         ]
         existing_refs = {}
 
-        loop_response = self._create_mock_loop_response()
-        _consolidate_references_in_place(messages, existing_refs, loop_response)  # type: ignore
+        _consolidate_references_in_place(messages, existing_refs)  # type: ignore
 
         # Should be processed in order 1, 2, 3 and assigned consecutive numbers
         assert existing_refs == {"source1": 1, "source2": 2, "source3": 3}
@@ -262,8 +256,7 @@ class TestConsolidateReferencesInPlace:
         ]
         existing_refs = {}
 
-        loop_response = self._create_mock_loop_response()
-        _consolidate_references_in_place(messages, existing_refs, loop_response)  # type: ignore
+        _consolidate_references_in_place(messages, existing_refs)  # type: ignore
 
         assert existing_refs == {}
         assert messages[0]["responses"][1]["text"] == "Text with no references"
@@ -314,8 +307,7 @@ class TestConsolidateReferencesInPlace:
         ]
         existing_refs = {}
 
-        loop_response = self._create_mock_loop_response()
-        _consolidate_references_in_place(messages, existing_refs, loop_response)  # type: ignore
+        _consolidate_references_in_place(messages, existing_refs)  # type: ignore
 
         assert existing_refs == {"source1": 1, "source2": 2}
         assert messages[0]["responses"][1]["text"] == "Assistant 1 text <sup>1</sup>"
@@ -368,8 +360,7 @@ class TestConsolidateReferencesInPlace:
         ]
         existing_refs = {"existing": 10}  # Start from 11
 
-        loop_response = self._create_mock_loop_response()
-        _consolidate_references_in_place(messages, existing_refs, loop_response)  # type: ignore
+        _consolidate_references_in_place(messages, existing_refs)  # type: ignore
 
         # sourceB gets 11 (first in sorted order), sourceA gets 12
         assert existing_refs == {"existing": 10, "sourceB": 11, "sourceA": 12}
@@ -416,8 +407,7 @@ class TestConsolidateReferencesInPlace:
         ]
         existing_refs = {}
 
-        loop_response = self._create_mock_loop_response()
-        _consolidate_references_in_place(messages, existing_refs, loop_response)  # type: ignore
+        _consolidate_references_in_place(messages, existing_refs)  # type: ignore
 
         # Should call _replace_references_in_text with the original text and ref mapping
         mock_replace.assert_called_once_with("Original text <sup>1</sup>", {1: 1})
@@ -446,8 +436,7 @@ class TestConsolidateReferencesInPlace:
         ]
         existing_refs = {}
 
-        loop_response = self._create_mock_loop_response()
-        _consolidate_references_in_place(messages, existing_refs, loop_response)  # type: ignore
+        _consolidate_references_in_place(messages, existing_refs)  # type: ignore
 
         # The original reference object should be modified in place
         assert original_ref["sequenceNumber"] == 1
@@ -488,8 +477,7 @@ class TestConsolidateReferencesInPlace:
         ]
 
         existing_refs_copy = existing_refs.copy()
-        loop_response = self._create_mock_loop_response()
-        _consolidate_references_in_place(messages, existing_refs_copy, loop_response)  # type: ignore
+        _consolidate_references_in_place(messages, existing_refs_copy)  # type: ignore
 
         assert existing_refs_copy["new_source"] == expected_start
 
@@ -505,8 +493,7 @@ class TestConsolidateReferencesInPlace:
         ]
         existing_refs = {}
 
-        loop_response = self._create_mock_loop_response()
-        _consolidate_references_in_place(messages, existing_refs, loop_response)  # type: ignore
+        _consolidate_references_in_place(messages, existing_refs)  # type: ignore
 
         assert existing_refs == {}
 
@@ -550,8 +537,7 @@ class TestConsolidateReferencesInPlace:
         ]
         existing_refs = {}
 
-        loop_response = self._create_mock_loop_response()
-        _consolidate_references_in_place(messages, existing_refs, loop_response)  # type: ignore
+        _consolidate_references_in_place(messages, existing_refs)  # type: ignore
 
         # Only valid messages should be processed
         assert existing_refs == {"source1": 1, "source3": 2}
@@ -611,8 +597,7 @@ class TestConsolidateReferencesInPlace:
         ]
         existing_refs = {}
 
-        loop_response = self._create_mock_loop_response()
-        _consolidate_references_in_place(messages, existing_refs, loop_response)  # type: ignore
+        _consolidate_references_in_place(messages, existing_refs)  # type: ignore
 
         # Should be processed in order 1, 2, 3 based on response sequence numbers
         assert existing_refs == {"source1": 1, "source2": 2, "source3": 3}
@@ -646,8 +631,7 @@ class TestConsolidateReferencesInPlace:
         ]
         existing_refs = {"existing_source": 99}
 
-        loop_response = self._create_mock_loop_response()
-        _consolidate_references_in_place(messages, existing_refs, loop_response)  # type: ignore
+        _consolidate_references_in_place(messages, existing_refs)  # type: ignore
 
         # Should use existing reference number and not add to new refs
         assert existing_refs == {"existing_source": 99}
@@ -685,8 +669,7 @@ class TestConsolidateReferencesInPlace:
         ]
         existing_refs = {"zero": 0, "negative": -5, "positive": 3}
 
-        loop_response = self._create_mock_loop_response()
-        _consolidate_references_in_place(messages, existing_refs, loop_response)  # type: ignore
+        _consolidate_references_in_place(messages, existing_refs)  # type: ignore
 
         # Should start from max(0, -5, 3) + 1 = 4
         assert existing_refs["new_source"] == 4

{unique_toolkit-1.17.3 → unique_toolkit-1.18.1}/unique_toolkit/agentic/tools/a2a/tool/config.py

@@ -1,3 +1,5 @@
+from typing import Literal
+
 from pydantic import Field
 
 from unique_toolkit._common.pydantic_helpers import get_configuration_dict
@@ -27,6 +29,10 @@ class SubAgentToolConfig(BaseToolConfig):
         default=True,
         description="Whether this sub agent's references should be used in the main agent's response.",
     )
+    forced_tools: list[str] | None = Field(
+        default=None,
+        description="The list of tool names that will be forced to be called for this sub-agent.",
+    )
 
     tool_description_for_system_prompt: str = Field(
         default="",
@@ -61,3 +67,7 @@ class SubAgentToolConfig(BaseToolConfig):
         default=120.0,
         description="Maximum time in seconds to wait for the sub-agent response before timing out.",
     )
+    stop_condition: Literal["stoppedStreamingAt", "completedAt"] = Field(
+        default="completedAt",
+        description="The condition that will be used to stop the polling for the sub-agent response.",
+    )

{unique_toolkit-1.17.3 → unique_toolkit-1.18.1}/unique_toolkit/agentic/tools/a2a/tool/service.py

@@ -282,8 +282,9 @@ class SubAgentTool(Tool[SubAgentToolConfig]):
                 text=tool_user_message,
                 chat_id=chat_id,
                 poll_interval=self.config.poll_interval,
+                tool_choices=self.config.forced_tools,
                 max_wait=self.config.max_wait,
-                stop_condition="completedAt",
+                stop_condition=self.config.stop_condition,
             )
         except TimeoutError as e:
             await self._notify_progress(

unique_toolkit-1.18.1/unique_toolkit/agentic/tools/utils/__init__.py

@@ -0,0 +1,19 @@
+"""Utilities for tools."""
+
+from unique_toolkit.agentic.tools.utils.execution.execution import (
+    Result,
+    SafeTaskExecutor,
+    failsafe,
+    failsafe_async,
+    safe_execute,
+    safe_execute_async,
+)
+
+__all__ = [
+    "failsafe",
+    "failsafe_async",
+    "safe_execute",
+    "safe_execute_async",
+    "SafeTaskExecutor",
+    "Result",
+]

unique_toolkit-1.17.3/unique_toolkit/agentic/tools/utils/__init__.py

@@ -1 +0,0 @@
-"""Utilities for tools."""