diff --git a/simplemind/logging.py b/simplemind/logging.py index c18d718..06defc3 100644 --- a/simplemind/logging.py +++ b/simplemind/logging.py @@ -14,13 +14,17 @@ def logger(func: Callable[..., Any]) -> Callable[..., Any]: def wrapper(*args, **kwargs) -> Any: if not settings.logging.is_enabled: return func(*args, **kwargs) + logfire.info(f"Calling {func.__name__} with args: {args}, kwargs: {kwargs}") t1 = time.perf_counter() + try: result = func(*args, **kwargs) t2 = time.perf_counter() logfire.info(f"{func.__name__} returned: {result} in {t2-t1} seconds") + return result + except Exception as e: t2 = time.perf_counter() logfire.error(f"Error in {func.__name__}: {e} in {t2-t1} seconds")