响应参数规范flhf模块

apps/rlhf/views.py

@@ -41,7 +41,11 @@ class ConversationViewSet(viewsets.ModelViewSet):
         conversation = self.get_object()
         messages = Message.objects.filter(conversation=conversation).order_by('timestamp')
         serializer = MessageSerializer(messages, many=True)
-        return Response(serializer.data)
+        return Response({
+            'code': 200,
+            'message': '成功',
+            'data': serializer.data
+        })
     
     @action(detail=True, methods=['post'])
     def message(self, request, pk=None):
@@ -49,7 +53,11 @@ class ConversationViewSet(viewsets.ModelViewSet):
         content = request.data.get('content')
         
         if not content:
-            return Response({'error': '消息内容不能为空'}, status=status.HTTP_400_BAD_REQUEST)
+            return Response({
+                'code': 400,
+                'message': '消息内容不能为空',
+                'data': None
+            }, status=status.HTTP_400_BAD_REQUEST)
         
         # 创建用户消息
         user_message = Message.objects.create(
@@ -79,7 +87,11 @@ class ConversationViewSet(viewsets.ModelViewSet):
             MessageSerializer(ai_message).data
         ]
         
-        return Response(messages)
+        return Response({
+            'code': 200,
+            'message': '成功',
+            'data': messages
+        })
     
     @action(detail=True, methods=['post'])
     def submit(self, request, pk=None):
@@ -88,7 +100,11 @@ class ConversationViewSet(viewsets.ModelViewSet):
         description = request.data.get('description', '')
         
         if conversation.is_submitted:
-            return Response({'error': '该对话已提交'}, status=status.HTTP_400_BAD_REQUEST)
+            return Response({
+                'code': 400,
+                'message': '该对话已提交',
+                'data': None
+            }, status=status.HTTP_400_BAD_REQUEST)
         
         # 更新对话为已提交状态
         conversation.is_submitted = True
@@ -115,8 +131,9 @@ class ConversationViewSet(viewsets.ModelViewSet):
         )
         
         return Response({
+            'code': 200,
             'message': '对话提交成功',
-            'submission_id': submission.id
+            'data': {'submission_id': submission.id}
         })
     
     @action(detail=True, methods=['post'])
@@ -124,7 +141,11 @@ class ConversationViewSet(viewsets.ModelViewSet):
         conversation = self.get_object()
         
         if not conversation.is_submitted:
-            return Response({'error': '该对话未提交，无需恢复'}, status=status.HTTP_400_BAD_REQUEST)
+            return Response({
+                'code': 400,
+                'message': '该对话未提交，无需恢复',
+                'data': None
+            }, status=status.HTTP_400_BAD_REQUEST)
         
         # 更新对话为未提交状态
         conversation.is_submitted = False
@@ -147,7 +168,11 @@ class ConversationViewSet(viewsets.ModelViewSet):
             target_id=str(conversation.id)
         )
         
-        return Response({'message': '对话已恢复为未提交状态'})
+        return Response({
+            'code': 200,
+            'message': '对话已恢复为未提交状态',
+            'data': None
+        })
     
     def _generate_ai_response(self, user_message, conversation):
         """
@@ -221,13 +246,21 @@ class FeedbackViewSet(viewsets.ModelViewSet):
         feedback_value = request.data.get('feedback_value')
         
         if not message_id or not conversation_id:
-            return Response({'error': '消息ID和对话ID不能为空'}, status=status.HTTP_400_BAD_REQUEST)
+            return Response({
+                'code': 400,
+                'message': '消息ID和对话ID不能为空',
+                'data': None
+            }, status=status.HTTP_400_BAD_REQUEST)
         
         try:
             message = Message.objects.get(id=message_id)
             conversation = Conversation.objects.get(id=conversation_id)
         except (Message.DoesNotExist, Conversation.DoesNotExist):
-            return Response({'error': '消息或对话不存在'}, status=status.HTTP_404_NOT_FOUND)
+            return Response({
+                'code': 404,
+                'message': '消息或对话不存在',
+                'data': None
+            }, status=status.HTTP_404_NOT_FOUND)
         
         # 创建或更新反馈
         feedback, created = Feedback.objects.update_or_create(
@@ -244,7 +277,11 @@ class FeedbackViewSet(viewsets.ModelViewSet):
         # 更新用户的标注统计
         self._update_annotation_stats(request.user.id, feedback_value)
         
-        return Response(FeedbackSerializer(feedback).data)
+        return Response({
+            'code': 200,
+            'message': '反馈提交成功',
+            'data': FeedbackSerializer(feedback).data
+        })
     
     def _update_annotation_stats(self, user_id, feedback_value):
         """更新用户的标注统计信息"""
@@ -308,14 +345,20 @@ class DetailedFeedbackViewSet(viewsets.ModelViewSet):
         
         if not message_id or not conversation_id or not feedback_type:
             return Response({
-                'error': '消息ID、对话ID和反馈类型不能为空'
+                'code': 400,
+                'message': '消息ID、对话ID和反馈类型不能为空',
+                'data': None
             }, status=status.HTTP_400_BAD_REQUEST)
         
         try:
             message = Message.objects.get(id=message_id)
             conversation = Conversation.objects.get(id=conversation_id)
         except (Message.DoesNotExist, Conversation.DoesNotExist):
-            return Response({'error': '消息或对话不存在'}, status=status.HTTP_404_NOT_FOUND)
+            return Response({
+                'code': 404,
+                'message': '消息或对话不存在',
+                'data': None
+            }, status=status.HTTP_404_NOT_FOUND)
         
         # 将标签列表转换为JSON字符串
         if isinstance(feedback_tags, list):
@@ -351,7 +394,11 @@ class DetailedFeedbackViewSet(viewsets.ModelViewSet):
         # 更新用户的标注统计
         self._update_annotation_stats(request.user.id, feedback_type)
         
-        return Response(DetailedFeedbackSerializer(detailed_feedback).data)
+        return Response({
+            'code': 200,
+            'message': '详细反馈提交成功',
+            'data': DetailedFeedbackSerializer(detailed_feedback).data
+        })
     
     def _update_annotation_stats(self, user_id, feedback_type):
         """更新用户的标注统计信息"""
@@ -407,7 +454,11 @@ class ConversationSubmissionViewSet(viewsets.ModelViewSet):
     @action(detail=True, methods=['post'])
     def review(self, request, pk=None):
         if request.user.role != 'admin':
-            return Response({'error': '只有管理员可以进行审核'}, status=status.HTTP_403_FORBIDDEN)
+            return Response({
+                'code': 403,
+                'message': '只有管理员可以进行审核',
+                'data': None
+            }, status=status.HTTP_403_FORBIDDEN)
         
         submission = self.get_object()
         status_value = request.data.get('status')
@@ -415,10 +466,18 @@ class ConversationSubmissionViewSet(viewsets.ModelViewSet):
         reviewer_notes = request.data.get('reviewer_notes', '')
         
         if not status_value or status_value not in ['accepted', 'rejected']:
-            return Response({'error': '状态值无效'}, status=status.HTTP_400_BAD_REQUEST)
+            return Response({
+                'code': 400,
+                'message': '状态值无效',
+                'data': None
+            }, status=status.HTTP_400_BAD_REQUEST)
         
         if quality_score is not None and (quality_score < 1 or quality_score > 5):
-            return Response({'error': '质量分数必须在1-5之间'}, status=status.HTTP_400_BAD_REQUEST)
+            return Response({
+                'code': 400,
+                'message': '质量分数必须在1-5之间',
+                'data': None
+            }, status=status.HTTP_400_BAD_REQUEST)
         
         # 更新提交状态
         submission.status = status_value
@@ -441,8 +500,9 @@ class ConversationSubmissionViewSet(viewsets.ModelViewSet):
         )
         
         return Response({
+            'code': 200,
             'message': '审核完成',
-            'submission': ConversationSubmissionSerializer(submission).data
+            'data': ConversationSubmissionSerializer(submission).data
         })
 
 
@@ -464,13 +524,19 @@ class ConversationEvaluationViewSet(viewsets.ModelViewSet):
         
         if not conversation_id or not has_logical_issues or not needs_satisfied:
             return Response({
-                'error': '对话ID、逻辑问题和需求满足度不能为空'
+                'code': 400,
+                'message': '对话ID、逻辑问题和需求满足度不能为空',
+                'data': None
             }, status=status.HTTP_400_BAD_REQUEST)
         
         try:
             conversation = Conversation.objects.get(id=conversation_id)
         except Conversation.DoesNotExist:
-            return Response({'error': '对话不存在'}, status=status.HTTP_404_NOT_FOUND)
+            return Response({
+                'code': 404,
+                'message': '对话不存在',
+                'data': None
+            }, status=status.HTTP_404_NOT_FOUND)
         
         # 创建或更新评估
         evaluation, created = ConversationEvaluation.objects.update_or_create(
@@ -497,7 +563,11 @@ class ConversationEvaluationViewSet(viewsets.ModelViewSet):
             }
         )
         
-        return Response(ConversationEvaluationSerializer(evaluation).data)
+        return Response({
+            'code': 200,
+            'message': '评估提交成功',
+            'data': ConversationEvaluationSerializer(evaluation).data
+        })
 
 
 class SystemConfigViewSet(viewsets.ModelViewSet):
@@ -519,17 +589,33 @@ class SystemConfigViewSet(viewsets.ModelViewSet):
             # 获取当前模型
             model_config = SystemConfig.objects.filter(config_key='current_model').first()
             if model_config:
-                return Response({'model': model_config.config_value})
+                return Response({
-            return Response({'model': '默认模型'})
+                    'code': 200,
+                    'message': '成功',
+                    'data': {'model': model_config.config_value}
+                })
+            return Response({
+                'code': 200,
+                'message': '成功',
+                'data': {'model': '默认模型'}
+            })
         
         elif request.method == 'POST':
             # 设置当前模型
             if request.user.role != 'admin':
-                return Response({'error': '只有管理员可以更改模型'}, status=status.HTTP_403_FORBIDDEN)
+                return Response({
+                    'code': 403,
+                    'message': '只有管理员可以更改模型',
+                    'data': None
+                }, status=status.HTTP_403_FORBIDDEN)
             
             model_name = request.data.get('model')
             if not model_name:
-                return Response({'error': '模型名称不能为空'}, status=status.HTTP_400_BAD_REQUEST)
+                return Response({
+                    'code': 400,
+                    'message': '模型名称不能为空',
+                    'data': None
+                }, status=status.HTTP_400_BAD_REQUEST)
             
             # 更新或创建配置
             config, created = SystemConfig.objects.update_or_create(
@@ -543,17 +629,25 @@ class SystemConfigViewSet(viewsets.ModelViewSet):
                 }
             )
             
-            return Response({'model': model_name})
+            return Response({
+                'code': 200,
+                'message': '模型设置成功',
+                'data': {'model': model_name}
+            })
     
     @action(detail=False, methods=['get'])
     def models(self, request):
         # 返回可用的模型列表
         return Response({
-            'models': [
+            'code': 200,
-                {'id': 'model1', 'name': 'GPT-3.5'},
+            'message': '成功',
-                {'id': 'model2', 'name': 'GPT-4'},
+            'data': {
-                {'id': 'model3', 'name': 'Claude'},
+                'models': [
-                {'id': 'model4', 'name': 'LLaMA'},
+                    {'id': 'model1', 'name': 'GPT-3.5'},
-                {'id': 'model5', 'name': 'Qwen'}
+                    {'id': 'model2', 'name': 'GPT-4'},
-            ]
+                    {'id': 'model3', 'name': 'Claude'},
+                    {'id': 'model4', 'name': 'LLaMA'},
+                    {'id': 'model5', 'name': 'Qwen'}
+                ]
+            }
         })