diff --git a/python/dify_plugin/integration/entities.py b/python/dify_plugin/integration/entities.py
index 4aece6a0..54580f42 100644
--- a/python/dify_plugin/integration/entities.py
+++ b/python/dify_plugin/integration/entities.py
@@ -6,10 +6,12 @@
 
 from dify_plugin.core.entities.plugin.request import (
     AgentActions,
+    DatasourceActions,
     EndpointActions,
     ModelActions,
     PluginInvokeType,
     ToolActions,
+    TriggerActions,
 )
 
 T = TypeVar("T", bound=BaseModel)
@@ -18,7 +20,7 @@
 class PluginInvokeRequest(BaseModel, Generic[T]):
     invoke_id: str
     type: PluginInvokeType
-    action: AgentActions | ToolActions | ModelActions | EndpointActions
+    action: AgentActions | ToolActions | ModelActions | EndpointActions | DatasourceActions | TriggerActions
     request: T