Simplify references and output instructions in LLM prompts

internal/framework/proposal_generation/generate.go

@@ -36,10 +36,10 @@ type InteractionArtifacts struct {
 
 // StructuredCorrectionProposal is one LLM response correction payload.
 type StructuredCorrectionProposal struct {
-	TargetSegmentID int     `json:"id"`
-	OriginalText    string  `json:"original_text"`
-	CorrectedText   string  `json:"corrected_text"`
-	Confidence      float64 `json:"confidence"`
+	SegmentReference int     `json:"id"`
+	OriginalText     string  `json:"original_text"`
+	CorrectedText    string  `json:"corrected_text"`
+	Confidence       float64 `json:"confidence"`
 }
 
 // StructuredCorrectionSet is the reusable structured LLM response model for
@@ -142,6 +142,7 @@ func GenerateCandidates(ctx context.Context, req Request) (Result, error) {
 		"replacement_policy": req.ReplacementPolicy,
 		"section":            req.Section,
 		"start_index":        req.StartIndex,
+		"segment_ids":        segmentIDs(req.WorkingTranscript),
 		"prompt":             completion.Prompt,
 	}
 	requestMetadata["response_schema"] = responseschema.MustLookup(responseschema.CorrectionSetKey).DiagnosticsMap()
@@ -185,8 +186,13 @@ func GenerateCandidates(ctx context.Context, req Request) (Result, error) {
 	corrections := make([]proposals.CorrectionProposal, 0, len(response.Corrections))
 	enriched := make([]proposals.EnrichedCorrectionProposal, 0, len(response.Corrections))
 	for i, raw := range response.Corrections {
+		// Unknown references remain invalid candidates for the existing domain validators.
+		targetID := 0
+		if req.WorkingTranscript != nil && raw.SegmentReference > 0 && raw.SegmentReference <= len(req.WorkingTranscript.Segments) {
+			targetID = req.WorkingTranscript.Segments[raw.SegmentReference-1].ID
+		}
 		candidate := proposals.CorrectionProposal{
-			TargetSegmentID: raw.TargetSegmentID,
+			TargetSegmentID: targetID,
 			OriginalText:    raw.OriginalText,
 			CorrectedText:   raw.CorrectedText,
 			Confidence:      raw.Confidence,
@@ -247,3 +253,15 @@ func (a diagnosticsWriterAdapter) WriteInteraction(stage string, requestMetadata
 		ErrorPayloadPath:    artifacts.ErrorPayloadPath,
 	}, nil
 }
+
+// segmentIDs records the request-local reference mapping in debug metadata.
+func segmentIDs(transcript *schema.Transcript) []int {
+	if transcript == nil {
+		return nil
+	}
+	ids := make([]int, len(transcript.Segments))
+	for i, segment := range transcript.Segments {
+		ids[i] = segment.ID
+	}
+	return ids
+}