diff --git a/apps/enp/views.py b/apps/enp/views.py
index 86432a0c..f862c60f 100644
--- a/apps/enp/views.py
+++ b/apps/enp/views.py
@@ -37,7 +37,7 @@ class DrainViewSet(CustomModelViewSet):
 
     @swagger_auto_schema(
         manual_parameters=[
-            openapi.Parameter(name="equip_data", in_=openapi.IN_QUERY, description="Include equip data in the response", type=openapi.TYPE_STRING, enum=["yes", "no"], required=False),
+            openapi.Parameter(name="equip_data", in_=openapi.IN_QUERY, description="Include equip data in the response", type=openapi.TYPE_STRING, enum=["no", "base", "enp"], required=False),
             openapi.Parameter(name="query", in_=openapi.IN_QUERY, description="定制返回数据", type=openapi.TYPE_STRING, required=False),
         ]
     )