62 lines
1.5 KiB
Go
62 lines
1.5 KiB
Go
package spells
|
|
|
|
import (
|
|
"encoding/json"
|
|
"testing"
|
|
|
|
"gitea.maximumdirect.net/eric/notarius/internal/core/source"
|
|
"gitea.maximumdirect.net/eric/notarius/internal/framework/contracts"
|
|
)
|
|
|
|
func promptExtractionRequest() contracts.ExtractionRequest {
|
|
doc := promptSourceDocument()
|
|
chunk := &contracts.SourceChunk{
|
|
ID: "session-alpha:chunk:0",
|
|
SourceID: doc.ID,
|
|
Index: 0,
|
|
Units: append([]source.SourceUnit(nil), doc.Units...),
|
|
Metadata: map[string]any{"ignored": "chunk metadata"},
|
|
}
|
|
return contracts.ExtractionRequest{
|
|
Source: doc,
|
|
Chunk: chunk,
|
|
}
|
|
}
|
|
|
|
func promptSourceDocument() *source.SourceDocument {
|
|
return &source.SourceDocument{
|
|
ID: "session-alpha",
|
|
Kind: "transcript",
|
|
Format: "application/vnd.seriatim.minimal+json",
|
|
Digest: "sha256:test",
|
|
Units: []source.SourceUnit{
|
|
{
|
|
ID: "seg-001",
|
|
Kind: "transcript_segment",
|
|
Text: "Aria raises her hand and casts Cure Wounds.",
|
|
Metadata: map[string]any{
|
|
"speaker": "Alice",
|
|
"start": json.Number("1.25"),
|
|
"end": json.Number("3.5"),
|
|
"ignored": "not rendered",
|
|
},
|
|
},
|
|
{
|
|
ID: "seg-002",
|
|
Kind: "transcript_segment",
|
|
Text: "The fighter's wounds begin to close.",
|
|
Metadata: map[string]any{"ignored": "not rendered"},
|
|
},
|
|
},
|
|
}
|
|
}
|
|
|
|
func mustJSON(t *testing.T, value any) string {
|
|
t.Helper()
|
|
encoded, err := json.Marshal(value)
|
|
if err != nil {
|
|
t.Fatalf("Marshal() error = %v, want nil", err)
|
|
}
|
|
return string(encoded)
|
|
}
|