modelpicker_providers.go RawHistoryFolder

1package agent
2
3import "github.com/patriceckhart/zot/packages/provider"
4
5// modelPickerProviders returns providers whose models can be selected without
6// prompting for credentials. Keyless custom endpoints need a configured URL.
7func modelPickerProviders() []string {
8	var out []string
9	seen := map[string]bool{}
10	for _, p := range knownProviders {
11		if CredentialAvailable(p) && !seen[p] {
12			out = append(out, p)
13			seen[p] = true
14		}
15	}
16	for p, cfg := range provider.CustomProviders() {
17		if seen[p] {
18			continue
19		}
20		if CredentialAvailable(p) || (!isBuiltinProvider(p) && (cfg.BaseURL != "" || customProviderHasModelURL(p))) {
21			out = append(out, p)
22			seen[p] = true
23		}
24	}
25	if !seen["ollama"] {
26		out = append(out, "ollama")
27	}
28	return out
29}
30
31func customProviderHasModelURL(name string) bool {
32	for _, m := range provider.ModelsForProvider(name) {
33		if m.Source == "user" && m.BaseURL != "" {
34			return true
35		}
36	}
37	return false
38}