From 53e9d4de0363f4ff5623c2254ccae548d130102e Mon Sep 17 00:00:00 2001
From: LittleSheep <littlesheep.code@hotmail.com>
Date: Thu, 30 Jan 2025 02:14:41 +0800
Subject: [PATCH] :bug: Did not use ollama server url set in config

---
 pkg/internal/services/ollama.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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
 	}