unique_orchestrator 2026.24.0.dev0__tar.gz → 2026.24.0.dev2__tar.gz

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: unique-orchestrator
-Version: 2026.24.0.dev0
+Version: 2026.24.0.dev2
 Summary: 
 Author: Andreas Hauri
 Author-email: Andreas Hauri <andreas.hauri@unique.ai>
@@ -9,13 +9,13 @@ 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.24.0.dev0,<2026.24.0rc0
-Requires-Dist: unique-stock-ticker>=2026.24.0.dev0,<2026.24.0rc0
-Requires-Dist: unique-follow-up-questions>=2026.24.0.dev0,<2026.24.0rc0
-Requires-Dist: unique-internal-search>=2026.24.0.dev0,<2026.24.0rc0
+Requires-Dist: unique-toolkit>=2026.24.0.dev11,<2026.24.0rc0
+Requires-Dist: unique-stock-ticker>=2026.24.0.dev1,<2026.24.0rc0
+Requires-Dist: unique-follow-up-questions>=2026.24.0.dev2,<2026.24.0rc0
+Requires-Dist: unique-internal-search>=2026.24.0.dev1,<2026.24.0rc0
 Requires-Dist: unique-deep-research>=2026.24.0.dev0,<2026.24.0rc0
-Requires-Dist: unique-web-search>=2026.24.0.dev0,<2026.24.0rc0
-Requires-Dist: unique-swot>=2026.24.0.dev0,<2026.24.0rc0
+Requires-Dist: unique-web-search>=2026.24.0.dev3,<2026.24.0rc0
+Requires-Dist: unique-swot>=2026.24.0.dev2,<2026.24.0rc0
 Requires-Dist: unique-skill-tool>=2026.24.0.dev0,<2026.24.0rc0
 Requires-Dist: openai>=1.109.1,<3
 Requires-Python: >=3.12, <4

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

@@ -1,6 +1,6 @@
 [project]
 name = "unique_orchestrator"
-version = "2026.24.0.dev0"
+version = "2026.24.0.dev2"
 description = ""
 readme = "README.md"
 license = { text = "Proprietary" }
@@ -13,13 +13,13 @@ dependencies = [
     "pydantic-settings>=2.10.1,<3",
     "typing-extensions>=4.9.0,<5",
     "jinja2>=3.1.0,<4",
-    "unique-toolkit>=2026.24.0.dev0,<2026.24.0rc0",
-    "unique-stock-ticker>=2026.24.0.dev0,<2026.24.0rc0",
-    "unique-follow-up-questions>=2026.24.0.dev0,<2026.24.0rc0",
-    "unique-internal-search>=2026.24.0.dev0,<2026.24.0rc0",
+    "unique-toolkit>=2026.24.0.dev11,<2026.24.0rc0",
+    "unique-stock-ticker>=2026.24.0.dev1,<2026.24.0rc0",
+    "unique-follow-up-questions>=2026.24.0.dev2,<2026.24.0rc0",
+    "unique-internal-search>=2026.24.0.dev1,<2026.24.0rc0",
     "unique-deep-research>=2026.24.0.dev0,<2026.24.0rc0",
-    "unique-web-search>=2026.24.0.dev0,<2026.24.0rc0",
-    "unique-swot>=2026.24.0.dev0,<2026.24.0rc0",
+    "unique-web-search>=2026.24.0.dev3,<2026.24.0rc0",
+    "unique-swot>=2026.24.0.dev2,<2026.24.0rc0",
     "unique-skill-tool>=2026.24.0.dev0,<2026.24.0rc0",
     "openai>=1.109.1,<3",
 ]

{unique_orchestrator-2026.24.0.dev0 → unique_orchestrator-2026.24.0.dev2}/unique_orchestrator/config.py

@@ -14,6 +14,7 @@ from unique_internal_search.uploaded_search.config import (
 )
 from unique_stock_ticker.config import StockTickerConfig
 from unique_swot import SwotAnalysisTool, SwotAnalysisToolConfig
+from unique_toolkit._common.pydantic.rjsf_tags import RJSFMetaTag
 from unique_toolkit._common.validators import (
     LMI,
     ClipInt,
@@ -76,6 +77,20 @@ class SpaceType(StrEnum):
 T = TypeVar("T", bound=SpaceType)
 
 
+class SwitchableLanguageModelConfig(BaseToolConfig):
+    """A language model that chat users may select for a single message."""
+
+    display_name: str = Field(
+        description="Human-readable label shown to chat users in the model picker.",
+    )
+    language_model: LMI = Field(
+        description=(
+            "Language-model configuration accepted from the platform payload as "
+            "`languageModel`."
+        ),
+    )
+
+
 class SpaceConfigBase(BaseToolConfig, Generic[T]):
     """Base class for space configuration."""
 
@@ -88,6 +103,18 @@ class SpaceConfigBase(BaseToolConfig, Generic[T]):
 
     language_model: LMI = get_LMI_default_field(DEFAULT_GPT_4o)
 
+    allow_model_switching: bool = Field(
+        default=False,
+        description=(
+            "Whether chat users may override the language model for a single message"
+        ),
+    )
+
+    switchable_language_models: list[SwitchableLanguageModelConfig] = Field(
+        default_factory=list,
+        description=("Language models selectable by chat users for a single message."),
+    )
+
     custom_instructions: str = Field(
         default="",
         description="A custom instruction provided by the system admin.",
@@ -210,14 +237,20 @@ class EvaluationConfig(BaseToolConfig):
 
 
 class UniqueAIPromptConfig(BaseToolConfig):
-    system_prompt_template: str = Field(
+    system_prompt_template: Annotated[
+        str,
+        RJSFMetaTag.StringWidget.textarea(rows=25),
+    ] = Field(
         default_factory=lambda: (
             Path(__file__).parent / "prompts" / "system_prompt.jinja2"
         ).read_text(),
         description="The system prompt template as a Jinja2 template string.",
     )
 
-    user_message_prompt_template: str = Field(
+    user_message_prompt_template: Annotated[
+        str,
+        RJSFMetaTag.StringWidget.textarea(rows=4),
+    ] = Field(
         default_factory=lambda: (
             Path(__file__).parent / "prompts" / "user_message_prompt.jinja2"
         ).read_text(),
@@ -421,6 +454,27 @@ class UniqueAIConfig(BaseToolConfig):
             self.agent.experimental.sub_agents_config.referencing_config = None
         return self
 
+    @model_validator(mode="after")
+    def remove_tools_with_unsupported_model_capabilities(self) -> "UniqueAIConfig":
+        if ModelCapabilities.RESPONSES_API in self.space.language_model.capabilities:
+            return self
+
+        self.space.tools = [
+            tool
+            for tool in self.space.tools
+            if tool.name != OpenAIBuiltInToolName.CODE_INTERPRETER
+        ]
+        return self
+
+    @model_validator(mode="after")
+    def disable_responses_api_when_model_does_not_support_it(self) -> "UniqueAIConfig":
+        if ModelCapabilities.RESPONSES_API in self.space.language_model.capabilities:
+            return self
+
+        self.agent.experimental.responses_api_config.use_responses_api = False
+        self.agent.experimental.use_responses_api = False
+        return self
+
     @model_validator(mode="after")
     def enable_responses_api_for_code_interpreter_tool(self) -> "UniqueAIConfig":
         """Auto-enable the Responses API when Code Interpreter is added directly as a tool.

unique_orchestrator-2026.24.0.dev2/unique_orchestrator/tests/test_agent_config_rjsf_ui_schema.py

@@ -0,0 +1,53 @@
+from humps import camelize
+from unique_toolkit._common.pydantic.rjsf_tags import ui_schema_for_model
+
+from unique_orchestrator.config import UniqueAIAgentConfig
+
+
+def _nested_get(data: dict, path: str) -> object:
+    current: object = data
+    for part in path.split("."):
+        assert isinstance(current, dict)
+        current = current[part]
+    return current
+
+
+def test_unique_ai_agent_config__prompt_fields__textarea_widgets() -> None:
+    ui_schema = ui_schema_for_model(UniqueAIAgentConfig, key_transform=camelize)
+    assert _nested_get(ui_schema, "promptConfig.systemPromptTemplate") == {
+        "ui:widget": "textarea",
+        "ui:disabled": False,
+        "ui:readonly": False,
+        "ui:options": {"rows": 25},
+        "ui:emptyValue": "",
+    }
+    assert _nested_get(ui_schema, "promptConfig.userMessagePromptTemplate") == {
+        "ui:widget": "textarea",
+        "ui:disabled": False,
+        "ui:readonly": False,
+        "ui:options": {"rows": 4},
+        "ui:emptyValue": "",
+    }
+    assert _nested_get(ui_schema, "services.followUpQuestionsConfig.userPrompt") == {
+        "ui:widget": "textarea",
+        "ui:disabled": False,
+        "ui:readonly": False,
+        "ui:options": {"rows": 10},
+        "ui:emptyValue": "",
+    }
+    assert _nested_get(ui_schema, "services.followUpQuestionsConfig.systemPrompt") == {
+        "ui:widget": "textarea",
+        "ui:disabled": False,
+        "ui:readonly": False,
+        "ui:options": {"rows": 15},
+        "ui:emptyValue": "",
+    }
+    assert _nested_get(
+        ui_schema, "services.followUpQuestionsConfig.suggestionsFormat"
+    ) == {
+        "ui:widget": "textarea",
+        "ui:disabled": False,
+        "ui:readonly": False,
+        "ui:options": {"rows": 8},
+        "ui:emptyValue": "",
+    }

{unique_orchestrator-2026.24.0.dev0 → unique_orchestrator-2026.24.0.dev2}/unique_orchestrator/tests/test_config_services_validators.py

@@ -29,6 +29,45 @@ from unique_orchestrator.config import (
 )
 
 
+class TestUniqueAISpaceConfigModelSwitching:
+    def test_accepts_camel_case_model_switching_fields(self):
+        config = UniqueAISpaceConfig.model_validate(
+            {
+                "allowModelSwitching": True,
+                "switchableLanguageModels": [
+                    {
+                        "displayName": "GPT-4o",
+                        "languageModel": LanguageModelName.AZURE_GPT_4o_2024_1120,
+                    }
+                ],
+            }
+        )
+
+        assert config.allow_model_switching is True
+        assert len(config.switchable_language_models) == 1
+        assert config.switchable_language_models[0].display_name == "GPT-4o"
+        assert (
+            config.switchable_language_models[0].language_model.name
+            == LanguageModelName.AZURE_GPT_4o_2024_1120
+        )
+
+    @pytest.mark.ai
+    def test_defaults_absent_switchable_language_models_from_node_backend(self):
+        """Purpose: Verify omitted node-backend model-switching payloads parse.
+        Why this matters: Existing spaces can omit the new JSON field before it is configured.
+        Setup summary: Validate a payload without switchableLanguageModels and assert it defaults to [].
+        """
+        config = UniqueAIConfig.model_validate(
+            {
+                "space": {
+                    "allowModelSwitching": False,
+                },
+            }
+        )
+
+        assert config.space.switchable_language_models == []
+
+
 class TestUniqueAIServicesStockTickerConfigValidator:
     """Test suite for UniqueAIServices.check_if_stock_ticker_config_is_none validator"""
 

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

@@ -68,7 +68,7 @@ from unique_toolkit.app.schemas import ChatEvent, McpServer
 from unique_toolkit.chat.service import ChatService
 from unique_toolkit.content import Content
 from unique_toolkit.content.service import ContentService
-from unique_toolkit.language_model.infos import ModelCapabilities
+from unique_toolkit.language_model.infos import LanguageModelInfo, ModelCapabilities
 from unique_toolkit.protocols.support import ResponsesSupportCompleteWithReferences
 
 from unique_orchestrator._builders import (
@@ -83,7 +83,11 @@ from unique_orchestrator._builders.skill_setup import (
     configure_skill_tool,
     normalize_available_skills_for_tool,
 )
-from unique_orchestrator.config import UniqueAIConfig, UploadedSearchToolConfig
+from unique_orchestrator.config import (
+    SwitchableLanguageModelConfig,
+    UniqueAIConfig,
+    UploadedSearchToolConfig,
+)
 from unique_orchestrator.unique_ai import UniqueAI
 from unique_orchestrator.utils import filter_uploaded_documents_by_selection
 
@@ -126,6 +130,11 @@ async def build_unique_ai(
     config: UniqueAIConfig,
     debug_info_manager: DebugInfoManager,
 ) -> UniqueAI:
+    config = _apply_model_choice_override(event=event, logger=logger, config=config)
+    _record_language_model_debug_info(
+        debug_info_manager=debug_info_manager,
+        config=config,
+    )
     common_components = await _build_common(event, logger, config)
 
     if (
@@ -170,6 +179,68 @@ class _CommonComponents(NamedTuple):
     mcp_servers: list[McpServer]
 
 
+def _apply_model_choice_override(
+    *,
+    event: ChatEvent,
+    logger: Logger,
+    config: UniqueAIConfig,
+) -> UniqueAIConfig:
+    if (
+        not config.space.allow_model_switching
+        or not event.payload.has_model_choice_override
+    ):
+        return config
+
+    selected_model = event.payload.model_choice
+    if (
+        config.space.switchable_language_models
+        and not _is_switchable_language_model_choice(
+            selected_model=selected_model,
+            switchable_language_models=config.space.switchable_language_models,
+        )
+    ):
+        raise ValueError(
+            f"User model choice {selected_model.display_name!r} is not "
+            "allowed for this space."
+        )
+
+    config_data = config.model_dump()
+    config_data["space"]["language_model"] = selected_model
+    validated_config = UniqueAIConfig.model_validate(config_data)
+
+    logger.info(
+        "Using user model choice %s.",
+        selected_model.display_name,
+    )
+
+    return validated_config
+
+
+def _record_language_model_debug_info(
+    *,
+    debug_info_manager: DebugInfoManager,
+    config: UniqueAIConfig,
+) -> None:
+    debug_info_manager.add(
+        "language_model",
+        config.space.language_model.model_dump(
+            mode="json",
+            include={"name", "provider", "family"},
+        ),
+    )
+
+
+def _is_switchable_language_model_choice(
+    *,
+    selected_model: LanguageModelInfo,
+    switchable_language_models: list[SwitchableLanguageModelConfig],
+) -> bool:
+    return any(
+        selected_model == switchable_model.language_model
+        for switchable_model in switchable_language_models
+    )
+
+
 async def _build_common(
     event: ChatEvent,
     logger: Logger,