We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent afc5164 commit 18025d0Copy full SHA for 18025d0
pkg/utils/list_models.go
@@ -4,6 +4,7 @@ import (
4
"encoding/json"
5
"fmt"
6
"net/http"
7
+ "net/url"
8
)
9
10
type Model struct {
@@ -12,6 +13,10 @@ type Model struct {
12
13
14
// ListModels lists available models from Ollama
15
func ListModels(addr string) ([]string, error) {
16
+ if _, err := url.Parse(addr); err != nil {
17
+ return nil, fmt.Errorf("invalid address provided: %v", err)
18
+ }
19
+
20
resp, err := http.Get(fmt.Sprintf("%s/api/tags", addr))
21
if err != nil {
22
return nil, fmt.Errorf("error fetching models from Ollama: %v", err)
0 commit comments