unique_toolkit 2026.34.3__tar.gz → 2026.36.0.dev0__tar.gz

{unique_toolkit-2026.34.3 → unique_toolkit-2026.36.0.dev0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 2026.34.3
+Version: 2026.36.0.dev0
 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.34.3
+Requires-Dist: unique-sdk>=2026.36.0.dev0,<2026.36.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.34.3 → unique_toolkit-2026.36.0.dev0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "2026.34.3"
+version = "2026.36.0.dev0"
 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.34.3",
+    "unique-sdk>=2026.36.0.dev0,<2026.36.0rc0",
     "jambo>=0.1.2,<0.2",
     "jinja2>=3.1.6,<4",
     "docxtpl>=0.20.1,<0.21",

{unique_toolkit-2026.34.3 → unique_toolkit-2026.36.0.dev0}/unique_toolkit/agentic/feature_flags/feature_flags.py

@@ -30,6 +30,7 @@ class FeatureFlagNames(StrEnum):
     enable_web_search_argument_screening_un_18741 = (
         "FEATURE_FLAG_ENABLE_WEB_SEARCH_ARGUMENT_SCREENING_UN_18741"
     )
+    enable_new_answers_ui_un_14411 = "FEATURE_FLAG_ENABLE_NEW_ANSWERS_UI_UN_14411"
 
 
 @deprecated(
@@ -82,25 +83,20 @@ class FeatureFlags(BaseSettings):
     """Feature flags loaded from environment variables.
 
     Environment variables are automatically loaded based on field names.
-    For example, `enable_html_rendering_un_15131` will be loaded from
-    `FEATURE_FLAG_ENABLE_HTML_RENDERING_UN_15131` (with the env_prefix).
+    For example, `enable_new_answers_ui_un_14411` will be loaded from
+    `FEATURE_FLAG_ENABLE_NEW_ANSWERS_UI_UN_14411` (with the env_prefix).
 
     Usage:
         >>> flags = FeatureFlags()
         >>> # Check if feature is enabled for a company
-        >>> flags.enable_html_rendering_un_15131.is_enabled("company123")
+        >>> flags.enable_new_answers_ui_un_14411.is_enabled("company123")
         >>> # Check global enablement
         >>> bool(flags.enable_elicitation_un_15809)
     """
 
     enable_new_answers_ui_un_14411: FeatureFlag = Field(
         default=FeatureFlag([]),
-        description=(
-            "Retired (UN-24507): the new answers UI is always on now, no code "
-            "reads this field. Kept only so removing it doesn't trip the "
-            "config-checker's breaking-change gate on this settings model; "
-            "delete once that checker allows field removal."
-        ),
+        description="Enable new answers UI (UN-14411)",
     )
 
     enable_html_rendering_un_15131: FeatureFlag = Field(

{unique_toolkit-2026.34.3 → unique_toolkit-2026.36.0.dev0}/unique_toolkit/agentic/tools/a2a/tool/service.py

@@ -22,6 +22,7 @@ from unique_toolkit._common.referencing import (
 )
 from unique_toolkit._common.utils.jinja.render import render_template
 from unique_toolkit.agentic.evaluation.schemas import EvaluationMetricName
+from unique_toolkit.agentic.feature_flags import FeatureFlagNames
 from unique_toolkit.agentic.tools.a2a.response_watcher import SubAgentResponseWatcher
 from unique_toolkit.agentic.tools.a2a.tool._memory import (
     get_sub_agent_short_term_memory_manager,
@@ -38,7 +39,10 @@ from unique_toolkit.agentic.tools.a2a.tool.config import (
 from unique_toolkit.agentic.tools.factory import ToolFactory
 from unique_toolkit.agentic.tools.schemas import ToolCallResponse
 from unique_toolkit.agentic.tools.tool import Tool
-from unique_toolkit.agentic.tools.tool_progress_reporter import ToolProgressReporter
+from unique_toolkit.agentic.tools.tool_progress_reporter import (
+    ProgressState,
+    ToolProgressReporter,
+)
 from unique_toolkit.app import ChatEvent
 from unique_toolkit.chat.schemas import (
     ChatMessageAssessmentLabel,
@@ -49,6 +53,7 @@ from unique_toolkit.chat.schemas import (
 )
 from unique_toolkit.chat.service import ChatService
 from unique_toolkit.content import ContentChunk, ContentReference
+from unique_toolkit.experimental.resources.feature_flags import is_flag_enabled
 from unique_toolkit.language_model import (
     LanguageModelFunction,
     LanguageModelService,
@@ -191,6 +196,12 @@ class SubAgentTool(Tool[SubAgentToolConfig]):
             timestamp = datetime.now()
 
             if self._lock.locked():
+                await self._notify_progress(
+                    tool_call=tool_call,
+                    message=f"Waiting for another run of `{self.display_name()}` to finish",
+                    state=ProgressState.STARTED,
+                )
+
                 active_message_log = self._create_or_update_message_log(
                     progress_message="_Waiting for another run of this sub agent to finish_",
                     active_message_log=active_message_log,
@@ -204,6 +215,12 @@ class SubAgentTool(Tool[SubAgentToolConfig]):
                 sequence_number = self._sequence_number
                 self._sequence_number += 1
 
+                await self._notify_progress(
+                    tool_call=tool_call,
+                    message=tool_input,
+                    state=ProgressState.RUNNING,
+                )
+
                 active_message_log = self._create_or_update_message_log(
                     progress_message=f"<em>Executing sub agent with input: {tool_input}</em>",
                     active_message_log=active_message_log,
@@ -268,6 +285,12 @@ class SubAgentTool(Tool[SubAgentToolConfig]):
                         has_refs=has_refs,
                     )
 
+                await self._notify_progress(
+                    tool_call=tool_call,
+                    message=tool_input,
+                    state=ProgressState.FINISHED,
+                )
+
                 # Update message log entry to completed
                 active_message_log = self._create_or_update_message_log(
                     progress_message=f"<em>Completed sub agent with input: {tool_input}</em>",
@@ -306,6 +329,11 @@ class SubAgentTool(Tool[SubAgentToolConfig]):
         except TimeoutError as e:
             raise e
         except Exception as e:
+            await self._notify_progress(
+                tool_call=tool_call,
+                message="Error while running sub agent",
+                state=ProgressState.FAILED,
+            )
             active_message_log = self._create_or_update_message_log(
                 progress_message="_Error while running sub agent_",
                 status=MessageLogStatus.FAILED,
@@ -398,6 +426,23 @@ class SubAgentTool(Tool[SubAgentToolConfig]):
             SubAgentShortTermMemorySchema(chat_id=chat_id)
         )
 
+    async def _notify_progress(
+        self,
+        tool_call: LanguageModelFunction,
+        message: str,
+        state: ProgressState,
+    ) -> None:
+        if self._tool_progress_reporter is not None and not await is_flag_enabled(
+            FeatureFlagNames.enable_new_answers_ui_un_14411,
+            company_id=self._company_id or "",
+        ):
+            await self._tool_progress_reporter.notify_from_tool_call(
+                tool_call=tool_call,
+                name=self._display_name,
+                message=message,
+                state=state,
+            )
+
     def _create_or_update_message_log(
         self,
         *,
@@ -466,6 +511,11 @@ class SubAgentTool(Tool[SubAgentToolConfig]):
                 wait_for_invocations=True,
             )
         except TimeoutError as e:
+            await self._notify_progress(
+                tool_call=tool_call,
+                message="Timeout while waiting for response from sub agent.",
+                state=ProgressState.FAILED,
+            )
             active_message_log = self._create_or_update_message_log(
                 progress_message="_Timeout while waiting for response from sub agent_",
                 status=MessageLogStatus.FAILED,

{unique_toolkit-2026.34.3 → unique_toolkit-2026.36.0.dev0}/unique_toolkit/agentic/tools/mcp/tool_wrapper.py

@@ -9,14 +9,19 @@ from typing import Any, Dict
 import unique_sdk
 
 from unique_toolkit.agentic.evaluation.schemas import EvaluationMetricName
+from unique_toolkit.agentic.feature_flags import FeatureFlagNames
 from unique_toolkit.agentic.tools.mcp.models import MCPToolConfig
 from unique_toolkit.agentic.tools.schemas import ToolCallResponse
 from unique_toolkit.agentic.tools.tool import Tool
-from unique_toolkit.agentic.tools.tool_progress_reporter import ToolProgressReporter
+from unique_toolkit.agentic.tools.tool_progress_reporter import (
+    ProgressState,
+    ToolProgressReporter,
+)
 from unique_toolkit.app.schemas import ChatEvent, McpServer, McpTool
 from unique_toolkit.chat.schemas import MessageLog, MessageLogStatus
 from unique_toolkit.chat.service import ChatService
 from unique_toolkit.content.functions import upload_content_from_bytes
+from unique_toolkit.experimental.resources.feature_flags import is_flag_enabled
 from unique_toolkit.language_model import LanguageModelService
 from unique_toolkit.language_model.schemas import (
     LanguageModelFunction,
@@ -128,6 +133,18 @@ class MCPToolWrapper(Tool[MCPToolConfig]):
             active_message_log=active_message_log,
         )
 
+        # Notify progress if reporter is available
+        if self._tool_progress_reporter and not await is_flag_enabled(
+            FeatureFlagNames.enable_new_answers_ui_un_14411,
+            company_id=self._event.company_id,
+        ):
+            await self._tool_progress_reporter.notify_from_tool_call(
+                tool_call=tool_call,
+                name=f"**{self.display_name()}**",
+                message=f"Executing MCP tool: {self.display_name()}",
+                state=ProgressState.RUNNING,
+            )
+
         try:
             # Robust argument extraction and validation
             arguments = self._extract_and_validate_arguments(tool_call)
@@ -153,6 +170,18 @@ class MCPToolWrapper(Tool[MCPToolConfig]):
                 image_data_urls=image_data_urls,
             )
 
+            # Notify completion
+            if self._tool_progress_reporter and not await is_flag_enabled(
+                FeatureFlagNames.enable_new_answers_ui_un_14411,
+                company_id=self._event.company_id,
+            ):
+                await self._tool_progress_reporter.notify_from_tool_call(
+                    tool_call=tool_call,
+                    name=f"**{self.display_name()}**",
+                    message=f"MCP tool completed: {self.display_name()}",
+                    state=ProgressState.FINISHED,
+                )
+
             # Update message log entry to completed
             active_message_log = self._create_or_update_message_log(
                 progress_message="_Completed MCP tool_",
@@ -165,6 +194,18 @@ class MCPToolWrapper(Tool[MCPToolConfig]):
         except Exception as e:
             self.logger.error(f"Error executing MCP tool {self.name}: {e}")
 
+            # Notify failure
+            if self._tool_progress_reporter and not await is_flag_enabled(
+                FeatureFlagNames.enable_new_answers_ui_un_14411,
+                company_id=self._event.company_id,
+            ):
+                await self._tool_progress_reporter.notify_from_tool_call(
+                    tool_call=tool_call,
+                    name=f"**{self.display_name()}**",
+                    message=f"MCP tool failed: {str(e)}",
+                    state=ProgressState.FAILED,
+                )
+
             # Update message log entry to failed
             active_message_log = self._create_or_update_message_log(
                 progress_message="_Failed executing MCP tool_",

{unique_toolkit-2026.34.3 → unique_toolkit-2026.36.0.dev0}/unique_toolkit/chat/responses_api.py

@@ -153,9 +153,7 @@ def _prepare_responses_params_util(
         _to_search_context(content_chunks) if content_chunks is not None else None
     )
 
-    model = (
-        model_name.value if isinstance(model_name, LanguageModelName) else model_name
-    )
+    model = model_name.name if isinstance(model_name, LanguageModelName) else model_name
 
     tools_res = _convert_tools_to_openai(tools) if tools is not None else None
 

{unique_toolkit-2026.34.3 → unique_toolkit-2026.36.0.dev0}/unique_toolkit/services/chat_service.py

@@ -22,6 +22,7 @@ from unique_toolkit._common.metadata_filter_scope import (
     merge_scope_clause_into_metadata_filter,
 )
 from unique_toolkit._common.utils.files import is_file_content, is_image_content
+from unique_toolkit.agentic.feature_flags import FeatureFlagNames
 from unique_toolkit.agentic.message_log_order import next_message_log_order
 from unique_toolkit.app.unique_settings import UniqueContext, UniqueSettings
 from unique_toolkit.chat.cancellation import CancellationWatcher
@@ -99,6 +100,7 @@ from unique_toolkit.content.schemas import (
     ContentSearchType,
 )
 from unique_toolkit.elicitation.service import ElicitationService
+from unique_toolkit.experimental.resources.feature_flags import is_flag_enabled
 from unique_toolkit.language_model.constants import (
     DEFAULT_COMPLETE_TEMPERATURE,
     DEFAULT_COMPLETE_TIMEOUT,
@@ -1804,6 +1806,11 @@ class ChatService(ChatServiceDeprecated):
 
             if not rate_limit_retry_config.log_message_on_retry:
                 return
+            if not await is_flag_enabled(
+                FeatureFlagNames.enable_new_answers_ui_un_14411,
+                company_id=self._company_id,
+            ):
+                return
 
             await _cancel_ticker()