xieyinlu/zhilu-admin
1
0
Fork 0
mirror of https://github.com/ccmjga/zhilu-admin synced 2026-04-10 16:07:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Long libraryId

Browse Source
This commit is contained in:
Chuck1sn
2025-06-27 22:57:00 +08:00
parent 4d70b49e61
commit 6ec07686a9
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
backend/src/main/java/com/zl/mjga/controller/LibraryController.java
View File

@@ -85,7 +85,7 @@ public class LibraryController {
Long libraryDocId = Long libraryDocId =
ragService.createLibraryDocBy( ragService.createLibraryDocBy(
Long.valueOf(libraryId), objectName, multipartFile.getOriginalFilename()); Long.valueOf(libraryId), objectName, multipartFile.getOriginalFilename());
ragService.embeddingAndCreateDocSegment(libraryDocId, objectName); ragService.embeddingAndCreateDocSegment(Long.valueOf(libraryId), libraryDocId, objectName);
return objectName; return objectName;
} }
} }

3
backend/src/main/java/com/zl/mjga/service/RagService.java
View File

@@ -109,7 +109,7 @@ public class RagService {
} }
@Async @Async
public void embeddingAndCreateDocSegment(Long libraryDocId, String objectName) { public void embeddingAndCreateDocSegment(Long libraryId, Long libraryDocId, String objectName) {
Document document = Document document =
amazonS3DocumentLoader.loadDocument( amazonS3DocumentLoader.loadDocument(
minIoConfig.getDefaultBucket(), objectName, new ApacheTikaDocumentParser()); minIoConfig.getDefaultBucket(), objectName, new ApacheTikaDocumentParser());
@@ -123,6 +123,7 @@ public class RagService {
Response<Embedding> embed = zhipuEmbeddingModel.embed(textSegment); Response<Embedding> embed = zhipuEmbeddingModel.embed(textSegment);
Integer tokenUsage = embed.tokenUsage().totalTokenCount(); Integer tokenUsage = embed.tokenUsage().totalTokenCount();
Embedding vector = embed.content(); Embedding vector = embed.content();
textSegment.metadata().put("libraryId", libraryId);
String embeddingId = zhiPuLibraryEmbeddingStore.add(vector, textSegment); String embeddingId = zhiPuLibraryEmbeddingStore.add(vector, textSegment);
LibraryDocSegment libraryDocSegment = new LibraryDocSegment(); LibraryDocSegment libraryDocSegment = new LibraryDocSegment();
libraryDocSegment.setEmbeddingId(embeddingId); libraryDocSegment.setEmbeddingId(embeddingId);