pengjunjie/ragflow
1
0
Fork 1
mirror of https://github.com/infiniflow/ragflow.git synced 2026-01-02 02:25:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix: pick message to delete (#12295)

Browse Source 
### What problem does this PR solve?

Pick unforgotten message when not found forgotten message to delete.

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
This commit is contained in:
Lynn
2025-12-29 17:10:46 +08:00
committed by GitHub
parent c2c079886f
commit d142b9095e
1 changed files with 9 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
memory/services/messages.py
View File

@ -194,19 +194,18 @@ class MessageService:
select_fields = ["message_id", "content", "content_embed"] select_fields = ["message_id", "content", "content_embed"]
_index_name = index_name(uid) _index_name = index_name(uid)
res = settings.msgStoreConn.get_forgotten_messages(select_fields, _index_name, memory_id) res = settings.msgStoreConn.get_forgotten_messages(select_fields, _index_name, memory_id)
if not res:
return []
message_list = settings.msgStoreConn.get_fields(res, select_fields)
current_size = 0 current_size = 0
ids_to_remove = [] ids_to_remove = []
for message in message_list.values(): if res:
if current_size < size_to_delete: message_list = settings.msgStoreConn.get_fields(res, select_fields)
current_size += cls.calculate_message_size(message) for message in message_list.values():
ids_to_remove.append(message["message_id"]) if current_size < size_to_delete:
else: current_size += cls.calculate_message_size(message)
ids_to_remove.append(message["message_id"])
else:
return ids_to_remove, current_size
if current_size >= size_to_delete:
return ids_to_remove, current_size return ids_to_remove, current_size
if current_size >= size_to_delete:
return ids_to_remove, current_size
order_by = OrderByExpr() order_by = OrderByExpr()
order_by.asc("valid_at") order_by.asc("valid_at")