fix(sandbox): handle structured command CI

Author: ramon-langchain

internal/cmd/sandbox_test.go

@@ -6,6 +6,7 @@ import (
 	"net/http"
 	"os"
 	"path/filepath"
+	"runtime"
 	"testing"
 
 	langsmith "github.com/langchain-ai/langsmith-go"
@@ -204,10 +205,16 @@ func TestSandboxCustomOutputCommands_Flags(t *testing.T) {
 		cmd  *cobra.Command
 		want []string
 	}{
-		{"console", newSandboxConsoleCmd(), []string{"shell", "forward-ssh-agent"}},
 		{"tunnel", newSandboxTunnelCmd(), []string{"url", "name", "remote-port", "local-port", "stdio", "log-level"}},
 		{"ssh-setup", newSandboxSSHSetupCmd(), []string{"identity"}},
 	}
+	if runtime.GOOS != "windows" {
+		tests = append(tests, struct {
+			name string
+			cmd  *cobra.Command
+			want []string
+		}{"console", newSandboxConsoleCmd(), []string{"shell", "forward-ssh-agent"}})
+	}
 	for _, tc := range tests {
 		t.Run(tc.name, func(t *testing.T) {
 			for _, name := range tc.want {

internal/cmd/sandbox_tunnel.go

@@ -65,7 +65,7 @@ Examples:
 			return nil, fmt.Errorf("provide a sandbox name or --url")
 		}
 		client := MustGetClient()
-		ctx, cancel := signal.NotifyContext(context.Background(), syscall.SIGTERM, syscall.SIGINT)
+		ctx, cancel := signal.NotifyContext(ctx, syscall.SIGTERM, syscall.SIGINT)
 		defer cancel()
 		sandboxURL := in.SandboxURL
 		if sandboxURL == "" {