Skip to content

Commit d5700ad

Browse files
author
harvey_xiang
committed
feat: add addMessage Stage log
1 parent 04858df commit d5700ad

3 files changed

Lines changed: 3 additions & 5 deletions

File tree

src/memos/chunkers/sentence_chunker.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,7 +50,8 @@ def chunk(self, text: str) -> list[str] | list[Chunk]:
5050
for c in chonkie_chunks:
5151
chunk = Chunk(text=c.text, token_count=c.token_count, sentences=c.sentences)
5252
chunk = self.restore_urls(chunk.text, url_map)
53-
5453
chunks.append(chunk)
5554

55+
logger.debug(f"Generated {len(chunks)} chunks from input text")
56+
5657
return chunks

src/memos/mem_feedback/feedback.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -725,7 +725,7 @@ def filter_fault_update(self, operations: list[dict]):
725725
for judge in all_judge:
726726
valid_update = None
727727
if judge["judgement"] == "UPDATE_APPROVED":
728-
valid_update = id2op.get(judge["id"])
728+
valid_update = id2op.get(judge["id"], None)
729729
if valid_update:
730730
valid_updates.append(valid_update)
731731

src/memos/mem_reader/multi_modal_struct.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -131,7 +131,6 @@ def _create_chunk_item(chunk):
131131
if split_item is not None:
132132
split_items.append(split_item)
133133

134-
logger.info(f"[split_large] → split_items={len(split_items)}")
135134
return split_items if split_items else [item]
136135
except Exception as e:
137136
logger.warning(
@@ -170,8 +169,6 @@ def _concat_multi_modal_memories(
170169
# Create a list to hold futures with their original index
171170
futures = []
172171
for idx, item in enumerate(all_memory_items):
173-
token_count = self._count_tokens(item.memory)
174-
logger.info(f"[token_count]: {token_count}, [max_tokens]: {max_tokens}")
175172
if (item.memory or "") and self._count_tokens(item.memory) > max_tokens:
176173
future = executor.submit(self._split_large_memory_item, item, max_tokens)
177174
futures.append(

0 commit comments

Comments
 (0)