Skip to content

Commit d04e2e4

Browse files
liuruibinCopilot
andcommitted
fix: Enhance chat user ID extraction to include counts and filter duplicates
Co-authored-by: Copilot <copilot@github.com>
1 parent 9b11c27 commit d04e2e4

1 file changed

Lines changed: 4 additions & 2 deletions

File tree

apps/application/long_term_memory/__init__.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
import re
22
import uuid_utils.compat as uuid
3-
from django.db.models import QuerySet
3+
from django.db.models import Count, QuerySet
44
from langchain_core.messages import HumanMessage
55

66
from application.models import Chat, ChatRecord, Application, ApplicationLongTermMemory
@@ -228,8 +228,10 @@ def _execute_scheduled_extract(workspace_id, application_id):
228228
chat_user_ids = list(
229229
QuerySet(Chat).filter(application_id=application_id)
230230
.exclude(chat_user_id__isnull=True)
231+
.values('chat_user_id')
232+
.annotate(count=Count('chat_user_id'))
233+
.filter(count__gt=1)
231234
.values_list('chat_user_id', flat=True)
232-
.distinct()
233235
)
234236
for chat_user_id in chat_user_ids:
235237
config = _get_long_term_config(application, chat_user_id)

0 commit comments

Comments
 (0)