மனோஜ்குமார் பழனிச்சாமி 1 gadu atpakaļ
vecāks
revīzija
a0fee17388
1 mainītis faili ar 4 papildinājumiem un 2 dzēšanām
  1. 4 2
      opendevin/logger.py

+ 4 - 2
opendevin/logger.py

@@ -45,10 +45,12 @@ class ColoredFormatter(logging.Formatter):
         if msg_type in LOG_COLORS and not DISABLE_COLOR_PRINTING:
             msg_type_color = colored(msg_type, LOG_COLORS[msg_type])
             msg = colored(record.msg, LOG_COLORS[msg_type])
-            time_str = colored(self.formatTime(record, self.datefmt), 'green')
+            time_str = colored(self.formatTime(record, self.datefmt), LOG_COLORS[msg_type])
             name_str = colored(record.name, 'cyan')
             level_str = colored(record.levelname, 'yellow')
-            return f'{time_str} - {name_str}:{level_str}: {record.filename}:{record.lineno}\n{msg_type_color}\n{msg}'
+            if msg_type in ['ERROR', 'INFO']:
+                return f'{time_str} - {name_str}:{level_str}: {record.filename}:{record.lineno}\n{msg_type_color}\n{msg}'
+            return f'{time_str} - {msg_type_color}\n{msg}'
         elif msg_type == 'STEP':
             msg = '\n\n==============\n' + record.msg + '\n'
             return f'{msg}'