Skip to content

Commit 690ae92

Browse files
committed
fix: The number of conversations has not been updated
1 parent 246123c commit 690ae92

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

apps/application/serializers/common.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
from application.chat_pipeline.step.chat_step.i_chat_step import PostResponseHandler
1717
from application.models import Application, ChatRecord, Chat, ApplicationVersion, ChatUserType, ApplicationTypeChoices, \
1818
ApplicationKnowledgeMapping
19+
from application.serializers.application_chat import ChatCountSerializer
1920
from common.constants.cache_version import Cache_Version
2021
from common.database_model_manage.database_model_manage import DatabaseModelManage
2122
from common.exception.app_exception import ChatException
@@ -93,8 +94,8 @@ def get_application(self):
9394
def get_chat_user(self, asker=None):
9495
if self.chat_user:
9596
return self.chat_user
96-
if self.chat_user_type == ChatUserType.CHAT_USER.value:
97-
chat_user_model = DatabaseModelManage.get_model("chat_user")
97+
chat_user_model = DatabaseModelManage.get_model("chat_user")
98+
if self.chat_user_type == ChatUserType.CHAT_USER.value and chat_user_model:
9899
chat_user = QuerySet(chat_user_model).filter(id=self.chat_user_id).first()
99100
return {
100101
'id': chat_user.id,
@@ -184,6 +185,7 @@ def append_chat_record(self, chat_record: ChatRecord):
184185
QuerySet(Chat).filter(id=self.chat_id).update(update_time=datetime.now())
185186
# 插入会话记录
186187
chat_record.save()
188+
ChatCountSerializer(data={'chat_id': self.chat_id}).update_chat()
187189

188190
def set_cache(self):
189191
cache.set(Cache_Version.CHAT.get_key(key=self.chat_id), self, version=Cache_Version.CHAT.get_version(),

0 commit comments

Comments
 (0)