10 minute timeout for dockerfile build

beam-cloud · Jan 3, 2025 · 3eef95d · 3eef95d
1 parent 8664934
commit 3eef95d
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/pkg/abstractions/image/build.go b/pkg/abstractions/image/build.go
@@ -210,14 +210,16 @@ func (i *BaseImage) String() string {
 // Build user image
 func (b *Builder) Build(ctx context.Context, opts *BuildOpts, outputChan chan common.OutputMsg) error {
 	var (
-		dockerfile  *string
-		authInfo, _ = auth.AuthInfoFromContext(ctx)
+		dockerfile             *string
+		authInfo, _            = auth.AuthInfoFromContext(ctx)
+		containerSpinupTimeout = defaultContainerSpinupTimeout
 	)
 
 	switch {
 	case opts.Dockerfile != "":
 		opts.addPythonRequirements()
 		dockerfile = &opts.Dockerfile
+		containerSpinupTimeout = 600 * time.Second
 	case opts.ExistingImageUri != "":
 		err := b.handleCustomBaseImage(opts, outputChan)
 		if err != nil {
@@ -329,7 +331,7 @@ func (b *Builder) Build(ctx context.Context, opts *BuildOpts, outputChan chan co
 			return errors.New(fmt.Sprintf("container exited with error: %s\n", msg))
 		}
 
-		if time.Since(start) > defaultContainerSpinupTimeout && opts.Dockerfile == "" {
+		if time.Since(start) > containerSpinupTimeout {
 			outputChan <- common.OutputMsg{Done: true, Success: false, Msg: "Timeout: container not running after 180 seconds.\n"}
 			return errors.New("timeout: container not running after 180 seconds")
 		}