Rrmove memory estimation logic and related structures (#504)

* remove memory estimation logic and related structures

* remove unused Elastic dependencies from go.mod and go.sum

* remove unnecessary field and helper

* update docs

* fix integration_test.go

Author: ilopezluna

cmd/cli/commands/compose.go

@@ -177,7 +177,7 @@ func downloadModelsOnlyIfNotFound(desktopClient *desktop.Client, models []string
 			printer := desktop.NewSimplePrinter(func(s string) {
 				_ = sendInfo(s)
 			})
-			_, _, err = desktopClient.Pull(model, false, printer)
+			_, _, err = desktopClient.Pull(model, printer)
 			if err != nil {
 				_ = sendErrorf("Failed to pull model: %v", err)
 				return fmt.Errorf("Failed to pull model: %w\n", err)

cmd/cli/commands/integration_test.go

@@ -371,7 +371,7 @@ func TestIntegration_PullModel(t *testing.T) {
 		t.Run(tc.name, func(t *testing.T) {
 			// Pull the model using the test case reference
 			t.Logf("Pulling model with reference: %s", tc.ref)
-			err := pullModel(newPullCmd(), env.client, tc.ref, true)
+			err := pullModel(newPullCmd(), env.client, tc.ref)
 			require.NoError(t, err, "Failed to pull model with reference: %s", tc.ref)
 
 			// List models and verify the expected model is present
@@ -426,7 +426,7 @@ func TestIntegration_InspectModel(t *testing.T) {
 	// Pull the model using a short reference
 	pullRef := "inspect-test"
 	t.Logf("Pulling model with reference: %s", pullRef)
-	err = pullModel(newPullCmd(), env.client, pullRef, true)
+	err = pullModel(newPullCmd(), env.client, pullRef)
 	require.NoError(t, err, "Failed to pull model")
 
 	// Verify the model was pulled
@@ -485,7 +485,7 @@ func TestIntegration_TagModel(t *testing.T) {
 	// Pull the model using a simple reference
 	pullRef := "tag-test"
 	t.Logf("Pulling model with reference: %s", pullRef)
-	err = pullModel(newPullCmd(), env.client, pullRef, true)
+	err = pullModel(newPullCmd(), env.client, pullRef)
 	require.NoError(t, err, "Failed to pull model")
 
 	// Verify the model was pulled
@@ -663,7 +663,7 @@ func TestIntegration_PushModel(t *testing.T) {
 	// Pull the model using a simple reference
 	pullRef := "tag-test"
 	t.Logf("Pulling model with reference: %s", pullRef)
-	err = pullModel(newPullCmd(), env.client, pullRef, true)
+	err = pullModel(newPullCmd(), env.client, pullRef)
 	require.NoError(t, err, "Failed to pull model")
 
 	// Verify the model was pulled
@@ -814,7 +814,7 @@ func TestIntegration_RemoveModel(t *testing.T) {
 				// Pull the model
 				pullRef := "rm-test"
 				t.Logf("Pulling model with reference: %s", pullRef)
-				err := pullModel(newPullCmd(), env.client, pullRef, true)
+				err := pullModel(newPullCmd(), env.client, pullRef)
 				require.NoError(t, err, "Failed to pull model")
 
 				// Verify model exists
@@ -848,11 +848,11 @@ func TestIntegration_RemoveModel(t *testing.T) {
 
 		// Pull both models
 		t.Logf("Pulling first model: rm-multi-1")
-		err := pullModel(newPullCmd(), env.client, "rm-multi-1", true)
+		err := pullModel(newPullCmd(), env.client, "rm-multi-1")
 		require.NoError(t, err, "Failed to pull first model")
 
 		t.Logf("Pulling second model: rm-multi-2")
-		err = pullModel(newPullCmd(), env.client, "rm-multi-2", true)
+		err = pullModel(newPullCmd(), env.client, "rm-multi-2")
 		require.NoError(t, err, "Failed to pull second model")
 
 		// Verify both models exist
@@ -878,7 +878,7 @@ func TestIntegration_RemoveModel(t *testing.T) {
 	t.Run("remove specific tag keeps other tags", func(t *testing.T) {
 		// Pull the model
 		t.Logf("Pulling model: rm-test")
-		err := pullModel(newPullCmd(), env.client, "rm-test", true)
+		err := pullModel(newPullCmd(), env.client, "rm-test")
 		require.NoError(t, err, "Failed to pull model")
 
 		// Add multiple tags to the same model
@@ -940,7 +940,7 @@ func TestIntegration_RemoveModel(t *testing.T) {
 	t.Run("remove by model ID removes all tags", func(t *testing.T) {
 		// Pull the model
 		t.Logf("Pulling model: rm-test")
-		err := pullModel(newPullCmd(), env.client, "rm-test", true)
+		err := pullModel(newPullCmd(), env.client, "rm-test")
 		require.NoError(t, err, "Failed to pull model")
 
 		// Add multiple tags
@@ -971,7 +971,7 @@ func TestIntegration_RemoveModel(t *testing.T) {
 	t.Run("force flag", func(t *testing.T) {
 		// Pull the model
 		t.Logf("Pulling model: rm-test")
-		err := pullModel(newPullCmd(), env.client, "rm-test", true)
+		err := pullModel(newPullCmd(), env.client, "rm-test")
 		require.NoError(t, err, "Failed to pull model")
 
 		// Test removal with force flag

cmd/cli/commands/pull.go

@@ -10,8 +10,6 @@ import (
 )
 
 func newPullCmd() *cobra.Command {
-	var ignoreRuntimeMemoryCheck bool
-
 	c := &cobra.Command{
 		Use:   "pull MODEL",
 		Short: "Pull a model from Docker Hub or HuggingFace to your local environment",
@@ -20,19 +18,17 @@ func newPullCmd() *cobra.Command {
 			if _, err := ensureStandaloneRunnerAvailable(cmd.Context(), asPrinter(cmd), false); err != nil {
 				return fmt.Errorf("unable to initialize standalone model runner: %w", err)
 			}
-			return pullModel(cmd, desktopClient, args[0], ignoreRuntimeMemoryCheck)
+			return pullModel(cmd, desktopClient, args[0])
 		},
 		ValidArgsFunction: completion.NoComplete,
 	}
 
-	c.Flags().BoolVar(&ignoreRuntimeMemoryCheck, "ignore-runtime-memory-check", false, "Do not block pull if estimated runtime memory for model exceeds system resources.")
-
 	return c
 }
 
-func pullModel(cmd *cobra.Command, desktopClient *desktop.Client, model string, ignoreRuntimeMemoryCheck bool) error {
+func pullModel(cmd *cobra.Command, desktopClient *desktop.Client, model string) error {
 	printer := asPrinter(cmd)
-	response, _, err := desktopClient.Pull(model, ignoreRuntimeMemoryCheck, printer)
+	response, _, err := desktopClient.Pull(model, printer)
 
 	if err != nil {
 		return handleClientError(err, "Failed to pull model")

cmd/cli/commands/run.go

@@ -570,7 +570,6 @@ func chatWithMarkdownContext(ctx context.Context, cmd *cobra.Command, client *de
 
 func newRunCmd() *cobra.Command {
 	var debug bool
-	var ignoreRuntimeMemoryCheck bool
 	var colorMode string
 	var detach bool
 
@@ -686,7 +685,7 @@ func newRunCmd() *cobra.Command {
 					return handleClientError(err, "Failed to inspect model")
 				}
 				cmd.Println("Unable to find model '" + model + "' locally. Pulling from the server.")
-				if err := pullModel(cmd, desktopClient, model, ignoreRuntimeMemoryCheck); err != nil {
+				if err := pullModel(cmd, desktopClient, model); err != nil {
 					return err
 				}
 			}
@@ -733,7 +732,6 @@ func newRunCmd() *cobra.Command {
 	c.Args = requireMinArgs(1, "run", cmdArgs)
 
 	c.Flags().BoolVar(&debug, "debug", false, "Enable debug logging")
-	c.Flags().BoolVar(&ignoreRuntimeMemoryCheck, "ignore-runtime-memory-check", false, "Do not block pull if estimated runtime memory for model exceeds system resources.")
 	c.Flags().StringVar(&colorMode, "color", "no", "Use colored output (auto|yes|no)")
 	c.Flags().BoolVarP(&detach, "detach", "d", false, "Load the model in the background without interaction")
 

cmd/cli/desktop/desktop.go

@@ -105,7 +105,7 @@ func (c *Client) Status() Status {
 	}
 }
 
-func (c *Client) Pull(model string, ignoreRuntimeMemoryCheck bool, printer standalone.StatusPrinter) (string, bool, error) {
+func (c *Client) Pull(model string, printer standalone.StatusPrinter) (string, bool, error) {
 	model = normalizeHuggingFaceModelName(model)
 
 	// Check if this is a Hugging Face model and if HF_TOKEN is set
@@ -116,9 +116,8 @@ func (c *Client) Pull(model string, ignoreRuntimeMemoryCheck bool, printer stand
 
 	return c.withRetries("download", 3, printer, func(attempt int) (string, bool, error, bool) {
 		jsonData, err := json.Marshal(dmrm.ModelCreateRequest{
-			From:                     model,
-			IgnoreRuntimeMemoryCheck: ignoreRuntimeMemoryCheck,
-			BearerToken:              hfToken,
+			From:        model,
+			BearerToken: hfToken,
 		})
 		if err != nil {
 			// Marshaling errors are not retryable

cmd/cli/desktop/desktop_test.go

@@ -39,7 +39,7 @@ func TestPullHuggingFaceModel(t *testing.T) {
 	}, nil)
 
 	printer := NewSimplePrinter(func(s string) {})
-	_, _, err := client.Pull(modelName, false, printer)
+	_, _, err := client.Pull(modelName, printer)
 	assert.NoError(t, err)
 }
 
@@ -126,7 +126,7 @@ func TestNonHuggingFaceModel(t *testing.T) {
 	}, nil)
 
 	printer := NewSimplePrinter(func(s string) {})
-	_, _, err := client.Pull(modelName, false, printer)
+	_, _, err := client.Pull(modelName, printer)
 	assert.NoError(t, err)
 }
 
@@ -250,7 +250,7 @@ func TestPullRetryOnNetworkError(t *testing.T) {
 	)
 
 	printer := NewSimplePrinter(func(s string) {})
-	_, _, err := client.Pull(modelName, false, printer)
+	_, _, err := client.Pull(modelName, printer)
 	assert.NoError(t, err)
 }
 
@@ -270,7 +270,7 @@ func TestPullNoRetryOn4xxError(t *testing.T) {
 	}, nil).Times(1)
 
 	printer := NewSimplePrinter(func(s string) {})
-	_, _, err := client.Pull(modelName, false, printer)
+	_, _, err := client.Pull(modelName, printer)
 	assert.Error(t, err)
 	assert.Contains(t, err.Error(), "Model not found")
 }
@@ -297,7 +297,7 @@ func TestPullRetryOn5xxError(t *testing.T) {
 	)
 
 	printer := NewSimplePrinter(func(s string) {})
-	_, _, err := client.Pull(modelName, false, printer)
+	_, _, err := client.Pull(modelName, printer)
 	assert.NoError(t, err)
 }
 
@@ -324,7 +324,7 @@ func TestPullRetryOnServiceUnavailable(t *testing.T) {
 	)
 
 	printer := NewSimplePrinter(func(s string) {})
-	_, _, err := client.Pull(modelName, false, printer)
+	_, _, err := client.Pull(modelName, printer)
 	assert.NoError(t, err)
 }
 
@@ -341,7 +341,7 @@ func TestPullMaxRetriesExhausted(t *testing.T) {
 	mockClient.EXPECT().Do(gomock.Any()).Return(nil, io.EOF).Times(4)
 
 	printer := NewSimplePrinter(func(s string) {})
-	_, _, err := client.Pull(modelName, false, printer)
+	_, _, err := client.Pull(modelName, printer)
 	assert.Error(t, err)
 	assert.Contains(t, err.Error(), "failed to download after 3 retries")
 }

cmd/cli/docs/reference/docker_model_pull.yaml

@@ -5,18 +5,6 @@ long: |
 usage: docker model pull MODEL
 pname: docker model
 plink: docker_model.yaml
-options:
-    - option: ignore-runtime-memory-check
-      value_type: bool
-      default_value: "false"
-      description: |
-        Do not block pull if estimated runtime memory for model exceeds system resources.
-      deprecated: false
-      hidden: false
-      experimental: false
-      experimentalcli: false
-      kubernetes: false
-      swarm: false
 examples: |-
     ### Pulling a model from Docker Hub
 

cmd/cli/docs/reference/docker_model_run.yaml

@@ -41,17 +41,6 @@ options:
       experimentalcli: false
       kubernetes: false
       swarm: false
-    - option: ignore-runtime-memory-check
-      value_type: bool
-      default_value: "false"
-      description: |
-        Do not block pull if estimated runtime memory for model exceeds system resources.
-      deprecated: false
-      hidden: false
-      experimental: false
-      experimentalcli: false
-      kubernetes: false
-      swarm: false
 examples: |-
     ### One-time prompt
 

cmd/cli/docs/reference/model_pull.md

@@ -3,12 +3,6 @@
 <!---MARKER_GEN_START-->
 Pull a model from Docker Hub or HuggingFace to your local environment
 
-### Options
-
-| Name                            | Type   | Default | Description                                                                       |
-|:--------------------------------|:-------|:--------|:----------------------------------------------------------------------------------|
-| `--ignore-runtime-memory-check` | `bool` |         | Do not block pull if estimated runtime memory for model exceeds system resources. |
-
 
 <!---MARKER_GEN_END-->
 

cmd/cli/docs/reference/model_run.md

@@ -5,12 +5,11 @@ Run a model and interact with it using a submitted prompt or chat mode
 
 ### Options
 
-| Name                            | Type     | Default | Description                                                                       |
-|:--------------------------------|:---------|:--------|:----------------------------------------------------------------------------------|
-| `--color`                       | `string` | `no`    | Use colored output (auto\|yes\|no)                                                |
-| `--debug`                       | `bool`   |         | Enable debug logging                                                              |
-| `-d`, `--detach`                | `bool`   |         | Load the model in the background without interaction                              |
-| `--ignore-runtime-memory-check` | `bool`   |         | Do not block pull if estimated runtime memory for model exceeds system resources. |
+| Name             | Type     | Default | Description                                          |
+|:-----------------|:---------|:--------|:-----------------------------------------------------|
+| `--color`        | `string` | `no`    | Use colored output (auto\|yes\|no)                   |
+| `--debug`        | `bool`   |         | Enable debug logging                                 |
+| `-d`, `--detach` | `bool`   |         | Load the model in the background without interaction |
 
 
 <!---MARKER_GEN_END-->