diff --git a/apps/chat/views/v2/chat_record.py b/apps/chat/views/v2/chat_record.py index db67eced891..38456f977ef 100644 --- a/apps/chat/views/v2/chat_record.py +++ b/apps/chat/views/v2/chat_record.py @@ -1,22 +1,32 @@ # coding=utf-8 """ - @project: MaxKB - @Author：虎虎 - @file： chat_record.py - @date：2025/6/23 10:42 - @desc: +@project: MaxKB +@Author：虎虎 +@file： chat_record.py +@date：2025/6/23 10:42 +@desc: """ + from django.utils.translation import gettext_lazy as _ from drf_spectacular.utils import extend_schema from rest_framework.request import Request from rest_framework.views import APIView from application.serializers.application_chat_record import ChatRecordOperateSerializer -from chat.api.chat_api import HistoricalConversationAPI, PageHistoricalConversationAPI, \ - PageHistoricalConversationRecordAPI, HistoricalConversationRecordAPI, HistoricalConversationOperateAPI +from chat.api.chat_api import ( + HistoricalConversationAPI, + PageHistoricalConversationAPI, + PageHistoricalConversationRecordAPI, + HistoricalConversationRecordAPI, + HistoricalConversationOperateAPI, +) from chat.api.vote_api import VoteAPI -from chat.serializers.chat_record import VoteSerializer, HistoricalConversationSerializer, \ - HistoricalConversationRecordSerializer, HistoricalConversationOperateSerializer +from chat.serializers.chat_record import ( + VoteSerializer, + HistoricalConversationSerializer, + HistoricalConversationRecordSerializer, + HistoricalConversationOperateSerializer, +) from common import result from common.auth import ChatTokenAuth @@ -25,177 +35,205 @@ class VoteView(APIView): authentication_classes = [ChatTokenAuth] @extend_schema( - methods=['PUT'], + methods=["PUT"], description=_("Like, Dislike"), summary=_("Like, Dislike"), operation_id=_("Like, Dislike"), # type: ignore parameters=VoteAPI.get_parameters(), request=VoteAPI.get_request(), responses=VoteAPI.get_response(), - tags=[_('Chat')] # type: ignore + tags=[_("Chat")], # type: ignore ) def put(self, request: Request, chat_id: str, chat_record_id: str): - return result.success(VoteSerializer( - data={'application_id': request.user.kwargs.get('application_id'), - 'chat_id': chat_id, - 'chat_record_id': chat_record_id - }).vote(request.data)) + return result.success( + VoteSerializer( + data={ + "application_id": request.user.kwargs.get("application_id"), + "chat_id": chat_id, + "chat_record_id": chat_record_id, + } + ).vote(request.data) + ) class HistoricalConversationView(APIView): authentication_classes = [ChatTokenAuth] @extend_schema( - methods=['GET'], + methods=["GET"], description=_("Get historical conversation"), summary=_("Get historical conversation"), operation_id=_("Get historical conversation"), # type: ignore parameters=HistoricalConversationAPI.get_parameters(), responses=HistoricalConversationAPI.get_response(), - tags=[_('Chat')] # type: ignore + tags=[_("Chat")], # type: ignore ) def get(self, request: Request): - return result.success(HistoricalConversationSerializer( - data={ - 'application_id': request.user.kwargs.get('application_id'), - 'chat_user_id': request.user.id, - }).list()) + return result.success( + HistoricalConversationSerializer( + data={ + "application_id": request.user.kwargs.get("application_id"), + "chat_user_id": request.user.id, + } + ).list() + ) class Operate(APIView): authentication_classes = [ChatTokenAuth] @extend_schema( - methods=['PUT'], + methods=["PUT"], description=_("Modify conversation about"), summary=_("Modify conversation about"), operation_id=_("Modify conversation about"), # type: ignore parameters=HistoricalConversationOperateAPI.get_parameters(), request=HistoricalConversationOperateAPI.get_request(), responses=HistoricalConversationOperateAPI.get_response(), - tags=[_('Chat')] # type: ignore + tags=[_("Chat")], # type: ignore ) def put(self, request: Request, chat_id: str): - return result.success(HistoricalConversationOperateSerializer( - data={ - 'application_id': request.user.kwargs.get('application_id'), - 'chat_user_id': request.user.id, - 'chat_id': chat_id, - }).edit_abstract(request.data) - ) + return result.success( + HistoricalConversationOperateSerializer( + data={ + "application_id": request.user.kwargs.get("application_id"), + "chat_user_id": request.user.id, + "chat_id": chat_id, + } + ).edit_abstract(request.data) + ) @extend_schema( - methods=['DELETE'], + methods=["DELETE"], description=_("Delete history conversation"), summary=_("Delete history conversation"), operation_id=_("Delete history conversation"), # type: ignore parameters=HistoricalConversationOperateAPI.get_parameters(), responses=HistoricalConversationOperateAPI.get_response(), - tags=[_('Chat')] # type: ignore + tags=[_("Chat")], # type: ignore ) def delete(self, request: Request, chat_id: str): - return result.success(HistoricalConversationOperateSerializer( - data={ - 'application_id': request.user.kwargs.get('application_id'), - 'chat_user_id': request.user.id, - 'chat_id': chat_id, - }).logic_delete()) + return result.success( + HistoricalConversationOperateSerializer( + data={ + "application_id": request.user.kwargs.get("application_id"), + "chat_user_id": request.user.id, + "chat_id": chat_id, + } + ).logic_delete() + ) class BatchDelete(APIView): authentication_classes = [ChatTokenAuth] @extend_schema( - methods=['DELETE'], + methods=["DELETE"], description=_("Batch delete history conversation"), summary=_("Batch delete history conversation"), operation_id=_("Batch delete history conversation"), # type: ignore - parameters=HistoricalConversationOperateAPI.get_parameters(), + parameters=[], responses=HistoricalConversationOperateAPI.get_response(), - tags=[_('Chat')] # type: ignore + tags=[_("Chat")], # type: ignore ) def delete(self, request: Request): - return result.success(HistoricalConversationOperateSerializer.Clear(data={ - 'application_id': request.user.kwargs.get('application_id'), - 'chat_user_id': request.user.id, - }).batch_logic_delete()) + return result.success( + HistoricalConversationOperateSerializer.Clear( + data={ + "application_id": request.user.kwargs.get("application_id"), + "chat_user_id": request.user.id, + } + ).batch_logic_delete() + ) class PageView(APIView): authentication_classes = [ChatTokenAuth] @extend_schema( - methods=['GET'], + methods=["GET"], description=_("Get historical conversation by page"), summary=_("Get historical conversation by page"), operation_id=_("Get historical conversation by page"), # type: ignore parameters=PageHistoricalConversationAPI.get_parameters(), responses=PageHistoricalConversationAPI.get_response(), - tags=[_('Chat')] # type: ignore + tags=[_("Chat")], # type: ignore ) def get(self, request: Request, current_page: int, page_size: int): - return result.success(HistoricalConversationSerializer( - data={ - 'application_id': request.user.kwargs.get('application_id'), - 'chat_user_id': request.user.id, - }).page(current_page, page_size)) + return result.success( + HistoricalConversationSerializer( + data={ + "application_id": request.user.kwargs.get("application_id"), + "chat_user_id": request.user.id, + } + ).page(current_page, page_size) + ) class HistoricalConversationRecordView(APIView): authentication_classes = [ChatTokenAuth] @extend_schema( - methods=['GET'], + methods=["GET"], description=_("Get historical conversation records"), summary=_("Get historical conversation records"), operation_id=_("Get historical conversation records"), # type: ignore parameters=HistoricalConversationRecordAPI.get_parameters(), responses=HistoricalConversationRecordAPI.get_response(), - tags=[_('Chat')] # type: ignore + tags=[_("Chat")], # type: ignore ) def get(self, request: Request, chat_id: str): - return result.success(HistoricalConversationRecordSerializer( - data={ - 'chat_id': chat_id, - 'application_id': request.user.kwargs.get('application_id'), - 'chat_user_id': request.user.id, - }).list()) + return result.success( + HistoricalConversationRecordSerializer( + data={ + "chat_id": chat_id, + "application_id": request.user.kwargs.get("application_id"), + "chat_user_id": request.user.id, + } + ).list() + ) class PageView(APIView): authentication_classes = [ChatTokenAuth] @extend_schema( - methods=['GET'], + methods=["GET"], description=_("Get historical conversation records by page "), summary=_("Get historical conversation records by page"), operation_id=_("Get historical conversation records by page"), # type: ignore parameters=PageHistoricalConversationRecordAPI.get_parameters(), responses=PageHistoricalConversationRecordAPI.get_response(), - tags=[_('Chat')] # type: ignore + tags=[_("Chat")], # type: ignore ) def get(self, request: Request, chat_id: str, current_page: int, page_size: int): - return result.success(HistoricalConversationRecordSerializer( - data={ - 'chat_id': chat_id, - 'application_id': request.user.kwargs.get('application_id'), - 'chat_user_id': request.user.id, - }).page(current_page, page_size)) + return result.success( + HistoricalConversationRecordSerializer( + data={ + "chat_id": chat_id, + "application_id": request.user.kwargs.get("application_id"), + "chat_user_id": request.user.id, + } + ).page(current_page, page_size) + ) class ChatRecordView(APIView): authentication_classes = [ChatTokenAuth] @extend_schema( - methods=['GET'], + methods=["GET"], description=_("Get conversation details"), summary=_("Get conversation details"), operation_id=_("Get conversation details"), # type: ignore parameters=PageHistoricalConversationRecordAPI.get_parameters(), responses=PageHistoricalConversationRecordAPI.get_response(), - tags=[_('Chat')] # type: ignore + tags=[_("Chat")], # type: ignore ) def get(self, request: Request, chat_id: str, chat_record_id: str): - return result.success(ChatRecordOperateSerializer( - data={ - 'chat_id': chat_id, - 'chat_record_id': chat_record_id, - 'application_id': request.user.kwargs.get('application_id'), - 'chat_user_id': request.user.id, - }).one(False)) + return result.success( + ChatRecordOperateSerializer( + data={ + "chat_id": chat_id, + "chat_record_id": chat_record_id, + "application_id": request.user.kwargs.get("application_id"), + "chat_user_id": request.user.id, + } + ).one(False) + ) diff --git a/apps/chat/views/v3/chat_record.py b/apps/chat/views/v3/chat_record.py index 15c0e1794c0..38539992d66 100644 --- a/apps/chat/views/v3/chat_record.py +++ b/apps/chat/views/v3/chat_record.py @@ -1,22 +1,32 @@ # coding=utf-8 """ - @project: MaxKB - @Author：虎虎 - @file： chat_record.py - @date：2025/6/23 10:42 - @desc: v3 chat record views —— application_id 从 path 获取，用户身份从 request.user(Principal) 获取 +@project: MaxKB +@Author：虎虎 +@file： chat_record.py +@date：2025/6/23 10:42 +@desc: v3 chat record views —— application_id 从 path 获取，用户身份从 request.user(Principal) 获取 """ + from django.utils.translation import gettext_lazy as _ from drf_spectacular.utils import extend_schema from rest_framework.request import Request from rest_framework.views import APIView from application.serializers.application_chat_record import ChatRecordOperateSerializer -from chat.api.chat_api import HistoricalConversationAPI, PageHistoricalConversationAPI, \ - PageHistoricalConversationRecordAPI, HistoricalConversationRecordAPI, HistoricalConversationOperateAPI +from chat.api.chat_api import ( + HistoricalConversationAPI, + PageHistoricalConversationAPI, + PageHistoricalConversationRecordAPI, + HistoricalConversationRecordAPI, + HistoricalConversationOperateAPI, +) from chat.api.vote_api import VoteAPI -from chat.serializers.chat_record import VoteSerializer, HistoricalConversationSerializer, \ - HistoricalConversationRecordSerializer, HistoricalConversationOperateSerializer +from chat.serializers.chat_record import ( + VoteSerializer, + HistoricalConversationSerializer, + HistoricalConversationRecordSerializer, + HistoricalConversationOperateSerializer, +) from common import result from common.auth import ChatTokenAuth from common.auth.authentication import has_permissions @@ -27,186 +37,210 @@ class VoteView(APIView): authentication_classes = [ChatTokenAuth] @extend_schema( - methods=['PUT'], + methods=["PUT"], description=_("Like, Dislike"), summary=_("Like, Dislike"), operation_id=_("V3 Like, Dislike"), # type: ignore parameters=VoteAPI.get_parameters(), request=VoteAPI.get_request(), responses=VoteAPI.get_response(), - tags=[_('V3 Chat')] # type: ignore + tags=[_("V3 Chat")], # type: ignore ) @has_permissions(ChatPermissionConstants.get_aggregate_permissions()) def put(self, request: Request, application_id: str, chat_id: str, chat_record_id: str): - return result.success(VoteSerializer( - data={'application_id': application_id, - 'chat_id': chat_id, - 'chat_record_id': chat_record_id - }).vote(request.data)) + return result.success( + VoteSerializer( + data={"application_id": application_id, "chat_id": chat_id, "chat_record_id": chat_record_id} + ).vote(request.data) + ) class HistoricalConversationView(APIView): authentication_classes = [ChatTokenAuth] @extend_schema( - methods=['GET'], + methods=["GET"], description=_("Get historical conversation"), summary=_("Get historical conversation"), operation_id=_("V3 Get historical conversation"), # type: ignore parameters=HistoricalConversationAPI.get_parameters(), responses=HistoricalConversationAPI.get_response(), - tags=[_('V3 Chat')] # type: ignore + tags=[_("V3 Chat")], # type: ignore ) @has_permissions(ChatPermissionConstants.get_aggregate_permissions()) def get(self, request: Request, application_id: str): - return result.success(HistoricalConversationSerializer( - data={ - 'application_id': application_id, - 'chat_user_id': request.user.id, - }).list()) + return result.success( + HistoricalConversationSerializer( + data={ + "application_id": application_id, + "chat_user_id": request.user.id, + } + ).list() + ) class Operate(APIView): authentication_classes = [ChatTokenAuth] @extend_schema( - methods=['PUT'], + methods=["PUT"], description=_("Modify conversation about"), summary=_("Modify conversation about"), operation_id=_("V3 Modify conversation about"), # type: ignore parameters=HistoricalConversationOperateAPI.get_parameters(), request=HistoricalConversationOperateAPI.get_request(), responses=HistoricalConversationOperateAPI.get_response(), - tags=[_('V3 Chat')] # type: ignore + tags=[_("V3 Chat")], # type: ignore ) @has_permissions(ChatPermissionConstants.get_aggregate_permissions()) def put(self, request: Request, application_id: str, chat_id: str): - return result.success(HistoricalConversationOperateSerializer( - data={ - 'application_id': application_id, - 'chat_user_id': request.user.id, - 'chat_id': chat_id, - }).edit_abstract(request.data) - ) + return result.success( + HistoricalConversationOperateSerializer( + data={ + "application_id": application_id, + "chat_user_id": request.user.id, + "chat_id": chat_id, + } + ).edit_abstract(request.data) + ) @extend_schema( - methods=['DELETE'], + methods=["DELETE"], description=_("Delete history conversation"), summary=_("Delete history conversation"), operation_id=_("V3 Delete history conversation"), # type: ignore parameters=HistoricalConversationOperateAPI.get_parameters(), responses=HistoricalConversationOperateAPI.get_response(), - tags=[_('V3 Chat')] # type: ignore + tags=[_("V3 Chat")], # type: ignore ) @has_permissions(ChatPermissionConstants.get_aggregate_permissions()) def delete(self, request: Request, application_id: str, chat_id: str): - return result.success(HistoricalConversationOperateSerializer( - data={ - 'application_id': application_id, - 'chat_user_id': request.user.id, - 'chat_id': chat_id, - }).logic_delete()) + return result.success( + HistoricalConversationOperateSerializer( + data={ + "application_id": application_id, + "chat_user_id": request.user.id, + "chat_id": chat_id, + } + ).logic_delete() + ) class BatchDelete(APIView): authentication_classes = [ChatTokenAuth] @extend_schema( - methods=['DELETE'], + methods=["DELETE"], description=_("Batch delete history conversation"), summary=_("Batch delete history conversation"), operation_id=_("V3 Batch delete history conversation"), # type: ignore - parameters=HistoricalConversationOperateAPI.get_parameters(), + parameters=[], responses=HistoricalConversationOperateAPI.get_response(), - tags=[_('V3 Chat')] # type: ignore + tags=[_("V3 Chat")], # type: ignore ) @has_permissions(ChatPermissionConstants.get_aggregate_permissions()) def delete(self, request: Request, application_id: str): - return result.success(HistoricalConversationOperateSerializer.Clear(data={ - 'application_id': application_id, - 'chat_user_id': request.user.id, - }).batch_logic_delete()) + return result.success( + HistoricalConversationOperateSerializer.Clear( + data={ + "application_id": application_id, + "chat_user_id": request.user.id, + } + ).batch_logic_delete() + ) class PageView(APIView): authentication_classes = [ChatTokenAuth] @extend_schema( - methods=['GET'], + methods=["GET"], description=_("Get historical conversation by page"), summary=_("Get historical conversation by page"), operation_id=_("V3 Get historical conversation by page"), # type: ignore parameters=PageHistoricalConversationAPI.get_parameters(), responses=PageHistoricalConversationAPI.get_response(), - tags=[_('V3 Chat')] # type: ignore + tags=[_("V3 Chat")], # type: ignore ) @has_permissions(ChatPermissionConstants.get_aggregate_permissions()) def get(self, request: Request, application_id: str, current_page: int, page_size: int): - return result.success(HistoricalConversationSerializer( - data={ - 'application_id': application_id, - 'chat_user_id': request.user.id, - }).page(current_page, page_size)) + return result.success( + HistoricalConversationSerializer( + data={ + "application_id": application_id, + "chat_user_id": request.user.id, + } + ).page(current_page, page_size) + ) class HistoricalConversationRecordView(APIView): authentication_classes = [ChatTokenAuth] @extend_schema( - methods=['GET'], + methods=["GET"], description=_("Get historical conversation records"), summary=_("Get historical conversation records"), operation_id=_("V3 Get historical conversation records"), # type: ignore parameters=HistoricalConversationRecordAPI.get_parameters(), responses=HistoricalConversationRecordAPI.get_response(), - tags=[_('V3 Chat')] # type: ignore + tags=[_("V3 Chat")], # type: ignore ) @has_permissions(ChatPermissionConstants.get_aggregate_permissions()) def get(self, request: Request, application_id: str, chat_id: str): - return result.success(HistoricalConversationRecordSerializer( - data={ - 'chat_id': chat_id, - 'application_id': application_id, - 'chat_user_id': request.user.id, - }).list()) + return result.success( + HistoricalConversationRecordSerializer( + data={ + "chat_id": chat_id, + "application_id": application_id, + "chat_user_id": request.user.id, + } + ).list() + ) class PageView(APIView): authentication_classes = [ChatTokenAuth] @extend_schema( - methods=['GET'], + methods=["GET"], description=_("Get historical conversation records by page "), summary=_("Get historical conversation records by page"), operation_id=_("V3 Get historical conversation records by page"), # type: ignore parameters=PageHistoricalConversationRecordAPI.get_parameters(), responses=PageHistoricalConversationRecordAPI.get_response(), - tags=[_('V3 Chat')] # type: ignore + tags=[_("V3 Chat")], # type: ignore ) @has_permissions(ChatPermissionConstants.get_aggregate_permissions()) def get(self, request: Request, application_id: str, chat_id: str, current_page: int, page_size: int): - return result.success(HistoricalConversationRecordSerializer( - data={ - 'chat_id': chat_id, - 'application_id': application_id, - 'chat_user_id': request.user.id, - }).page(current_page, page_size)) + return result.success( + HistoricalConversationRecordSerializer( + data={ + "chat_id": chat_id, + "application_id": application_id, + "chat_user_id": request.user.id, + } + ).page(current_page, page_size) + ) class ChatRecordView(APIView): authentication_classes = [ChatTokenAuth] @extend_schema( - methods=['GET'], + methods=["GET"], description=_("Get conversation details"), summary=_("Get conversation details"), operation_id=_("V3 Get conversation details"), # type: ignore parameters=PageHistoricalConversationRecordAPI.get_parameters(), responses=PageHistoricalConversationRecordAPI.get_response(), - tags=[_('V3 Chat')] # type: ignore + tags=[_("V3 Chat")], # type: ignore ) @has_permissions(ChatPermissionConstants.get_aggregate_permissions()) def get(self, request: Request, application_id: str, chat_id: str, chat_record_id: str): - return result.success(ChatRecordOperateSerializer( - data={ - 'chat_id': chat_id, - 'chat_record_id': chat_record_id, - 'application_id': application_id, - 'chat_user_id': request.user.id, - }).one(False)) + return result.success( + ChatRecordOperateSerializer( + data={ + "chat_id": chat_id, + "chat_record_id": chat_record_id, + "application_id": application_id, + "chat_user_id": request.user.id, + } + ).one(False) + )