diff --git a/aigc/process.go b/aigc/process.go index de01b950..3e044d33 100644 --- a/aigc/process.go +++ b/aigc/process.go @@ -15,7 +15,7 @@ func processAigcs(process *process.Process) interface{} { process.ValidateArgNums(1) aigc, err := Select(process.ID) if err != nil { - exception.New("scripts.%s not loaded", 404, process.ID).Throw() + exception.New("aigcs.%s not loaded", 404, process.ID).Throw() return nil } diff --git a/engine/load.go b/engine/load.go index 518fc2c8..73cf2cbf 100644 --- a/engine/load.go +++ b/engine/load.go @@ -9,6 +9,7 @@ import ( "github.com/fatih/color" "github.com/yaoapp/gou/application" "github.com/yaoapp/kun/exception" + "github.com/yaoapp/yao/aigc" "github.com/yaoapp/yao/api" "github.com/yaoapp/yao/cert" "github.com/yaoapp/yao/config" @@ -177,6 +178,11 @@ func Load(cfg config.Config) (err error) { printErr(cfg.Mode, "Widget", err) } + err = aigc.Load(cfg) + if err != nil { + printErr(cfg.Mode, "AIGC", err) + } + return nil }