From 0b6569446e5c425602ef8e30a2f6fc760d087104 Mon Sep 17 00:00:00 2001
From: Tvrtko Sternak <sternakt@gmail.com>
Date: Tue, 19 Nov 2024 14:07:55 +0000
Subject: [PATCH] Change system message to text message

---
 context_leakage_team/workflow/scenarios/scenario.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/context_leakage_team/workflow/scenarios/scenario.py b/context_leakage_team/workflow/scenarios/scenario.py
index 4183540..c92b5cb 100644
--- a/context_leakage_team/workflow/scenarios/scenario.py
+++ b/context_leakage_team/workflow/scenarios/scenario.py
@@ -105,9 +105,10 @@ def _validate_tool_call(
         self, ui: UI, messages: list[dict[str, Any]], agent: Agent, action: str
     ) -> Agent | None:
         if "tool_calls" not in messages[-1] and len(messages) > 1:
-            ui.system_message(
+            ui.text_message(
                 sender="Context leakage team",
-                message=f"Please call the function to {action}.",
+                recipient=agent.name,
+                body=f"Please call the function to {action}.",
             )
             return agent
         return None