unique-orchestrator 2026.28.0.dev2__tar.gz → 2026.28.0.dev3__tar.gz

{unique_orchestrator-2026.28.0.dev2 → unique_orchestrator-2026.28.0.dev3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-orchestrator
-Version: 2026.28.0.dev2
+Version: 2026.28.0.dev3
 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.28.0.dev0,<2026.28.0rc0
+Requires-Dist: unique-toolkit>=2026.28.0.dev2,<2026.28.0rc0
 Requires-Dist: unique-stock-ticker>=2026.28.0.dev0,<2026.28.0rc0
 Requires-Dist: unique-follow-up-questions>=2026.28.0.dev0,<2026.28.0rc0
 Requires-Dist: unique-internal-search>=2026.28.0.dev0,<2026.28.0rc0

{unique_orchestrator-2026.28.0.dev2 → unique_orchestrator-2026.28.0.dev3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_orchestrator"
-version = "2026.28.0.dev2"
+version = "2026.28.0.dev3"
 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.28.0.dev0,<2026.28.0rc0",
+    "unique-toolkit>=2026.28.0.dev2,<2026.28.0rc0",
     "unique-stock-ticker>=2026.28.0.dev0,<2026.28.0rc0",
     "unique-follow-up-questions>=2026.28.0.dev0,<2026.28.0rc0",
     "unique-internal-search>=2026.28.0.dev0,<2026.28.0rc0",

{unique_orchestrator-2026.28.0.dev2 → unique_orchestrator-2026.28.0.dev3}/unique_orchestrator/config.py

@@ -37,6 +37,12 @@ from unique_toolkit.agentic.tools.a2a import (
     REFERENCING_INSTRUCTIONS_FOR_USER_PROMPT,
 )
 from unique_toolkit.agentic.tools.a2a.evaluation import SubAgentEvaluationServiceConfig
+from unique_toolkit.agentic.tools.experimental.ask_user_tool import (
+    AskUserTool,
+)
+from unique_toolkit.agentic.tools.experimental.ask_user_tool import (
+    AskUserToolConfig as BaseAskUserToolConfig,
+)
 from unique_toolkit.agentic.tools.experimental.open_file_tool.config import (
     OpenFileToolConfig,
 )
@@ -391,6 +397,13 @@ class UploadedSearchToolConfig(BaseToolConfig):
         return self
 
 
+class AskUserToolConfig(BaseAskUserToolConfig):
+    enabled: bool = Field(
+        default=False,
+        description="Enable the AskUser elicitation tool.",
+    )
+
+
 class ExperimentalConfig(BaseToolConfig):
     """Experimental features this part of the configuration might evolve in the future continuously"""
 
@@ -431,6 +444,12 @@ class ExperimentalConfig(BaseToolConfig):
 
     uploaded_search_tool_config: UploadedSearchToolConfig = UploadedSearchToolConfig()
 
+    ask_user_tool_config: AskUserToolConfig = Field(
+        title="Ask User Tool (Elicitation)",
+        description="Configuration for the AskUser tool",
+        default_factory=AskUserToolConfig,
+    )
+
     use_responses_api: bool = Field(
         default=False,
         description="If set, the main agent will use the Responses API from OpenAI",
@@ -593,6 +612,27 @@ class UniqueAIConfig(BaseToolConfig):
 
         return self
 
+    @model_validator(mode="after")
+    def inject_ask_user_tool(self) -> "UniqueAIConfig":
+        tool_names = [t.name for t in self.space.tools]
+        has_tool = AskUserTool.name in tool_names
+        config = self.agent.experimental.ask_user_tool_config
+
+        if config.enabled and not has_tool:
+            self.space.tools.append(
+                ToolBuildConfig(
+                    name=AskUserTool.name,
+                    display_name=AskUserTool.DISPLAY_NAME,
+                    configuration=config,
+                )
+            )
+        elif not config.enabled and has_tool:
+            self.space.tools = [
+                t for t in self.space.tools if t.name != AskUserTool.name
+            ]
+
+        return self
+
     @property
     def effective_max_loop_iterations(self) -> int:
         """Effective max loop iterations, accounting for model-specific overrides."""

{unique_orchestrator-2026.28.0.dev2 → unique_orchestrator-2026.28.0.dev3}/unique_orchestrator/tests/test_config_services_validators.py

@@ -1,6 +1,9 @@
 import pytest
 from unique_follow_up_questions.config import FollowUpQuestionsConfig
 from unique_stock_ticker.config import StockTickerConfig
+from unique_toolkit.agentic.tools.experimental.ask_user_tool import (
+    AskUserTool,
+)
 from unique_toolkit.agentic.tools.experimental.open_file_tool.config import (
     OpenFileToolConfig,
 )
@@ -23,6 +26,7 @@ from unique_toolkit.language_model.schemas import LanguageModelTokenLimits
 from unique_user_memory.config import UserMemoryConfig
 
 from unique_orchestrator.config import (
+    AskUserToolConfig,
     EvaluationConfig,
     UniqueAIConfig,
     UniqueAIServices,
@@ -496,3 +500,68 @@ class TestUniqueAIConfigUserMemory:
         config = UniqueAIConfig(space={"allowUserMemory": True})
 
         assert config.space.allow_user_memory is True
+
+
+class TestUniqueAIConfigInjectAskUserToolValidator:
+    """Tests for UniqueAIConfig.inject_ask_user_tool (AskUser)."""
+
+    def _entries(self, config: UniqueAIConfig) -> list[ToolBuildConfig]:
+        return [t for t in config.space.tools if t.name == AskUserTool.name]
+
+    def test_appends_when_enabled(self) -> None:
+        config = UniqueAIConfig(
+            space=UniqueAISpaceConfig(
+                language_model=_make_model(supports_responses_api=True),
+                tools=[],
+            ),
+            agent={
+                "experimental": {
+                    "ask_user_tool_config": AskUserToolConfig(enabled=True)
+                }
+            },
+        )
+
+        entries = self._entries(config)
+        assert len(entries) == 1
+        assert entries[0].name == AskUserTool.name
+        assert isinstance(entries[0].configuration, AskUserToolConfig)
+        assert entries[0].configuration.enabled is True
+
+    def test_removes_when_disabled(self) -> None:
+        config = UniqueAIConfig(
+            space=UniqueAISpaceConfig(
+                language_model=_make_model(supports_responses_api=True),
+                tools=[
+                    ToolBuildConfig(
+                        name=AskUserTool.name,
+                        configuration=AskUserToolConfig(enabled=False),
+                    )
+                ],
+            ),
+            agent={
+                "experimental": {
+                    "ask_user_tool_config": AskUserToolConfig(enabled=False)
+                }
+            },
+        )
+
+        assert self._entries(config) == []
+
+    def test_idempotent_when_already_present(self) -> None:
+        existing = ToolBuildConfig(
+            name=AskUserTool.name,
+            configuration=AskUserToolConfig(enabled=True),
+        )
+        config = UniqueAIConfig(
+            space=UniqueAISpaceConfig(
+                language_model=_make_model(supports_responses_api=True),
+                tools=[existing],
+            ),
+            agent={
+                "experimental": {
+                    "ask_user_tool_config": AskUserToolConfig(enabled=True)
+                }
+            },
+        )
+
+        assert len(self._entries(config)) == 1

{unique_orchestrator-2026.28.0.dev2 → unique_orchestrator-2026.28.0.dev3}/unique_orchestrator/unique_ai.py

@@ -701,7 +701,7 @@ class UniqueAI:
         # (see ``unique_skill_tool.SkillTool._log_skill_loaded``), so it is
         # redundant and noisy to also list it here.
 
-        tool_names_not_to_log: set[str] = {"DeepResearch", "Skill"}
+        tool_names_not_to_log: set[str] = {"DeepResearch", "Skill", "AskUser"}
 
         used_tools: dict[str, int] = {}
         for tool_call in tool_calls: