From b210aa7ab5ed474a1f50e7b403ee6222af79a064 Mon Sep 17 00:00:00 2001
From: Max <shadow.iqka@qq.com>
Date: Thu, 27 Apr 2023 22:48:20 +0800
Subject: [PATCH] [_]

---
 aigc/process.go | 2 +-
 engine/load.go  | 6 ++++++
 2 files changed, 7 insertions(+), 1 deletion(-)

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
 }