浏览代码

Merge pull request #2526 from dhiltgen/harden_for_quotes

Harden the OLLAMA_HOST lookup for quotes
Daniel Hiltgen 1 年之前
父节点
当前提交
c9ab1aead3
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      cmd/cmd.go

+ 1 - 1
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 {