fix: Type errors in sandbox_files.py

- Convert bytearray to bytes for ExtractedFile.content
- Wrap data_uri with MediaFileType for store_media_file
This commit is contained in:
Nick Tindle
2026-02-11 11:54:14 -06:00
parent 2f2a031b2c
commit 9c4c29b096
2 changed files with 8 additions and 2 deletions

View File

@@ -20,7 +20,10 @@ from backend.data.model import (
SchemaField,
)
from backend.integrations.providers import ProviderName
from backend.util.sandbox_files import SandboxFileOutput, extract_and_store_sandbox_files
from backend.util.sandbox_files import (
SandboxFileOutput,
extract_and_store_sandbox_files,
)
if TYPE_CHECKING:
from backend.executor.utils import ExecutionContext

View File

@@ -13,6 +13,7 @@ from typing import TYPE_CHECKING
from pydantic import BaseModel
from backend.util.file import store_media_file
from backend.util.type import MediaFileType
if TYPE_CHECKING:
from e2b import AsyncSandbox as BaseAsyncSandbox
@@ -160,6 +161,8 @@ async def extract_sandbox_files(
content = await sandbox.files.read(file_path, format="bytes")
if isinstance(content, str):
content = content.encode("utf-8")
elif isinstance(content, bytearray):
content = bytes(content)
# Extract filename from path
file_name = file_path.split("/")[-1]
@@ -229,7 +232,7 @@ async def store_sandbox_files(
)
result = await store_media_file(
file=data_uri,
file=MediaFileType(data_uri),
execution_context=execution_context,
return_format="for_block_output",
)