refactored the dir structure
This commit is contained in:
@@ -0,0 +1,35 @@
|
||||
package sidekick
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"text/template"
|
||||
|
||||
"github.com/BurntSushi/toml"
|
||||
)
|
||||
|
||||
// LoadPrompts reads a TOML file containing multiline prompts and compiles them into a template registry.
|
||||
func LoadPrompts(path string) (*template.Template, error) {
|
||||
var prompts map[string]string
|
||||
if _, err := toml.DecodeFile(path, &prompts); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
tmpl := template.New("prompts")
|
||||
for name, content := range prompts {
|
||||
_, err := tmpl.New(name).Parse(content)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to parse prompt template '%s': %w", name, err)
|
||||
}
|
||||
}
|
||||
return tmpl, nil
|
||||
}
|
||||
|
||||
// RenderPrompt executes a specific prompt template by name.
|
||||
func RenderPrompt(tmpl *template.Template, name string) (string, error) {
|
||||
var buf bytes.Buffer
|
||||
if err := tmpl.ExecuteTemplate(&buf, name, nil); err != nil {
|
||||
return "", err
|
||||
}
|
||||
return buf.String(), nil
|
||||
}
|
||||
Reference in New Issue
Block a user