Merge pull request #72 from mainframecomputer/feature/task-context-propagation

Feature/task context propagation

Author: philippe-page

packages/python/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "mainframe-orchestra"
-version = "0.0.31"
+version = "0.0.32"
 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. <[email protected]>",

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.31"
+__version__ = "0.0.32"
 
 import importlib
 

packages/python/src/mainframe_orchestra/orchestration.py

@@ -46,6 +46,9 @@ async def conduct_tool(
                 messages = kwargs.get("messages", [])
                 current_time = datetime.now().isoformat()
 
+                # Get the parent task's context if available
+                parent_context = kwargs.get("parent_context")
+
                 # Track agent iterations
                 agent_call_counts = {}  # Track {agent_id: count}
 
@@ -108,6 +111,7 @@ async def conduct_tool(
                                 f"You are {target_agent.role}. "
                                 f"Your goal is {target_agent.goal}"
                                 f"{' Your attributes are: ' + target_agent.attributes if target_agent.attributes and target_agent.attributes.strip() else ''}"
+                                f"{'\nAdditional context: ' + parent_context if parent_context else ''}"
                             ).strip(),
                         }
                     ]
@@ -199,7 +203,8 @@ async def nested_callback(result):
                         event_queue=event_queue,
                         messages=messages,
                         tool_summaries=tool_summaries,
-                        pre_execute=kwargs.get("pre_execute")
+                        pre_execute=kwargs.get("pre_execute"),
+                        context=parent_context  # Pass the parent context to the delegated task
                     )
 
                     # Generate a delegation result event after task completion

packages/python/src/mainframe_orchestra/task.py

@@ -847,14 +847,13 @@ async def callback_wrapper(event):
 
                                 if tool_name == "conduct_tool":
                                     # Store callback-related parameters separately
-                                    special_params.update(
-                                        {
-                                            "callback": callback,
-                                            "thread_id": self.thread_id,
-                                            "event_queue": self.event_queue,
-                                            "pre_execute": pre_execute,
-                                        }
-                                    )
+                                    special_params.update({
+                                        "callback": callback,
+                                        "thread_id": self.thread_id,
+                                        "event_queue": self.event_queue,
+                                        "pre_execute": pre_execute,
+                                        "parent_context": self.context,  # Pass the parent task's context
+                                    })
 
                                 # Log only the serializable parameters
                                 logger.info(