Simplify references and output instructions in LLM prompts

2026-09-25 12:38:23 +00:00
parent 325c4ea7d3
commit 011b021d47
31 changed files with 248 additions and 129 deletions
--- a/internal/modules/grammar/module_test.go
+++ b/internal/modules/grammar/module_test.go
@@ -42,10 +42,10 @@ func (f *fakeLLMClient) CompleteStructured(ctx context.Context, req contracts.St
 	items, _ := payload["corrections"].([]map[string]any)
 	for _, item := range items {
 		target.Corrections = append(target.Corrections, proposal_generation.StructuredCorrectionProposal{
-			TargetSegmentID: item["id"].(int),
-			OriginalText:    item["original_text"].(string),
-			CorrectedText:   item["corrected_text"].(string),
-			Confidence:      item["confidence"].(float64),
+			SegmentReference: item["id"].(int),
+			OriginalText:     item["original_text"].(string),
+			CorrectedText:    item["corrected_text"].(string),
+			Confidence:       item["confidence"].(float64),
 		})
 	}
 	return contracts.StructuredCompletionResponse{}, nil