J/daren
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

响应参数规范了flhf模块

Browse Source
This commit is contained in:
wanjia 2025-06-09 16:56:16 +08:00
parent a4e2a76d97
commit 63b2aebf84
1 changed files with 199 additions and 138 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

337
apps/rlhf/views.py
View File

@ -23,7 +23,75 @@ from django.db import transaction
from django.db.models.functions import TruncDate
from apps.user.authentication import CustomTokenAuthentication
class ConversationViewSet(viewsets.ModelViewSet):
# 创建统一响应格式的基类
class StandardResponseMixin:
"""标准响应格式的混合类，用于统一API响应格式"""
def get_standard_response(self, data=None, message="成功", code=200, status_code=None):
"""返回标准格式的响应"""
response_data = {
"code": code,
"message": message,
"data": data
}
return Response(response_data, status=status_code or status.HTTP_200_OK)
def create(self, request, *args, **kwargs):
serializer = self.get_serializer(data=request.data)
serializer.is_valid(raise_exception=True)
self.perform_create(serializer)
headers = self.get_success_headers(serializer.data)
return self.get_standard_response(
data=serializer.data,
message="创建成功",
code=200,
status_code=status.HTTP_201_CREATED
)
def list(self, request, *args, **kwargs):
queryset = self.filter_queryset(self.get_queryset())
page = self.paginate_queryset(queryset)
if page is not None:
serializer = self.get_serializer(page, many=True)
return self.get_paginated_response(serializer.data)
serializer = self.get_serializer(queryset, many=True)
return self.get_standard_response(data=serializer.data)
def retrieve(self, request, *args, **kwargs):
instance = self.get_object()
serializer = self.get_serializer(instance)
return self.get_standard_response(data=serializer.data)
def update(self, request, *args, **kwargs):
partial = kwargs.pop('partial', False)
instance = self.get_object()
serializer = self.get_serializer(instance, data=request.data, partial=partial)
serializer.is_valid(raise_exception=True)
self.perform_update(serializer)
return self.get_standard_response(data=serializer.data, message="更新成功")
def destroy(self, request, *args, **kwargs):
instance = self.get_object()
self.perform_destroy(instance)
return self.get_standard_response(message="删除成功", data=None)
def get_paginated_response(self, data):
"""
重写分页响应格式
"""
assert self.paginator is not None
return self.get_standard_response(
data={
'count': self.paginator.page.paginator.count,
'next': self.paginator.get_next_link(),
'previous': self.paginator.get_previous_link(),
'results': data
}
)
class ConversationViewSet(StandardResponseMixin, viewsets.ModelViewSet):
queryset = Conversation.objects.all()
serializer_class = ConversationSerializer
authentication_classes = [CustomTokenAuthentication]
@ -41,11 +109,7 @@ class ConversationViewSet(viewsets.ModelViewSet):
conversation = self.get_object()
messages = Message.objects.filter(conversation=conversation).order_by('timestamp')
serializer = MessageSerializer(messages, many=True)
return Response({
'code': 200,
'message': '成功',
'data': serializer.data
})
return self.get_standard_response(data=serializer.data)
@action(detail=True, methods=['post'])
def message(self, request, pk=None):
@ -53,11 +117,12 @@ class ConversationViewSet(viewsets.ModelViewSet):
content = request.data.get('content')
if not content:
return Response({
'code': 400,
'message': '消息内容不能为空',
'data': None
}, status=status.HTTP_400_BAD_REQUEST)
return self.get_standard_response(
code=400,
message='消息内容不能为空',
data=None,
status_code=status.HTTP_400_BAD_REQUEST
)
# 创建用户消息
user_message = Message.objects.create(
@ -87,11 +152,7 @@ class ConversationViewSet(viewsets.ModelViewSet):
MessageSerializer(ai_message).data
]
return Response({
'code': 200,
'message': '成功',
'data': messages
})
return self.get_standard_response(data=messages)
@action(detail=True, methods=['post'])
def submit(self, request, pk=None):
@ -100,11 +161,12 @@ class ConversationViewSet(viewsets.ModelViewSet):
description = request.data.get('description', '')
if conversation.is_submitted:
return Response({
'code': 400,
'message': '该对话已提交',
'data': None
}, status=status.HTTP_400_BAD_REQUEST)
return self.get_standard_response(
code=400,
message='该对话已提交',
data=None,
status_code=status.HTTP_400_BAD_REQUEST
)
# 更新对话为已提交状态
conversation.is_submitted = True
@ -130,22 +192,22 @@ class ConversationViewSet(viewsets.ModelViewSet):
details={'title': title}
)
return Response({
'code': 200,
'message': '对话提交成功',
'data': {'submission_id': submission.id}
})
return self.get_standard_response(
message='对话提交成功',
data={'submission_id': submission.id}
)
@action(detail=True, methods=['post'])
def resume(self, request, pk=None):
conversation = self.get_object()
if not conversation.is_submitted:
return Response({
'code': 400,
'message': '该对话未提交，无需恢复',
'data': None
}, status=status.HTTP_400_BAD_REQUEST)
return self.get_standard_response(
code=400,
message='该对话未提交，无需恢复',
data=None,
status_code=status.HTTP_400_BAD_REQUEST
)
# 更新对话为未提交状态
conversation.is_submitted = False
@ -168,11 +230,10 @@ class ConversationViewSet(viewsets.ModelViewSet):
target_id=str(conversation.id)
)
return Response({
'code': 200,
'message': '对话已恢复为未提交状态',
'data': None
})
return self.get_standard_response(
message='对话已恢复为未提交状态',
data=None
)
def _generate_ai_response(self, user_message, conversation):
"""
@ -219,7 +280,7 @@ class ConversationViewSet(viewsets.ModelViewSet):
stats.save()
class MessageViewSet(viewsets.ModelViewSet):
class MessageViewSet(StandardResponseMixin, viewsets.ModelViewSet):
queryset = Message.objects.all()
serializer_class = MessageSerializer
authentication_classes = [CustomTokenAuthentication]
@ -230,7 +291,7 @@ class MessageViewSet(viewsets.ModelViewSet):
return Message.objects.filter(conversation__user=user).order_by('timestamp')
class FeedbackViewSet(viewsets.ModelViewSet):
class FeedbackViewSet(StandardResponseMixin, viewsets.ModelViewSet):
queryset = Feedback.objects.all()
serializer_class = FeedbackSerializer
authentication_classes = [CustomTokenAuthentication]
@ -246,21 +307,23 @@ class FeedbackViewSet(viewsets.ModelViewSet):
feedback_value = request.data.get('feedback_value')
if not message_id or not conversation_id:
return Response({
'code': 400,
'message': '消息ID和对话ID不能为空',
'data': None
}, status=status.HTTP_400_BAD_REQUEST)
return self.get_standard_response(
code=400,
message='消息ID和对话ID不能为空',
data=None,
status_code=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({
'code': 404,
'message': '消息或对话不存在',
'data': None
}, status=status.HTTP_404_NOT_FOUND)
return self.get_standard_response(
code=404,
message='消息或对话不存在',
data=None,
status_code=status.HTTP_404_NOT_FOUND
)
# 创建或更新反馈
feedback, created = Feedback.objects.update_or_create(
@ -277,11 +340,10 @@ class FeedbackViewSet(viewsets.ModelViewSet):
# 更新用户的标注统计
self._update_annotation_stats(request.user.id, feedback_value)
return Response({
'code': 200,
'message': '反馈提交成功',
'data': FeedbackSerializer(feedback).data
})
return self.get_standard_response(
message='反馈提交成功',
data=FeedbackSerializer(feedback).data
)
def _update_annotation_stats(self, user_id, feedback_value):
"""更新用户的标注统计信息"""
@ -311,7 +373,7 @@ class FeedbackViewSet(viewsets.ModelViewSet):
stats.save()
class FeedbackTagViewSet(viewsets.ModelViewSet):
class FeedbackTagViewSet(StandardResponseMixin, viewsets.ModelViewSet):
queryset = FeedbackTag.objects.all()
serializer_class = FeedbackTagSerializer
authentication_classes = [CustomTokenAuthentication]
@ -324,7 +386,7 @@ class FeedbackTagViewSet(viewsets.ModelViewSet):
return FeedbackTag.objects.all()
class DetailedFeedbackViewSet(viewsets.ModelViewSet):
class DetailedFeedbackViewSet(StandardResponseMixin, viewsets.ModelViewSet):
queryset = DetailedFeedback.objects.all()
serializer_class = DetailedFeedbackSerializer
authentication_classes = [CustomTokenAuthentication]
@ -344,21 +406,23 @@ class DetailedFeedbackViewSet(viewsets.ModelViewSet):
is_inline = request.data.get('is_inline', True)
if not message_id or not conversation_id or not feedback_type:
return Response({
'code': 400,
'message': '消息ID、对话ID和反馈类型不能为空',
'data': None
}, status=status.HTTP_400_BAD_REQUEST)
return self.get_standard_response(
code=400,
message='消息ID、对话ID和反馈类型不能为空',
data=None,
status_code=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({
'code': 404,
'message': '消息或对话不存在',
'data': None
}, status=status.HTTP_404_NOT_FOUND)
return self.get_standard_response(
code=404,
message='消息或对话不存在',
data=None,
status_code=status.HTTP_404_NOT_FOUND
)
# 将标签列表转换为JSON字符串
if isinstance(feedback_tags, list):
@ -394,11 +458,10 @@ class DetailedFeedbackViewSet(viewsets.ModelViewSet):
# 更新用户的标注统计
self._update_annotation_stats(request.user.id, feedback_type)
return Response({
'code': 200,
'message': '详细反馈提交成功',
'data': DetailedFeedbackSerializer(detailed_feedback).data
})
return self.get_standard_response(
message='详细反馈提交成功',
data=DetailedFeedbackSerializer(detailed_feedback).data
)
def _update_annotation_stats(self, user_id, feedback_type):
"""更新用户的标注统计信息"""
@ -428,7 +491,7 @@ class DetailedFeedbackViewSet(viewsets.ModelViewSet):
stats.save()
class ConversationSubmissionViewSet(viewsets.ModelViewSet):
class ConversationSubmissionViewSet(StandardResponseMixin, viewsets.ModelViewSet):
queryset = ConversationSubmission.objects.all()
serializer_class = ConversationSubmissionSerializer
authentication_classes = [CustomTokenAuthentication]
@ -454,11 +517,12 @@ class ConversationSubmissionViewSet(viewsets.ModelViewSet):
@action(detail=True, methods=['post'])
def review(self, request, pk=None):
if request.user.role != 'admin':
return Response({
'code': 403,
'message': '只有管理员可以进行审核',
'data': None
}, status=status.HTTP_403_FORBIDDEN)
return self.get_standard_response(
code=403,
message='只有管理员可以进行审核',
data=None,
status_code=status.HTTP_403_FORBIDDEN
)
submission = self.get_object()
status_value = request.data.get('status')
@ -466,18 +530,20 @@ class ConversationSubmissionViewSet(viewsets.ModelViewSet):
reviewer_notes = request.data.get('reviewer_notes', '')
if not status_value or status_value not in ['accepted', 'rejected']:
return Response({
'code': 400,
'message': '状态值无效',
'data': None
}, status=status.HTTP_400_BAD_REQUEST)
return self.get_standard_response(
code=400,
message='状态值无效',
data=None,
status_code=status.HTTP_400_BAD_REQUEST
)
if quality_score is not None and (quality_score < 1 or quality_score > 5):
return Response({
'code': 400,
'message': '质量分数必须在1-5之间',
'data': None
}, status=status.HTTP_400_BAD_REQUEST)
return self.get_standard_response(
code=400,
message='质量分数必须在1-5之间',
data=None,
status_code=status.HTTP_400_BAD_REQUEST
)
# 更新提交状态
submission.status = status_value
@ -499,14 +565,13 @@ class ConversationSubmissionViewSet(viewsets.ModelViewSet):
}
)
return Response({
'code': 200,
'message': '审核完成',
'data': ConversationSubmissionSerializer(submission).data
})
return self.get_standard_response(
message='审核完成',
data=ConversationSubmissionSerializer(submission).data
)
class ConversationEvaluationViewSet(viewsets.ModelViewSet):
class ConversationEvaluationViewSet(StandardResponseMixin, viewsets.ModelViewSet):
queryset = ConversationEvaluation.objects.all()
serializer_class = ConversationEvaluationSerializer
authentication_classes = [CustomTokenAuthentication]
@ -523,20 +588,22 @@ class ConversationEvaluationViewSet(viewsets.ModelViewSet):
needs_satisfied = request.data.get('needs_satisfied')
if not conversation_id or not has_logical_issues or not needs_satisfied:
return Response({
'code': 400,
'message': '对话ID、逻辑问题和需求满足度不能为空',
'data': None
}, status=status.HTTP_400_BAD_REQUEST)
return self.get_standard_response(
code=400,
message='对话ID、逻辑问题和需求满足度不能为空',
data=None,
status_code=status.HTTP_400_BAD_REQUEST
)
try:
conversation = Conversation.objects.get(id=conversation_id)
except Conversation.DoesNotExist:
return Response({
'code': 404,
'message': '对话不存在',
'data': None
}, status=status.HTTP_404_NOT_FOUND)
return self.get_standard_response(
code=404,
message='对话不存在',
data=None,
status_code=status.HTTP_404_NOT_FOUND
)
# 创建或更新评估
evaluation, created = ConversationEvaluation.objects.update_or_create(
@ -563,14 +630,13 @@ class ConversationEvaluationViewSet(viewsets.ModelViewSet):
}
)
return Response({
'code': 200,
'message': '评估提交成功',
'data': ConversationEvaluationSerializer(evaluation).data
})
return self.get_standard_response(
message='评估提交成功',
data=ConversationEvaluationSerializer(evaluation).data
)
class SystemConfigViewSet(viewsets.ModelViewSet):
class SystemConfigViewSet(StandardResponseMixin, viewsets.ModelViewSet):
queryset = SystemConfig.objects.all()
serializer_class = SystemConfigSerializer
authentication_classes = [CustomTokenAuthentication]
@ -589,33 +655,31 @@ class SystemConfigViewSet(viewsets.ModelViewSet):
# 获取当前模型
model_config = SystemConfig.objects.filter(config_key='current_model').first()
if model_config:
return Response({
'code': 200,
'message': '成功',
'data': {'model': model_config.config_value}
})
return Response({
'code': 200,
'message': '成功',
'data': {'model': '默认模型'}
})
return self.get_standard_response(
data={'model': model_config.config_value}
)
return self.get_standard_response(
data={'model': '默认模型'}
)
elif request.method == 'POST':
# 设置当前模型
if request.user.role != 'admin':
return Response({
'code': 403,
'message': '只有管理员可以更改模型',
'data': None
}, status=status.HTTP_403_FORBIDDEN)
return self.get_standard_response(
code=403,
message='只有管理员可以更改模型',
data=None,
status_code=status.HTTP_403_FORBIDDEN
)
model_name = request.data.get('model')
if not model_name:
return Response({
'code': 400,
'message': '模型名称不能为空',
'data': None
}, status=status.HTTP_400_BAD_REQUEST)
return self.get_standard_response(
code=400,
message='模型名称不能为空',
data=None,
status_code=status.HTTP_400_BAD_REQUEST
)
# 更新或创建配置
config, created = SystemConfig.objects.update_or_create(
@ -629,19 +693,16 @@ class SystemConfigViewSet(viewsets.ModelViewSet):
}
)
return Response({
'code': 200,
'message': '模型设置成功',
'data': {'model': model_name}
})
return self.get_standard_response(
message='模型设置成功',
data={'model': model_name}
)
@action(detail=False, methods=['get'])
def models(self, request):
# 返回可用的模型列表
return Response({
'code': 200,
'message': '成功',
'data': {
return self.get_standard_response(
data={
'models': [
{'id': 'model1', 'name': 'GPT-3.5'},
{'id': 'model2', 'name': 'GPT-4'},
@ -650,4 +711,4 @@ class SystemConfigViewSet(viewsets.ModelViewSet):
{'id': 'model5', 'name': 'Qwen'}
]
}
})
)