vikarti.anatra/Skyvern
1
0
Fork 0
mirror of https://github.com/Skyvern-AI/skyvern.git synced 2025-09-17 02:39:40 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

use skyvern temp to save tempfiles (#1262)

Browse source
This commit is contained in:
LawyZheng 2024-11-26 12:27:58 +08:00 • committed by GitHub
parent 6b417d0e83
commit af25aeed77
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
7 changed files with 50 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

4
skyvern/forge/sdk/workflow/models/block.py
View file

@ -11,7 +11,6 @@ from dataclasses import dataclass
from email.message import EmailMessage
from enum import StrEnum
from pathlib import Path
from tempfile import NamedTemporaryFile
from typing import Annotated, Any, Literal, Union
import filetype
@ -36,6 +35,7 @@ from skyvern.forge.prompts import prompt_engine
from skyvern.forge.sdk.api.aws import AsyncAWSClient
from skyvern.forge.sdk.api.files import (
calculate_sha256_for_file,
create_named_temporary_file,
download_file,
download_from_s3,
get_path_for_workflow_download_directory,
@ -1056,7 +1056,7 @@ class SendEmailBlock(Block):
async def _download_from_s3(self, s3_uri: str) -> str:
client = self.get_async_aws_client()
downloaded_bytes = await client.download_file(uri=s3_uri)
file_path = NamedTemporaryFile(delete=False)
file_path = create_named_temporary_file(delete=False)
file_path.write(downloaded_bytes)
return file_path.name