mirror of
https://github.com/charmbracelet/crush.git
synced 2026-05-30 18:47:33 +00:00
Ayman BagabasandGitHub
5ff8d68760
* refactor(config): introduce ConfigStore and Scope for better config management This makes config.Config immutable and introduces a ConfigStore that manages the config and provides helper methods for accessing config values with proper scoping (global, workspace). This allows us to avoid passing around mutable config objects and ensures that all parts of the code are accessing the most up-to-date config values. It also lays the groundwork for future features like per-workspace config overrides. * fixt: lint
111 lines
2.5 KiB
Go
111 lines
2.5 KiB
Go
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"slices"
|
|
"sort"
|
|
"strings"
|
|
|
|
"charm.land/catwalk/pkg/catwalk"
|
|
"charm.land/lipgloss/v2/tree"
|
|
"github.com/charmbracelet/crush/internal/config"
|
|
"github.com/mattn/go-isatty"
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
var modelsCmd = &cobra.Command{
|
|
Use: "models",
|
|
Short: "List all available models from configured providers",
|
|
Long: `List all available models from configured providers. Shows provider name and model IDs.`,
|
|
Example: `# List all available models
|
|
crush models
|
|
|
|
# Search models
|
|
crush models gpt5`,
|
|
Args: cobra.ArbitraryArgs,
|
|
RunE: func(cmd *cobra.Command, args []string) error {
|
|
cwd, err := ResolveCwd(cmd)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
dataDir, _ := cmd.Flags().GetString("data-dir")
|
|
debug, _ := cmd.Flags().GetBool("debug")
|
|
|
|
cfg, err := config.Init(cwd, dataDir, debug)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if !cfg.Config().IsConfigured() {
|
|
return fmt.Errorf("no providers configured - please run 'crush' to set up a provider interactively")
|
|
}
|
|
|
|
term := strings.ToLower(strings.Join(args, " "))
|
|
filter := func(p config.ProviderConfig, m catwalk.Model) bool {
|
|
for _, s := range []string{p.ID, p.Name, m.ID, m.Name} {
|
|
if term == "" || strings.Contains(strings.ToLower(s), term) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
var providerIDs []string
|
|
providerModels := make(map[string][]string)
|
|
|
|
for providerID, provider := range cfg.Config().Providers.Seq2() {
|
|
if provider.Disable {
|
|
continue
|
|
}
|
|
var found bool
|
|
for _, model := range provider.Models {
|
|
if !filter(provider, model) {
|
|
continue
|
|
}
|
|
providerModels[providerID] = append(providerModels[providerID], model.ID)
|
|
found = true
|
|
}
|
|
if !found {
|
|
continue
|
|
}
|
|
slices.Sort(providerModels[providerID])
|
|
providerIDs = append(providerIDs, providerID)
|
|
}
|
|
sort.Strings(providerIDs)
|
|
|
|
if len(providerIDs) == 0 && len(args) == 0 {
|
|
return fmt.Errorf("no enabled providers found")
|
|
}
|
|
if len(providerIDs) == 0 {
|
|
return fmt.Errorf("no enabled providers found matching %q", term)
|
|
}
|
|
|
|
if !isatty.IsTerminal(os.Stdout.Fd()) {
|
|
for _, providerID := range providerIDs {
|
|
for _, modelID := range providerModels[providerID] {
|
|
fmt.Println(providerID + "/" + modelID)
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
t := tree.New()
|
|
for _, providerID := range providerIDs {
|
|
providerNode := tree.Root(providerID)
|
|
for _, modelID := range providerModels[providerID] {
|
|
providerNode.Child(modelID)
|
|
}
|
|
t.Child(providerNode)
|
|
}
|
|
|
|
cmd.Println(t)
|
|
return nil
|
|
},
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(modelsCmd)
|
|
}
|