feat: RND-72: Add vendor extensions support in openapi autoschema

label_studio/core/settings/base.py

@@ -345,6 +345,13 @@
     },
 }
 
+# specify the list of the extensions that are allowed to be presented in auto generated OpenAPI schema
+# for example, by specifying in swagger_auto_schema(..., x_fern_sdk_group_name='projects') we can group endpoints
+# /api/projects/:
+#   get:
+#     x-fern-sdk-group-name: projects
+X_VENDOR_OPENAPI_EXTENSIONS = ['x-fern']
+
 # Swagger: automatic API documentation
 SWAGGER_SETTINGS = {
     'SECURITY_DEFINITIONS': {
@@ -362,6 +369,7 @@
     'APIS_SORTER': 'alpha',
     'SUPPORTED_SUBMIT_METHODS': ['get', 'post', 'put', 'delete', 'patch'],
     'OPERATIONS_SORTER': 'alpha',
+    'DEFAULT_AUTO_SCHEMA_CLASS': 'core.utils.openapi_extensions.XVendorExtensionsAutoSchema',
 }
 
 SENTRY_DSN = get_env('SENTRY_DSN', None)

label_studio/core/utils/openapi_extensions.py

@@ -0,0 +1,13 @@
+from django.conf import settings
+from drf_yasg.inspectors import SwaggerAutoSchema
+
+
+class XVendorExtensionsAutoSchema(SwaggerAutoSchema):
+    allowed_extensions = tuple([e.replace('-', '_') for e in settings.X_VENDOR_OPENAPI_EXTENSIONS])
+
+    def get_operation(self, operation_keys=None):
+        operation = super(XVendorExtensionsAutoSchema, self).get_operation(operation_keys)
+        for key, value in self.overrides.items():
+            if key.startswith(self.allowed_extensions):
+                operation[key.replace('_', '-')] = value
+        return operation

label_studio/data_import/api.py

@@ -111,6 +111,8 @@
     name='post',
     decorator=swagger_auto_schema(
         tags=['Import'],
+        x_fern_sdk_group_name='projects',
+        x_fern_sdk_method_name='import_tasks',
         responses=task_create_response_scheme,
         manual_parameters=[
             openapi.Parameter(
@@ -490,6 +492,8 @@ def post(self, *args, **kwargs):
     name='get',
     decorator=swagger_auto_schema(
         tags=['Import'],
+        x_fern_sdk_group_name='files',
+        x_fern_sdk_method_name='list',
         operation_summary='Get files list',
         manual_parameters=[
             openapi.Parameter(
@@ -515,6 +519,8 @@ def post(self, *args, **kwargs):
     name='delete',
     decorator=swagger_auto_schema(
         tags=['Import'],
+        x_fern_sdk_group_name='files',
+        x_fern_sdk_method_name='delete_many',
         operation_summary='Delete files',
         operation_description="""
         Delete uploaded files for a specific project.
@@ -561,6 +567,8 @@ def delete(self, request, *args, **kwargs):
     name='get',
     decorator=swagger_auto_schema(
         tags=['Import'],
+        x_fern_sdk_group_name='files',
+        x_fern_sdk_method_name='get',
         operation_summary='Get file upload',
         operation_description='Retrieve details about a specific uploaded file.',
     ),
@@ -569,6 +577,8 @@ def delete(self, request, *args, **kwargs):
     name='patch',
     decorator=swagger_auto_schema(
         tags=['Import'],
+        x_fern_sdk_group_name='files',
+        x_fern_sdk_method_name='update',
         operation_summary='Update file upload',
         operation_description='Update a specific uploaded file.',
         request_body=FileUploadSerializer,
@@ -578,6 +588,8 @@ def delete(self, request, *args, **kwargs):
     name='delete',
     decorator=swagger_auto_schema(
         tags=['Import'],
+        x_fern_sdk_group_name='files',
+        x_fern_sdk_method_name='delete',
         operation_summary='Delete file upload',
         operation_description='Delete a specific uploaded file.',
     ),

label_studio/ml/api.py

@@ -188,11 +188,10 @@ def perform_update(self, serializer):
             },
         ),
         responses={
-            200: openapi.Response(title='Training OK', description='Training has successfully started.'),
+            200: openapi.Response(description='Training has successfully started.'),
             500: openapi.Response(
                 description='Training error',
                 schema=openapi.Schema(
-                    title='Error message',
                     description='Error message',
                     type=openapi.TYPE_STRING,
                     example='Server responded with an error.',
@@ -230,11 +229,10 @@ def post(self, request, *args, **kwargs):
             ),
         ],
         responses={
-            200: openapi.Response(title='Predicting OK', description='Predicting has successfully started.'),
+            200: openapi.Response(description='Predicting has successfully started.'),
             500: openapi.Response(
                 description='Predicting error',
                 schema=openapi.Schema(
-                    title='Error message',
                     description='Error message',
                     type=openapi.TYPE_STRING,
                     example='Server responded with an error.',
@@ -287,7 +285,7 @@ def post(self, request, *args, **kwargs):
         ],
         request_body=MLInteractiveAnnotatingRequest,
         responses={
-            200: openapi.Response(title='Annotating OK', description='Interactive annotation has succeeded.'),
+            200: openapi.Response(description='Interactive annotation has succeeded.'),
         },
     ),
 )

label_studio/tasks/api.py

@@ -241,6 +241,8 @@ def put(self, request, *args, **kwargs):
         tags=['Annotations'],
         operation_summary='Get annotation by its ID',
         operation_description='Retrieve a specific annotation for a task using the annotation result ID.',
+        x_fern_sdk_group_name='annotations',
+        x_fern_sdk_method_name='get',
     ),
 )
 @method_decorator(

label_studio/users/api.py

@@ -205,8 +205,11 @@ def post(self, request, *args, **kwargs):
         responses={
             200: openapi.Response(
                 description='User token response',
-                type=openapi.TYPE_OBJECT,
-                properties={'detail': openapi.Schema(description='Token', type=openapi.TYPE_STRING)},
+                schema=openapi.Schema(
+                    description='User token',
+                    type=openapi.TYPE_OBJECT,
+                    properties={'detail': openapi.Schema(description='Token', type=openapi.TYPE_STRING)},
+                ),
             )
         },
     ),