Add feedback-aware correction contracts

internal/modules/dnd/extract/combatturns/extractor_test.go

@@ -592,6 +592,11 @@ func (client *fakeCombatTurnsLLMClient) CompleteStructured(_ context.Context, re
 
 func cloneStructuredCompletionRequest(req contracts.StructuredCompletionRequest) contracts.StructuredCompletionRequest {
 	req.Inputs = req.Inputs.Clone()
+	correction, err := contracts.CloneSemanticCorrection(req.Correction)
+	if err != nil {
+		panic(err)
+	}
+	req.Correction = correction
 	if len(req.Vars) == 0 {
 		req.Vars = nil
 		return req

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

@@ -58,6 +58,11 @@ func newExtractor(t *testing.T, client contracts.StructuredLLMClient, references
 
 func cloneStructuredCompletionRequest(req contracts.StructuredCompletionRequest) contracts.StructuredCompletionRequest {
 	req.Inputs = req.Inputs.Clone()
+	correction, err := contracts.CloneSemanticCorrection(req.Correction)
+	if err != nil {
+		panic(err)
+	}
+	req.Correction = correction
 	return req
 }
 

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

@@ -88,5 +88,10 @@ func (client *fakeItemsLLMClient) CompleteStructured(ctx context.Context, req co
 
 func cloneStructuredCompletionRequest(req contracts.StructuredCompletionRequest) contracts.StructuredCompletionRequest {
 	req.Inputs = req.Inputs.Clone()
+	correction, err := contracts.CloneSemanticCorrection(req.Correction)
+	if err != nil {
+		panic(err)
+	}
+	req.Correction = correction
 	return req
 }

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

@@ -78,5 +78,10 @@ func (client *fakeOccurrencesLLMClient) CompleteStructured(_ context.Context, re
 
 func cloneStructuredCompletionRequest(req contracts.StructuredCompletionRequest) contracts.StructuredCompletionRequest {
 	req.Inputs = req.Inputs.Clone()
+	correction, err := contracts.CloneSemanticCorrection(req.Correction)
+	if err != nil {
+		panic(err)
+	}
+	req.Correction = correction
 	return req
 }

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

@@ -85,5 +85,10 @@ func (client *fakeLocationsLLMClient) CompleteStructured(_ context.Context, req
 
 func cloneStructuredCompletionRequest(req contracts.StructuredCompletionRequest) contracts.StructuredCompletionRequest {
 	req.Inputs = req.Inputs.Clone()
+	correction, err := contracts.CloneSemanticCorrection(req.Correction)
+	if err != nil {
+		panic(err)
+	}
+	req.Correction = correction
 	return req
 }

internal/modules/dnd/extract/npcoccurrences/extractor_test.go

@@ -452,5 +452,10 @@ func (client *fakeOccurrencesLLMClient) CompleteStructured(_ context.Context, re
 
 func cloneStructuredCompletionRequest(req contracts.StructuredCompletionRequest) contracts.StructuredCompletionRequest {
 	req.Inputs = req.Inputs.Clone()
+	correction, err := contracts.CloneSemanticCorrection(req.Correction)
+	if err != nil {
+		panic(err)
+	}
+	req.Correction = correction
 	return req
 }

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

@@ -65,6 +65,11 @@ func mismatchedSourceInputRequest(req contracts.TypedExtractionRequest) contract
 
 func cloneStructuredCompletionRequest(req contracts.StructuredCompletionRequest) contracts.StructuredCompletionRequest {
 	req.Inputs = req.Inputs.Clone()
+	correction, err := contracts.CloneSemanticCorrection(req.Correction)
+	if err != nil {
+		panic(err)
+	}
+	req.Correction = correction
 	if len(req.Vars) == 0 {
 		req.Vars = nil
 		return req

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

@@ -61,6 +61,11 @@ func mismatchedSourceInputRequest(req contracts.TypedExtractionRequest) contract
 
 func cloneStructuredCompletionRequest(req contracts.StructuredCompletionRequest) contracts.StructuredCompletionRequest {
 	req.Inputs = req.Inputs.Clone()
+	correction, err := contracts.CloneSemanticCorrection(req.Correction)
+	if err != nil {
+		panic(err)
+	}
+	req.Correction = correction
 	if len(req.Vars) == 0 {
 		req.Vars = nil
 		return req

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

@@ -163,6 +163,11 @@ func (client *fakeSpellsLLMClient) CompleteStructured(_ context.Context, req con
 
 func cloneStructuredCompletionRequest(req contracts.StructuredCompletionRequest) contracts.StructuredCompletionRequest {
 	req.Inputs = req.Inputs.Clone()
+	correction, err := contracts.CloneSemanticCorrection(req.Correction)
+	if err != nil {
+		panic(err)
+	}
+	req.Correction = correction
 	if len(req.Vars) == 0 {
 		req.Vars = nil
 		return req