fix: The conversation user is not authorized to use (#3581)

Author: shaohuzhang1

apps/application/serializers/application_chat_record.py

@@ -125,9 +125,10 @@ def reset_chat_record(chat_record, show_source, show_exec):
                     'paragraph_list') or [])
 
             if item.get('type') == 'reranker-node' and item.get('show_knowledge', False):
-                paragraph_list = paragraph_list + [rl.get('metadata') for rl in item.get('result_list') if
-                                                   'document_id' in rl.get('metadata') and 'knowledge_id' in rl.get(
-                                                       'metadata')]
+                paragraph_list = paragraph_list + [rl.get('metadata') for rl in (item.get('result_list') or []) if
+                                                   'document_id' in (rl.get('metadata') or {}) and 'knowledge_id' in (
+                                                               rl.get(
+                                                                   'metadata') or {})]
         paragraph_list = list({p.get('id'): p for p in paragraph_list}.values())
         knowledge_list = knowledge_list + [{'id': knowledge_id, **knowledge} for knowledge_id, knowledge in
                                            reduce(lambda x, y: {**x, **y},

apps/chat/api/chat_authentication_api.py

@@ -52,17 +52,4 @@ def get_parameters():
 class ChatOpenAPI(APIMixin):
     @staticmethod
     def get_parameters():
-        return [OpenApiParameter(
-            name="workspace_id",
-            description="工作空间id",
-            type=OpenApiTypes.STR,
-            location='path',
-            required=True,
-        ),
-            OpenApiParameter(
-                name="application_id",
-                description="应用id",
-                type=OpenApiTypes.STR,
-                location='path',
-                required=True,
-            )]
+        return []

apps/chat/serializers/chat.py

@@ -29,6 +29,7 @@
     ChatUserType, ApplicationChatUserStats, ApplicationAccessToken, ChatRecord, Chat, ApplicationVersion
 from application.serializers.application import ApplicationOperateSerializer
 from application.serializers.common import ChatInfo
+from common.database_model_manage.database_model_manage import DatabaseModelManage
 from common.exception.app_exception import AppApiException, AppChatNumOutOfBoundsFailed, ChatException
 from common.handle.base_to_response import BaseToResponse
 from common.handle.impl.response.openai_to_response import OpenaiToResponse
@@ -308,13 +309,23 @@ def chat_work_flow(self, chat_info: ChatInfo, instance: dict, base_to_response):
         r = work_flow_manage.run()
         return r
 
+    def is_valid_chat_user(self):
+        chat_user_id = self.data.get('chat_user_id')
+        application_id = self.data.get('application_id')
+        is_auth_chat_user = DatabaseModelManage.get_model("is_auth_chat_user")
+        if self.chat_user_type == ChatUserType.CHAT_USER.value and is_auth_chat_user:
+            is_auth = is_auth_chat_user(chat_user_id, application_id)
+            if not is_auth:
+                raise ChatException(500, _("The chat user is not authorized."))
+
     def chat(self, instance: dict, base_to_response: BaseToResponse = SystemToResponse()):
         super().is_valid(raise_exception=True)
         ChatMessageSerializers(data=instance).is_valid(raise_exception=True)
         chat_info = self.get_chat_info()
         chat_info.get_application()
         chat_info.get_chat_user()
         self.is_valid_chat_id(chat_info)
+        self.is_valid_chat_user()
         if chat_info.application.type == ApplicationTypeChoices.SIMPLE:
             self.is_valid_application_simple(raise_exception=True, chat_info=chat_info)
             return self.chat_simple(chat_info, instance, base_to_response)