mirror of
https://github.com/crewAIInc/crewAI.git
synced 2026-01-13 18:18:29 +00:00
bb622bf747
* fix: 'from datetime import datetime for logging' to print the timestamp * fix: correct default model (gpt-4o), correct token counts, and correct TaskOutput attributes (added agent) * test: verify Task callback data is an instance of TaskOutput
21 lines
694 B
Python
21 lines
694 B
Python
from datetime import datetime
|
|
from crewai.utilities.printer import Printer
|
|
|
|
|
|
class Logger:
|
|
_printer = Printer()
|
|
|
|
def __init__(self, verbose_level=0):
|
|
verbose_level = (
|
|
2 if isinstance(verbose_level, bool) and verbose_level else verbose_level
|
|
)
|
|
self.verbose_level = verbose_level
|
|
|
|
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:
|
|
timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
|
self._printer.print(
|
|
f"[{timestamp}][{level.upper()}]: {message}", color=color
|
|
)
|