72 lines
2.2 KiB
Go
72 lines
2.2 KiB
Go
package glossary
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"gitea.maximumdirect.net/eric/audita/internal/core/schema"
|
|
"gitea.maximumdirect.net/eric/audita/internal/framework/contracts"
|
|
"gitea.maximumdirect.net/eric/audita/internal/framework/promptcontext"
|
|
"gitea.maximumdirect.net/eric/audita/internal/prompts"
|
|
)
|
|
|
|
type promptSegment struct {
|
|
ID int `json:"id"`
|
|
Speaker string `json:"speaker"`
|
|
Start float64 `json:"start"`
|
|
End float64 `json:"end"`
|
|
Text string `json:"text"`
|
|
Categories []string `json:"categories,omitempty"`
|
|
}
|
|
|
|
type promptTranscriptSection struct {
|
|
SectionIndex int `json:"section_index"`
|
|
Segments []promptSegment `json:"segments"`
|
|
}
|
|
|
|
func BuildProposalMessages(transcript *schema.Transcript, glossary *schema.Glossary, sectionIndex int, transcriptDescription string) ([]contracts.LLMMessage, error) {
|
|
glossaryJSON, err := json.MarshalIndent(glossary, "", " ")
|
|
if err != nil {
|
|
return nil, fmt.Errorf("marshal glossary prompt context: %w", err)
|
|
}
|
|
|
|
sectionPayload := promptTranscriptSection{
|
|
SectionIndex: sectionIndex,
|
|
Segments: make([]promptSegment, 0),
|
|
}
|
|
if transcript != nil {
|
|
for _, s := range transcript.Segments {
|
|
sectionPayload.Segments = append(sectionPayload.Segments, promptSegment{
|
|
ID: s.ID,
|
|
Speaker: s.Speaker,
|
|
Start: s.Start,
|
|
End: s.End,
|
|
Text: s.Text,
|
|
Categories: append([]string(nil), s.Categories...),
|
|
})
|
|
}
|
|
}
|
|
sectionJSON, err := json.MarshalIndent(sectionPayload, "", " ")
|
|
if err != nil {
|
|
return nil, fmt.Errorf("marshal transcript prompt context: %w", err)
|
|
}
|
|
|
|
system, user, _, err := prompts.RenderUserSystem(prompts.PromptIDModuleGlossaryProposal, map[string]string{
|
|
"TranscriptDescriptionBlock": promptcontext.TranscriptDescriptionBlock(transcriptDescription),
|
|
"GlossaryJSON": string(glossaryJSON),
|
|
"SectionJSON": string(sectionJSON),
|
|
})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return []contracts.LLMMessage{
|
|
{Role: "system", Content: system},
|
|
{Role: "user", Content: user},
|
|
}, nil
|
|
}
|
|
|
|
func proposalPromptMetadata() prompts.Metadata {
|
|
return prompts.MustLookupMetadata(prompts.PromptIDModuleGlossaryProposal)
|
|
}
|