Skip to content

Commit 18025d0

Browse files
committed
feat: add url validation
1 parent afc5164 commit 18025d0

File tree

1 file changed

+5
-0
lines changed

1 file changed

+5
-0
lines changed

pkg/utils/list_models.go

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ import (
44
"encoding/json"
55
"fmt"
66
"net/http"
7+
"net/url"
78
)
89

910
type Model struct {
@@ -12,6 +13,10 @@ type Model struct {
1213

1314
// ListModels lists available models from Ollama
1415
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+
1520
resp, err := http.Get(fmt.Sprintf("%s/api/tags", addr))
1621
if err != nil {
1722
return nil, fmt.Errorf("error fetching models from Ollama: %v", err)

0 commit comments

Comments
 (0)