Removes unused code (#519)

Author: ilopezluna

pkg/inference/models/api.go

@@ -18,17 +18,6 @@ type ModelCreateRequest struct {
 	BearerToken string `json:"bearer-token,omitempty"`
 }
 
-// ModelPackageRequest represents a model package request, which creates a new model
-// from an existing one with modified properties (e.g., context size).
-type ModelPackageRequest struct {
-	// From is the name of the source model to package from.
-	From string `json:"from"`
-	// Tag is the name to give the new packaged model.
-	Tag string `json:"tag"`
-	// ContextSize specifies the context size to set for the new model.
-	ContextSize *int32 `json:"context-size,omitempty"`
-}
-
 // SimpleModel is a wrapper that allows creating a model with modified configuration
 type SimpleModel struct {
 	types.Model

pkg/inference/models/http_handler.go

@@ -15,7 +15,6 @@ import (
 	"github.com/docker/model-runner/pkg/distribution/distribution"
 	"github.com/docker/model-runner/pkg/distribution/registry"
 	"github.com/docker/model-runner/pkg/inference"
-	"github.com/docker/model-runner/pkg/internal/utils"
 	"github.com/docker/model-runner/pkg/logging"
 	"github.com/docker/model-runner/pkg/middleware"
 	"github.com/sirupsen/logrus"
@@ -81,7 +80,6 @@ func (h *HTTPHandler) routeHandlers() map[string]http.HandlerFunc {
 	return map[string]http.HandlerFunc{
 		"POST " + inference.ModelsPrefix + "/create":                          h.handleCreateModel,
 		"POST " + inference.ModelsPrefix + "/load":                            h.handleLoadModel,
-		"POST " + inference.ModelsPrefix + "/package":                         h.handlePackageModel,
 		"GET " + inference.ModelsPrefix:                                       h.handleGetModels,
 		"GET " + inference.ModelsPrefix + "/{name...}":                        h.handleGetModel,
 		"DELETE " + inference.ModelsPrefix + "/{name...}":                     h.handleDeleteModel,
@@ -436,44 +434,6 @@ func (h *HTTPHandler) handlePushModel(w http.ResponseWriter, r *http.Request, mo
 	}
 }
 
-// handlePackageModel handles POST <inference-prefix>/models/package requests.
-func (h *HTTPHandler) handlePackageModel(w http.ResponseWriter, r *http.Request) {
-
-	// Decode the request
-	var request ModelPackageRequest
-	if err := json.NewDecoder(r.Body).Decode(&request); err != nil {
-		http.Error(w, "invalid request body", http.StatusBadRequest)
-		return
-	}
-
-	// Validate required fields
-	if request.From == "" || request.Tag == "" {
-		http.Error(w, "both 'from' and 'tag' fields are required", http.StatusBadRequest)
-		return
-	}
-
-	err := h.manager.Package(request.From, request.Tag, request.ContextSize)
-	if err != nil {
-		if errors.Is(err, distribution.ErrModelNotFound) {
-			h.log.Warnf("Failed to package model from %q: %v", utils.SanitizeForLog(request.From, -1), err)
-			http.Error(w, "Model not found", http.StatusNotFound)
-			return
-		}
-		http.Error(w, err.Error(), http.StatusInternalServerError)
-		return
-	}
-
-	// Return success response
-	w.Header().Set("Content-Type", "application/json")
-	response := map[string]string{
-		"message": fmt.Sprintf("Successfully packaged model from %s with tag %s", request.From, request.Tag),
-		"model":   request.Tag,
-	}
-	if err := json.NewEncoder(w).Encode(response); err != nil {
-		h.log.Warnln("Error while encoding package response:", err)
-	}
-}
-
 // handlePurge handles DELETE <inference-prefix>/models/purge requests.
 func (h *HTTPHandler) handlePurge(w http.ResponseWriter, _ *http.Request) {
 	err := h.manager.Purge()

pkg/inference/models/manager.go

@@ -9,7 +9,6 @@ import (
 	"strings"
 
 	"github.com/docker/model-runner/pkg/diskusage"
-	"github.com/docker/model-runner/pkg/distribution/builder"
 	"github.com/docker/model-runner/pkg/distribution/distribution"
 	"github.com/docker/model-runner/pkg/distribution/registry"
 	"github.com/docker/model-runner/pkg/distribution/types"
@@ -410,48 +409,6 @@ func (m *Manager) Push(model string, r *http.Request, w http.ResponseWriter) err
 	return nil
 }
 
-func (m *Manager) Package(ref string, tag string, contextSize *int32) error {
-	// Create a builder from an existing model by getting the bundle first
-	// Since ModelArtifact interface is needed to work with the builder
-	bundle, err := m.distributionClient.GetBundle(ref)
-	if err != nil {
-		return fmt.Errorf("error while getting model bundle: %w", err)
-	}
-
-	// Create a builder from the existing model artifact (from the bundle)
-	modelArtifact, ok := bundle.(types.ModelArtifact)
-	if !ok {
-		return fmt.Errorf("model bundle is not a valid model artifact")
-	}
-
-	// Create a builder from the existing model
-	bldr, err := builder.FromModel(modelArtifact)
-	if err != nil {
-		return fmt.Errorf("error while building model bundle: %w", err)
-	}
-
-	// Apply context size if specified
-	if contextSize != nil {
-		bldr = bldr.WithContextSize(*contextSize)
-	}
-
-	// Get the built model artifact
-	builtModel := bldr.Model()
-
-	// Check if we can use lightweight repackaging (config-only changes from existing model)
-	useLightweight := bldr.HasOnlyConfigChanges()
-
-	if useLightweight {
-		// Use the lightweight method to avoid re-transferring layers
-		if err := m.distributionClient.WriteLightweightModel(builtModel, []string{tag}); err != nil {
-			return fmt.Errorf("error writing model: %w", err)
-		}
-	} else {
-		return err
-	}
-	return nil
-}
-
 func (m *Manager) Purge() error {
 	if m.distributionClient == nil {
 		return fmt.Errorf("model distribution service unavailable")