andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-01-10 00:28:31 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

printing max rpm message in different color

Browse Source
This commit is contained in:
João Moura
2024-09-18 04:35:18 -03:00
parent 5a3b888f43
commit 5e8322b272
2 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/crewai/utilities/logger.py
View File

@@ -9,9 +9,9 @@ class Logger(BaseModel):
verbose: bool = Field(default=False) verbose: bool = Field(default=False)
_printer: Printer = PrivateAttr(default_factory=Printer) _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: if self.verbose:
timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S") timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
self._printer.print( self._printer.print(
f"[{timestamp}][{level.upper()}]: {message}", color=color f"\n[{timestamp}][{level.upper()}]: {message}", color=color
) )

5
src/crewai/utilities/printer.py
View File

@@ -15,6 +15,8 @@ class Printer:
self._print_bold_blue(content) self._print_bold_blue(content)
elif color == "yellow": elif color == "yellow":
self._print_yellow(content) self._print_yellow(content)
elif color == "bold_yellow":
self._print_bold_yellow(content)
else: else:
print(content) print(content)
@@ -35,3 +37,6 @@ class Printer:
def _print_yellow(self, content): def _print_yellow(self, content):
print("\033[93m {}\033[00m".format(content)) print("\033[93m {}\033[00m".format(content))
def _print_bold_yellow(self, content):
print("\033[1m\033[93m {}\033[00m".format(content))