unique_orchestrator 1.1.0__tar.gz → 1.1.1__tar.gz

{unique_orchestrator-1.1.0 → unique_orchestrator-1.1.1}/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.1.1] - 2025-10-03
+- Adapt orchestrator to toolkit 1.8.0.
+
 ## [1.1.0] - 2025-09-29
 - Add ability to display sub agent's answers in main agent.
 - Add ability to consolidate sub agent's assessment's in main agent.

{unique_orchestrator-1.1.0 → unique_orchestrator-1.1.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_orchestrator
-Version: 1.1.0
+Version: 1.1.1
 Summary: 
 License: Proprietary
 Author: Andreas Hauri
@@ -19,7 +19,7 @@ Requires-Dist: unique-follow-up-questions (>=1.0.0,<2.0.0)
 Requires-Dist: unique-internal-search (>=1.0.0,<2.0.0)
 Requires-Dist: unique-sdk (>=0.10.24,<0.11.0)
 Requires-Dist: unique-stock-ticker (>=1.0.0,<2.0.0)
-Requires-Dist: unique-toolkit (>=1.4.0,<2.0.0)
+Requires-Dist: unique-toolkit (>=1.8.0,<2.0.0)
 Requires-Dist: unique-web-search (>=1.0.0,<2.0.0)
 Description-Content-Type: text/markdown
 
@@ -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.1.1] - 2025-10-03
+- Adapt orchestrator to toolkit 1.8.0.
+
 ## [1.1.0] - 2025-09-29
 - Add ability to display sub agent's answers in main agent.
 - Add ability to consolidate sub agent's assessment's in main agent.

{unique_orchestrator-1.1.0 → unique_orchestrator-1.1.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "unique_orchestrator"
-version = "1.1.0"
+version = "1.1.1"
 description = ""
 authors = ["Andreas Hauri <andreas.hauri@unique.ai>"]
 readme = ["README.md", "CHANGELOG.md"]
@@ -15,7 +15,7 @@ python-dotenv = "^1.0.1"
 pytest = "^8.4.1"
 unique-sdk = "^0.10.24"
 
-unique-toolkit = "^1.4.0"
+unique-toolkit = "^1.8.0"
 unique-stock-ticker = "^1.0.0"
 unique-follow-up-questions = "^1.0.0"
 unique-internal-search = "^1.0.0"

{unique_orchestrator-1.1.0 → unique_orchestrator-1.1.1}/unique_orchestrator/config.py

@@ -22,7 +22,7 @@ from unique_toolkit.agentic.evaluation.schemas import EvaluationMetricName
 from unique_toolkit.agentic.history_manager.history_manager import (
     UploadedContentConfig,
 )
-from unique_toolkit.agentic.tools.a2a.evaluation import SubAgentEvaluationConfig
+from unique_toolkit.agentic.tools.a2a.evaluation import SubAgentEvaluationServiceConfig
 from unique_toolkit.agentic.tools.config import get_configuration_dict
 from unique_toolkit.agentic.tools.tool import ToolBuildConfig
 from unique_web_search.config import WebSearchConfig
@@ -117,7 +117,9 @@ class EvaluationConfig(BaseModel):
     model_config = get_configuration_dict()
     max_review_steps: int = 3
     hallucination_config: HallucinationConfig = HallucinationConfig()
-    sub_agents_config: SubAgentEvaluationConfig = SubAgentEvaluationConfig()
+    sub_agents_config: SubAgentEvaluationServiceConfig | None = (
+        SubAgentEvaluationServiceConfig()
+    )
 
 
 # ------------------------------------------------------------

{unique_orchestrator-1.1.0 → unique_orchestrator-1.1.1}/unique_orchestrator/unique_ai_builder.py

@@ -35,9 +35,10 @@ from unique_toolkit.agentic.thinking_manager.thinking_manager import (
     ThinkingManager,
     ThinkingManagerConfig,
 )
-from unique_toolkit.agentic.tools.a2a.evaluation import SubAgentsEvaluation
-from unique_toolkit.agentic.tools.a2a.manager import A2AManager
-from unique_toolkit.agentic.tools.a2a.postprocessing import (
+from unique_toolkit.agentic.tools.a2a import (
+    A2AManager,
+    ExtendedSubAgentToolConfig,
+    SubAgentEvaluationService,
     SubAgentResponsesPostprocessor,
 )
 from unique_toolkit.agentic.tools.config import ToolBuildConfig
@@ -166,25 +167,33 @@ def build_unique_ai(
         )
 
     if len(tool_manager.sub_agents) > 0:
-        postprocessor_manager.add_postprocessor(
-            SubAgentResponsesPostprocessor(
-                user_id=event.user_id,
-                agent_chat_id=event.payload.chat_id,
-                company_id=event.company_id,
-                sub_agent_tools=tool_manager.sub_agents,
-            )
+        sub_agent_responses_postprocessor = SubAgentResponsesPostprocessor(
+            user_id=event.user_id,
+            main_agent_chat_id=event.payload.chat_id,
+            company_id=event.company_id,
         )
-        if config.agent.services.evaluation_config is not None:
-            logger.info("Adding sub agents evaluation")
-            evaluation_manager.add_evaluation(
-                SubAgentsEvaluation(
-                    config.agent.services.evaluation_config.sub_agents_config,
-                    tool_manager.sub_agents,
-                    LanguageModelService.from_event(event),
-                )
+        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
+        ):
+            sub_agent_evaluation = SubAgentEvaluationService(
+                config=config.agent.services.evaluation_config.sub_agents_config,
+                language_model_service=LanguageModelService.from_event(event),
             )
-        else:
-            logger.warning("No evaluation config found for sub agents")
+            evaluation_manager.add_evaluation(sub_agent_evaluation)
+
+        for tool in tool_manager.sub_agents:
+            assert isinstance(tool.config, ExtendedSubAgentToolConfig)
+            sub_agent_responses_postprocessor.register_sub_agent_tool(
+                tool, tool.config.response_display_config
+            )
+            if sub_agent_evaluation is not None:
+                sub_agent_evaluation.register_sub_agent_tool(
+                    tool, tool.config.evaluation_config
+                )
 
     return UniqueAI(
         event=event,