Merge pull request #2526 from dhiltgen/harden_for_quotes

Harden the OLLAMA_HOST lookup for quotes

Merge pull request #2526 from dhiltgen/harden_for_quotes
Harden the OLLAMA_HOST lookup for quotes
c9ab1aea · Daniel Hiltgen · GitHub · fefb3e77 · 4a10e7a7 · c9ab1aea
Unverified Commit c9ab1aea authored Feb 15, 2024 by Daniel Hiltgen Committed by GitHub Feb 15, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

cmd/cmd.go cmd/cmd.go +1 -1

No files found.
--- a/cmd/cmd.go
+++ b/cmd/cmd.go
@@ -685,7 +685,7 @@ func generate(cmd *cobra.Command, opts runOptions) error {
 }

 func RunServer(cmd *cobra.Command, _ []string) error {
-	host, port, err := net.SplitHostPort(os.Getenv("OLLAMA_HOST"))
+	host, port, err := net.SplitHostPort(strings.Trim(os.Getenv("OLLAMA_HOST"), "\"'"))
 	if err != nil {
 		host, port = "127.0.0.1", "11434"
 		if ip := net.ParseIP(strings.Trim(os.Getenv("OLLAMA_HOST"), "[]")); ip != nil {