From 6ea21e95b6fd6dc136d92f0e2dcb96eaa90ef99d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jo=C3=A3o=20Moura?= Date: Sat, 10 Feb 2024 00:57:04 -0800 Subject: [PATCH] Adding printer logic --- src/crewai/utilities/__init__.py | 1 + src/crewai/utilities/printer.py | 9 +++++++++ 2 files changed, 10 insertions(+) create mode 100644 src/crewai/utilities/printer.py diff --git a/src/crewai/utilities/__init__.py b/src/crewai/utilities/__init__.py index 1a7af0727..86c41fa44 100644 --- a/src/crewai/utilities/__init__.py +++ b/src/crewai/utilities/__init__.py @@ -1,4 +1,5 @@ from .i18n import I18N from .logger import Logger +from .printer import Printer from .prompts import Prompts from .rpm_controller import RPMController diff --git a/src/crewai/utilities/printer.py b/src/crewai/utilities/printer.py new file mode 100644 index 000000000..c00063833 --- /dev/null +++ b/src/crewai/utilities/printer.py @@ -0,0 +1,9 @@ +class Printer: + def print(self, content: str, color: str): + if color == "yellow": + self._print_yellow(content) + else: + print(content) + + def _print_yellow(self, content): + print("\033[93m {}\033[00m".format(content))