From 5e8322b272898edfda73a1c1f228b05adf87c030 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jo=C3=A3o=20Moura?= Date: Wed, 18 Sep 2024 04:35:18 -0300 Subject: [PATCH] printing max rpm message in different color --- src/crewai/utilities/logger.py | 4 ++-- src/crewai/utilities/printer.py | 5 +++++ 2 files changed, 7 insertions(+), 2 deletions(-) diff --git a/src/crewai/utilities/logger.py b/src/crewai/utilities/logger.py index 9cf06757a..9b883f617 100644 --- a/src/crewai/utilities/logger.py +++ b/src/crewai/utilities/logger.py @@ -9,9 +9,9 @@ class Logger(BaseModel): verbose: bool = Field(default=False) _printer: Printer = PrivateAttr(default_factory=Printer) - def log(self, level, message, color="bold_green"): + def log(self, level, message, color="bold_yellow"): if self.verbose: timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S") self._printer.print( - f"[{timestamp}][{level.upper()}]: {message}", color=color + f"\n[{timestamp}][{level.upper()}]: {message}", color=color ) diff --git a/src/crewai/utilities/printer.py b/src/crewai/utilities/printer.py index f766003e4..edb339c29 100644 --- a/src/crewai/utilities/printer.py +++ b/src/crewai/utilities/printer.py @@ -15,6 +15,8 @@ class Printer: self._print_bold_blue(content) elif color == "yellow": self._print_yellow(content) + elif color == "bold_yellow": + self._print_bold_yellow(content) else: print(content) @@ -35,3 +37,6 @@ class Printer: def _print_yellow(self, content): print("\033[93m {}\033[00m".format(content)) + + def _print_bold_yellow(self, content): + print("\033[1m\033[93m {}\033[00m".format(content))