unique_orchestrator 1.2.3__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-1.2.3.dist-info → unique_orchestrator-1.2.4.dist-info}/METADATA

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

{unique_orchestrator-1.2.3.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.py,sha256=CBOd1eQuasvt-A1azsGZJbHlExm9UGsgy19_l613PoQ,16798
 unique_orchestrator/unique_ai_builder.py,sha256=FUIJAVv0lPTWDgWeMx-KuCyrIc8w8lWyEciFs3_7a3Q,7608
-unique_orchestrator-1.2.3.dist-info/LICENSE,sha256=GlN8wHNdh53xwOPg44URnwag6TEolCjoq3YD_KrWgss,193
-unique_orchestrator-1.2.3.dist-info/METADATA,sha256=8AZVFC-ygH-hJlGryJr8To5D1qQGxVX-I-7a_oY5EQk,2386
-unique_orchestrator-1.2.3.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
-unique_orchestrator-1.2.3.dist-info/RECORD,,
+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,,