From 40519e84294fb8c3e26a95dcd94a4e1fb2a7f62b Mon Sep 17 00:00:00 2001
From: Max <shadow.iqka@qq.com>
Date: Wed, 23 Jul 2025 17:56:51 +0800
Subject: [PATCH] Refactor Knowledge Base loading in Load and Reload functions

- Moved Knowledge Base loading logic from the Load function to ensure it is consistently handled in both Load and Reload functions.
- Improved error handling by printing errors during Knowledge Base loading in the Reload function, enhancing user feedback.
---
 engine/load.go | 21 ++++++++++++++-------
 1 file changed, 14 insertions(+), 7 deletions(-)

diff --git a/engine/load.go b/engine/load.go
index 7f20f3c6..71b2deee 100644
--- a/engine/load.go
+++ b/engine/load.go
@@ -266,6 +266,13 @@ func Load(cfg config.Config, options LoadOption) (warnings []Warning, err error)
 		warnings = append(warnings, Warning{Widget: "Pipe", Error: err})
 	}
 
+	// Load Knowledge Base
+	_, err = kb.Load(cfg)
+	if err != nil {
+		// printErr(cfg.Mode, "Knowledge Base", err)
+		warnings = append(warnings, Warning{Widget: "Knowledge Base", Error: err})
+	}
+
 	// Load Neo
 	err = neo.Load(cfg)
 	if err != nil {
@@ -281,13 +288,6 @@ func Load(cfg config.Config, options LoadOption) (warnings []Warning, err error)
 		}
 	}
 
-	// Load Knowledge Base
-	_, err = kb.Load(cfg)
-	if err != nil {
-		// printErr(cfg.Mode, "Knowledge Base", err)
-		warnings = append(warnings, Warning{Widget: "Knowledge Base", Error: err})
-	}
-
 	// Load OpenAPI
 	_, err = openapi.Load(cfg)
 	if err != nil {
@@ -478,6 +478,13 @@ func Reload(cfg config.Config, options LoadOption) (err error) {
 		printErr(cfg.Mode, "AIGC", err)
 	}
 
+	// Load Knowledge Base
+	_, err = kb.Load(cfg)
+	if err != nil {
+		printErr(cfg.Mode, "Knowledge Base", err)
+
+	}
+
 	// Load Neo
 	err = neo.Load(cfg)
 	if err != nil {