autobyteus 1.1.7__py3-none-any.whl → 1.1.8__py3-none-any.whl

autobyteus/agent/handlers/tool_result_event_handler.py

@@ -16,10 +16,10 @@ logger = logging.getLogger(__name__)
 
 class ToolResultEventHandler(AgentEventHandler):
     """
-    Handles ToolResultEvents. It immediately processes and notifies for each
-    individual tool result. If a multi-tool call turn is active, it accumulates
-    these processed results until the turn is complete, then sends a single
-    aggregated message to the LLM.
+    Handles ToolResultEvents. It processes and notifies for each individual tool
+    result as it arrives. If a multi-tool call turn is active, it accumulates
+    these results until the turn is complete, re-orders them to match the original
+    invocation sequence, and then sends a single aggregated message to the LLM.
     """
     def __init__(self):
         logger.info("ToolResultEventHandler initialized.")
@@ -28,8 +28,8 @@ class ToolResultEventHandler(AgentEventHandler):
                                                   processed_events: List[ToolResultEvent],
                                                   context: 'AgentContext'):
         """
-        Aggregates a list of PRE-PROCESSED tool results into a single message and
-        dispatches it to the LLM.
+        Aggregates a list of PRE-PROCESSED and ORDERED tool results into a single
+        message and dispatches it to the LLM.
         """
         agent_id = context.agent_id
         
@@ -133,9 +133,29 @@ class ToolResultEventHandler(AgentEventHandler):
         if not active_turn.is_complete():
             return
             
-        # If all results are in, dispatch them to the LLM and clean up the turn state.
-        logger.info(f"Agent '{agent_id}': All tool results for the turn collected. Aggregating for LLM.")
-        await self._dispatch_aggregated_results_to_llm(active_turn.results, context)
+        # If all results are in, re-order them and then dispatch to the LLM.
+        logger.info(f"Agent '{agent_id}': All tool results for the turn collected. Re-ordering to match invocation sequence.")
+        
+        # --- NEW RE-ORDERING LOGIC ---
+        results_by_id = {res.tool_invocation_id: res for res in active_turn.results}
+        sorted_results: List[ToolResultEvent] = []
+        for original_invocation in active_turn.invocations:
+            result = results_by_id.get(original_invocation.id)
+            if result:
+                sorted_results.append(result)
+            else:
+                # This should not happen if the logic is correct, but it's a good safeguard.
+                logger.error(f"Agent '{agent_id}': Missing result for invocation ID '{original_invocation.id}' during re-ordering.")
+                # Add a synthetic error result to maintain sequence length.
+                sorted_results.append(ToolResultEvent(
+                    tool_name=original_invocation.name,
+                    result=None,
+                    error=f"Critical Error: Result for this tool call was lost.",
+                    tool_invocation_id=original_invocation.id
+                ))
+
+        await self._dispatch_aggregated_results_to_llm(sorted_results, context)
         
         context.state.active_multi_tool_call_turn = None
         logger.info(f"Agent '{agent_id}': Multi-tool call turn state has been cleared.")
+

{autobyteus-1.1.7.dist-info → autobyteus-1.1.8.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: autobyteus
-Version: 1.1.7
+Version: 1.1.8
 Summary: Multi-Agent framework
 Home-page: https://github.com/AutoByteus/autobyteus
 Author: Ryan Zheng

{autobyteus-1.1.7.dist-info → autobyteus-1.1.8.dist-info}/RECORD

@@ -36,7 +36,7 @@ autobyteus/agent/handlers/llm_complete_response_received_event_handler.py,sha256
 autobyteus/agent/handlers/llm_user_message_ready_event_handler.py,sha256=4WgDXUY4rmjpP44X2THKMAt2NpAZ5GukPqVNmLt6XJM,9064
 autobyteus/agent/handlers/tool_execution_approval_event_handler.py,sha256=Tda_LrlIPbEwVzf1I6u4Vb9sPmAbGQRqT_2-Q9ukLhw,4489
 autobyteus/agent/handlers/tool_invocation_request_event_handler.py,sha256=AjNGQgRCQkjXvYThd_AaPj0Lzh1adEkhYJO83alxVAY,10957
-autobyteus/agent/handlers/tool_result_event_handler.py,sha256=MmT567LCZ60sK0fgeULCR0wM7PIJEAsRESCdDZNhwC8,7292
+autobyteus/agent/handlers/tool_result_event_handler.py,sha256=wecvtww_MuS1ekv_WexWGlR11bP96N46dK6uoWkXpHo,8370
 autobyteus/agent/handlers/user_input_message_event_handler.py,sha256=WBV_yFILqEEGbwhzPxKRKdTZcbJosvRIXAs8m1jOHi4,4937
 autobyteus/agent/hooks/__init__.py,sha256=a1do0Ribb2Hpf9t0Xqxhf3Ls7R6EQuWJtfTGQK7VjUM,495
 autobyteus/agent/hooks/base_phase_hook.py,sha256=7JU3FgPzX06Yg6eJgB4GXzXcaWTpOcEMmyw5Ku0mwJk,2040
@@ -465,7 +465,7 @@ autobyteus/workflow/streaming/workflow_stream_event_payloads.py,sha256=jv1bVYg-7
 autobyteus/workflow/streaming/workflow_stream_events.py,sha256=75P29jNgcL7Go7D9wVz236KTwPfmqc5K7hUvVnc94K0,2221
 autobyteus/workflow/utils/__init__.py,sha256=SzaMZHnJBIJKcT_r-HOeyIcuxzRu2bGeFkOcMLJaalk,222
 autobyteus/workflow/utils/wait_for_idle.py,sha256=FgHtz59DN0eg8Na1PkkVR55Ihdd2e5Gn_mr7RVHl4qI,2001
-autobyteus-1.1.7.dist-info/licenses/LICENSE,sha256=Ompok_c8HRsXRwmax-pGR9OZRRxZC9RPp4JB6eTJd0M,1360
+autobyteus-1.1.8.dist-info/licenses/LICENSE,sha256=Ompok_c8HRsXRwmax-pGR9OZRRxZC9RPp4JB6eTJd0M,1360
 examples/__init__.py,sha256=BtTQJ6yeHyksK5GC3kfN6RFR6Gcrwq1TBmp6FIIj3Z8,40
 examples/discover_phase_transitions.py,sha256=NiFK_XzDCpWwmNsQqf0Ou2w6L5bofKIKODq7sH5uPzk,3679
 examples/run_browser_agent.py,sha256=tpBJGIYYvVsNZAUo_WsZbhV_8fdeORUoHlQ8uQvnXPM,11737
@@ -476,7 +476,7 @@ examples/run_mcp_list_tools.py,sha256=-dOM-7xyyDM2gp5e_8KZVGbX5ZxWqFQB9l-fHfR8Xx
 examples/run_poem_writer.py,sha256=wJsT4ZHwXw3MbPAESwyCkAMWWYt7KH5BLhEDQxA06XM,13145
 examples/run_sqlite_agent.py,sha256=wy1Mp_F7RR0WmvfmxsPLG9JFPi6SpTVd0x_Ep76bUQ8,13159
 examples/agent_team/__init__.py,sha256=WIg0HENp1TUClJ3p2gIRn0C-VW9Qr7Ttqtedr4xQ3Jo,51
-autobyteus-1.1.7.dist-info/METADATA,sha256=pgphbfB80cOcXlJd43NMVhIxaSSqOdJilDhTdo2ZyhE,10173
-autobyteus-1.1.7.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-autobyteus-1.1.7.dist-info/top_level.txt,sha256=vNmK1Y8Irbc0iDPdRtr9gIx5eLM-c2v1ntItkzICzHU,20
-autobyteus-1.1.7.dist-info/RECORD,,
+autobyteus-1.1.8.dist-info/METADATA,sha256=cvBLfLpv5ukmQrFsn7HL9P13ZIOoUNzhpeOXfRZMC0g,10173
+autobyteus-1.1.8.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+autobyteus-1.1.8.dist-info/top_level.txt,sha256=vNmK1Y8Irbc0iDPdRtr9gIx5eLM-c2v1ntItkzICzHU,20
+autobyteus-1.1.8.dist-info/RECORD,,