From fa8058b5ffe91a4ec0839ccf8fcf1b36d33dcd89 Mon Sep 17 00:00:00 2001 From: bobo Date: Sat, 18 May 2024 14:55:17 +0800 Subject: [PATCH] chore: Refactor Step class to improve logging of step duration --- lib/chatmark/step.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/lib/chatmark/step.py b/lib/chatmark/step.py index 93de1d4..349c097 100644 --- a/lib/chatmark/step.py +++ b/lib/chatmark/step.py @@ -1,8 +1,9 @@ -from contextlib import AbstractContextManager import time +from contextlib import AbstractContextManager from lib.ide_service import IDEService + class Step(AbstractContextManager): """ Show a running step in the TUI. @@ -30,7 +31,6 @@ def __exit__(self, exc_type, exc_val, exc_tb): # close the step end_time = time.time() IDEService().ide_logging( - "debug", - f"Step {self.title} took {end_time - self.enter_time:.2f} seconds" + "debug", f"Step {self.title} took {end_time - self.enter_time:.2f} seconds" ) print("\n```", flush=True)