mirror of
https://github.com/mudler/LocalAI.git
synced 2024-06-07 19:40:48 +00:00
6559ac11b1
* feat(ui): allow to set system prompt for chat Make also the models in the index clickable, and display as table Fixes #2257 Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * feat(vision): support also png with base64 input Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * feat(ui): support vision and upload of files Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * display the processed image Signed-off-by: Ettore Di Giacinto <mudler@localai.io> * make trust remote code stand out Signed-off-by: mudler <mudler@localai.io> * feat(ui): track in progress job across index/model gallery Signed-off-by: mudler <mudler@localai.io> * minor fixups Signed-off-by: mudler <mudler@localai.io> --------- Signed-off-by: Ettore Di Giacinto <mudler@localai.io> Signed-off-by: mudler <mudler@localai.io>
50 lines
2.1 KiB
Go
50 lines
2.1 KiB
Go
package gallery
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
)
|
|
|
|
// GalleryModel is the struct used to represent a model in the gallery returned by the endpoint.
|
|
// It is used to install the model by resolving the URL and downloading the files.
|
|
// The other fields are used to override the configuration of the model.
|
|
type GalleryModel struct {
|
|
URL string `json:"url,omitempty" yaml:"url,omitempty"`
|
|
Name string `json:"name,omitempty" yaml:"name,omitempty"`
|
|
Description string `json:"description,omitempty" yaml:"description,omitempty"`
|
|
License string `json:"license,omitempty" yaml:"license,omitempty"`
|
|
URLs []string `json:"urls,omitempty" yaml:"urls,omitempty"`
|
|
Icon string `json:"icon,omitempty" yaml:"icon,omitempty"`
|
|
Tags []string `json:"tags,omitempty" yaml:"tags,omitempty"`
|
|
// config_file is read in the situation where URL is blank - and therefore this is a base config.
|
|
ConfigFile map[string]interface{} `json:"config_file,omitempty" yaml:"config_file,omitempty"`
|
|
// Overrides are used to override the configuration of the model located at URL
|
|
Overrides map[string]interface{} `json:"overrides,omitempty" yaml:"overrides,omitempty"`
|
|
// AdditionalFiles are used to add additional files to the model
|
|
AdditionalFiles []File `json:"files,omitempty" yaml:"files,omitempty"`
|
|
// Gallery is a reference to the gallery which contains the model
|
|
Gallery Gallery `json:"gallery,omitempty" yaml:"gallery,omitempty"`
|
|
// Installed is used to indicate if the model is installed or not
|
|
Installed bool `json:"installed,omitempty" yaml:"installed,omitempty"`
|
|
}
|
|
|
|
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
|
|
}
|