Merge pull request #70 from mainframecomputer/bugfix/update-context-to-system-prompt

Bugfix/update context to system prompt

Author: philippe-page

packages/python/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "mainframe-orchestra"
-version = "0.0.30"
+version = "0.0.31"
 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.30"
+__version__ = "0.0.31"
 
 import importlib
 

packages/python/src/mainframe_orchestra/task.py

@@ -312,20 +312,14 @@ async def _create_async(
                         f"You are {role or (agent.role if agent else None)}. "
                         f"Your goal is {goal or (agent.goal if agent else None)}"
                         f"{' Your attributes are: ' + (attributes or (agent.attributes if agent else '')) if attributes or (agent.attributes if agent else '') else ''}"
+                        f"{'\nAdditional context: ' + (context or '') if context else ''}"
                     ).strip(),
                 }
                 messages.insert(0, system_message)
 
-            # Combine context and instruction
-            user_content = []
-            if context:
-                user_content.append(context)
-            user_content.append(instruction)
-            user_message_content = "\n\n".join(user_content)
-
             # Only append if different from last message
-            if not messages or messages[-1].get("content") != user_message_content:
-                messages.append({"role": "user", "content": user_message_content})
+            if not messages or messages[-1].get("content") != instruction:
+                messages.append({"role": "user", "content": instruction})
 
             task_data = {
                 "agent_id": agent.agent_id if agent else None,
@@ -615,8 +609,6 @@ def hash_tool_call(tool_call: dict) -> str:
 
                 # Include tool results in context if we have any
                 context_parts = []
-                if self.context:
-                    context_parts.append(self.context)
                 context_parts.append(tool_descriptions)
 
                 if tool_results:  # Using existing tool_results list