Implement SendEmailBlock

pyproject.toml

@@ -48,6 +48,7 @@ typer = "^0.9.0"
 types-toml = "^0.10.8.7"
 apscheduler = "^3.10.4"
 httpx = "^0.27.0"
+filetype = "^1.2.0"
 
 
 [tool.poetry.group.dev.dependencies]

skyvern/exceptions.py

@@ -179,3 +179,9 @@ def __init__(self, error_message: str) -> None:
 class WorkflowRunContextNotInitialized(SkyvernException):
     def __init__(self, workflow_run_id: str) -> None:
         super().__init__(f"WorkflowRunContext not initialized for workflow run {workflow_run_id}")
+
+
+class DownloadFileMaxSizeExceeded(SkyvernException):
+    def __init__(self, max_size: int) -> None:
+        self.max_size = max_size
+        super().__init__(f"Download file size exceeded the maximum allowed size of {max_size} MB.")

skyvern/forge/sdk/api/files.py

@@ -3,37 +3,54 @@
 import zipfile
 from urllib.parse import urlparse
 
-import requests
+import aiohttp
 import structlog
 
+from skyvern.exceptions import DownloadFileMaxSizeExceeded
+
 LOG = structlog.get_logger()
 
 
-def download_file(url: str) -> str | None:
-    # Send an HTTP request to the URL of the file, stream=True to prevent loading the content at once into memory
-    r = requests.get(url, stream=True)
+async def download_file(url: str, max_size_mb: int | None = None) -> str:
+    try:
+        async with aiohttp.ClientSession(raise_for_status=True) as session:
+            LOG.info("Starting to download file")
+            async with session.get(url) as response:
+                # Check the content length if available
+                if max_size_mb and response.content_length and response.content_length > max_size_mb * 1024 * 1024:
+                    # todo: move to root exception.py
+                    raise DownloadFileMaxSizeExceeded(max_size_mb)
+
+                # Parse the URL
+                a = urlparse(url)
 
-    # Check if the request is successful
-    if r.status_code == 200:
-        # Parse the URL
-        a = urlparse(url)
+                # Get the file name
+                temp_dir = tempfile.mkdtemp(prefix="skyvern_downloads_")
 
-        # Get the file name
-        temp_dir = tempfile.mkdtemp(prefix="skyvern_downloads_")
+                file_name = os.path.basename(a.path)
+                file_path = os.path.join(temp_dir, file_name)
 
-        file_name = os.path.basename(a.path)
-        file_path = os.path.join(temp_dir, file_name)
+                LOG.info(f"Downloading file to {file_path}")
+                with open(file_path, "wb") as f:
+                    # Write the content of the request into the file
+                    total_bytes_downloaded = 0
+                    async for chunk in response.content.iter_chunked(1024):
+                        f.write(chunk)
+                        total_bytes_downloaded += len(chunk)
+                        if max_size_mb and total_bytes_downloaded > max_size_mb * 1024 * 1024:
+                            raise DownloadFileMaxSizeExceeded(max_size_mb)
 
-        LOG.info(f"Downloading file to {file_path}")
-        with open(file_path, "wb") as f:
-            # Write the content of the request into the file
-            for chunk in r.iter_content(1024):
-                f.write(chunk)
-        LOG.info(f"File downloaded successfully to {file_path}")
-        return file_path
-    else:
-        LOG.error(f"Failed to download file, status code: {r.status_code}")
-        return None
+                LOG.info(f"File downloaded successfully to {file_path}")
+                return file_path
+    except aiohttp.ClientResponseError as e:
+        LOG.error(f"Failed to download file, status code: {e.status}")
+        raise
+    except DownloadFileMaxSizeExceeded as e:
+        LOG.error(f"Failed to download file, max size exceeded: {e.max_size}")
+        raise
+    except Exception:
+        LOG.exception("Failed to download file")
+        raise
 
 
 def zip_files(files_path: str, zip_file_path: str) -> str:

skyvern/forge/sdk/workflow/exceptions.py

@@ -31,6 +31,6 @@ def __init__(self, duplicate_keys: set[str]) -> None:
         )
 
 
-class DownloadFileMaxSizeExceeded(BaseWorkflowException):
-    def __init__(self, max_size: int) -> None:
-        super().__init__(f"Download file size exceeded the maximum allowed size of {max_size} MB.")
+class InvalidEmailClientConfiguration(BaseWorkflowException):
+    def __init__(self, problems: list[str]) -> None:
+        super().__init__(f"Email client configuration is invalid. These parameters are missing or invalid: {problems}")