diff --git a/envconfig/config.go b/envconfig/config.go index 14c7c617..d155bd8f 100644 --- a/envconfig/config.go +++ b/envconfig/config.go @@ -24,6 +24,9 @@ func Host() *url.URL { switch { case !ok: scheme, hostport = "http", s + if s == "ollama.com" { + scheme, hostport = "https", "ollama.com:443" + } case scheme == "http": defaultPort = "80" case scheme == "https": diff --git a/envconfig/config_test.go b/envconfig/config_test.go index f232f1cd..ddd86a11 100644 --- a/envconfig/config_test.go +++ b/envconfig/config_test.go @@ -37,6 +37,7 @@ func TestHost(t *testing.T) { "https": {"https://1.2.3.4", "https://1.2.3.4:443"}, "https port": {"https://1.2.3.4:4321", "https://1.2.3.4:4321"}, "proxy path": {"https://example.com/ollama", "https://example.com:443/ollama"}, + "ollama.com": {"ollama.com", "https://ollama.com:443"}, } for name, tt := range cases {