69 lines
1.2 KiB
Go
69 lines
1.2 KiB
Go
package aigc
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/yaoapp/gou/application"
|
|
"github.com/yaoapp/yao/config"
|
|
"github.com/yaoapp/yao/share"
|
|
)
|
|
|
|
// Load load AIGC
|
|
func Load(cfg config.Config) error {
|
|
exts := []string{"*.ai.yml", "*.ai.yaml"}
|
|
return application.App.Walk("aigcs", func(root, file string, isdir bool) error {
|
|
if isdir {
|
|
return nil
|
|
}
|
|
|
|
id := share.ID(root, file)
|
|
_, err := LoadFile(file, id)
|
|
return err
|
|
}, exts...)
|
|
}
|
|
|
|
// 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
|
|
}
|