refactored the dir structure
This commit is contained in:
@@ -0,0 +1,42 @@
|
||||
package sidekick
|
||||
|
||||
import (
|
||||
"os"
|
||||
"testing"
|
||||
)
|
||||
|
||||
func TestPromptsTemplating(t *testing.T) {
|
||||
content := `
|
||||
base_rules = "You are helpful."
|
||||
coordinator = "{{template \"base_rules\"}} Route tasks."
|
||||
`
|
||||
tmpFile, err := os.CreateTemp("", "prompts-*.toml")
|
||||
if err != nil {
|
||||
t.Fatalf("failed to create temp file: %v", err)
|
||||
}
|
||||
defer os.Remove(tmpFile.Name())
|
||||
|
||||
if _, err := tmpFile.WriteString(content); err != nil {
|
||||
t.Fatalf("failed to write temp file: %v", err)
|
||||
}
|
||||
tmpFile.Close()
|
||||
|
||||
tmpl, err := LoadPrompts(tmpFile.Name())
|
||||
if err != nil {
|
||||
t.Fatalf("LoadPrompts failed: %v", err)
|
||||
}
|
||||
|
||||
if tmpl.Lookup("coordinator") == nil {
|
||||
t.Fatalf("template 'coordinator' not found")
|
||||
}
|
||||
|
||||
rendered, err := RenderPrompt(tmpl, "coordinator")
|
||||
if err != nil {
|
||||
t.Fatalf("RenderPrompt failed: %v", err)
|
||||
}
|
||||
|
||||
expected := "You are helpful. Route tasks."
|
||||
if rendered != expected {
|
||||
t.Errorf("expected %q, got %q", expected, rendered)
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user