diff --git a/app/server/appsmith-server/src/main/java/com/appsmith/server/services/ce/NewActionServiceCEImpl.java b/app/server/appsmith-server/src/main/java/com/appsmith/server/services/ce/NewActionServiceCEImpl.java index f4f020dca3..00447d4735 100644 --- a/app/server/appsmith-server/src/main/java/com/appsmith/server/services/ce/NewActionServiceCEImpl.java +++ b/app/server/appsmith-server/src/main/java/com/appsmith/server/services/ce/NewActionServiceCEImpl.java @@ -682,7 +682,7 @@ public class NewActionServiceCEImpl extends BaseService getCachedDatasourceForActionExecution(Mono actionDTOMono) { + protected Mono getCachedDatasourceForActionExecution(Mono actionDTOMono, String environmentName) { return actionDTOMono .flatMap(actionDTO -> datasourceService.getValidDatasourceFromActionMono(actionDTO, @@ -976,7 +976,7 @@ public class NewActionServiceCEImpl extends BaseService actionDTOMono = getCachedActionDTOForActionExecution(actionMono, executeActionDTO, actionId); // 3. Instantiate the implementation class based on the query type - Mono datasourceMono = getCachedDatasourceForActionExecution(actionDTOMono); + Mono datasourceMono = getCachedDatasourceForActionExecution(actionDTOMono, environmentName); Mono pluginMono = getCachedPluginForActionExecution(datasourceMono, actionId); Mono pluginExecutorMono = pluginExecutorHelper.getPluginExecutor(pluginMono);