unique_orchestrator 1.5.0__tar.gz → 1.5.2__tar.gz

{unique_orchestrator-1.5.0 → unique_orchestrator-1.5.2}/CHANGELOG.md

@@ -5,6 +5,15 @@ 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.5.2] - 2025-10-23
+- Run evaluation and post processing in parallel
+
+## [1.5.1] - 2025-10-17
+- revert behavior of unique ai upload and chat to 
+1. Add upload and chat tool to forced tools if there are tool choices
+2. Simply force it if there are no tool choices.
+3. Tool not available when no uploaded documents
+
 ## [1.5.0] - 2025-10-16
 - Make code interpreter configurable through spaces 2.0.
 

{unique_orchestrator-1.5.0 → unique_orchestrator-1.5.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_orchestrator
-Version: 1.5.0
+Version: 1.5.2
 Summary: 
 License: Proprietary
 Author: Andreas Hauri
@@ -33,6 +33,15 @@ 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.5.2] - 2025-10-23
+- Run evaluation and post processing in parallel
+
+## [1.5.1] - 2025-10-17
+- revert behavior of unique ai upload and chat to 
+1. Add upload and chat tool to forced tools if there are tool choices
+2. Simply force it if there are no tool choices.
+3. Tool not available when no uploaded documents
+
 ## [1.5.0] - 2025-10-16
 - Make code interpreter configurable through spaces 2.0.
 

{unique_orchestrator-1.5.0 → unique_orchestrator-1.5.2}/pyproject.toml

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

{unique_orchestrator-1.5.0 → unique_orchestrator-1.5.2}/unique_orchestrator/unique_ai.py

@@ -1,3 +1,4 @@
+import asyncio
 from datetime import datetime
 from logging import Logger
 
@@ -14,6 +15,7 @@ from unique_toolkit.agentic.reference_manager.reference_manager import Reference
 from unique_toolkit.agentic.thinking_manager.thinking_manager import ThinkingManager
 from unique_toolkit.agentic.tools.tool_manager import (
     ResponsesApiToolManager,
+    SafeTaskExecutor,
     ToolManager,
 )
 from unique_toolkit.app.schemas import ChatEvent, McpServer
@@ -338,14 +340,31 @@ class UniqueAI:
         self, loop_response: LanguageModelStreamResponse
     ) -> bool:
         """Handle the case where no tool calls are returned."""
+        task_executor = SafeTaskExecutor(
+            logger=self._logger,
+        )
+
         selected_evaluation_names = self._tool_manager.get_evaluation_check_list()
-        evaluation_results = await self._evaluation_manager.run_evaluations(
-            selected_evaluation_names, loop_response, self._latest_assistant_id
+        evaluation_results = task_executor.execute_async(
+            self._evaluation_manager.run_evaluations,
+            selected_evaluation_names,
+            loop_response,
+            self._latest_assistant_id,
+        )
+
+        postprocessor_result = task_executor.execute_async(
+            self._postprocessor_manager.run_postprocessors,
+            loop_response.model_copy(deep=True),
         )
 
-        await self._postprocessor_manager.run_postprocessors(loop_response)
+        _, evaluation_results = await asyncio.gather(
+            postprocessor_result,
+            evaluation_results,
+        )
 
-        if not all(result.is_positive for result in evaluation_results):
+        if evaluation_results.success and not all(
+            result.is_positive for result in evaluation_results.unpack()
+        ):
             self._logger.warning(
                 "we should add here the retry counter add an instruction and retry the loop for now we just exit the loop"
             )  # TODO: add retry counter and instruction

{unique_orchestrator-1.5.0 → unique_orchestrator-1.5.2}/unique_orchestrator/unique_ai_builder.py

@@ -375,17 +375,24 @@ def _build_completions(
     common_components: _CommonComponents,
     debug_info_manager: DebugInfoManager,
 ) -> UniqueAI:
-    if len(common_components.uploaded_documents) > 0:
+    # Uploaded content behavior is always to force uploaded search tool:
+    # 1. Add it to forced tools if there are tool choices.
+    # 2. Simply force it if there are no tool choices.
+    # 3. Not available if not uploaded documents.
+    UPLOADED_DOCUMENTS = len(common_components.uploaded_documents) > 0
+    TOOL_CHOICES = len(event.payload.tool_choices) > 0
+    if UPLOADED_DOCUMENTS:
         logger.info(
             f"Adding UploadedSearchTool with {len(common_components.uploaded_documents)} documents"
         )
-        config.space.tools.append(
+        common_components.tool_manager_config.tools.append(
             ToolBuildConfig(
                 name=UploadedSearchTool.name,
                 display_name=UploadedSearchTool.name,
                 configuration=UploadedSearchConfig(),
-            ),
+            )
         )
+    if TOOL_CHOICES and UPLOADED_DOCUMENTS:
         event.payload.tool_choices.append(str(UploadedSearchTool.name))
 
     tool_manager = ToolManager(
@@ -396,6 +403,8 @@ def _build_completions(
         mcp_manager=common_components.mcp_manager,
         a2a_manager=common_components.a2a_manager,
     )
+    if not TOOL_CHOICES and UPLOADED_DOCUMENTS:
+        tool_manager.add_forced_tool(UploadedSearchTool.name)
 
     postprocessor_manager = PostprocessorManager(
         logger=logger,