diff --git a/benchmarks/multi_turn/bench_utils.py b/benchmarks/multi_turn/bench_utils.py index d4d3c1ca8c52f..e959a4be711c9 100644 --- a/benchmarks/multi_turn/bench_utils.py +++ b/benchmarks/multi_turn/bench_utils.py @@ -4,7 +4,7 @@ import logging from enum import Enum -class Color(str, Enum): +class Color(Enum): RED = "\033[91m" GREEN = "\033[92m" BLUE = "\033[94m" @@ -13,6 +13,9 @@ class Color(str, Enum): YELLOW = "\033[93m" RESET = "\033[0m" + def __str__(self): + return self.value + TEXT_SEPARATOR = "-" * 100