Skip to content

Commit d6f6100

Browse files
Refactor(scheduler): Safely access file_ids for KB log source_doc_id
1 parent cc28316 commit d6f6100

File tree

1 file changed

+9
-4
lines changed

1 file changed

+9
-4
lines changed

src/memos/mem_scheduler/general_scheduler.py

Lines changed: 9 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -519,7 +519,8 @@ def send_add_log_messages_to_cloud_env(
519519

520520
# Process added items
521521
for item in prepared_add_items:
522-
file_ids = getattr(item.metadata, "file_ids", None)
522+
metadata = getattr(item, "metadata", None)
523+
file_ids = getattr(metadata, "file_ids", None) if metadata else None
523524
source_doc_id = file_ids[0] if isinstance(file_ids, list) and file_ids else None
524525
kb_log_content.append(
525526
{
@@ -536,7 +537,8 @@ def send_add_log_messages_to_cloud_env(
536537
# Process updated items
537538
for item_data in prepared_update_items_with_original:
538539
item = item_data["new_item"]
539-
file_ids = getattr(item.metadata, "file_ids", None)
540+
metadata = getattr(item, "metadata", None)
541+
file_ids = getattr(metadata, "file_ids", None) if metadata else None
540542
source_doc_id = file_ids[0] if isinstance(file_ids, list) and file_ids else None
541543
kb_log_content.append(
542544
{
@@ -893,8 +895,11 @@ def _process_memories_with_reader(
893895
# New: Knowledge Base Logging (Cloud Service)
894896
kb_log_content = []
895897
for item in flattened_memories:
896-
file_ids = getattr(item.metadata, "file_ids", None)
897-
source_doc_id = file_ids[0] if isinstance(file_ids, list) and file_ids else None
898+
metadata = getattr(item, "metadata", None)
899+
file_ids = getattr(metadata, "file_ids", None) if metadata else None
900+
source_doc_id = (
901+
file_ids[0] if isinstance(file_ids, list) and file_ids else None
902+
)
898903
kb_log_content.append(
899904
{
900905
"log_source": "KNOWLEDGE_BASE_LOG",

0 commit comments

Comments
 (0)