feat: add support for cancel api

jira: cq-1123
risk: low

Author: fiedlr

gooddata-sdk/gooddata_sdk/compute/service.py

@@ -3,6 +3,7 @@
 
 import logging
 
+from gooddata_api_client.model.afm_cancel_tokens import AfmCancelTokens
 from gooddata_api_client.model.chat_history_request import ChatHistoryRequest
 from gooddata_api_client.model.chat_history_result import ChatHistoryResult
 from gooddata_api_client.model.chat_request import ChatRequest
@@ -108,3 +109,22 @@ def ai_chat_history_reset(self, workspace_id: str) -> None:
         """
         chat_history_request = ChatHistoryRequest(reset=True)
         self._actions_api.ai_chat_history(workspace_id, chat_history_request, _check_return_type=False)
+
+    def cancel_executions(self, executions: list[Execution]) -> None:
+        """
+        Try to cancel given executions using the cancel api endpoint.
+
+        Args:
+            executions: list of executions to cancel
+        """
+        workspace_to_tokens: dict[str, AfmCancelTokens] = {}
+
+        for execution in executions:
+            if not workspace_to_tokens[execution.workspace_id]:
+                workspace_to_tokens[execution.workspace_id] = set()
+
+            if execution.cancel_token:
+                workspace_to_tokens[execution.workspace_id].token_ids.add(execution.cancel_token)
+
+        for workspace_id, token_ids in workspace_to_tokens.items():
+            self._actions_api.cancel_executions(workspace_id, AfmCancelTokens(list(token_ids)))