unique_orchestrator 2026.28.0.dev1__tar.gz → 2026.28.0.dev3__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-orchestrator
-Version: 2026.28.0.dev1
+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.dev1 → unique_orchestrator-2026.28.0.dev3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "unique_orchestrator"
-version = "2026.28.0.dev1"
+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.dev1 → 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.dev1 → 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.dev1 → unique_orchestrator-2026.28.0.dev3}/unique_orchestrator/tests/test_unique_ai_loop_debug_params.py

@@ -352,6 +352,7 @@ class TestRunLoopDebugParams:
             call.args[0] for call in ua._debug_info_manager.add.call_args_list
         ]
         assert "loop_params" in keys_written
+        assert "skills" in keys_written
 
     @pytest.mark.ai
     @pytest.mark.asyncio

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

@@ -6,6 +6,7 @@ from typing import Any, cast, overload
 
 import jinja2
 from typing_extensions import deprecated
+from unique_skill_tool.service import SkillTool
 from unique_toolkit.agentic.debug_info_manager.debug_info_manager import (
     DebugInfoManager,
 )
@@ -35,7 +36,7 @@ from unique_toolkit.agentic.tools.tool_manager import (
     SafeTaskExecutor,
     ToolManager,
 )
-from unique_toolkit.app.schemas import ChatEvent, McpServer
+from unique_toolkit.app.schemas import ChatEvent, McpServer, SkillReference
 from unique_toolkit.chat.cancellation import CancellationEvent
 from unique_toolkit.chat.service import ChatService
 from unique_toolkit.content import Content
@@ -143,7 +144,9 @@ class UniqueAI:
         self._content_service = content_service
         self._uploaded_documents = uploaded_documents or []
         self._user_memory_text = user_memory_text
-        self._skill_choices = getattr(event.payload, "skill_choices", [])
+        self._skill_choices: list[SkillReference] = getattr(
+            event.payload, "skill_choices", []
+        )
 
         self._debug_info_manager = debug_info_manager
         self._reference_manager = reference_manager
@@ -313,6 +316,10 @@ class UniqueAI:
                 },
             )
             self._debug_info_manager.add("loop_params", self._loop_debug_params)
+            self._debug_info_manager.add(
+                "skills",
+                self._get_activated_skills_debug_info(),
+            )
 
             tool_names = [
                 tool["name"] for tool in self._debug_info_manager.get()["tools"]
@@ -349,6 +356,33 @@ class UniqueAI:
             }
         )
 
+    def _get_activated_skills_debug_info(self) -> list[dict[str, str | bool]]:
+        skill_tool = self._tool_manager.get_tool_by_name(SkillTool.name)
+        if not isinstance(skill_tool, SkillTool):
+            return []
+
+        forced_content_ids = {
+            choice.content_id for choice in self._skill_choices if choice.content_id
+        }
+        forced_names = {choice.name for choice in self._skill_choices if choice.name}
+
+        skills_debug_info: dict[str, dict[str, str | bool]] = {}
+        for skill in skill_tool.activated_skills:
+            skills_debug_info.setdefault(
+                skill.name,
+                {
+                    "name": skill.name,
+                    "content_id": skill.content_id,
+                    "is_forced": (
+                        skill.content_id in forced_content_ids
+                        if skill.content_id
+                        else skill.name in forced_names
+                    ),
+                },
+            )
+
+        return list(skills_debug_info.values())
+
     # @track()
     async def _plan_or_execute(self) -> LanguageModelStreamResponse:
         self._logger.info("Planning or executing the loop.")
@@ -667,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: