From f1f2459d6015bf81a30b645598db519596b20040 Mon Sep 17 00:00:00 2001
From: Wenjie Zhang <xerrors@163.com>
Date: Thu, 20 Mar 2025 20:11:27 +0800
Subject: [PATCH] fix embedding bug

---
 src/core/knowledgebase.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/core/knowledgebase.py b/src/core/knowledgebase.py
index dbce831b..a79380f5 100644
--- a/src/core/knowledgebase.py
+++ b/src/core/knowledgebase.py
@@ -184,7 +184,7 @@ class KnowledgeBase:
         return {"message": "删除成功"}
 
     def restart(self):
-        self.embed_model = get_embedding_model(config)
+        self._load_models()
         self._load_databases()
 
     ################################