Skip to content

Commit 4e68899

Browse files
committed
fix bugs: change name
1 parent b3600ae commit 4e68899

File tree

2 files changed

+7
-5
lines changed

2 files changed

+7
-5
lines changed

examples/mem_scheduler/try_schedule_modules.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,8 @@
55
from queue import Queue
66
from typing import TYPE_CHECKING
77

8+
from tqdm import tqdm
9+
810
from memos.configs.mem_cube import GeneralMemCubeConfig
911
from memos.configs.mem_os import MOSConfig
1012
from memos.configs.mem_scheduler import AuthConfig
@@ -169,7 +171,7 @@ def show_web_logs(mem_scheduler: GeneralScheduler):
169171

170172
mos.add(conversations, user_id=user_id, mem_cube_id=mem_cube_id)
171173

172-
for item in questions:
174+
for item in tqdm(questions, desc="processing queries"):
173175
query = item["question"]
174176

175177
# test process_session_turn

src/memos/mem_scheduler/modules/retriever.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,8 @@
1313
)
1414
from memos.mem_scheduler.utils import (
1515
extract_json_dict,
16-
is_all_Chinese,
17-
is_all_English,
16+
is_all_chinese,
17+
is_all_english,
1818
transform_name_to_key,
1919
)
2020
from memos.memories.textual.tree import TextualMemoryItem, TreeTextMemory
@@ -163,9 +163,9 @@ def filter_too_short_memories(
163163
continue
164164

165165
# Determine measurement method based on language
166-
if is_all_English(stripped_memory):
166+
if is_all_english(stripped_memory):
167167
length = len(stripped_memory.split()) # Word count for English
168-
elif is_all_Chinese(stripped_memory):
168+
elif is_all_chinese(stripped_memory):
169169
length = len(stripped_memory) # Character count for Chinese
170170
else:
171171
logger.debug(

0 commit comments

Comments
 (0)