fix(cli): unable to use CLI within the container

In the container, `OLLAMA_HOST` is set by default to `0.0.0.0` (ref: [Dockerfile#L137]), which is fine when starting the server. However, as a client, it is necessary to use `127.0.0.1` or `localhost` for requests.

fix: #3521 #1337
maybe fix: #3526

[Dockerfile#L137]: https://github.com/ollama/ollama/blob/7e432cdfac51583459e7bfa8fdd485c74a6597e7/Dockerfile#L137

Signed-off-by: Kevin Cui <bh@bugs.cc>

api/client.go

@@ -82,6 +82,13 @@ func ClientFromEnvironment() (*Client, error) {
 		}
 	}
 
+	// 0.0.0.0 => 127.0.0.1
+	if ip := net.ParseIP(host); ip != nil {
+		if ip.IsUnspecified() {
+			host = "127.0.0.1"
+		}
+	}
+
 	return &Client{
 		base: &url.URL{
 			Scheme: scheme,

api/client_test.go

@@ -24,6 +24,8 @@ func TestClientFromEnvironment(t *testing.T) {
 		"scheme, hostname, and port": {value: "https://example.com:1234", expect: "https://example.com:1234"},
 		"trailing slash":             {value: "example.com/", expect: "http://example.com:11434"},
 		"trailing slash port":        {value: "example.com:1234/", expect: "http://example.com:1234"},
+		"unspecified ipv4 host":      {value: "0.0.0.0:1234", expect: "http://127.0.0.1:1234"},
+		"unspecified ipv6 host":      {value: "[::]:1234", expect: "http://127.0.0.1:1234"},
 	}
 
 	for k, v := range testCases {