Revert "Use workers type as-is" to test internal worker

This reverts commit 0cbd8ae.

Author: Inokinoki

backend/src/main/java/ai/giskard/service/ModelService.java

@@ -44,6 +44,9 @@ public class ModelService {
 
     public MLWorkerWSRunModelForDataFrameDTO predict(ProjectModel model, Dataset dataset, Map<String, String> features) {
         MLWorkerID workerID = model.getProject().isUsingInternalWorker() ? MLWorkerID.INTERNAL : MLWorkerID.EXTERNAL;
+        if (GeneralSettingsService.IS_RUNNING_IN_DEMO_HF_SPACES) {
+            workerID = MLWorkerID.INTERNAL;
+        }
         if (mlWorkerWSService.isWorkerConnected(workerID)) {
             return getRunModelForDataFrameResponse(model, dataset, features);
         }
@@ -81,6 +84,9 @@ private MLWorkerWSRunModelForDataFrameDTO getRunModelForDataFrameResponse(Projec
         }
 
         MLWorkerID workerID = model.getProject().isUsingInternalWorker() ? MLWorkerID.INTERNAL : MLWorkerID.EXTERNAL;
+        if (GeneralSettingsService.IS_RUNNING_IN_DEMO_HF_SPACES) {
+            workerID = MLWorkerID.INTERNAL;
+        }
         // Perform the runModelForDataFrame action and parse the reply
         MLWorkerWSBaseDTO result = mlWorkerWSCommService.performAction(
             workerID,
@@ -102,6 +108,9 @@ public boolean shouldDrop(String columnDtype, String value) {
 
     public MLWorkerWSExplainDTO explain(ProjectModel model, Dataset dataset, Map<String, String> features) {
         MLWorkerID workerID = model.getProject().isUsingInternalWorker() ? MLWorkerID.INTERNAL : MLWorkerID.EXTERNAL;
+        if (GeneralSettingsService.IS_RUNNING_IN_DEMO_HF_SPACES) {
+            workerID = MLWorkerID.INTERNAL;
+        }
         if (mlWorkerWSService.isWorkerConnected(workerID)) {
             MLWorkerWSExplainParamDTO param = MLWorkerWSExplainParamDTO.builder()
                 .model(MLWorkerWSArtifactRefDTO.fromModel(model))
@@ -126,6 +135,9 @@ public MLWorkerWSExplainDTO explain(ProjectModel model, Dataset dataset, Map<Str
 
     public MLWorkerWSExplainTextDTO explainText(ProjectModel model, Dataset dataset, String featureName, Map<String, String> features) {
         MLWorkerID workerID = model.getProject().isUsingInternalWorker() ? MLWorkerID.INTERNAL : MLWorkerID.EXTERNAL;
+        if (GeneralSettingsService.IS_RUNNING_IN_DEMO_HF_SPACES) {
+            workerID = MLWorkerID.INTERNAL;
+        }
         if (mlWorkerWSService.isWorkerConnected(workerID)) {
             MLWorkerWSExplainTextParamDTO param = MLWorkerWSExplainTextParamDTO.builder()
                 .model(MLWorkerWSArtifactRefDTO.fromModel(model))
@@ -174,6 +186,9 @@ public Inspection createInspection(String name, UUID modelId, UUID datasetId, bo
 
     protected void predictSerializedDataset(ProjectModel model, Dataset dataset, Long inspectionId, boolean sample) {
         MLWorkerID workerID = model.getProject().isUsingInternalWorker() ? MLWorkerID.INTERNAL : MLWorkerID.EXTERNAL;
+        if (GeneralSettingsService.IS_RUNNING_IN_DEMO_HF_SPACES) {
+            workerID = MLWorkerID.INTERNAL;
+        }
         if (mlWorkerWSService.isWorkerConnected(workerID)) {
             // Initialize params
             MLWorkerWSRunModelParamDTO param = MLWorkerWSRunModelParamDTO.builder()

backend/src/main/java/ai/giskard/service/TestSuiteExecutionService.java

@@ -57,6 +57,9 @@ public void executeScheduledTestSuite(TestSuiteExecution execution,
             }));
 
         MLWorkerID workerID = suite.getProject().isUsingInternalWorker() ? MLWorkerID.INTERNAL : MLWorkerID.EXTERNAL;
+        if (GeneralSettingsService.IS_RUNNING_IN_DEMO_HF_SPACES) {
+            workerID = MLWorkerID.INTERNAL;
+        }
         if (mlWorkerWSService.isWorkerConnected(workerID)) {
             Map<String, FunctionInput> suiteInputsAndShared = Stream.concat(
                 execution.getInputs().stream(),

backend/src/main/java/ai/giskard/service/TestSuiteService.java

@@ -202,6 +202,9 @@ public Long generateTestSuite(String projectKey, GenerateTestSuiteDTO dto) {
 
         Project project = projectRepository.getOneByKey(projectKey);
         MLWorkerID workerID = project.isUsingInternalWorker() ? MLWorkerID.INTERNAL : MLWorkerID.EXTERNAL;
+        if (GeneralSettingsService.IS_RUNNING_IN_DEMO_HF_SPACES) {
+            workerID = MLWorkerID.INTERNAL;
+        }
         if (mlWorkerWSService.isWorkerConnected(workerID)) {
             MLWorkerWSGenerateTestSuiteParamDTO param = MLWorkerWSGenerateTestSuiteParamDTO.builder()
                     .projectKey(projectKey)

backend/src/main/java/ai/giskard/web/rest/controllers/DatasetsController.java

@@ -139,6 +139,9 @@ public DatasetProcessingResultDTO datasetProcessing(@PathVariable("projectId") @
             .collect(Collectors.toMap(Callable::getUuid, Function.identity(), (l, r) -> l));
 
         MLWorkerID workerID = project.isUsingInternalWorker() ? MLWorkerID.INTERNAL : MLWorkerID.EXTERNAL;
+        if (GeneralSettingsService.IS_RUNNING_IN_DEMO_HF_SPACES) {
+            workerID = MLWorkerID.INTERNAL;
+        }
         if (!mlWorkerWSService.isWorkerConnected(workerID)) {
             throw new MLWorkerNotConnectedException(workerID, log);
         }

backend/src/main/java/ai/giskard/web/rest/controllers/MLWorkerController.java

@@ -5,6 +5,7 @@
 import ai.giskard.ml.dto.MLWorkerWSBaseDTO;
 import ai.giskard.ml.dto.MLWorkerWSGetInfoDTO;
 import ai.giskard.ml.dto.MLWorkerWSGetInfoParamDTO;
+import ai.giskard.service.GeneralSettingsService;
 import ai.giskard.service.ml.MLWorkerService;
 import ai.giskard.service.ml.MLWorkerWSCommService;
 import ai.giskard.service.ml.MLWorkerWSService;
@@ -71,6 +72,9 @@ public List<MLWorkerInfoDTO> getMLWorkerInfo()
     @PostMapping("/stop")
     public void stopWorker(boolean internal) {
         MLWorkerID workerID = internal ? MLWorkerID.INTERNAL : MLWorkerID.EXTERNAL;
+        if (GeneralSettingsService.IS_RUNNING_IN_DEMO_HF_SPACES) {
+            workerID = MLWorkerID.INTERNAL;
+        }
         if (mlWorkerWSService.isWorkerConnected(workerID)) {
             mlWorkerWSCommService.performAction(workerID, MLWorkerWSAction.STOP_WORKER, null);
         }

backend/src/main/java/ai/giskard/web/rest/controllers/testing/TestController.java

@@ -14,6 +14,7 @@
 import ai.giskard.ml.dto.MLWorkerWSRunAdHocTestParamDTO;
 import ai.giskard.repository.ProjectRepository;
 import ai.giskard.repository.ml.TestFunctionRepository;
+import ai.giskard.service.GeneralSettingsService;
 import ai.giskard.service.TestArgumentService;
 import ai.giskard.service.ml.MLWorkerWSCommService;
 import ai.giskard.service.ml.MLWorkerWSService;
@@ -50,6 +51,9 @@ public TestTemplateExecutionResultDTO runAdHocTest(@RequestBody RunAdhocTestRequ
         Project project = projectRepository.getMandatoryById(request.getProjectId());
 
         MLWorkerID workerID = project.isUsingInternalWorker() ? MLWorkerID.INTERNAL : MLWorkerID.EXTERNAL;
+        if (GeneralSettingsService.IS_RUNNING_IN_DEMO_HF_SPACES) {
+            workerID = MLWorkerID.INTERNAL;
+        }
         if (mlWorkerWSService.isWorkerConnected(workerID)) {
             MLWorkerWSRunAdHocTestParamDTO.MLWorkerWSRunAdHocTestParamDTOBuilder builder =
                 MLWorkerWSRunAdHocTestParamDTO.builder()

backend/src/main/java/ai/giskard/web/socket/WorkerStatusSocketService.java

@@ -2,6 +2,7 @@
 
 import ai.giskard.event.UpdateWorkerStatusEvent;
 import ai.giskard.ml.MLWorkerID;
+import ai.giskard.service.GeneralSettingsService;
 import ai.giskard.service.ml.MLWorkerWSService;
 import lombok.RequiredArgsConstructor;
 import org.springframework.context.event.EventListener;
@@ -33,7 +34,8 @@ public void handleWorkerStatusChangeEvent(UpdateWorkerStatusEvent event) {
 
 
     public void sendCurrentStatus() {
-        MLWorkerID workerID = MLWorkerID.EXTERNAL;
+        MLWorkerID workerID = GeneralSettingsService.IS_RUNNING_IN_DEMO_HF_SPACES ?
+            MLWorkerID.INTERNAL : MLWorkerID.EXTERNAL;
         Map<String, Boolean> data = new HashMap<>();
         data.put("connected", mlWorkerWSService.isWorkerConnected(workerID));
         sendData(data);