98 lines
2.2 KiB
Go
98 lines
2.2 KiB
Go
package scenes
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"gitea.maximumdirect.net/eric/notarius/internal/core/source"
|
|
"gitea.maximumdirect.net/eric/notarius/internal/framework/contracts"
|
|
"gitea.maximumdirect.net/eric/notarius/internal/framework/prompt"
|
|
)
|
|
|
|
type promptData struct {
|
|
SourceID string
|
|
Units []promptUnit
|
|
}
|
|
|
|
type promptUnit struct {
|
|
ID string
|
|
Text string
|
|
Metadata []promptMetadata
|
|
}
|
|
|
|
type promptMetadata struct {
|
|
Key string
|
|
Value string
|
|
}
|
|
|
|
var scenesPromptBundle = mustLoadPromptBundle()
|
|
|
|
func mustLoadPromptBundle() *prompt.Bundle {
|
|
bundle, err := prompt.LoadBundle(embeddedAssets, prompt.Definition{
|
|
PromptID: PromptID,
|
|
Version: ResponseSchemaVersion,
|
|
EmbeddedPath: "assets/prompts",
|
|
SystemPath: "assets/prompts/system.md",
|
|
UserPath: "assets/prompts/user.md",
|
|
})
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
return bundle
|
|
}
|
|
|
|
func buildPromptData(req contracts.ChunkRequest) (promptData, error) {
|
|
if req.Source == nil {
|
|
return promptData{}, fmt.Errorf("dnd scenes prompt: source must not be nil")
|
|
}
|
|
|
|
data := promptData{
|
|
SourceID: req.Source.ID,
|
|
Units: make([]promptUnit, 0, len(req.Source.Units)),
|
|
}
|
|
for _, unit := range req.Source.Units {
|
|
data.Units = append(data.Units, promptUnit{
|
|
ID: unit.ID,
|
|
Text: unit.Text,
|
|
Metadata: selectedMetadata(unit),
|
|
})
|
|
}
|
|
return data, nil
|
|
}
|
|
|
|
func renderPrompt(req contracts.ChunkRequest) (system string, user string, metadata prompt.Metadata, err error) {
|
|
data, err := buildPromptData(req)
|
|
if err != nil {
|
|
return "", "", prompt.Metadata{}, err
|
|
}
|
|
system, user, metadata, err = scenesPromptBundle.RenderUserSystem(data)
|
|
if err != nil {
|
|
return "", "", prompt.Metadata{}, fmt.Errorf("dnd scenes prompt: %w", err)
|
|
}
|
|
return system, user, metadata, nil
|
|
}
|
|
|
|
func selectedMetadata(unit source.SourceUnit) []promptMetadata {
|
|
if len(unit.Metadata) == 0 {
|
|
return nil
|
|
}
|
|
|
|
keys := []string{"speaker", "start", "end"}
|
|
metadata := make([]promptMetadata, 0, len(keys))
|
|
for _, key := range keys {
|
|
value, ok := unit.Metadata[key]
|
|
if !ok {
|
|
continue
|
|
}
|
|
rendered := strings.TrimSpace(fmt.Sprint(value))
|
|
if rendered == "" {
|
|
continue
|
|
}
|
|
metadata = append(metadata, promptMetadata{
|
|
Key: key,
|
|
Value: rendered,
|
|
})
|
|
}
|
|
return metadata
|
|
}
|