feat: Expand variables in model path

Allow define model path in config file as `${HOME}/model.gguf`.
macie · Dec 30, 2023 · 0379822 · 0379822
1 parent 230e1d8
commit 0379822
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cmd/boludo/config.go b/cmd/boludo/config.go
@@ -10,6 +10,7 @@ import (
 	"fmt"
 	"io"
 	"io/fs"
+	"os"
 	"os/signal"
 	"strings"
 	"time"
@@ -162,7 +163,7 @@ func (c *ConfigFile) UnmarshalTOML(data interface{}) error {
 		for k, v := range definedConfigs[configId].(map[string]interface{}) {
 			switch k {
 			case "model":
-				defaultSpec.Model = v.(string)
+				defaultSpec.Model = os.ExpandEnv(v.(string))
 			case "creativity":
 				defaultSpec.Creativity = (float32)(v.(float64))
 			case "cutoff":