remove extra characters

EmergenceAI · Jun 21, 2024 · 375ab90 · 375ab90
1 parent 2223db4
commit 375ab90
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/ae/core/playwright_manager.py b/ae/core/playwright_manager.py
@@ -285,10 +285,16 @@ async def notify_user(self, message: str, message_type: MessageType = MessageTyp
         if self.ui_manager.overlay_show_details == True:
             if message_type not in (MessageType.PLAN,  MessageType.QUESTION , MessageType.ANSWER,  MessageType.INFO, MessageType.STEP):
                 return
-
+
+        if message.startswith(":"):
+            message = message[1:]
+
+        if message.endswith(","):
+            message = message[:-1]
+
         if message_type == MessageType.PLAN:
             message = beautify_plan_message(message)
-            message = "Plan: \n" + message
+            message = "Plan:\n" + message
         if message_type == MessageType.STEP:
             if "confirm" in message.lower():
                 message = "Verify: " + message