unique_toolkit 1.73.0__tar.gz → 1.74.1__tar.gz

{unique_toolkit-1.73.0 → unique_toolkit-1.74.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-toolkit
-Version: 1.73.0
+Version: 1.74.1
 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>

{unique_toolkit-1.73.0 → unique_toolkit-1.74.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_toolkit"
-version = "1.73.0"
+version = "1.74.1"
 description = ""
 readme = "README.md"
 requires-python = ">=3.12"

{unique_toolkit-1.73.0 → unique_toolkit-1.74.1}/unique_toolkit/agentic/loop_runner/middleware/planning/planning.py

@@ -141,6 +141,7 @@ class ResponsesPlanningMiddleware(ResponsesLoopIterationRunner):
                             "type": "json_schema",
                             "name": planning_schema.get("title", "Plan"),
                             "schema": planning_schema,
+                            "strict": False,
                         }
                     )
                 }

{unique_toolkit-1.73.0 → unique_toolkit-1.74.1}/unique_toolkit/agentic/tools/openai_builtin/code_interpreter/postprocessors/code_display.py

@@ -31,6 +31,10 @@ logger = logging.getLogger(__name__)
 
 class ShowExecutedCodePostprocessorConfig(BaseModel):
     model_config = get_configuration_dict()
+    enable: bool = Field(
+        default=True,
+        description="Enable display of executed code before the assistant message",
+    )
     remove_from_history: SkipJsonSchema[bool] = (
         Field(  # At the moment, it's not possible to keep executed code in the history
             default=True,
@@ -51,23 +55,23 @@ class ShowExecutedCodePostprocessor(ResponsesApiPostprocessor):
     ):
         super().__init__(self.__class__.__name__)
         self._config = config
-        self._company_id = company_id
+        self._is_enabled = (
+            self._config.enable
+            and not feature_flags.enable_code_execution_fence_un_17972.is_enabled(
+                company_id
+            )
+        )
 
     @override
     async def run(self, loop_response: ResponsesLanguageModelStreamResponse) -> None:
-        if feature_flags.enable_code_execution_fence_un_17972.is_enabled(
-            self._company_id
-        ):
-            return
-        await asyncio.sleep(self._config.sleep_time_before_display)
+        if self._is_enabled:
+            await asyncio.sleep(self._config.sleep_time_before_display)
 
     @override
     def apply_postprocessing_to_response(
         self, loop_response: ResponsesLanguageModelStreamResponse
     ) -> bool:
-        if feature_flags.enable_code_execution_fence_un_17972.is_enabled(
-            self._company_id
-        ):
+        if not self._is_enabled:
             return False
 
         prepended_text = ""