remove WriteData type in python-sdk

e2b-dev · Dec 13, 2024 · 724fbc6 · 724fbc6
1 parent 502c414
commit 724fbc6
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 13 deletions.
diff --git a/packages/python-sdk/e2b/sandbox/filesystem/filesystem.py b/packages/python-sdk/e2b/sandbox/filesystem/filesystem.py
@@ -47,14 +47,11 @@ class EntryInfo:
     """
 
 
-WriteData = Union[str, bytes, IO]
-
-
-@dataclass
+dataclass
 class WriteEntry:
     """
     Contains path and data of the file to be written to the filesystem.
     """
 
     path: str
-    data: WriteData
+    data: Union[str, bytes, IO]
diff --git a/packages/python-sdk/e2b/sandbox_async/filesystem/filesystem.py b/packages/python-sdk/e2b/sandbox_async/filesystem/filesystem.py
@@ -1,6 +1,6 @@
 from io import TextIOBase
-from typing import AsyncIterator, List, Literal, Optional, overload, Union
-from e2b.sandbox.filesystem.filesystem import WriteData, WriteEntry
+from typing import AsyncIterator, IO, List, Literal, Optional, overload, Union
+from e2b.sandbox.filesystem.filesystem import WriteEntry
 
 import e2b_connect as connect
 import httpcore
@@ -134,7 +134,7 @@ async def read(
     async def write(
         self,
         path: str,
-        data: WriteData,
+        data: Union[str, bytes, IO],
         user: Username = "user",
         request_timeout: Optional[float] = None,
     ) -> EntryInfo:
@@ -174,7 +174,7 @@ async def write(
     async def write(
         self,
         path_or_files: Union[str, List[WriteEntry]],
-        data_or_user: Union[WriteData, Username] = "user",
+        data_or_user: Union[str, bytes, IO, Username] = "user",
         user_or_request_timeout: Optional[Union[float, Username]] = None,
         request_timeout_or_none: Optional[float] = None
     ) -> Union[EntryInfo, List[EntryInfo]]:

diff --git a/packages/python-sdk/e2b/sandbox_sync/filesystem/filesystem.py b/packages/python-sdk/e2b/sandbox_sync/filesystem/filesystem.py
@@ -1,6 +1,6 @@
 from io import TextIOBase
-from typing import Iterator, List, Literal, Optional, overload, Union
-from e2b.sandbox.filesystem.filesystem import WriteData, WriteEntry
+from typing import IO, Iterator, List, Literal, Optional, overload, Union
+from e2b.sandbox.filesystem.filesystem import WriteEntry
 
 import e2b_connect
 import httpcore
@@ -131,7 +131,7 @@ def read(
     def write(
         self,
         path: str,
-        data: WriteData,
+        data: Union[str, bytes, IO],
         user: Username = "user",
         request_timeout: Optional[float] = None,
     ) -> EntryInfo:
@@ -174,7 +174,7 @@ def write(
     def write(
         self,
         path_or_files: Union[str, List[WriteEntry]],
-        data_or_user: Union[WriteData, Username] = "user",
+        data_or_user: Union[str, bytes, IO, Username] = "user",
         user_or_request_timeout: Optional[Union[float, Username]] = None,
         request_timeout_or_none: Optional[float] = None
     ) -> Union[EntryInfo, List[EntryInfo]]: