fix(ui): do not show duplicate entries if not installed by gallery

Signed-off-by: Ettore Di Giacinto <[email protected]>
mudler · Aug 1, 2024 · 3b9532c · 3b9532c
1 parent e4b91e9
commit 3b9532c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/core/http/endpoints/localai/welcome.go b/core/http/endpoints/localai/welcome.go
@@ -17,7 +17,10 @@ func WelcomeEndpoint(appConfig *config.ApplicationConfig,
 		backendConfigs := cl.GetAllBackendConfigs()
 
 		galleryConfigs := map[string]*gallery.Config{}
+		modelsWithBackendConfig := map[string]interface{}{}
+
 		for _, m := range backendConfigs {
+			modelsWithBackendConfig[m.Name] = nil
 
 			cfg, err := gallery.GetLocalModelConfiguration(ml.ModelPath, m.Name)
 			if err != nil {
@@ -32,7 +35,7 @@ func WelcomeEndpoint(appConfig *config.ApplicationConfig,
 		modelsWithoutConfig := []string{}
 
 		for _, m := range models {
-			if _, ok := galleryConfigs[m]; !ok {
+			if _, ok := modelsWithBackendConfig[m]; !ok {
 				modelsWithoutConfig = append(modelsWithoutConfig, m)
 			}
 		}