Use references in D&D spell extraction
This commit is contained in:
@@ -101,6 +101,50 @@ func TestRenderPromptIncludesSourceContext(t *testing.T) {
|
||||
if metadata.EmbeddedPath != "assets/prompts" {
|
||||
t.Fatalf("metadata.EmbeddedPath = %q, want assets/prompts", metadata.EmbeddedPath)
|
||||
}
|
||||
if strings.Contains(user, "Roster reference material") || strings.Contains(user, "Glossary reference material") {
|
||||
t.Fatalf("user prompt = %q, want no optional reference sections without bindings", user)
|
||||
}
|
||||
}
|
||||
|
||||
func TestRenderPromptIncludesBoundReferences(t *testing.T) {
|
||||
req := promptExtractionRequest()
|
||||
req.References = contracts.ReferenceSet{
|
||||
Slots: map[string]contracts.ResolvedReferenceSlot{
|
||||
"roster": {
|
||||
Slot: contracts.ReferenceSlot{Name: "roster"},
|
||||
Items: []contracts.ReferenceItem{
|
||||
{SlotName: "roster", Content: []byte("Aria: cleric, also known as Sister Aria")},
|
||||
},
|
||||
},
|
||||
"glossary": {
|
||||
Slot: contracts.ReferenceSlot{Name: "glossary"},
|
||||
Items: []contracts.ReferenceItem{
|
||||
{SlotName: "glossary", Content: []byte("Cure Wounds: healing spell")},
|
||||
},
|
||||
},
|
||||
},
|
||||
}
|
||||
|
||||
_, user, metadata, err := renderPrompt(req)
|
||||
if err != nil {
|
||||
t.Fatalf("renderPrompt() error = %v, want nil", err)
|
||||
}
|
||||
|
||||
for _, want := range []string{
|
||||
"Roster reference material:",
|
||||
"Aria: cleric, also known as Sister Aria",
|
||||
"Glossary reference material:",
|
||||
"Cure Wounds: healing spell",
|
||||
"Use roster and glossary reference material only to clarify source text.",
|
||||
"mentioned only in reference material.",
|
||||
} {
|
||||
if !strings.Contains(user, want) {
|
||||
t.Fatalf("user prompt = %q, want substring %q", user, want)
|
||||
}
|
||||
}
|
||||
if metadata.SHA256 != spellsPromptBundle.Metadata().SHA256 {
|
||||
t.Fatalf("metadata.SHA256 = %q, want template hash %q", metadata.SHA256, spellsPromptBundle.Metadata().SHA256)
|
||||
}
|
||||
}
|
||||
|
||||
func TestBuildPromptDataRejectsMissingSourceContext(t *testing.T) {
|
||||
|
||||
Reference in New Issue
Block a user