audita/internal/framework/proposal_generation/module_proposal.go

package proposal_generation

import (
	"context"
	"fmt"
	"strings"

	"gitea.maximumdirect.net/eric/audita/v3/internal/framework/contracts"
	"gitea.maximumdirect.net/eric/audita/v3/internal/framework/promptcontext"
	"gitea.maximumdirect.net/eric/audita/v3/internal/framework/stagename"
	"gitea.maximumdirect.net/eric/audita/v3/internal/prompts"
)

type ModuleProposalRequest struct {
	ProposalRequest contracts.ProposalRequest
	PromptID        string
}

// ExecuteModuleProposal prepares and runs a module's selected prompt.
func ExecuteModuleProposal(ctx context.Context, req ModuleProposalRequest) (contracts.ProposalResult, error) {
	if strings.TrimSpace(req.PromptID) == "" {
		return contracts.ProposalResult{}, fmt.Errorf("prompt ID must not be empty")
	}

	sectionIndex := 0
	if req.ProposalRequest.Section != nil {
		sectionIndex = req.ProposalRequest.Section.Index
	}

	transcriptDescription := req.ProposalRequest.Settings.Context.Description

	glossary := req.ProposalRequest.GlossaryVariables
	if glossary == nil {
		prepared, err := promptcontext.PrepareGlossaryVariables(req.ProposalRequest.Glossary, req.ProposalRequest.ProtectedTerms)
		if err != nil {
			return contracts.ProposalResult{}, err
		}
		glossary = &prepared
	}
	variables, err := buildProposalVariables(
		req.ProposalRequest.WorkingTranscript,
		*glossary,
		sectionIndex,
		transcriptDescription,
	)
	if err != nil {
		return contracts.ProposalResult{}, err
	}

	promptMetadata, ok := prompts.LookupMetadata(req.PromptID)
	if !ok {
		return contracts.ProposalResult{}, fmt.Errorf("unknown prompt ID %q", req.PromptID)
	}

	generated, err := GenerateCandidates(ctx, Request{
		StageIndex: req.ProposalRequest.StageIndex, StageID: req.ProposalRequest.StageID,
		ModuleKey:             req.ProposalRequest.RunSpec.ModuleKey,
		ModuleInstance:        req.ProposalRequest.RunSpec.InstanceName,
		ReplacementPolicy:     req.ProposalRequest.RunSpec.ReplacementPolicy,
		WorkingTranscript:     req.ProposalRequest.WorkingTranscript,
		Section:               req.ProposalRequest.Section,
		PromptID:              promptMetadata.PromptID,
		Variables:             variables,
		StageName:             stagename.ModuleProposal(req.ProposalRequest.RunSpec.InstanceName, sectionIndexPtr(req.ProposalRequest.Section)),
		StartIndex:            0,
		LLMClient:             req.ProposalRequest.LLMClient,
		DiagnosticsDir:        req.ProposalRequest.DiagnosticsDir,
		ReportDiagnosticError: req.ProposalRequest.ReportDiagnosticError,
	})
	if err != nil {
		return contracts.ProposalResult{Operations: generated.Operations}, err
	}

	return contracts.ProposalResult{
		Proposals:  generated.Corrections,
		Operations: generated.Operations,
	}, nil
}