diff --git a/apps/trigger/api/trigger.py b/apps/trigger/api/trigger.py
index a854322e91d..1117ebb7feb 100644
--- a/apps/trigger/api/trigger.py
+++ b/apps/trigger/api/trigger.py
@@ -12,6 +12,7 @@
 
 from common.mixins.api_mixin import APIMixin
 from knowledge.serializers.common import BatchSerializer
+from trigger.serializers.task_source_trigger import TaskSourceTriggerEditRequest
 from trigger.serializers.trigger import TriggerCreateRequest, TriggerResponse, BatchActiveSerializer
 
 
@@ -196,4 +197,8 @@ def get_parameters():
                 location='path',
                 required=True,
             ),
-        ]
\ No newline at end of file
+        ]
+
+    @staticmethod
+    def get_request():
+        return TaskSourceTriggerEditRequest