unique_orchestrator 1.2.4__tar.gz → 1.3.0__tar.gz

{unique_orchestrator-1.2.4 → unique_orchestrator-1.3.0}/CHANGELOG.md

@@ -5,6 +5,9 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), 
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [1.3.0] - 2025-10-14
+- Re-organize sub-agents configuration for clarity.
+
 ## [1.2.4] - 2025-10-14
 - Let control taking tool itself set the message state to completed
 

{unique_orchestrator-1.2.4 → unique_orchestrator-1.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_orchestrator
-Version: 1.2.4
+Version: 1.3.0
 Summary: 
 License: Proprietary
 Author: Andreas Hauri
@@ -33,6 +33,9 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/), 
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [1.3.0] - 2025-10-14
+- Re-organize sub-agents configuration for clarity.
+
 ## [1.2.4] - 2025-10-14
 - Let control taking tool itself set the message state to completed
 

{unique_orchestrator-1.2.4 → unique_orchestrator-1.3.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_orchestrator"
-version = "1.2.4"
+version = "1.3.0"
 description = ""
 authors = ["Andreas Hauri <andreas.hauri@unique.ai>"]
 readme = ["README.md", "CHANGELOG.md"]

{unique_orchestrator-1.2.4 → unique_orchestrator-1.3.0}/unique_orchestrator/config.py

@@ -32,6 +32,11 @@ from unique_toolkit.language_model.default_language_model import DEFAULT_GPT_4o
 from unique_web_search.config import WebSearchConfig
 from unique_web_search.service import WebSearchTool
 
+DeactivatedNone = Annotated[
+    None,
+    Field(title="Deactivated", description="None"),
+]
+
 
 class SpaceType(StrEnum):
     UNIQUE_CUSTOM = "unique_custom"
@@ -121,9 +126,6 @@ class EvaluationConfig(BaseModel):
     model_config = get_configuration_dict()
     max_review_steps: int = 3
     hallucination_config: HallucinationConfig = HallucinationConfig()
-    sub_agents_config: SubAgentEvaluationServiceConfig | None = (
-        SubAgentEvaluationServiceConfig()
-    )
 
 
 # ------------------------------------------------------------
@@ -149,12 +151,6 @@ class UniqueAIPromptConfig(BaseModel):
     )
 
 
-DeactivatedNone = Annotated[
-    None,
-    Field(title="Deactivated", description="None"),
-]
-
-
 class UniqueAIServices(BaseModel):
     """Determine the services the agent is using
 
@@ -205,12 +201,9 @@ class InputTokenDistributionConfig(BaseModel):
         return int(self.percent_for_history * max_input_token)
 
 
-class SubAgentsConfig(BaseModel):
+class SubAgentsReferencingConfig(BaseModel):
     model_config = get_configuration_dict()
-    use_sub_agent_references: bool = Field(
-        default=True,
-        description="Whether to use sub agent references in the main agent's response. Only has an effect if sub agents are used.",
-    )
+
     referencing_instructions_for_system_prompt: str = Field(
         default=REFERENCING_INSTRUCTIONS_FOR_SYSTEM_PROMPT,
         description="Referencing instructions for the main agent's system prompt.",
@@ -221,6 +214,18 @@ class SubAgentsConfig(BaseModel):
     )
 
 
+class SubAgentsConfig(BaseModel):
+    model_config = get_configuration_dict()
+
+    referencing_config: (
+        Annotated[SubAgentsReferencingConfig, Field(title="Active")] | DeactivatedNone
+    ) = SubAgentsReferencingConfig()
+    evaluation_config: (
+        Annotated[SubAgentEvaluationServiceConfig, Field(title="Active")]
+        | DeactivatedNone
+    ) = SubAgentEvaluationServiceConfig()
+
+
 class ExperimentalConfig(BaseModel):
     """Experimental features this part of the configuration might evolve in the future continuously"""
 
@@ -282,5 +287,5 @@ class UniqueAIConfig(BaseModel):
     @model_validator(mode="after")
     def disable_sub_agent_referencing_if_not_used(self) -> "UniqueAIConfig":
         if not any(tool.is_sub_agent for tool in self.space.tools):
-            self.agent.experimental.sub_agents_config.use_sub_agent_references = False
+            self.agent.experimental.sub_agents_config.referencing_config = None
         return self

{unique_orchestrator-1.2.4 → unique_orchestrator-1.3.0}/unique_orchestrator/unique_ai.py

@@ -260,10 +260,15 @@ class UniqueAI:
 
         query = self._event.payload.user_message.text
 
-        use_sub_agent_references = (
-            self._config.agent.experimental.sub_agents_config.use_sub_agent_references
-        )
-        sub_agent_referencing_instructions = self._config.agent.experimental.sub_agents_config.referencing_instructions_for_user_prompt
+        if (
+            self._config.agent.experimental.sub_agents_config.referencing_config
+            is not None
+        ):
+            use_sub_agent_references = True
+            sub_agent_referencing_instructions = self._config.agent.experimental.sub_agents_config.referencing_config.referencing_instructions_for_user_prompt
+        else:
+            use_sub_agent_references = False
+            sub_agent_referencing_instructions = None
 
         user_msg = user_message_template.render(
             query=query,
@@ -294,10 +299,15 @@ class UniqueAI:
             mcp_server.system_prompt for mcp_server in self._mcp_servers
         ]
 
-        use_sub_agent_references = (
-            self._config.agent.experimental.sub_agents_config.use_sub_agent_references
-        )
-        sub_agent_referencing_instructions = self._config.agent.experimental.sub_agents_config.referencing_instructions_for_system_prompt
+        if (
+            self._config.agent.experimental.sub_agents_config.referencing_config
+            is not None
+        ):
+            use_sub_agent_references = True
+            sub_agent_referencing_instructions = self._config.agent.experimental.sub_agents_config.referencing_config.referencing_instructions_for_system_prompt
+        else:
+            use_sub_agent_references = False
+            sub_agent_referencing_instructions = None
 
         system_message = system_prompt_template.render(
             model_info=self._config.space.language_model.model_dump(mode="json"),

{unique_orchestrator-1.2.4 → unique_orchestrator-1.3.0}/unique_orchestrator/unique_ai_builder.py

@@ -178,12 +178,9 @@ def build_unique_ai(
         postprocessor_manager.add_postprocessor(sub_agent_responses_postprocessor)
 
         sub_agent_evaluation = None
-        if (
-            config.agent.services.evaluation_config is not None
-            and config.agent.services.evaluation_config.sub_agents_config is not None
-        ):
+        if config.agent.experimental.sub_agents_config.evaluation_config is not None:
             sub_agent_evaluation = SubAgentEvaluationService(
-                config=config.agent.services.evaluation_config.sub_agents_config,
+                config=config.agent.experimental.sub_agents_config.evaluation_config,
                 language_model_service=LanguageModelService.from_event(event),
             )
             evaluation_manager.add_evaluation(sub_agent_evaluation)