Share module proposal execution and transcript-section prompt payload helpers

internal/modules/grammar/module.go

@@ -3,11 +3,10 @@ package grammar
 import (
 	"context"
 
-	"gitea.maximumdirect.net/eric/audita/internal/core/schema"
 	"gitea.maximumdirect.net/eric/audita/internal/framework/contracts"
 	"gitea.maximumdirect.net/eric/audita/internal/framework/proposal_generation"
 	"gitea.maximumdirect.net/eric/audita/internal/framework/proposals"
-	"gitea.maximumdirect.net/eric/audita/internal/framework/stagename"
+	"gitea.maximumdirect.net/eric/audita/internal/prompts"
 	builtinvalidators "gitea.maximumdirect.net/eric/audita/internal/validators"
 )
 
@@ -37,62 +36,9 @@ func (m *Module) Validators() []contracts.Validator {
 }
 
 func (m *Module) Propose(ctx context.Context, req contracts.ProposalRequest) (contracts.ProposalResult, error) {
-	sectionTranscript := transcriptForSection(req.WorkingTranscript, req.Section)
-	sectionIndex := 0
-	if req.Section != nil {
-		sectionIndex = req.Section.Index
-	}
-	transcriptDescription := ""
-	if req.Config != nil {
-		transcriptDescription = req.Config.TranscriptDescription
-	}
-	messages, err := BuildProposalMessages(sectionTranscript, req.Glossary, sectionIndex, transcriptDescription)
-	if err != nil {
-		return contracts.ProposalResult{}, err
-	}
-
-	generated, err := proposal_generation.GenerateCandidates(ctx, proposal_generation.Request{
-		ModuleKey:         req.RunSpec.ModuleKey,
-		ModuleInstance:    req.RunSpec.InstanceName,
-		ReplacementPolicy: req.RunSpec.ReplacementPolicy,
-		WorkingTranscript: req.WorkingTranscript,
-		Section:           req.Section,
-		Glossary:          req.Glossary,
-		Config:            req.Config,
-		Messages:          messages,
-		PromptMetadata:    proposalPromptMetadata().DiagnosticsMap(),
-		StageName:         stagename.ModuleProposal(req.RunSpec.InstanceName, sectionIndexPtr(req.Section)),
-		StartIndex:        0,
-		LLMClient:         req.LLMClient,
-		Scheduler:         req.LLMScheduler,
-		DiagnosticsDir:    req.DiagnosticsDir,
+	return proposal_generation.ExecuteModuleProposal(ctx, proposal_generation.ModuleProposalRequest{
+		ProposalRequest: req,
+		PromptID:        prompts.PromptIDModuleGrammarProposal,
+		BuildMessages:   BuildProposalMessages,
 	})
-	if err != nil {
-		return contracts.ProposalResult{}, err
-	}
-	return contracts.ProposalResult{
-		Proposals: generated.Corrections,
-		Warnings:  generated.Warnings,
-	}, nil
-}
-
-func transcriptForSection(transcript *schema.Transcript, section *contracts.SectionMetadata) *schema.Transcript {
-	if section == nil || transcript == nil {
-		return transcript
-	}
-	segments := make([]schema.Segment, 0, len(transcript.Segments))
-	for _, seg := range transcript.Segments {
-		if seg.ID >= section.StartSegmentID && seg.ID <= section.EndSegmentID {
-			segments = append(segments, seg)
-		}
-	}
-	return &schema.Transcript{Segments: segments}
-}
-
-func sectionIndexPtr(section *contracts.SectionMetadata) *int {
-	if section == nil {
-		return nil
-	}
-	index := section.Index
-	return &index
 }