69 lines
2.4 KiB
Go
69 lines
2.4 KiB
Go
package consumers
|
|
|
|
import (
|
|
"strings"
|
|
|
|
configkit "gitea.maximumdirect.net/eric/ConfigKit"
|
|
)
|
|
|
|
// PromptCatalogConfig is an application-owned prompt collection shape.
|
|
type PromptCatalogConfig struct {
|
|
Prompts []PromptRecord `yaml:"prompts"`
|
|
}
|
|
|
|
// PromptRecord keeps its identity and message meaning in the consumer.
|
|
type PromptRecord struct {
|
|
ID string `yaml:"id"`
|
|
Name string `yaml:"name"`
|
|
Messages []string `yaml:"messages"`
|
|
Resource string `yaml:"resource"`
|
|
}
|
|
|
|
// LoadPromptCatalog discovers explicitly registered prompt records in the main
|
|
// tier and replaces the lower-priority default collection as a whole.
|
|
func LoadPromptCatalog(mainPath string) (*configkit.Result[PromptCatalogConfig], error) {
|
|
return configkit.LoadLayered(configkit.Input{
|
|
Source: configkit.File(mainPath),
|
|
Imports: true,
|
|
}, configkit.LayeredOptions[PromptCatalogConfig]{
|
|
Underlays: []configkit.Underlay{{Inputs: []configkit.Input{{
|
|
Source: configkit.Bytes("default prompts", []byte("prompts:\n - id: default\n name: Default prompt\n messages: [Default message]\n")),
|
|
}}}},
|
|
ImportDefaults: configkit.ImportDefaults{
|
|
Pattern: []string{"curated*.yml", "generated-*.yml"},
|
|
},
|
|
Definition: promptCatalogDefinition(),
|
|
})
|
|
}
|
|
|
|
// LoadRequiredPromptCatalog keeps the common one-file import shape small.
|
|
func LoadRequiredPromptCatalog(mainPath string) (*configkit.Result[PromptCatalogConfig], error) {
|
|
return configkit.Load(configkit.File(mainPath), configkit.Options[PromptCatalogConfig]{
|
|
Imports: true, Definition: promptCatalogDefinition(),
|
|
})
|
|
}
|
|
|
|
func promptCatalogDefinition() configkit.Definition[PromptCatalogConfig] {
|
|
return configkit.Definition[PromptCatalogConfig]{
|
|
SequenceCompositions: []configkit.SequenceComposition{{
|
|
Path: configkit.Keys("prompts"),
|
|
IdentityField: "id",
|
|
}},
|
|
Validate: validatePromptCatalog,
|
|
}
|
|
}
|
|
|
|
func validatePromptCatalog(configuration *PromptCatalogConfig, validation *configkit.Validation) {
|
|
for index, prompt := range configuration.Prompts {
|
|
if strings.TrimSpace(prompt.Name) == "" {
|
|
validation.Error(configkit.Keys("prompts").Index(index).Key("name"), "must not be empty")
|
|
}
|
|
if len(prompt.Messages) == 0 {
|
|
validation.Error(configkit.Keys("prompts").Index(index).Key("messages"), "must contain at least one message")
|
|
}
|
|
}
|
|
if len(configuration.Prompts) == 0 {
|
|
validation.Error(configkit.Keys("prompts"), "must contain at least one prompt")
|
|
}
|
|
}
|