Skip to content
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 3 additions & 5 deletions src/memos/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,9 +6,6 @@

logger = get_logger(__name__)

# Global threshold (seconds) for timing logs
DEFAULT_TIME_BAR = 10.0


def timed_with_status(
func=None,
Expand Down Expand Up @@ -97,7 +94,7 @@ def wrapper(*args, **kwargs):
return decorator(func)


def timed(func=None, *, log=False, log_prefix=""):
def timed(func=None, *, log=True, log_prefix=""):
def decorator(fn):
def wrapper(*args, **kwargs):
start = time.perf_counter()
Expand All @@ -107,7 +104,8 @@ def wrapper(*args, **kwargs):
if log is not True:
return result

if elapsed_ms >= (DEFAULT_TIME_BAR * 1000.0):
# 100ms threshold
if elapsed_ms >= 100.0:
logger.info(f"[TIMER] {log_prefix or fn.__name__} took {elapsed_ms:.0f} ms")

return result
Expand Down