Function ImportModelEndpoint

core/http/endpoints/localai/import_model.go:128–263

ImportModelEndpoint handles creating new model configurations

(cl *config.ModelConfigLoader, gs *galleryop.GalleryService, appConfig *config.ApplicationConfig)

// ImportModelEndpoint handles creating new model configurations
func ImportModelEndpoint(cl *config.ModelConfigLoader, gs *galleryop.GalleryService, appConfig *config.ApplicationConfig) echo.HandlerFunc {
	return func(c echo.Context) error {
		// Get the raw body
		body, err := io.ReadAll(c.Request().Body)
		if err != nil {
			response := ModelResponse{
				Success: false,
				Error:   "Failed to read request body: " + err.Error(),
			}
			return c.JSON(http.StatusBadRequest, response)
		}
		if len(body) == 0 {
			response := ModelResponse{
				Success: false,
				Error:   "Request body is empty",
			}
			return c.JSON(http.StatusBadRequest, response)
		}

		// Detect format once and reuse for both typed and map parsing
		contentType := c.Request().Header.Get("Content-Type")
		trimmed := strings.TrimSpace(string(body))
		isJSON := strings.Contains(contentType, "application/json") ||
			(!strings.Contains(contentType, "yaml") && len(trimmed) > 0 && trimmed[0] == '{')

		var modelConfig config.ModelConfig
		if isJSON {
			if err := json.Unmarshal(body, &modelConfig); err != nil {
				return c.JSON(http.StatusBadRequest, ModelResponse{Success: false, Error: "Failed to parse JSON: " + err.Error()})
			}
		} else {
			if err := yaml.Unmarshal(body, &modelConfig); err != nil {
				return c.JSON(http.StatusBadRequest, ModelResponse{Success: false, Error: "Failed to parse YAML: " + err.Error()})
			}
		}

		// Validate required fields
		if modelConfig.Name == "" {
			response := ModelResponse{
				Success: false,
				Error:   "Name is required",
			}
			return c.JSON(http.StatusBadRequest, response)
		}

		// Validate without calling SetDefaults() — runtime defaults should not
		// be persisted to disk. SetDefaults() is called when loading configs
		// for inference via LoadModelConfigsFromPath().
		if valid, vErr := modelConfig.Validate(); !valid {
			msg := "Invalid configuration"
			if vErr != nil {
				msg = vErr.Error()
			}
			return c.JSON(http.StatusBadRequest, ModelResponse{Success: false, Error: msg})
		}

		// Reject aliases whose target is missing, chained, or disabled so a
		// dangling alias can't be persisted and surface as a runtime error later.