From 4728708c184479c70bb97a6eb87c3ec6b3b6eb00 Mon Sep 17 00:00:00 2001
From: Elias Freider <elias@modal.com>
Date: Tue, 19 Nov 2024 09:57:12 +0100
Subject: [PATCH] Fix for older proto generators not having ValueType

---
 modal/io_streams.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/modal/io_streams.py b/modal/io_streams.py
index d939a5167..59d2f8008 100644
--- a/modal/io_streams.py
+++ b/modal/io_streams.py
@@ -30,7 +30,7 @@
 
 
 async def _sandbox_logs_iterator(
-    sandbox_id: str, file_descriptor: api_pb2.FileDescriptor.ValueType, last_entry_id: str, client: _Client
+    sandbox_id: str, file_descriptor: "api_pb2.FileDescriptor.ValueType", last_entry_id: str, client: _Client
 ) -> AsyncGenerator[Tuple[Optional[bytes], str], None]:
     req = api_pb2.SandboxGetLogsRequest(
         sandbox_id=sandbox_id,
@@ -49,7 +49,7 @@ async def _sandbox_logs_iterator(
 
 
 async def _container_process_logs_iterator(
-    process_id: str, file_descriptor: api_pb2.FileDescriptor.ValueType, client: _Client
+    process_id: str, file_descriptor: "api_pb2.FileDescriptor.ValueType", client: _Client
 ) -> AsyncGenerator[Optional[bytes], None]:
     req = api_pb2.ContainerExecGetOutputRequest(
         exec_id=process_id, timeout=55, file_descriptor=file_descriptor, get_raw_bytes=True
@@ -90,7 +90,7 @@ class _StreamReader(Generic[T]):
 
     def __init__(
         self,
-        file_descriptor: api_pb2.FileDescriptor.ValueType,
+        file_descriptor: "api_pb2.FileDescriptor.ValueType",
         object_id: str,
         object_type: Literal["sandbox", "container_process"],
         client: _Client,