83 lines
2.9 KiB
Go
83 lines
2.9 KiB
Go
package proposal_generation
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
|
|
"gitea.maximumdirect.net/eric/audita/internal/core/schema"
|
|
"gitea.maximumdirect.net/eric/audita/internal/framework/contracts"
|
|
"gitea.maximumdirect.net/eric/audita/internal/framework/stagename"
|
|
"gitea.maximumdirect.net/eric/audita/internal/prompts"
|
|
)
|
|
|
|
type ProposalVariableBuilder func(transcript *schema.Transcript, glossary *schema.Glossary, sectionIndex int, transcriptDescription string) (map[string]string, error)
|
|
|
|
type ModuleProposalRequest struct {
|
|
ProposalRequest contracts.ProposalRequest
|
|
PromptID string
|
|
BuildVariables ProposalVariableBuilder
|
|
}
|
|
|
|
// ExecuteModuleProposal runs shared proposal generation plumbing for one
|
|
// module, leaving only module-specific prompt message building at call sites.
|
|
func ExecuteModuleProposal(ctx context.Context, req ModuleProposalRequest) (contracts.ProposalResult, error) {
|
|
if req.BuildVariables == nil {
|
|
return contracts.ProposalResult{}, fmt.Errorf("proposal variable builder is required")
|
|
}
|
|
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 := ""
|
|
if req.ProposalRequest.Config != nil {
|
|
transcriptDescription = req.ProposalRequest.Config.Context.Description
|
|
}
|
|
|
|
variables, err := req.BuildVariables(
|
|
req.ProposalRequest.WorkingTranscript,
|
|
req.ProposalRequest.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{
|
|
ModuleKey: req.ProposalRequest.RunSpec.ModuleKey,
|
|
ModuleInstance: req.ProposalRequest.RunSpec.InstanceName,
|
|
ReplacementPolicy: req.ProposalRequest.RunSpec.ReplacementPolicy,
|
|
WorkingTranscript: req.ProposalRequest.WorkingTranscript,
|
|
Section: req.ProposalRequest.Section,
|
|
Glossary: req.ProposalRequest.Glossary,
|
|
Config: req.ProposalRequest.Config,
|
|
PromptID: promptMetadata.PromptID,
|
|
PromptVersion: promptMetadata.PromptVersion,
|
|
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
|
|
}
|