Function EditModelEndpoint

core/http/endpoints/localai/edit_model.go:59–96

EditModelEndpoint handles updating existing model configurations

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

// EditModelEndpoint handles updating existing model configurations
func EditModelEndpoint(cl *config.ModelConfigLoader, ml *model.ModelLoader, gs *galleryop.GalleryService, appConfig *config.ApplicationConfig) echo.HandlerFunc {
	svc := modeladmin.NewConfigService(cl, appConfig)
	return func(c echo.Context) error {
		modelName := c.Param("name")
		if decoded, err := url.PathUnescape(modelName); err == nil {
			modelName = decoded
		}
		body, err := io.ReadAll(c.Request().Body)
		if err != nil {
			return c.JSON(http.StatusBadRequest, ModelResponse{Success: false, Error: "Failed to read request body: " + err.Error()})
		}
		result, err := svc.EditYAML(c.Request().Context(), modelName, body, ml)
		if err != nil {
			return c.JSON(httpStatusForModelAdminError(err), ModelResponse{Success: false, Error: err.Error()})
		}

		// Tell peer replicas to refresh their in-memory config: this endpoint
		// only reloaded the local loader. A rename is a delete of the old name
		// plus an install of the new one. No-op in standalone mode.
		if gs != nil {
			if result.Renamed {
				gs.BroadcastModelsChanged(result.OldName, "delete")
			}
			gs.BroadcastModelsChanged(result.NewName, "install")
		}

		msg := fmt.Sprintf("Model '%s' updated successfully. Model has been reloaded with new configuration.", result.NewName)
		if result.Renamed {
			msg = fmt.Sprintf("Model '%s' renamed to '%s' and updated successfully.", result.OldName, result.NewName)
		}
		return c.JSON(http.StatusOK, ModelResponse{
			Success:  true,
			Message:  msg,
			Filename: result.Filename,
			Config:   result.Config,
		})
	}
}

// httpStatusForModelAdminError maps the typed errors from modeladmin to
// the HTTP status codes the existing endpoints used to return — keeps the