Implement sandbox exec through command router - Go

modal-go/client.go

@@ -275,6 +275,7 @@ type retryCallOption struct {
 const (
 	apiEndpoint            = "api.modal.com:443"
 	maxMessageSize         = 100 * 1024 * 1024 // 100 MB
+	windowSize             = 64 * 1024 * 1024  // 64 MiB
 	defaultRetryAttempts   = 3
 	defaultRetryBaseDelay  = 100 * time.Millisecond
 	defaultRetryMaxDelay   = 1 * time.Second
@@ -334,6 +335,8 @@ func newClient(ctx context.Context, profile Profile, c *Client, customUnaryInter
 	conn, err := grpc.NewClient(
 		target,
 		grpc.WithTransportCredentials(creds),
+		grpc.WithInitialWindowSize(windowSize),
+		grpc.WithInitialConnWindowSize(windowSize),
 		grpc.WithDefaultCallOptions(
 			grpc.MaxCallRecvMsgSize(maxMessageSize),
 			grpc.MaxCallSendMsgSize(maxMessageSize),

modal-go/config.go

@@ -8,29 +8,32 @@ import (
 	"fmt"
 	"os"
 	"path/filepath"
+	"strings"
 
 	"github.com/pelletier/go-toml/v2"
 )
 
 // Profile holds a fully-resolved configuration ready for use by the client.
 type Profile struct {
-	ServerURL           string
-	TokenID             string
-	TokenSecret         string
-	Environment         string
-	ImageBuilderVersion string
-	LogLevel            string
+	ServerURL                 string
+	TokenID                   string
+	TokenSecret               string
+	Environment               string
+	ImageBuilderVersion       string
+	LogLevel                  string
+	TaskCommandRouterInsecure bool
 }
 
 // rawProfile mirrors the TOML structure on disk.
 type rawProfile struct {
-	ServerURL           string `toml:"server_url"`
-	TokenID             string `toml:"token_id"`
-	TokenSecret         string `toml:"token_secret"`
-	Environment         string `toml:"environment"`
-	ImageBuilderVersion string `toml:"image_builder_version"`
-	LogLevel            string `toml:"loglevel"`
-	Active              bool   `toml:"active"`
+	ServerURL                 string `toml:"server_url"`
+	TokenID                   string `toml:"token_id"`
+	TokenSecret               string `toml:"token_secret"`
+	Environment               string `toml:"environment"`
+	ImageBuilderVersion       string `toml:"image_builder_version"`
+	LogLevel                  string `toml:"loglevel"`
+	Active                    bool   `toml:"active"`
+	TaskCommandRouterInsecure bool   `toml:"task_command_router_insecure"`
 }
 
 type config map[string]rawProfile
@@ -96,13 +99,19 @@ func getProfile(name string, cfg config) Profile {
 	imageBuilderVersion := firstNonEmpty(os.Getenv("MODAL_IMAGE_BUILDER_VERSION"), raw.ImageBuilderVersion)
 	logLevel := firstNonEmpty(os.Getenv("MODAL_LOGLEVEL"), raw.LogLevel)
 
+	taskCommandRouterInsecure := raw.TaskCommandRouterInsecure
+	if envVal := os.Getenv("MODAL_TASK_COMMAND_ROUTER_INSECURE"); envVal != "" {
+		taskCommandRouterInsecure = strings.ToLower(envVal) == "true" || envVal == "1"
+	}
+
 	return Profile{
-		ServerURL:           serverURL,
-		TokenID:             tokenID,
-		TokenSecret:         tokenSecret,
-		Environment:         environment,
-		ImageBuilderVersion: imageBuilderVersion,
-		LogLevel:            logLevel,
+		ServerURL:                 serverURL,
+		TokenID:                   tokenID,
+		TokenSecret:               tokenSecret,
+		Environment:               environment,
+		ImageBuilderVersion:       imageBuilderVersion,
+		LogLevel:                  logLevel,
+		TaskCommandRouterInsecure: taskCommandRouterInsecure,
 	}
 }
 

modal-go/errors.go

@@ -100,3 +100,12 @@ type SandboxTimeoutError struct {
 func (e SandboxTimeoutError) Error() string {
 	return "SandboxTimeoutError: " + e.Exception
 }
+
+// ExecTimeoutError is returned when a container exec exceeds its execution duration limit.
+type ExecTimeoutError struct {
+	Exception string
+}
+
+func (e ExecTimeoutError) Error() string {
+	return "ExecTimeoutError: " + e.Exception
+}