1
0
mirror of https://github.com/coalaura/whiskr.git synced 2025-09-09 09:19:54 +00:00

some more prompts

This commit is contained in:
Laura
2025-08-15 03:00:59 +02:00
parent 3251b297d4
commit 3adaa69bc0
8 changed files with 164 additions and 13 deletions

View File

@@ -18,20 +18,38 @@ var (
//go:embed prompts/normal.txt
PromptNormal string
PromptNormalTmpl = template.Must(template.New("normal").Parse(PromptNormal))
//go:embed prompts/reviewer.txt
PromptReviewer string
//go:embed prompts/engineer.txt
PromptEngineer string
//go:embed prompts/scripts.txt
PromptScripts string
//go:embed prompts/physics.txt
PromptPhysics string
Templates = map[string]*template.Template{
"normal": NewTemplate("normal", PromptNormal),
"reviewer": NewTemplate("reviewer", PromptReviewer),
"engineer": NewTemplate("engineer", PromptEngineer),
"scripts": NewTemplate("scripts", PromptScripts),
"physics": NewTemplate("physics", PromptPhysics),
}
)
func NewTemplate(name, text string) *template.Template {
return template.Must(template.New(name).Parse(text))
}
func BuildPrompt(name string, model *Model) (string, error) {
if name == "" {
return "", nil
}
var tmpl *template.Template
switch name {
case "normal":
tmpl = PromptNormalTmpl
default:
tmpl, ok := Templates[name]
if !ok {
return "", fmt.Errorf("unknown prompt: %q", name)
}