fix(internal): ensure prompt filenames have .yml suffix

nomihq · Oct 5, 2024 · 2ac9b38 · 2ac9b38
1 parent ff309cb
commit 2ac9b38
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/internal/prompt/loader.go b/internal/prompt/loader.go
@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"os"
 	"path/filepath"
+	"strings"
 
 	"github.com/nullswan/golem/internal/config"
 	"gopkg.in/yaml.v2"
@@ -13,7 +14,11 @@ import (
 var ErrPromptNotFound = errors.New("prompt not found")
 
 func LoadPrompt(filename string) (*Prompt, error) {
-	fp := filepath.Join(config.GetPromptDirectory(), filename+".yml")
+	if !strings.HasSuffix(filename, ".yml") {
+		filename += ".yml"
+	}
+
+	fp := filepath.Join(config.GetPromptDirectory(), filename)
 	if _, err := os.Stat(fp); os.IsNotExist(err) {
 		return nil, ErrPromptNotFound
 	}