Skip to content

GitLab

Commit 4a10e7a7 authored by Daniel Hiltgen's avatar Daniel Hiltgen
Browse files

Harden the OLLAMA_HOST lookup for quotes

parent fefb3e77
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
cmd/cmd.go
View file @ 4a10e7a7
...@@ -685,7 +685,7 @@ func generate(cmd *cobra.Command, opts runOptions) error { ...@@ -685,7 +685,7 @@ func generate(cmd *cobra.Command, opts runOptions) error {
} }
func RunServer(cmd *cobra.Command, _ []string) 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 { if err != nil {
host, port = "127.0.0.1", "11434" 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(os.Getenv("OLLAMA_HOST"), "[]")); ip != nil {
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment