Merge pull request #859 from jmorganca/mxyng/fix-hostname

fix: ollama host for hostname

Merge pull request #859 from jmorganca/mxyng/fix-hostname
fix: ollama host for hostname
533d7636 · Michael Yang · GitHub · cbfff4f8 · 459f4a78 · 533d7636
Unverified Commit 533d7636 authored Oct 20, 2023 by Michael Yang Committed by GitHub Oct 20, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 1 deletion

api/client.go api/client.go +3 -1

No files found.
--- a/api/client.go
+++ b/api/client.go
@@ -52,8 +52,10 @@ func ClientFromEnvironment() (*Client, error) {
 	host, port, err := net.SplitHostPort(hostport)
 	if err != nil {
 		host, port = "127.0.0.1", "11434"
-		if ip := net.ParseIP(strings.Trim(os.Getenv("OLLAMA_HOST"), "[]")); ip != nil {
+		if ip := net.ParseIP(strings.Trim(hostport, "[]")); ip != nil {
 			host = ip.String()
+		} else if hostport != "" {
+			host = hostport
 		}
 	}