From 5124e2f89b2aad981fe410382371bc78afdc7260 Mon Sep 17 00:00:00 2001 From: Lukas Werner Date: Tue, 28 Jul 2026 20:53:54 -0700 Subject: [PATCH] embed: respect OLLAMA_HOST env var --- store/embedding.go | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/store/embedding.go b/store/embedding.go index 16fbd08..b446789 100644 --- a/store/embedding.go +++ b/store/embedding.go @@ -4,13 +4,21 @@ import ( "context" "net/http" "net/url" + "os" sqlite_vec "github.com/lukasmwerner/sqlite-vec-go/cgo" ollama "github.com/ollama/ollama/api" ) func ollama_embedding(model, s string) ([]byte, error) { - host, _ := url.Parse("http://localhost:11434") + ollama_host := os.Getenv("OLLAMA_HOST") + if ollama_host == "" { + ollama_host = "http://localhost:11434" + } + host, err := url.Parse(ollama_host) + if err != nil { + return nil, err + } cli := ollama.NewClient(host, http.DefaultClient) req := &ollama.EmbedRequest{ Model: model, -- 2.51.2