Relax private D&D response schemas

internal/modules/dnd/extract/spells/test_helpers_test.go

@@ -8,7 +8,6 @@ import (
 
 	"gitea.maximumdirect.net/eric/notarius/internal/core/source"
 	"gitea.maximumdirect.net/eric/notarius/internal/framework/contracts"
-	"gitea.maximumdirect.net/eric/notarius/internal/modules/dnd/shared"
 	spellcatalog "gitea.maximumdirect.net/eric/notarius/internal/modules/dnd/spells/catalog"
 )
 
@@ -76,8 +75,8 @@ func mustJSON(t *testing.T, value any) string {
 func responseSourceRefs(startUnitID int, endUnitID int) []spellSourceRefResponse {
 	return []spellSourceRefResponse{
 		{
-			StartUnitID: shared.UnitRefFromInt(startUnitID),
-			EndUnitID:   shared.UnitRefFromInt(endUnitID),
+			StartUnitID: startUnitID,
+			EndUnitID:   endUnitID,
 		},
 	}
 }
@@ -151,9 +150,13 @@ func (client *fakeSpellsLLMClient) CompleteStructured(_ context.Context, req con
 	if !ok {
 		return contracts.StructuredCompletionResponse{}, errors.New("unexpected output target")
 	}
-	*target = client.response
 	content := append([]byte(nil), client.content...)
-	if len(content) == 0 {
+	if len(content) != 0 {
+		if err := json.Unmarshal(content, target); err != nil {
+			return contracts.StructuredCompletionResponse{}, err
+		}
+	} else {
+		*target = client.response
 		var err error
 		content, err = json.Marshal(client.response)
 		if err != nil {