mirror of
https://github.com/danielmiessler/Fabric.git
synced 2026-02-12 23:15:05 -05:00
cc3e4226d7
CHANGES - Add -V/--vendor flag to specify model vendor - Implement vendor-aware model resolution and availability validation - Warn on ambiguous models; suggest --vendor to disambiguate - Update bash, zsh, fish completions with vendor suggestions - Extend --listmodels to print vendor|model when interactive - Add VendorsModels.PrintWithVendor; sort vendors and models alphabetically - Pass vendor through API; update server chat handler - Standardize docs and errors to --yt-dlp-args="..." syntax - Add test covering ambiguous model warning across multiple vendors - Promote go-shellquote to direct dependency in go.mod
50 lines
1.3 KiB
Go
50 lines
1.3 KiB
Go
package ai
|
|
|
|
import (
|
|
"fmt"
|
|
"sort"
|
|
"strings"
|
|
|
|
"github.com/danielmiessler/fabric/internal/util"
|
|
)
|
|
|
|
func NewVendorsModels() *VendorsModels {
|
|
return &VendorsModels{GroupsItemsSelectorString: util.NewGroupsItemsSelectorString("Available models")}
|
|
}
|
|
|
|
type VendorsModels struct {
|
|
*util.GroupsItemsSelectorString
|
|
}
|
|
|
|
// PrintWithVendor prints models including their vendor on each line.
|
|
// When shellCompleteList is true, output is suitable for shell completion.
|
|
func (o *VendorsModels) PrintWithVendor(shellCompleteList bool) {
|
|
if !shellCompleteList {
|
|
fmt.Printf("\n%v:\n", o.SelectionLabel)
|
|
}
|
|
|
|
var currentItemIndex int
|
|
|
|
sortedGroups := make([]*util.GroupItems[string], len(o.GroupsItems))
|
|
copy(sortedGroups, o.GroupsItems)
|
|
sort.SliceStable(sortedGroups, func(i, j int) bool {
|
|
return strings.ToLower(sortedGroups[i].Group) < strings.ToLower(sortedGroups[j].Group)
|
|
})
|
|
|
|
for _, groupItems := range sortedGroups {
|
|
items := make([]string, len(groupItems.Items))
|
|
copy(items, groupItems.Items)
|
|
sort.SliceStable(items, func(i, j int) bool {
|
|
return strings.ToLower(items[i]) < strings.ToLower(items[j])
|
|
})
|
|
for _, item := range items {
|
|
currentItemIndex++
|
|
if shellCompleteList {
|
|
fmt.Printf("%s|%s\n", groupItems.Group, item)
|
|
} else {
|
|
fmt.Printf("\t[%d]\t%s|%s\n", currentItemIndex, groupItems.Group, item)
|
|
}
|
|
}
|
|
}
|
|
}
|