unique_orchestrator 1.2.2__py3-none-any.whl → 1.2.4__py3-none-any.whl

unique_orchestrator/unique_ai.py

@@ -71,6 +71,9 @@ class UniqueAI:
         self._latest_assistant_id: str = event.payload.assistant_message.id
         self._mcp_servers = mcp_servers
 
+        # Helper variable to support control loop
+        self._tool_took_control = False
+
     ############################################################
     # Override of base methods
     ############################################################
@@ -125,8 +128,9 @@ class UniqueAI:
                 loop_response
             )
 
+        # Only set completed_at if no tool took control. Tools that take control will set the message state to completed themselves.
         await self._chat_service.modify_assistant_message_async(
-            set_completed_at=True,
+            set_completed_at=not self._tool_took_control,
         )
 
     # @track()
@@ -352,7 +356,10 @@ class UniqueAI:
         self._reference_manager.extract_referenceable_chunks(tool_call_responses)
         self._debug_info_manager.extract_tool_debug_info(tool_call_responses)
 
-        return self._tool_manager.does_a_tool_take_control(tool_calls)
+        self._tool_took_control = self._tool_manager.does_a_tool_take_control(
+            tool_calls
+        )
+        return self._tool_took_control
 
     async def _create_new_assistant_message_if_loop_response_contains_content(
         self, loop_response: LanguageModelStreamResponse

unique_orchestrator/unique_ai_builder.py

@@ -155,7 +155,10 @@ def build_unique_ai(
             )
         )
 
-    if config.agent.services.follow_up_questions_config:
+    if (
+        config.agent.services.follow_up_questions_config
+        and config.agent.services.follow_up_questions_config.number_of_questions > 0
+    ):
         postprocessor_manager.add_postprocessor(
             FollowUpPostprocessor(
                 logger=logger,

{unique_orchestrator-1.2.2.dist-info → unique_orchestrator-1.2.4.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: unique_orchestrator
-Version: 1.2.2
+Version: 1.2.4
 Summary: 
 License: Proprietary
 Author: Andreas Hauri
@@ -33,6 +33,12 @@ 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.2.4] - 2025-10-14
+- Let control taking tool itself set the message state to completed
+
+## [1.2.3] - 2025-10-13
+- Fix bug where follow-up questions were being generated even if the number of questions is set to 0 in the config.
+
 ## [1.2.2] - 2025-10-09
 - update loading path of `DEFAULT_GPT_4o` from `unique_toolkit`
 

{unique_orchestrator-1.2.2.dist-info → unique_orchestrator-1.2.4.dist-info}/RECORD

@@ -3,9 +3,9 @@ unique_orchestrator/prompts/generic_reference_prompt.jinja2,sha256=fYPaiE-N1gSoO
 unique_orchestrator/prompts/system_prompt.jinja2,sha256=YXFdx3PG2p4TKfjEpz7guIw2GaKoY-4zRMEzXaKhHXE,7213
 unique_orchestrator/prompts/user_message_prompt.jinja2,sha256=BQokpBh3H2J-rFk8i-PRph3jy4T1gAJPPb1mxxRWNuM,878
 unique_orchestrator/tests/test_unique_ai_reference_order.py,sha256=8mZeVP1k8neH4qrFW3oa3zwIdaq2c7R1VvurC7kjBU8,4445
-unique_orchestrator/unique_ai.py,sha256=8sBD5P16XcJ_3BXywBuZRRa4h89ANAi8h3VA4V2G_sI,16471
-unique_orchestrator/unique_ai_builder.py,sha256=m9-k9oVY7RR9ntrWEzBs0gcycNk4JcbqNfqW-DlJXrc,7507
-unique_orchestrator-1.2.2.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_orchestrator-1.2.2.dist-info/METADATA,sha256=XIbO6S33sxXWD5OiMNCfzLsK7x2EYXVg5RdR0LzgMfE,2245
-unique_orchestrator-1.2.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_orchestrator-1.2.2.dist-info/RECORD,,
+unique_orchestrator/unique_ai.py,sha256=CBOd1eQuasvt-A1azsGZJbHlExm9UGsgy19_l613PoQ,16798
+unique_orchestrator/unique_ai_builder.py,sha256=FUIJAVv0lPTWDgWeMx-KuCyrIc8w8lWyEciFs3_7a3Q,7608
+unique_orchestrator-1.2.4.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
+unique_orchestrator-1.2.4.dist-info/METADATA,sha256=QkwdL4PgwDsEX2c-rSanDPDoN9AaIV11FVnmEkacK50,2479
+unique_orchestrator-1.2.4.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
+unique_orchestrator-1.2.4.dist-info/RECORD,,