From be2b24a1559a52f7eb0e272ec19db757274d6d60 Mon Sep 17 00:00:00 2001
From: Abhimanyu Saharan <desk.abhimanyu@gmail.com>
Date: Fri, 25 Aug 2023 14:46:51 +0530
Subject: [PATCH] fixes the swagger schema for token provisioning #13557

---
 netbox/users/api/views.py | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/netbox/users/api/views.py b/netbox/users/api/views.py
index 04b3ae336..8be1f85a2 100644
--- a/netbox/users/api/views.py
+++ b/netbox/users/api/views.py
@@ -70,7 +70,13 @@ class TokenProvisionView(APIView):
     """
     permission_classes = []
 
-    # @extend_schema(methods=["post"], responses={201: serializers.TokenSerializer})
+    @extend_schema(
+        request=serializers.TokenProvisionSerializer,
+        responses={
+            201: serializers.TokenSerializer,
+            401: OpenApiTypes.OBJECT,
+        }
+    )
     def post(self, request):
         serializer = serializers.TokenProvisionSerializer(data=request.data)
         serializer.is_valid()