andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-03 16:22:49 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Improving agent logging

Browse Source
This commit is contained in:
João Moura
2024-03-11 17:05:54 -03:00
parent 27914bd841
commit 815606f4c2
3 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -10,7 +10,7 @@ class Logger:
)
self.verbose_level = verbose_level
def log(self, level, message):
def log(self, level, message, color="bold_green"):
level_map = {"debug": 1, "info": 2}
if self.verbose_level and level_map.get(level, 0) <= self.verbose_level:
self._printer.print(f"[{level.upper()}]: {message}", color="bold_green")
self._printer.print(f"[{level.upper()}]: {message}", color=color)

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

@@ -6,9 +6,14 @@ class Printer:
self._print_red(content)
elif color == "bold_green":
self._print_bold_green(content)
elif color == "bold_yellow":
self._print_bold_yellow(content)
else:
print(content)
def _print_bold_yellow(self, content):
print("\033[1m\033[93m {}\033[00m".format(content))
def _print_bold_green(self, content):
print("\033[1m\033[92m {}\033[00m".format(content))