Add docker model start and stop commands

To start and stop the docker model runner container.

Signed-off-by: Eric Curtin <eric.curtin@docker.com>

Author: ericcurtin

cmd/cli/commands/root.go

@@ -106,6 +106,8 @@ func NewRootCmd(cli *command.DockerCli) *cobra.Command {
 		newTagCmd(),
 		newInstallRunner(),
 		newUninstallRunner(),
+		newStartRunner(),
+		newStopRunner(),
 		newConfigureCmd(),
 		newPSCmd(),
 		newDFCmd(),

cmd/cli/commands/start-runner.go

@@ -0,0 +1,44 @@
+package commands
+
+import (
+	"fmt"
+	"github.com/docker/model-runner/cmd/cli/pkg/types"
+
+	"github.com/docker/model-runner/cmd/cli/commands/completion"
+	"github.com/docker/model-runner/cmd/cli/desktop"
+	"github.com/docker/model-runner/cmd/cli/pkg/standalone"
+	"github.com/spf13/cobra"
+)
+
+func newStartRunner() *cobra.Command {
+	c := &cobra.Command{
+		Use:   "start",
+		Short: "Start Docker Model Runner container",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			// Ensure that we're running in a supported model runner context.
+			if kind := modelRunner.EngineKind(); kind == types.ModelRunnerEngineKindDesktop {
+				cmd.Println("Standalone start not supported with Docker Desktop")
+				cmd.Println("Use `docker desktop enable model-runner` instead")
+				return nil
+			} else if kind == types.ModelRunnerEngineKindMobyManual {
+				cmd.Println("Standalone start not supported with MODEL_RUNNER_HOST set")
+				return nil
+			}
+
+			// Create a Docker client for the active context.
+			dockerClient, err := desktop.DockerClientForContext(dockerCLI, dockerCLI.CurrentContext())
+			if err != nil {
+				return fmt.Errorf("failed to create Docker client: %w", err)
+			}
+
+			// Start the model runner container.
+			if err := standalone.StartControllerContainer(cmd.Context(), dockerClient, cmd); err != nil {
+				return fmt.Errorf("unable to start model runner container: %w", err)
+			}
+
+			return nil
+		},
+		ValidArgsFunction: completion.NoComplete,
+	}
+	return c
+}

cmd/cli/commands/stop-runner.go

@@ -0,0 +1,44 @@
+package commands
+
+import (
+	"fmt"
+	"github.com/docker/model-runner/cmd/cli/pkg/types"
+
+	"github.com/docker/model-runner/cmd/cli/commands/completion"
+	"github.com/docker/model-runner/cmd/cli/desktop"
+	"github.com/docker/model-runner/cmd/cli/pkg/standalone"
+	"github.com/spf13/cobra"
+)
+
+func newStopRunner() *cobra.Command {
+	c := &cobra.Command{
+		Use:   "stop",
+		Short: "Stop Docker Model Runner container",
+		RunE: func(cmd *cobra.Command, args []string) error {
+			// Ensure that we're running in a supported model runner context.
+			if kind := modelRunner.EngineKind(); kind == types.ModelRunnerEngineKindDesktop {
+				cmd.Println("Standalone stop not supported with Docker Desktop")
+				cmd.Println("Use `docker desktop disable model-runner` instead")
+				return nil
+			} else if kind == types.ModelRunnerEngineKindMobyManual {
+				cmd.Println("Standalone stop not supported with MODEL_RUNNER_HOST set")
+				return nil
+			}
+
+			// Create a Docker client for the active context.
+			dockerClient, err := desktop.DockerClientForContext(dockerCLI, dockerCLI.CurrentContext())
+			if err != nil {
+				return fmt.Errorf("failed to create Docker client: %w", err)
+			}
+
+			// Stop the model runner container.
+			if err := standalone.StopControllerContainer(cmd.Context(), dockerClient, cmd); err != nil {
+				return fmt.Errorf("unable to stop model runner container: %w", err)
+			}
+
+			return nil
+		},
+		ValidArgsFunction: completion.NoComplete,
+	}
+	return c
+}

cmd/cli/docs/reference/docker_model_start.yaml

@@ -0,0 +1,13 @@
+command: docker model start
+short: Start Docker Model Runner container
+long: |
+    Start the Docker Model Runner container. This command starts a previously stopped container without pulling images or recreating it.
+usage: docker model start
+pname: docker model
+plink: docker_model.yaml
+deprecated: false
+hidden: false
+experimental: false
+experimentalcli: false
+kubernetes: false
+swarm: false

cmd/cli/docs/reference/docker_model_stop.yaml

@@ -0,0 +1,13 @@
+command: docker model stop
+short: Stop Docker Model Runner container
+long: |
+    Stop the Docker Model Runner container. This command stops the running container without removing it.
+usage: docker model stop
+pname: docker model
+plink: docker_model.yaml
+deprecated: false
+hidden: false
+experimental: false
+experimentalcli: false
+kubernetes: false
+swarm: false

cmd/cli/docs/reference/model_start.md

@@ -0,0 +1,10 @@
+# docker model start
+
+<!---MARKER_GEN_START-->
+Start Docker Model Runner container
+
+<!---MARKER_GEN_END-->
+
+## Description
+
+Start the Docker Model Runner container. This command starts a previously stopped container without pulling images or recreating it.

cmd/cli/docs/reference/model_stop.md

@@ -0,0 +1,10 @@
+# docker model stop
+
+<!---MARKER_GEN_START-->
+Stop Docker Model Runner container
+
+<!---MARKER_GEN_END-->
+
+## Description
+
+Stop the Docker Model Runner container. This command stops the running container without removing it.

cmd/cli/pkg/standalone/containers.go

@@ -347,6 +347,57 @@ func CreateControllerContainer(ctx context.Context, dockerClient *client.Client,
 	return nil
 }
 
+// StartControllerContainer starts the model runner controller container.
+func StartControllerContainer(ctx context.Context, dockerClient client.ContainerAPIClient, printer StatusPrinter) error {
+	// Find the controller container.
+	containerID, containerName, _, err := FindControllerContainer(ctx, dockerClient)
+	if err != nil {
+		return fmt.Errorf("unable to find model runner container: %w", err)
+	}
+	if containerID == "" {
+		return errors.New("model runner container not found")
+	}
+
+	// Start the container.
+	if containerName != "" {
+		printer.Printf("Starting model runner container %s (%s)...\n", containerName, containerID[:12])
+	} else {
+		printer.Printf("Starting model runner container %s...\n", containerID[:12])
+	}
+	if err := ensureContainerStarted(ctx, dockerClient, containerID); err != nil {
+		return fmt.Errorf("failed to start container: %w", err)
+	}
+
+	printer.Printf("Model runner container started successfully\n")
+	return nil
+}
+
+// StopControllerContainer stops the model runner controller container.
+func StopControllerContainer(ctx context.Context, dockerClient client.ContainerAPIClient, printer StatusPrinter) error {
+	// Find the controller container.
+	containerID, containerName, _, err := FindControllerContainer(ctx, dockerClient)
+	if err != nil {
+		return fmt.Errorf("unable to find model runner container: %w", err)
+	}
+	if containerID == "" {
+		return errors.New("model runner container not found")
+	}
+
+	// Stop the container.
+	if containerName != "" {
+		printer.Printf("Stopping model runner container %s (%s)...\n", containerName, containerID[:12])
+	} else {
+		printer.Printf("Stopping model runner container %s...\n", containerID[:12])
+	}
+	stopTimeout := 10 // seconds
+	if err := dockerClient.ContainerStop(ctx, containerID, container.StopOptions{Timeout: &stopTimeout}); err != nil {
+		return fmt.Errorf("failed to stop container: %w", err)
+	}
+
+	printer.Printf("Model runner container stopped successfully\n")
+	return nil
+}
+
 // PruneControllerContainers stops and removes any model runner controller
 // containers.
 func PruneControllerContainers(ctx context.Context, dockerClient client.ContainerAPIClient, skipRunning bool, printer StatusPrinter) error {