35 lines
944 B
Python
35 lines
944 B
Python
import json
|
|
import re
|
|
|
|
from Agent_E.ae.utils.logger import logger
|
|
|
|
|
|
def escape_js_message(message: str) -> str:
|
|
"""
|
|
Escape a message for use in JavaScript code.
|
|
|
|
Args:
|
|
message (str): The message to escape.
|
|
|
|
Returns:
|
|
str: The escaped message.
|
|
"""
|
|
return json.dumps(message)
|
|
|
|
|
|
def beautify_plan_message(message:str) -> str:
|
|
"""
|
|
Add a newline between each numbered step in the plan message if it does not already exist.
|
|
|
|
Args:
|
|
message (str): The plan message.
|
|
|
|
Returns:
|
|
str: The plan message with newlines added between each numbered step.
|
|
"""
|
|
logger.debug(f"beautify_plan_message original:\n{message}")
|
|
# Add a newline before each numbered step that is not already preceded by a newline
|
|
plan_with_newlines = re.sub(r'(?<!\n)( \d+\.)', r'\n\1', message)
|
|
logger.debug(f"beautify_plan_message modified:\n{plan_with_newlines}")
|
|
return plan_with_newlines
|