Merge pull request #68 from mainframecomputer/MFE-562-preexecute-in-tool-loop

Mfe 562 preexecute in tool loop

Author: philippe-page

packages/python/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "mainframe-orchestra"
-version = "0.0.29"
+version = "0.0.30"
 description = "Mainframe-Orchestra is a lightweight, open-source agentic framework for building LLM based pipelines and self-orchestrating multi-agent teams"
 authors = [
     "Mainframe Computer Inc. <hi@mainfra.me>",

packages/python/src/mainframe_orchestra/__init__.py

@@ -3,7 +3,7 @@
 """
 # Copyright 2024 Mainframe-Orchestra Contributors. Licensed under Apache License 2.0.
 
-__version__ = "0.0.29"
+__version__ = "0.0.30"
 
 import importlib
 

packages/python/src/mainframe_orchestra/orchestration.py

@@ -199,6 +199,7 @@ async def nested_callback(result):
                         event_queue=event_queue,
                         messages=messages,
                         tool_summaries=tool_summaries,
+                        pre_execute=kwargs.get("pre_execute")
                     )
 
                     # Generate a delegation result event after task completion

packages/python/src/mainframe_orchestra/task.py

@@ -609,6 +609,10 @@ def hash_tool_call(tool_call: dict) -> str:
                 logger.debug(f"Starting iteration {iteration_count + 1}/{MAX_ITERATIONS}")
                 iteration_count += 1
 
+                # Call pre_execute at the beginning of each tool loop iteration
+                if pre_execute:
+                    await pre_execute({"agent_id": self.agent_id})
+
                 # Include tool results in context if we have any
                 context_parts = []
                 if self.context:
@@ -757,7 +761,7 @@ def hash_tool_call(tool_call: dict) -> str:
 
                         initial_prompt = (
                             f"Given the task instruction: '{self.instruction}' and the planned tool calls: {json.dumps(response_data['tool_calls'], indent=2)}, "
-                            "please provide an initial response explaining your planned approach before executing the tools."
+                            "please provide an initial response explaining your planned approach. Provide only a conversational response. You cannot call tools *in* this response, you will be calling them after this response, so don't attempt to call them here. Do not attempt to fully answer the task, and don't ask questions because this will be immediately followed by tool calls."
                         )
 
                         original_messages = self.messages