diff --git a/pkg/internal/services/ollama.go b/pkg/internal/services/ollama.go index eff5504..4a35fb4 100644 --- a/pkg/internal/services/ollama.go +++ b/pkg/internal/services/ollama.go @@ -29,7 +29,7 @@ var LargeModel *ollama.LLM func ConnectOllama() error { model := viper.GetString("ollama.model") - llm, err := ollama.New(ollama.WithModel(model)) + llm, err := ollama.New(ollama.WithModel(model), ollama.WithServerURL(viper.GetString("ollama.url"))) if err != nil { return err }