unique_orchestrator 1.5.1__tar.gz → 1.5.2__tar.gz

{unique_orchestrator-1.5.1 → unique_orchestrator-1.5.2}/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.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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_orchestrator
-Version: 1.5.1
+Version: 1.5.2
 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.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

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

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

{unique_orchestrator-1.5.1 → 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