fabced291c
- Added checks in various loading functions to ignore processing if the respective directories do not exist, improving error handling and preventing unnecessary operations. - Updated the Load function signature to return warnings alongside errors, allowing for better feedback on loading issues across different components.
95 lines
1.6 KiB
Go
95 lines
1.6 KiB
Go
package aigc
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/yaoapp/gou/application"
|
|
"github.com/yaoapp/yao/config"
|
|
"github.com/yaoapp/yao/share"
|
|
)
|
|
|
|
// Load load AIGC
|
|
func Load(cfg config.Config) error {
|
|
|
|
// Ignore if the aigcs directory does not exist
|
|
exists, err := application.App.Exists("aigcs")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if !exists {
|
|
return nil
|
|
}
|
|
|
|
exts := []string{"*.ai.yml", "*.ai.yaml"}
|
|
messages := []string{}
|
|
err = application.App.Walk("aigcs", func(root, file string, isdir bool) error {
|
|
if isdir {
|
|
return nil
|
|
}
|
|
|
|
id := share.ID(root, file)
|
|
_, err := LoadFile(file, id)
|
|
if err != nil {
|
|
messages = append(messages, err.Error())
|
|
}
|
|
|
|
return nil
|
|
}, exts...)
|
|
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if len(messages) > 0 {
|
|
return fmt.Errorf("%s", strings.Join(messages, ";\n"))
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// LoadFile load AIGC by file
|
|
func LoadFile(file string, id string) (*DSL, error) {
|
|
|
|
data, err := application.App.Read(file)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return LoadSource(data, file, id)
|
|
}
|
|
|
|
// LoadSource load AIGC
|
|
func LoadSource(data []byte, file, id string) (*DSL, error) {
|
|
|
|
dsl := DSL{
|
|
ID: id,
|
|
Optional: Optional{
|
|
Autopilot: false,
|
|
JSON: false,
|
|
},
|
|
}
|
|
|
|
err := application.Parse(file, data, &dsl)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if dsl.Prompts == nil || len(dsl.Prompts) == 0 {
|
|
return nil, fmt.Errorf("%s prompts is required", id)
|
|
}
|
|
|
|
// create AI interface
|
|
dsl.AI, err = dsl.newAI()
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
// add to autopilots
|
|
if dsl.Optional.Autopilot {
|
|
Autopilots = append(Autopilots, id)
|
|
}
|
|
|
|
// add to AIGCs
|
|
AIGCs[id] = &dsl
|
|
return AIGCs[id], nil
|
|
}
|