diff --git a/pkg/lmcli/lmcli.go b/pkg/lmcli/lmcli.go index b5389c1..a841d84 100644 --- a/pkg/lmcli/lmcli.go +++ b/pkg/lmcli/lmcli.go @@ -66,7 +66,7 @@ func (c *Context) GetModels() (models []string) { for _, p := range c.Config.Providers { for _, m := range *p.Models { modelCounts[m]++ - models = append(models, *p.Name+"/"+m) + models = append(models, fmt.Sprintf("%s@%s", m, *p.Name)) } } @@ -80,12 +80,12 @@ func (c *Context) GetModels() (models []string) { } func (c *Context) GetModelProvider(model string) (string, api.ChatCompletionClient, error) { - parts := strings.Split(model, "/") + parts := strings.Split(model, "@") var provider string if len(parts) > 1 { - provider = parts[0] - model = parts[1] + model = parts[0] + provider = parts[1] } for _, p := range c.Config.Providers {