mirror of
https://github.com/mudler/LocalAI.git
synced 2026-01-04 01:30:08 -06:00
b3c2a3c257
* migrate core/system to pkg/system - it has no dependencies FROM core, and IS USED in pkg Signed-off-by: Dave Lee <dave@gray101.com> * move pkg/templates up to core/templates -- nothing in pkg references it, but it does reference core. Signed-off-by: Dave Lee <dave@gray101.com> * remove extra check, len of nil is 0 Signed-off-by: Dave Lee <dave@gray101.com> * move pkg/startup to core/startup -- it does have important and unfixable dependencies on core Signed-off-by: Dave Lee <dave@gray101.com> --------- Signed-off-by: Dave Lee <dave@gray101.com>
51 lines
1.5 KiB
Go
51 lines
1.5 KiB
Go
package startup
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"path/filepath"
|
|
"strings"
|
|
|
|
"github.com/mudler/LocalAI/core/config"
|
|
"github.com/mudler/LocalAI/core/gallery"
|
|
"github.com/mudler/LocalAI/pkg/downloader"
|
|
"github.com/mudler/LocalAI/pkg/system"
|
|
"github.com/rs/zerolog/log"
|
|
)
|
|
|
|
func InstallExternalBackends(galleries []config.Gallery, backendPath string, downloadStatus func(string, string, string, float64), backends ...string) error {
|
|
var errs error
|
|
systemState, err := system.GetSystemState()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to get system state: %w", err)
|
|
}
|
|
for _, backend := range backends {
|
|
uri := downloader.URI(backend)
|
|
switch {
|
|
case uri.LooksLikeOCI():
|
|
name, err := uri.FilenameFromUrl()
|
|
if err != nil {
|
|
return fmt.Errorf("failed to get filename from URL: %w", err)
|
|
}
|
|
// strip extension if any
|
|
name = strings.TrimSuffix(name, filepath.Ext(name))
|
|
|
|
log.Info().Str("backend", backend).Str("name", name).Msg("Installing backend from OCI image")
|
|
if err := gallery.InstallBackend(backendPath, &gallery.GalleryBackend{
|
|
Metadata: gallery.Metadata{
|
|
Name: name,
|
|
},
|
|
URI: backend,
|
|
}, downloadStatus); err != nil {
|
|
errs = errors.Join(err, fmt.Errorf("error installing backend %s", backend))
|
|
}
|
|
default:
|
|
err := gallery.InstallBackendFromGallery(galleries, systemState, backend, backendPath, downloadStatus, true)
|
|
if err != nil {
|
|
errs = errors.Join(err, fmt.Errorf("error installing backend %s", backend))
|
|
}
|
|
}
|
|
}
|
|
return errs
|
|
}
|