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);