Skip to content

Commit a443f62

Browse files
committed
Merge remote-tracking branch 'refs/remotes/origin/dev' into dev_zdy_1221_01_user_names
2 parents 3760c5b + 15b475b commit a443f62

File tree

3 files changed

+23
-29
lines changed

3 files changed

+23
-29
lines changed

src/memos/mem_reader/read_multi_modal/utils.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
import os
55
import re
66

7-
from datetime import datetime, timezone
7+
from datetime import datetime
88
from typing import Any, TypeAlias
99
from urllib.parse import urlparse
1010

@@ -245,8 +245,8 @@ def coerce_scene_data(scene_data: SceneDataInput, scene_type: str) -> list[Messa
245245

246246
# Default timestamp
247247
if chat_time_value is None:
248-
session_date = datetime.now(timezone.utc)
249-
date_format = "%I:%M %p on %d %B, %Y UTC"
248+
session_date = datetime.now()
249+
date_format = "%I:%M %p on %d %B, %Y"
250250
chat_time_value = session_date.strftime(date_format)
251251

252252
# Inject chat_time

src/memos/mem_scheduler/task_schedule_modules/redis_queue.py

Lines changed: 17 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -699,27 +699,23 @@ def _batch_claim_pending_messages(
699699
results = []
700700
try:
701701
results = pipe.execute()
702-
except Exception as e:
703-
err_msg = str(e).lower()
704-
if "nogroup" in err_msg or "no such key" in err_msg:
705-
# Fallback: attempt sequential xautoclaim for robustness
706-
for stream_key, need_count, label in claims_spec:
707-
try:
708-
self._ensure_consumer_group(stream_key=stream_key)
709-
res = self._redis_conn.xautoclaim(
710-
name=stream_key,
711-
groupname=self.consumer_group,
712-
consumername=self.consumer_name,
713-
min_idle_time=self.orchestrator.get_task_idle_min(task_label=label),
714-
start_id="0-0",
715-
count=need_count,
716-
justid=False,
717-
)
718-
results.append(res)
719-
except Exception:
720-
continue
721-
else:
722-
logger.error(f"Pipeline xautoclaim failed: {e}")
702+
except Exception:
703+
# Fallback: attempt sequential xautoclaim for robustness
704+
for stream_key, need_count, label in claims_spec:
705+
try:
706+
self._ensure_consumer_group(stream_key=stream_key)
707+
res = self._redis_conn.xautoclaim(
708+
name=stream_key,
709+
groupname=self.consumer_group,
710+
consumername=self.consumer_name,
711+
min_idle_time=self.orchestrator.get_task_idle_min(task_label=label),
712+
start_id="0-0",
713+
count=need_count,
714+
justid=False,
715+
)
716+
results.append(res)
717+
except Exception:
718+
continue
723719

724720
claimed_pairs: list[tuple[str, list[tuple[str, dict]]]] = []
725721
for (stream_key, _need_count, _label), claimed_result in zip(

src/memos/utils.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -6,9 +6,6 @@
66

77
logger = get_logger(__name__)
88

9-
# Global threshold (seconds) for timing logs
10-
DEFAULT_TIME_BAR = 10.0
11-
129

1310
def timed_with_status(
1411
func=None,
@@ -97,7 +94,7 @@ def wrapper(*args, **kwargs):
9794
return decorator(func)
9895

9996

100-
def timed(func=None, *, log=False, log_prefix=""):
97+
def timed(func=None, *, log=True, log_prefix=""):
10198
def decorator(fn):
10299
def wrapper(*args, **kwargs):
103100
start = time.perf_counter()
@@ -107,7 +104,8 @@ def wrapper(*args, **kwargs):
107104
if log is not True:
108105
return result
109106

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

113111
return result

0 commit comments

Comments
 (0)