diff --git a/core/http/routes/ui.go b/core/http/routes/ui.go index 3c805422b131..4618f032598e 100644 --- a/core/http/routes/ui.go +++ b/core/http/routes/ui.go @@ -26,6 +26,7 @@ func RegisterUIRoutes(app *fiber.App, appConfig *config.ApplicationConfig, galleryService *services.GalleryService, auth func(*fiber.Ctx) error) { + tmpLMS := services.NewListModelsService(ml, cl, appConfig) // TODO: once createApplication() is fully in use, reference the central instance. // keeps the state of models that are being installed from the UI var processingModels = xsync.NewSyncedMap[string, string]() @@ -235,7 +236,7 @@ func RegisterUIRoutes(app *fiber.App, // Show the Chat page app.Get("/chat/:model", auth, func(c *fiber.Ctx) error { - backendConfigs := cl.GetAllBackendConfigs() + backendConfigs, _ := tmpLMS.ListModels("", true) summary := fiber.Map{ "Title": "LocalAI - Chat with " + c.Params("model"), @@ -249,7 +250,7 @@ func RegisterUIRoutes(app *fiber.App, }) app.Get("/talk/", auth, func(c *fiber.Ctx) error { - backendConfigs := cl.GetAllBackendConfigs() + backendConfigs, _ := tmpLMS.ListModels("", true) if len(backendConfigs) == 0 { // If no model is available redirect to the index which suggests how to install models @@ -259,7 +260,7 @@ func RegisterUIRoutes(app *fiber.App, summary := fiber.Map{ "Title": "LocalAI - Talk", "ModelsConfig": backendConfigs, - "Model": backendConfigs[0].Name, + "Model": backendConfigs[0].ID, "Version": internal.PrintableVersion(), } @@ -269,7 +270,7 @@ func RegisterUIRoutes(app *fiber.App, app.Get("/chat/", auth, func(c *fiber.Ctx) error { - backendConfigs := cl.GetAllBackendConfigs() + backendConfigs, _ := tmpLMS.ListModels("", true) if len(backendConfigs) == 0 { // If no model is available redirect to the index which suggests how to install models @@ -277,9 +278,9 @@ func RegisterUIRoutes(app *fiber.App, } summary := fiber.Map{ - "Title": "LocalAI - Chat with " + backendConfigs[0].Name, + "Title": "LocalAI - Chat with " + backendConfigs[0].ID, "ModelsConfig": backendConfigs, - "Model": backendConfigs[0].Name, + "Model": backendConfigs[0].ID, "Version": internal.PrintableVersion(), } diff --git a/core/http/views/chat.html b/core/http/views/chat.html index 7f13c7bd0aa1..1e490b659481 100644 --- a/core/http/views/chat.html +++ b/core/http/views/chat.html @@ -100,10 +100,10 @@

Chat wit {{ $model:=.Model}} {{ range .ModelsConfig }} - {{ if eq .Name $model }} - + {{ if eq .ID $model }} + {{ else }} - + {{ end }} {{ end }} diff --git a/core/http/views/talk.html b/core/http/views/talk.html index 0846a0c83cb4..14e6b7a5eada 100644 --- a/core/http/views/talk.html +++ b/core/http/views/talk.html @@ -62,7 +62,7 @@ {{ range .ModelsConfig }} - + {{ end }}