UI

Author: mudler

core/gallery/backend_types.go

@@ -25,3 +25,11 @@ func (m *GalleryBackend) GetName() string {
 func (m *GalleryBackend) GetGallery() config.Gallery {
 	return m.Gallery
 }
+
+func (m *GalleryBackend) GetDescription() string {
+	return m.Description
+}
+
+func (m *GalleryBackend) GetTags() []string {
+	return m.Tags
+}

core/gallery/backends_test.go

@@ -24,7 +24,7 @@ var _ = Describe("Gallery Backends", func() {
 		galleries = []config.Gallery{
 			{
 				Name: "test-gallery",
-				URL:  "https://raw.githubusercontent.com/mudler/LocalAI/refs/heads/master/gallery/index.yaml",
+				URL:  "https://gist.githubusercontent.com/mudler/71d5376bc2aa168873fa519fa9f4bd56/raw/0557f9c640c159fa8e4eab29e8d98df6a3d6e80f/backend-gallery.yaml",
 			},
 		}
 	})
@@ -40,8 +40,11 @@ var _ = Describe("Gallery Backends", func() {
 			Expect(err.Error()).To(ContainSubstring("no backend found with name"))
 		})
 
-		// Note: Full installation test would require mocking the OCI client
-		// which is beyond the scope of this basic test setup
+		It("should install backend from gallery", func() {
+			err := InstallBackendFromGallery(galleries, "test-backend", tempDir, nil)
+			Expect(err).ToNot(HaveOccurred())
+			Expect(filepath.Join(tempDir, "test-backend", "run.sh")).To(BeARegularFile())
+		})
 	})
 
 	Describe("InstallBackend", func() {

core/gallery/gallery.go

@@ -46,9 +46,49 @@ type GalleryElement interface {
 	SetGallery(gallery config.Gallery)
 	SetInstalled(installed bool)
 	GetName() string
+	GetDescription() string
+	GetTags() []string
 	GetGallery() config.Gallery
 }
 
+type GalleryElements[T GalleryElement] []T
+
+func (gm GalleryElements[T]) Search(term string) GalleryElements[T] {
+	var filteredModels GalleryElements[T]
+
+	for _, m := range gm {
+		if strings.Contains(m.GetName(), term) ||
+			strings.Contains(m.GetDescription(), term) ||
+			strings.Contains(m.GetGallery().Name, term) ||
+			strings.Contains(strings.Join(m.GetTags(), ","), term) {
+			filteredModels = append(filteredModels, m)
+		}
+	}
+	return filteredModels
+}
+
+func (gm GalleryElements[T]) FindByName(name string) T {
+	for _, m := range gm {
+		if strings.EqualFold(m.GetName(), name) {
+			return m
+		}
+	}
+	var zero T
+	return zero
+}
+
+func (gm GalleryElements[T]) Paginate(pageNum int, itemsNum int) GalleryElements[T] {
+	start := (pageNum - 1) * itemsNum
+	end := start + itemsNum
+	if start > len(gm) {
+		start = len(gm)
+	}
+	if end > len(gm) {
+		end = len(gm)
+	}
+	return gm[start:end]
+}
+
 func FindGalleryElement[T GalleryElement](models []T, name string, basePath string) T {
 	var model T
 	name = strings.ReplaceAll(name, string(os.PathSeparator), "__")
@@ -76,7 +116,7 @@ func FindGalleryElement[T GalleryElement](models []T, name string, basePath stri
 // List available models
 // Models galleries are a list of yaml files that are hosted on a remote server (for example github).
 // Each yaml file contains a list of models that can be downloaded and optionally overrides to define a new model setting.
-func AvailableGalleryModels(galleries []config.Gallery, basePath string) (GalleryModels, error) {
+func AvailableGalleryModels(galleries []config.Gallery, basePath string) (GalleryElements[*GalleryModel], error) {
 	var models []*GalleryModel
 
 	// Get models from galleries
@@ -92,7 +132,7 @@ func AvailableGalleryModels(galleries []config.Gallery, basePath string) (Galler
 }
 
 // List available backends
-func AvailableBackends(galleries []config.Gallery, basePath string) (GalleryBackends, error) {
+func AvailableBackends(galleries []config.Gallery, basePath string) (GalleryElements[*GalleryBackend], error) {
 	var models []*GalleryBackend
 
 	// Get models from galleries
@@ -149,9 +189,13 @@ func getGalleryElements[T GalleryElement](gallery config.Gallery, basePath strin
 		model.SetGallery(gallery)
 		// we check if the model was already installed by checking if the config file exists
 		// TODO: (what to do if the model doesn't install a config file?)
+		// TODO: This is sub-optimal now that the gallery handles both backends and models - we need to abstract this away
 		if _, err := os.Stat(filepath.Join(basePath, fmt.Sprintf("%s.yaml", model.GetName()))); err == nil {
 			model.SetInstalled(true)
 		}
+		if _, err := os.Stat(filepath.Join(basePath, model.GetName())); err == nil {
+			model.SetInstalled(true)
+		}
 	}
 	return models, nil
 }

core/gallery/models_types.go

@@ -2,7 +2,6 @@ package gallery
 
 import (
 	"fmt"
-	"strings"
 
 	"github.com/mudler/LocalAI/core/config"
 )
@@ -38,39 +37,10 @@ func (m GalleryModel) ID() string {
 	return fmt.Sprintf("%s@%s", m.Gallery.Name, m.Name)
 }
 
-type GalleryModels []*GalleryModel
-
-func (gm GalleryModels) Search(term string) GalleryModels {
-	var filteredModels GalleryModels
-
-	for _, m := range gm {
-		if strings.Contains(m.Name, term) ||
-			strings.Contains(m.Description, term) ||
-			strings.Contains(m.Gallery.Name, term) ||
-			strings.Contains(strings.Join(m.Tags, ","), term) {
-			filteredModels = append(filteredModels, m)
-		}
-	}
-	return filteredModels
-}
-
-func (gm GalleryModels) FindByName(name string) *GalleryModel {
-	for _, m := range gm {
-		if strings.EqualFold(m.Name, name) {
-			return m
-		}
-	}
-	return nil
+func (m *GalleryModel) GetTags() []string {
+	return m.Tags
 }
 
-func (gm GalleryModels) Paginate(pageNum int, itemsNum int) GalleryModels {
-	start := (pageNum - 1) * itemsNum
-	end := start + itemsNum
-	if start > len(gm) {
-		start = len(gm)
-	}
-	if end > len(gm) {
-		end = len(gm)
-	}
-	return gm[start:end]
+func (m *GalleryModel) GetDescription() string {
+	return m.Description
 }