Add D&D extraction fallback profile

This commit is contained in:
2026-08-03 16:39:45 +00:00
parent b05634ee86
commit a3bd0c1867
16 changed files with 145 additions and 16 deletions

View File

@@ -1,6 +1,6 @@
id: dnd.scenes
version: "v1"
default_profile: gemini-2-flash
default_profile: dnd-extraction
inputs:
- name: transcript
required: true

View File

@@ -1,6 +1,6 @@
id: dnd.combat_turns
version: "v1"
default_profile: gemini-2-flash
default_profile: dnd-extraction
inputs:
- name: transcript
required: true

View File

@@ -1,6 +1,6 @@
id: dnd.item_events
version: "v1"
default_profile: gemini-2-flash
default_profile: dnd-extraction
inputs:
- name: transcript
required: true

View File

@@ -1,6 +1,6 @@
id: dnd.npc_interactions
version: "v1"
default_profile: gemini-2-flash
default_profile: dnd-extraction
inputs:
- name: transcript
required: true

View File

@@ -1,6 +1,6 @@
id: dnd.npcs
version: "v1"
default_profile: gemini-2-flash
default_profile: dnd-extraction
inputs:
- name: transcript
required: true

View File

@@ -1,6 +1,6 @@
id: dnd.scene_descriptions
version: "v1"
default_profile: gemini-2-flash
default_profile: dnd-extraction
inputs:
- name: transcript
required: true

View File

@@ -1,6 +1,6 @@
id: dnd.spells
version: "v1"
default_profile: gemini-2-flash
default_profile: dnd-extraction
inputs:
- name: transcript
required: true

View File

@@ -1,6 +1,6 @@
id: dnd.npcs.normalize
version: "v1"
default_profile: gemini-2-flash
default_profile: dnd-extraction
inputs:
- name: candidates
required: true

View File

@@ -0,0 +1,5 @@
id: dnd-extraction
backend: openrouter
model: openai/gpt-5.6-luna
timeout_seconds: 240
service_tier: flex

View File

@@ -0,0 +1,14 @@
package register
import (
"embed"
"gitea.maximumdirect.net/eric/notarius/internal/framework/llm"
)
//go:embed assets/profiles/*.yaml
var embeddedProfileAssets embed.FS
func registerFallbackProfiles(assets *llm.AssetRegistry) error {
return assets.RegisterFallbackProfileFS(embeddedProfileAssets, "assets/profiles")
}

View File

@@ -30,6 +30,9 @@ func Register(registries pipeline.Registries, assets *llm.AssetRegistry) error {
if err := registerPromptAssets(assets); err != nil {
return err
}
if err := registerFallbackProfiles(assets); err != nil {
return err
}
return registerDefaultChains(registries.ValidatorChains)
}

View File

@@ -39,6 +39,31 @@ func TestRegisterAddsDNDFamily(t *testing.T) {
if _, err := fs.ReadFile(promptFS, "dnd.npcs.normalize/dnd.npcs.normalize.yaml"); err != nil {
t.Fatalf("normalization prompt asset = %v, want registered private prompt", err)
}
for _, name := range []string{
"dnd.scenes/dnd.scenes.yaml",
"dnd.spells/dnd.spells.yaml",
"dnd.npcs/dnd.npcs.yaml",
"dnd.combat_turns/dnd.combat_turns.yaml",
"dnd.item_events/dnd.item_events.yaml",
"dnd.npc_interactions/dnd.npc_interactions.yaml",
"dnd.scene_descriptions/dnd.scene_descriptions.yaml",
"dnd.npcs.normalize/dnd.npcs.normalize.yaml",
} {
content, err := fs.ReadFile(promptFS, name)
if err != nil {
t.Fatalf("read prompt asset %q: %v", name, err)
}
if !strings.Contains(string(content), "default_profile: dnd-extraction") {
t.Fatalf("prompt asset %q does not select dnd-extraction", name)
}
}
fallbackFS, err := assets.FallbackProfileFS()
if err != nil {
t.Fatalf("FallbackProfileFS() error = %v", err)
}
if _, err := fs.ReadFile(fallbackFS, "dnd-extraction.yaml"); err != nil {
t.Fatalf("fallback profile asset = %v, want registered D&D profile", err)
}
schemaFS, err := assets.SchemaFS()
if err != nil {
t.Fatalf("SchemaFS() error = %v", err)