diff --git a/modal/image.py b/modal/image.py index 39b462fb0..27c2a3fc5 100644 --- a/modal/image.py +++ b/modal/image.py @@ -1679,7 +1679,9 @@ def run_inside(self): async def logs(self) -> AsyncGenerator[str, None]: last_entry_id: Optional[str] = None - request = api_pb2.ImageJoinStreamingRequest(image_id=self._object_id, timeout=55, last_entry_id=last_entry_id) + request = api_pb2.ImageJoinStreamingRequest( + image_id=self._object_id, timeout=55, last_entry_id=last_entry_id, include_logs_for_finished=True + ) async for response in unary_stream(self._client.stub.ImageJoinStreaming, request): if response.result.status: return diff --git a/modal_proto/api.proto b/modal_proto/api.proto index 2fc0a6020..44ccf773e 100644 --- a/modal_proto/api.proto +++ b/modal_proto/api.proto @@ -1359,6 +1359,7 @@ message ImageJoinStreamingRequest { string image_id = 1; float timeout = 2; string last_entry_id = 3; + bool include_logs_for_finished = 4; } message ImageJoinStreamingResponse {