79 lines
2.6 KiB
Go
79 lines
2.6 KiB
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
|
|
}
|