unique-toolkit 2026.28.0.dev9__tar.gz → 2026.28.0.dev11__tar.gz

{unique_toolkit-2026.28.0.dev9 → unique_toolkit-2026.28.0.dev11}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.28.0.dev9
+Version: 2026.28.0.dev11
 Summary: 
 Author: Cedric Klinkert, Martin Fadler, Sadique Sheik, Fabian Schläpfer, Pascal Hauri, Ejiroghene Oki-Peter, Rami Azouz, Ricardo Visini, Sanzio Monti
 Author-email: Cedric Klinkert <cedric.klinkert@unique.ch>, Martin Fadler <martin.fadler@unique.ch>, Sadique Sheik <sadique@unique.ch>, Fabian Schläpfer <fabian@unique.ch>, Pascal Hauri <pascal@unique.ch>, Ejiroghene Oki-Peter <oki-peter.ejiroghene@unique.ai>, Rami Azouz <rami.ext@unique.ch>, Ricardo Visini <ricardo.visini@unique.ch>, Sanzio Monti <sanzio.monti@unique.ai>
@@ -16,7 +16,7 @@ Requires-Dist: sseclient>=0.0.27,<0.0.28
 Requires-Dist: openai>=1.105.0,<3
 Requires-Dist: platformdirs>=4.0.0,<5
 Requires-Dist: pillow>=12.1.1
-Requires-Dist: unique-sdk>=2026.28.0.dev11,<2026.28.0rc0
+Requires-Dist: unique-sdk>=2026.28.0.dev12,<2026.28.0rc0
 Requires-Dist: jambo>=0.1.2,<0.2
 Requires-Dist: jinja2>=3.1.6,<4
 Requires-Dist: docxtpl>=0.20.1,<0.21

{unique_toolkit-2026.28.0.dev9 → unique_toolkit-2026.28.0.dev11}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "2026.28.0.dev9"
+version = "2026.28.0.dev11"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12"
@@ -28,7 +28,7 @@ dependencies = [
     "openai>=1.105.0,<3",
     "platformdirs>=4.0.0,<5",
     "pillow>=12.1.1",
-    "unique-sdk>=2026.28.0.dev11,<2026.28.0rc0",
+    "unique-sdk>=2026.28.0.dev12,<2026.28.0rc0",
     "jambo>=0.1.2,<0.2",
     "jinja2>=3.1.6,<4",
     "docxtpl>=0.20.1,<0.21",

{unique_toolkit-2026.28.0.dev9 → unique_toolkit-2026.28.0.dev11}/unique_toolkit/agentic/tools/a2a/manager.py

@@ -29,6 +29,13 @@ class A2AManager:
             if not tool_config.is_sub_agent:
                 continue
 
+            if not tool_config.is_enabled:
+                self._logger.info(
+                    "Skipping disabled sub-agent '%s'",
+                    tool_config.name,
+                )
+                continue
+
             if not isinstance(tool_config.configuration, SubAgentToolConfig):
                 self._logger.error(
                     "tool_config.configuration must be of type SubAgentToolConfig"
@@ -37,16 +44,23 @@ class A2AManager:
 
             sub_agent_tool_config = tool_config.configuration
 
-            sub_agents.append(
-                SubAgentTool(
-                    configuration=sub_agent_tool_config,
-                    event=event,
-                    tool_progress_reporter=self._tool_progress_reporter,
-                    name=tool_config.name,
-                    display_name=tool_config.display_name,
-                    response_watcher=self._response_watcher,
+            try:
+                sub_agents.append(
+                    SubAgentTool(
+                        configuration=sub_agent_tool_config,
+                        event=event,
+                        tool_progress_reporter=self._tool_progress_reporter,
+                        name=tool_config.name,
+                        display_name=tool_config.display_name,
+                        response_watcher=self._response_watcher,
+                    )
+                )
+            except Exception:
+                self._logger.warning(
+                    "Skipping sub-agent '%s' due to initialization failure.",
+                    tool_config.name,
+                    exc_info=True,
                 )
-            )
 
         filtered_tool_config = [
             tool_config for tool_config in tool_configs if not tool_config.is_sub_agent

{unique_toolkit-2026.28.0.dev9 → unique_toolkit-2026.28.0.dev11}/unique_toolkit/agentic/tools/config.py

@@ -1,3 +1,4 @@
+import logging
 from enum import StrEnum
 from typing import Annotated, Any, Generic
 
@@ -14,9 +15,36 @@ from typing_extensions import TypeVar
 from unique_toolkit._common.pydantic_helpers import get_configuration_dict
 from unique_toolkit.agentic.tools.schemas import BaseToolConfig
 
+logger = logging.getLogger(__name__)
+
 T = TypeVar("T", bound=BaseToolConfig, default=BaseToolConfig)
 
 
+def _ensure_base_tool_config(value: dict[str, Any]) -> None:
+    """Make configuration parseable; disabled/demoted tools are never built at runtime."""
+    configuration = value.get("configuration")
+    if not isinstance(configuration, BaseToolConfig):
+        value["configuration"] = BaseToolConfig()
+
+
+def _non_empty_str(value: dict[str, Any], snake_key: str, camel_key: str) -> str:
+    raw = value.get(snake_key, value.get(camel_key, ""))
+    if raw is None:
+        return ""
+    return str(raw)
+
+
+def _is_mcp_tool_payload(value: dict[str, Any]) -> bool:
+    if _non_empty_str(value, "mcp_source_id", "mcpSourceId"):
+        return True
+
+    configuration = value.get("configuration", {})
+    if isinstance(configuration, dict):
+        if _non_empty_str(configuration, "mcp_source_id", "mcpSourceId"):
+            return True
+    return False
+
+
 class ToolIcon(StrEnum):
     ANALYTICS = "IconAnalytics"
     BOOK = "IconBook"
@@ -76,11 +104,26 @@ class ToolBuildConfig(BaseModel, Generic[T]):
         value: Any,
         info: ValidationInfo,
     ) -> Any:
-        """Check the given values for."""
+        """Validate and resolve tool configuration based on tool type and name.
+
+        Disabled tools skip validation entirely.
+        Enabled tools that fail validation are demoted to disabled.
+        """
         if not isinstance(value, dict):
             return value
 
-        is_mcp_tool = value.get("mcp_source_id", "") != ""
+        if "isEnabled" in value:
+            is_enabled = bool(value["isEnabled"])
+        elif "is_enabled" in value:
+            is_enabled = bool(value["is_enabled"])
+        else:
+            is_enabled = True
+
+        if not is_enabled:
+            _ensure_base_tool_config(value)
+            return value
+
+        is_mcp_tool = _is_mcp_tool_payload(value)
         mcp_configuration = value.get("configuration", {})
 
         # Import at runtime to avoid circular imports
@@ -102,26 +145,39 @@ class ToolBuildConfig(BaseModel, Generic[T]):
 
         configuration = value.get("configuration", {})
 
-        if is_sub_agent_tool:
-            from unique_toolkit.agentic.tools.a2a import ExtendedSubAgentToolConfig
-
-            config = ExtendedSubAgentToolConfig.model_validate(configuration)
-        elif isinstance(configuration, dict):
-            # Local import to avoid circular import at module import time
-            from unique_toolkit.agentic.tools.factory import ToolFactory
-
-            config = ToolFactory.build_tool_config(
-                value["name"],
-                **configuration,
+        try:
+            if is_sub_agent_tool:
+                from unique_toolkit.agentic.tools.a2a import ExtendedSubAgentToolConfig
+
+                config = ExtendedSubAgentToolConfig.model_validate(configuration)
+            elif isinstance(configuration, dict):
+                # Local import to avoid circular import at module import time
+                from unique_toolkit.agentic.tools.factory import ToolFactory
+
+                config = ToolFactory.build_tool_config(
+                    value["name"],
+                    **configuration,
+                )
+            else:
+                # Check that the type of config matches the tool name
+                from unique_toolkit.agentic.tools.factory import ToolFactory
+
+                assert isinstance(
+                    configuration,
+                    ToolFactory.tool_config_map[value["name"]],  # pyright: ignore[reportArgumentType]
+                )
+                config = configuration
+        except Exception:
+            tool_name = value.get("name", "<unknown>")
+            logger.warning(
+                "Tool '%s' has invalid configuration and will be disabled.",
+                tool_name,
+                exc_info=True,
             )
-        else:
-            # Check that the type of config matches the tool name
-            from unique_toolkit.agentic.tools.factory import ToolFactory
+            value["is_enabled"] = False
+            value["isEnabled"] = False
+            _ensure_base_tool_config(value)
+            return value
 
-            assert isinstance(
-                configuration,
-                ToolFactory.tool_config_map[value["name"]],  # pyright: ignore[reportArgumentType]
-            )
-            config = configuration
         value["configuration"] = config
         return value

{unique_toolkit-2026.28.0.dev9 → unique_toolkit-2026.28.0.dev11}/unique_toolkit/agentic/tools/tool_manager.py

@@ -147,7 +147,9 @@ class _ToolManager(Generic[_ApiMode]):
         self._tool_choices = run_context.tool_choices
         self._disabled_tools = run_context.disabled_tools
         self._exclusive_tools = [
-            tool.name for tool in self._config.tools if tool.is_exclusive
+            tool.name
+            for tool in self._config.tools
+            if tool.is_exclusive and tool.is_enabled
         ]
         # this needs to be a set of strings to avoid duplicates
         self._tool_evaluation_check_list: set[EvaluationMetricName] = set()
@@ -199,28 +201,39 @@ class _ToolManager(Generic[_ApiMode]):
 
         registered_tool_names.update(t.name for t in self._mcp_tools)
 
-        # Build internal tools from configurations
+        # Build internal tools from configurations, skipping disabled and failing tools
         self._internal_tools.clear()
+        safe_executor = SafeTaskExecutor(logger=self._logger, log_exceptions=False)
         for t in tool_configs:
             if t.name in registered_tool_names:
                 continue
             if t.name in OpenAIBuiltInToolName:
                 continue
-            if tool_init_event is not None:
-                built_tool = ToolFactory.build_tool_with_settings(
-                    t.name,
-                    t,
-                    t.configuration,
-                    tool_init_event,
-                    tool_progress_reporter=self._tool_progress_reporter,
-                )
-            else:
+            if not t.is_enabled:
+                self._logger.info("Skipping disabled tool '%s'", t.name)
+                continue
+            if tool_init_event is None:
                 self._logger.info(
                     "Skipping internal tool '%s' (requires chat event for initialization)",
                     t.name,
                 )
                 continue
-            self._internal_tools.append(built_tool)
+            result = safe_executor.execute(
+                ToolFactory.build_tool_with_settings,
+                t.name,
+                t,
+                t.configuration,
+                tool_init_event,
+                tool_progress_reporter=self._tool_progress_reporter,
+            )
+            if result.success:
+                self._internal_tools.append(result.unpack())
+            else:
+                self._logger.warning(
+                    "Skipping tool '%s' due to initialization failure.",
+                    t.name,
+                    exc_info=result.exception,
+                )
 
         # Combine all types of tools
         self.available_tools = (

{unique_toolkit-2026.28.0.dev9 → unique_toolkit-2026.28.0.dev11}/unique_toolkit/chat/functions.py

@@ -267,6 +267,9 @@ def create_message(
     references: list[ContentReference] | None = None,
     debug_info: dict[str, Any] | None = None,
     set_completed_at: bool | None = False,
+    segment_kind: str | None = None,
+    response_turn_id: str | None = None,
+    segment_index: int | None = None,
 ) -> ChatMessage:
     """Creates a message in the chat session synchronously.
 
@@ -281,6 +284,9 @@ def create_message(
         references (list[ContentReference], optional): list of ContentReference objects. Defaults to None.
         debug_info (dict[str, Any]], optional): Debug information. Defaults to None.
         set_completed_at (Optional[bool]): Whether to set the completedAt field with the current date time. Defaults to False.
+        segment_kind (str, optional): The segment kind (e.g. "PROCESS", "PREFACE", "ELICITATION", "ANSWER"). Defaults to None.
+        response_turn_id (str, optional): The response turn ID grouping segments of the same assistant turn. Defaults to None.
+        segment_index (int, optional): The segment index within the turn. Defaults to None (derived by node-chat).
 
     Returns:
         ChatMessage: The created message.
@@ -304,6 +310,9 @@ def create_message(
             references=references,
             debug_info=debug_info,
             set_completed_at=set_completed_at,
+            segment_kind=segment_kind,
+            response_turn_id=response_turn_id,
+            segment_index=segment_index,
         )
 
         message = unique_sdk.Message.create(**params)
@@ -324,8 +333,11 @@ async def create_message_async(
     references: list[ContentReference] | None = None,
     debug_info: dict[str, Any] | None = None,
     set_completed_at: bool | None = False,
+    segment_kind: str | None = None,
+    response_turn_id: str | None = None,
+    segment_index: int | None = None,
 ):
-    """Creates a message in the chat session synchronously.
+    """Creates a message in the chat session asynchronously.
 
     Args:
         user_id (str): The user ID.
@@ -338,6 +350,9 @@ async def create_message_async(
         references (list[ContentReference], optional): list of ContentReference objects. Defaults to None.
         debug_info (dict[str, Any]], optional): Debug information. Defaults to None.
         set_completed_at (Optional[bool]): Whether to set the completedAt field with the current date time. Defaults to False.
+        segment_kind (str, optional): The segment kind (e.g. "PROCESS", "PREFACE", "ELICITATION", "ANSWER"). Defaults to None.
+        response_turn_id (str, optional): The response turn ID grouping segments of the same assistant turn. Defaults to None.
+        segment_index (int, optional): The segment index within the turn. Defaults to None (derived by node-chat).
 
     Returns:
         ChatMessage: The created message.
@@ -361,6 +376,9 @@ async def create_message_async(
             references=references,
             debug_info=debug_info,
             set_completed_at=set_completed_at,
+            segment_kind=segment_kind,
+            response_turn_id=response_turn_id,
+            segment_index=segment_index,
         )
 
         message = await unique_sdk.Message.create_async(**params)
@@ -381,6 +399,9 @@ def _construct_message_create_params(
     references: list[ContentReference] | None = None,
     debug_info: dict[str, Any] | None = None,
     set_completed_at: bool | None = False,
+    segment_kind: str | None = None,
+    response_turn_id: str | None = None,
+    segment_index: int | None = None,
 ) -> dict[str, Any]:
     if original_content is None:
         original_content = content
@@ -399,6 +420,12 @@ def _construct_message_create_params(
         "debugInfo": debug_info or {},
         "completedAt": _time_utils.get_datetime_now() if set_completed_at else None,
     }
+    if segment_kind is not None:
+        params["segmentKind"] = segment_kind
+    if response_turn_id is not None:
+        params["responseTurnId"] = response_turn_id
+    if segment_index is not None:
+        params["segmentIndex"] = segment_index
     return params
 
 

{unique_toolkit-2026.28.0.dev9 → unique_toolkit-2026.28.0.dev11}/unique_toolkit/chat/schemas.py

@@ -146,6 +146,9 @@ class ChatMessage(BaseModel):
     updated_at: datetime | None = None
     references: list[ContentReference] | None = []
     assessment: list[ChatMessageAssessment] | None = None
+    segment_kind: str | None = None
+    response_turn_id: str | None = None
+    segment_index: int | None = None
 
     # TODO make sdk return role consistently in lowercase
     # Currently needed as sdk returns role in uppercase
@@ -213,8 +216,9 @@ class MessageLogUncitedReferences(BaseModel):
 
 class MessageLogEvent(BaseModel):
     model_config = model_config
-    type: Literal["WebSearch", "InternalSearch"]
+    type: Literal["WebSearch", "InternalSearch", "Elicitation", "Thinking", "ToolCall"]
     text: str
+    step_type: str | None = None
 
 
 class MessageLogDetails(BaseModel):
@@ -223,6 +227,10 @@ class MessageLogDetails(BaseModel):
     status: str | None = Field(
         default=None, description="Overarching status of the current message log"
     )
+    debug: dict[str, Any] | None = Field(
+        default=None,
+        description="Raw tool tracing: tool name, input args (stdin), truncated output (stdout) for debugging",
+    )
 
 
 class MessageLog(BaseModel):

{unique_toolkit-2026.28.0.dev9 → unique_toolkit-2026.28.0.dev11}/unique_toolkit/services/chat_service.py

@@ -471,6 +471,9 @@ class ChatService(ChatServiceDeprecated):
         references: list[ContentReference] | None = None,
         debug_info: dict[str, Any] | None = None,
         set_completed_at: bool | None = False,
+        segment_kind: str | None = None,
+        response_turn_id: str | None = None,
+        segment_index: int | None = None,
     ) -> ChatMessage:
         """Creates a message in the chat session synchronously.
 
@@ -480,6 +483,9 @@ class ChatService(ChatServiceDeprecated):
             references (list[ContentReference]): list of ContentReference objects. Defaults to None.
             debug_info (dict[str, Any]]): Debug information. Defaults to None.
             set_completed_at (Optional[bool]): Whether to set the completedAt field with the current date time. Defaults to False.
+            segment_kind (str, optional): The segment kind (e.g. "PROCESS", "PREFACE", "ELICITATION", "ANSWER"). Defaults to None.
+            response_turn_id (str, optional): The response turn ID grouping segments of the same assistant turn. Defaults to None.
+            segment_index (int, optional): The segment index within the turn. Defaults to None (derived by node-chat).
 
         Returns:
             ChatMessage: The created message.
@@ -499,6 +505,9 @@ class ChatService(ChatServiceDeprecated):
             references=references,
             debug_info=debug_info,
             set_completed_at=set_completed_at,
+            segment_kind=segment_kind,
+            response_turn_id=response_turn_id,
+            segment_index=segment_index,
         )
         # Update the assistant message id
         self._assistant_message_id = chat_message.id or "unknown"
@@ -511,6 +520,9 @@ class ChatService(ChatServiceDeprecated):
         references: list[ContentReference] | None = None,
         debug_info: dict[str, Any] | None = None,
         set_completed_at: bool | None = False,
+        segment_kind: str | None = None,
+        response_turn_id: str | None = None,
+        segment_index: int | None = None,
     ) -> ChatMessage:
         """Creates a message in the chat session asynchronously.
 
@@ -520,6 +532,9 @@ class ChatService(ChatServiceDeprecated):
             references (list[ContentReference]): list of references. Defaults to None.
             debug_info (dict[str, Any]]): Debug information. Defaults to None.
             set_completed_at (Optional[bool]): Whether to set the completedAt field with the current date time. Defaults to False.
+            segment_kind (str, optional): The segment kind (e.g. "PROCESS", "PREFACE", "ELICITATION", "ANSWER"). Defaults to None.
+            response_turn_id (str, optional): The response turn ID grouping segments of the same assistant turn. Defaults to None.
+            segment_index (int, optional): The segment index within the turn. Defaults to None (derived by node-chat).
 
         Returns:
             ChatMessage: The created message.
@@ -539,6 +554,9 @@ class ChatService(ChatServiceDeprecated):
             references=references,
             debug_info=debug_info,
             set_completed_at=set_completed_at,
+            segment_kind=segment_kind,
+            response_turn_id=response_turn_id,
+            segment_index=segment_index,
         )
         # Update the assistant message id
         self._assistant_message_id = chat_message.id or "unknown"