Skip to content

Commit dd84da4

Browse files
committed
fix: swagger show more args
1 parent 72db45b commit dd84da4

File tree

2 files changed

+23
-2
lines changed

2 files changed

+23
-2
lines changed

apps/application/views/chat_views.py

Lines changed: 20 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,8 +8,10 @@
88
"""
99

1010
from django.utils.translation import gettext_lazy as _
11+
from drf_yasg import openapi
1112
from drf_yasg.utils import swagger_auto_schema
1213
from rest_framework.decorators import action
14+
from rest_framework.parsers import MultiPartParser
1315
from rest_framework.request import Request
1416
from rest_framework.views import APIView
1517

@@ -448,11 +450,28 @@ def delete(self, request: Request, application_id: str, chat_id: str, chat_recor
448450

449451
class UploadFile(APIView):
450452
authentication_classes = [TokenAuth]
453+
parser_classes = [MultiPartParser]
451454

452455
@action(methods=['POST'], detail=False)
453456
@swagger_auto_schema(operation_summary=_("Upload files"),
454457
operation_id=_("Upload files"),
455-
manual_parameters=ChatRecordApi.get_request_params_api(),
458+
manual_parameters=[
459+
openapi.Parameter(name='application_id',
460+
in_=openapi.IN_PATH,
461+
type=openapi.TYPE_STRING,
462+
required=True,
463+
description=_('Application ID')),
464+
openapi.Parameter(name='chat_id',
465+
in_=openapi.IN_PATH,
466+
type=openapi.TYPE_STRING,
467+
required=True,
468+
description=_('Conversation ID')),
469+
openapi.Parameter(name='file',
470+
in_=openapi.IN_FORM,
471+
type=openapi.TYPE_FILE,
472+
required=True,
473+
description=_('Upload file'))
474+
],
456475
tags=[_("Application/Conversation Log")]
457476
)
458477
@has_permissions(

apps/dataset/serializers/file_serializers.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,9 @@ class FileSerializer(serializers.Serializer):
6262
def upload(self, with_valid=True):
6363
if with_valid:
6464
self.is_valid(raise_exception=True)
65-
meta = self.data.get('meta', {'debug': True})
65+
meta = self.data.get('meta', None)
66+
if not meta:
67+
meta = {'debug': True}
6668
file_id = meta.get('file_id', uuid.uuid1())
6769
file = File(id=file_id, file_name=self.data.get('file').name, meta=meta)
6870
file.save(self.data.get('file').read())

0 commit comments

Comments
 (0)