From 92c339c22257319ee3d6d424c09fcccf323f486e Mon Sep 17 00:00:00 2001
From: moyangzhan <moyangzhan@qq.com>
Date: Fri, 22 Mar 2024 00:18:34 +0800
Subject: [PATCH] fix: knowledge base embedding

---
 .../moyz/adi/common/service/KnowledgeBaseItemService.java    | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/adi-common/src/main/java/com/moyz/adi/common/service/KnowledgeBaseItemService.java b/adi-common/src/main/java/com/moyz/adi/common/service/KnowledgeBaseItemService.java
index 3b9bef2..85db6ba 100644
--- a/adi-common/src/main/java/com/moyz/adi/common/service/KnowledgeBaseItemService.java
+++ b/adi-common/src/main/java/com/moyz/adi/common/service/KnowledgeBaseItemService.java
@@ -116,6 +116,11 @@ public class KnowledgeBaseItemService extends ServiceImpl<KnowledgeBaseItemMappe
         metadata.add("kb_item_uuid", kbItem.getUuid());
         Document document = new Document(kbItem.getRemark(), metadata);
         ragService.ingest(document);
+
+        ChainWrappers.lambdaUpdateChain(baseMapper)
+                .eq(KnowledgeBaseItem::getId, kbItem.getId())
+                .set(KnowledgeBaseItem::getIsEmbedded, true)
+                .update();
         return true;
     }