Change system message to text message

airtai · Nov 19, 2024 · 0b65694 · 0b65694
1 parent a3e9b44
commit 0b65694
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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