Organize D&D extensions by domain

internal/modules/integration/dnd_spells_helpers_test.go

@@ -0,0 +1,63 @@
+package integration_test
+
+import (
+	"context"
+	"encoding/json"
+	"fmt"
+
+	"gitea.maximumdirect.net/eric/notarius/internal/framework/contracts"
+	"gitea.maximumdirect.net/eric/notarius/internal/modules/dnd/shared"
+)
+
+type extractionResponse struct {
+	SpellCasts []spellCastResponse `json:"spell_casts"`
+}
+
+type spellCastResponse struct {
+	Caster               string                     `json:"caster"`
+	Spell                string                     `json:"spell"`
+	Effect               string                     `json:"effect"`
+	NarrativeDescription string                     `json:"narrative_description"`
+	SourceRefs           []shared.SourceRefResponse `json:"source_refs"`
+}
+
+type fakeSpellsLLMClient struct {
+	response extractionResponse
+	requests []contracts.StructuredCompletionRequest
+}
+
+func (client *fakeSpellsLLMClient) CompleteStructured(_ context.Context, req contracts.StructuredCompletionRequest, out any) (contracts.StructuredCompletionResponse, error) {
+	client.requests = append(client.requests, cloneStructuredCompletionRequest(req))
+	content, err := json.Marshal(client.response)
+	if err != nil {
+		return contracts.StructuredCompletionResponse{}, err
+	}
+	if err := json.Unmarshal(content, out); err != nil {
+		return contracts.StructuredCompletionResponse{}, fmt.Errorf("populate structured target: %w", err)
+	}
+	return contracts.StructuredCompletionResponse{Content: content}, nil
+}
+
+func responseSourceRefs(sourceID string, startUnitID int, endUnitID int) []shared.SourceRefResponse {
+	return []shared.SourceRefResponse{
+		{
+			SourceID:    sourceID,
+			StartUnitID: shared.UnitRefFromInt(startUnitID),
+			EndUnitID:   shared.UnitRefFromInt(endUnitID),
+		},
+	}
+}
+
+func cloneStructuredCompletionRequest(req contracts.StructuredCompletionRequest) contracts.StructuredCompletionRequest {
+	req.Inputs = req.Inputs.Clone()
+	if len(req.Vars) == 0 {
+		req.Vars = nil
+		return req
+	}
+	vars := make(map[string]any, len(req.Vars))
+	for key, value := range req.Vars {
+		vars[key] = value
+	}
+	req.Vars = vars
+	return req
+}