mirror of
https://github.com/mudler/LocalAI.git
synced 2025-01-01 10:26:39 +00:00
a3d69872e3
feat(api): list loaded models in /system Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
32 lines
911 B
Go
32 lines
911 B
Go
package localai
|
|
|
|
import (
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/mudler/LocalAI/core/config"
|
|
"github.com/mudler/LocalAI/core/schema"
|
|
"github.com/mudler/LocalAI/pkg/model"
|
|
)
|
|
|
|
// SystemInformations returns the system informations
|
|
// @Summary Show the LocalAI instance information
|
|
// @Success 200 {object} schema.SystemInformationResponse "Response"
|
|
// @Router /system [get]
|
|
func SystemInformations(ml *model.ModelLoader, appConfig *config.ApplicationConfig) func(*fiber.Ctx) error {
|
|
return func(c *fiber.Ctx) error {
|
|
availableBackends, err := ml.ListAvailableBackends(appConfig.AssetsDestination)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
loadedModels := ml.ListModels()
|
|
for b := range appConfig.ExternalGRPCBackends {
|
|
availableBackends = append(availableBackends, b)
|
|
}
|
|
return c.JSON(
|
|
schema.SystemInformationResponse{
|
|
Backends: availableBackends,
|
|
Models: loadedModels,
|
|
},
|
|
)
|
|
}
|
|
}
|