From 6ec07686a98be083c20e5e88dd43aea52d6c6529 Mon Sep 17 00:00:00 2001
From: Chuck1sn <chuck1inzl@gmail.com>
Date: Fri, 27 Jun 2025 22:57:00 +0800
Subject: [PATCH] Long libraryId

---
 .../main/java/com/zl/mjga/controller/LibraryController.java    | 2 +-
 backend/src/main/java/com/zl/mjga/service/RagService.java      | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/backend/src/main/java/com/zl/mjga/controller/LibraryController.java b/backend/src/main/java/com/zl/mjga/controller/LibraryController.java
index 0adc032..88d4da7 100644
--- a/backend/src/main/java/com/zl/mjga/controller/LibraryController.java
+++ b/backend/src/main/java/com/zl/mjga/controller/LibraryController.java
@@ -85,7 +85,7 @@ public class LibraryController {
     Long libraryDocId =
         ragService.createLibraryDocBy(
             Long.valueOf(libraryId), objectName, multipartFile.getOriginalFilename());
-    ragService.embeddingAndCreateDocSegment(libraryDocId, objectName);
+    ragService.embeddingAndCreateDocSegment(Long.valueOf(libraryId), libraryDocId, objectName);
     return objectName;
   }
 }
diff --git a/backend/src/main/java/com/zl/mjga/service/RagService.java b/backend/src/main/java/com/zl/mjga/service/RagService.java
index ed46275..b2e5b26 100644
--- a/backend/src/main/java/com/zl/mjga/service/RagService.java
+++ b/backend/src/main/java/com/zl/mjga/service/RagService.java
@@ -109,7 +109,7 @@ public class RagService {
   }
 
   @Async
-  public void embeddingAndCreateDocSegment(Long libraryDocId, String objectName) {
+  public void embeddingAndCreateDocSegment(Long libraryId, Long libraryDocId, String objectName) {
     Document document =
         amazonS3DocumentLoader.loadDocument(
             minIoConfig.getDefaultBucket(), objectName, new ApacheTikaDocumentParser());
@@ -123,6 +123,7 @@ public class RagService {
               Response<Embedding> embed = zhipuEmbeddingModel.embed(textSegment);
               Integer tokenUsage = embed.tokenUsage().totalTokenCount();
               Embedding vector = embed.content();
+              textSegment.metadata().put("libraryId", libraryId);
               String embeddingId = zhiPuLibraryEmbeddingStore.add(vector, textSegment);
               LibraryDocSegment libraryDocSegment = new LibraryDocSegment();
               libraryDocSegment.setEmbeddingId(embeddingId);