unique-orchestrator 2026.40.0.dev0__tar.gz → 2026.40.0.dev2__tar.gz

{unique_orchestrator-2026.40.0.dev0 → unique_orchestrator-2026.40.0.dev2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-orchestrator
-Version: 2026.40.0.dev0
+Version: 2026.40.0.dev2
 Summary: 
 Author: Andreas Hauri
 Author-email: Andreas Hauri <andreas.hauri@unique.ai>
@@ -9,7 +9,7 @@ Requires-Dist: pydantic>=2.8.2,<3
 Requires-Dist: pydantic-settings>=2.10.1,<3
 Requires-Dist: typing-extensions>=4.9.0,<5
 Requires-Dist: jinja2>=3.1.0,<4
-Requires-Dist: unique-toolkit>=2026.40.0.dev0,<2026.40.0rc0
+Requires-Dist: unique-toolkit>=2026.40.0.dev5,<2026.40.0rc0
 Requires-Dist: unique-stock-ticker>=2026.40.0.dev0,<2026.40.0rc0
 Requires-Dist: unique-follow-up-questions>=2026.40.0.dev0,<2026.40.0rc0
 Requires-Dist: unique-internal-search>=2026.40.0.dev0,<2026.40.0rc0

{unique_orchestrator-2026.40.0.dev0 → unique_orchestrator-2026.40.0.dev2}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_orchestrator"
-version = "2026.40.0.dev0"
+version = "2026.40.0.dev2"
 description = ""
 readme = "README.md"
 license = { text = "Proprietary" }
@@ -13,7 +13,7 @@ dependencies = [
     "pydantic-settings>=2.10.1,<3",
     "typing-extensions>=4.9.0,<5",
     "jinja2>=3.1.0,<4",
-    "unique-toolkit>=2026.40.0.dev0,<2026.40.0rc0",
+    "unique-toolkit>=2026.40.0.dev5,<2026.40.0rc0",
     "unique-stock-ticker>=2026.40.0.dev0,<2026.40.0rc0",
     "unique-follow-up-questions>=2026.40.0.dev0,<2026.40.0rc0",
     "unique-internal-search>=2026.40.0.dev0,<2026.40.0rc0",

{unique_orchestrator-2026.40.0.dev0 → unique_orchestrator-2026.40.0.dev2}/unique_orchestrator/tests/test_register_code_interpreter_postprocessors.py

@@ -218,25 +218,78 @@ class TestConfigPassthrough:
         assert postprocessor._config is custom_code_display_config
 
     @pytest.mark.ai
-    def test_show_executed_code_postprocessor_receives_company_id(self):
-        """ShowExecutedCodePostprocessor must receive company_id for fence FF checks (UN-17972)."""
+    def test_show_executed_code_postprocessor_receives_fence_config(self):
+        """The enable_code_execution_fence config value must be threaded through.
+
+        The field lives on the code-display config. When fences are disabled
+        there, the legacy executed-code display must become active (given the
+        enable=True default), and the files postprocessor must keep fences off.
+        """
+        from unique_toolkit.agentic.tools.openai_builtin.code_interpreter.postprocessors.code_display import (
+            ShowExecutedCodePostprocessorConfig,
+        )
+
+        ci_config = CodeInterpreterExtendedConfig(
+            executed_code_display_config=ShowExecutedCodePostprocessorConfig(
+                enable_code_execution_fence=False
+            )
+        )
         mgr = _make_postprocessor_manager()
 
         _register_code_interpreter_postprocessors(
-            tools=[_make_code_interpreter_tool()],
+            tools=[_make_code_interpreter_tool(config=ci_config)],
             postprocessor_manager=mgr,
             client=MagicMock(),
             content_service=MagicMock(),
             user_id="u1",
-            company_id="company-fence",
+            company_id="c1",
             chat_id="ch1",
             chat_service=MagicMock(),
         )
 
-        postprocessor: ShowExecutedCodePostprocessor = (
+        show_code_pp: ShowExecutedCodePostprocessor = (
             mgr.add_postprocessor.call_args_list[0][0][0]
         )
-        assert postprocessor._company_id == "company-fence"
+        display_files_pp: DisplayCodeInterpreterFilesPostProcessor = (
+            mgr.add_postprocessor.call_args_list[1][0][0]
+        )
+        # Fence off → legacy code display is active; files postprocessor keeps fences off.
+        assert show_code_pp._is_enabled is True
+        assert display_files_pp._fence_enabled is False
+
+    @pytest.mark.ai
+    def test_display_files_postprocessor_receives_html_fence_config(self):
+        """The enable_html_with_fence config value must be threaded through.
+
+        The field lives on the code-display config, but only the files
+        postprocessor reads it (UN-17927).
+        """
+        from unique_toolkit.agentic.tools.openai_builtin.code_interpreter.postprocessors.code_display import (
+            ShowExecutedCodePostprocessorConfig,
+        )
+
+        ci_config = CodeInterpreterExtendedConfig(
+            executed_code_display_config=ShowExecutedCodePostprocessorConfig(
+                enable_html_with_fence=False
+            )
+        )
+        mgr = _make_postprocessor_manager()
+
+        _register_code_interpreter_postprocessors(
+            tools=[_make_code_interpreter_tool(config=ci_config)],
+            postprocessor_manager=mgr,
+            client=MagicMock(),
+            content_service=MagicMock(),
+            user_id="u1",
+            company_id="c1",
+            chat_id="ch1",
+            chat_service=MagicMock(),
+        )
+
+        display_files_pp: DisplayCodeInterpreterFilesPostProcessor = (
+            mgr.add_postprocessor.call_args_list[1][0][0]
+        )
+        assert display_files_pp._html_fence_enabled is False
 
     @pytest.mark.ai
     def test_display_files_postprocessor_receives_client_and_company_id(self):
@@ -261,9 +314,11 @@ class TestConfigPassthrough:
             mgr.add_postprocessor.call_args_list[1][0][0]
         )
         assert pp._client is client
-        assert pp._company_id == "company-99"
         # All three IDs were passed, so short-term memory manager is initialised
         assert pp._short_term_memory_manager is not None
+        # The postprocessor does not keep the company_id; it only scopes the memory.
+        memory_service = pp._short_term_memory_manager._short_term_memory_service
+        assert memory_service.company_id == "company-99"
 
 
 # ---------------------------------------------------------------------------

{unique_orchestrator-2026.40.0.dev0 → unique_orchestrator-2026.40.0.dev2}/unique_orchestrator/unique_ai_builder.py

@@ -507,7 +507,6 @@ def _register_code_interpreter_postprocessors(
     postprocessor_manager.add_postprocessor(
         ShowExecutedCodePostprocessor(
             config=code_interpreter_config.executed_code_display_config,
-            company_id=company_id,
         )
     )
     postprocessor_manager.add_postprocessor(
@@ -519,6 +518,12 @@ def _register_code_interpreter_postprocessors(
             company_id=company_id,
             chat_id=chat_id,
             chat_service=chat_service,
+            code_execution_fence_enabled=(
+                code_interpreter_config.executed_code_display_config.enable_code_execution_fence
+            ),
+            html_with_fence_enabled=(
+                code_interpreter_config.executed_code_display_config.enable_html_with_fence
+            ),
         )
     )