xieyinlu/ruoyi-ai
1
0
Fork 0
mirror of https://gitcode.com/ageerle/ruoyi-ai.git synced 2026-04-02 14:33:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: 新增pdf图片解析分支

Browse Source
This commit is contained in:
ageerle
2025-05-19 15:33:29 +08:00
parent 57693059a1
commit a4e995d46c
2 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
ruoyi-modules-api/ruoyi-knowledge-api/src/main/java/org/ruoyi/service/impl/VectorStoreServiceImpl.java
View File

@@ -1,13 +1,11 @@
package org.ruoyi.service.impl;
import cn.hutool.core.util.RandomUtil;
import com.google.protobuf.ServiceException;
import dev.langchain4j.data.embedding.Embedding;
import dev.langchain4j.data.segment.TextSegment;
import dev.langchain4j.model.embedding.EmbeddingModel;
import dev.langchain4j.model.ollama.OllamaEmbeddingModel;
import dev.langchain4j.model.openai.OpenAiEmbeddingModel;
import dev.langchain4j.model.output.Response;
import dev.langchain4j.store.embedding.EmbeddingMatch;
import dev.langchain4j.store.embedding.EmbeddingSearchRequest;
import dev.langchain4j.store.embedding.EmbeddingStore;
@@ -31,6 +29,7 @@ import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;
/**
* 向量库管理

8
ruoyi-modules/ruoyi-chat/src/main/java/org/ruoyi/chat/service/knowledge/KnowledgeInfoServiceImpl.java
View File

@@ -354,7 +354,7 @@ public class KnowledgeInfoServiceImpl implements IKnowledgeInfoService {
/**
* 第二步 定时 解析图片内容
*/
//@Scheduled(fixedDelay = 15000)
@Scheduled(fixedDelay = 15000)
public void dealKnowledgeAttachPicAnys() throws Exception {
//获取未处理的图片记录
List<KnowledgeAttachPic> knowledgeAttachPics = picMapper.selectList(
@@ -371,7 +371,7 @@ public class KnowledgeInfoServiceImpl implements IKnowledgeInfoService {
/**
* 第三步 定时 处理 附件上传后上传向量数据库
*/
//@Scheduled(fixedDelay = 30000) // 每3秒执行一次
@Scheduled(fixedDelay = 30000) // 每3秒执行一次
public void dealKnowledgeAttachVector() throws Exception {
//处理 需要上传向量数据库的记录
List<KnowledgeAttach> knowledgeAttaches = attachMapper.selectList(
@@ -390,7 +390,7 @@ public class KnowledgeInfoServiceImpl implements IKnowledgeInfoService {
/**
* 第四步 定时 处理 失败数据
*/
//@Scheduled(fixedDelay = 30 * 60 * 1000)
@Scheduled(fixedDelay = 30 * 60 * 1000)
public void dealKnowledge40Status() throws Exception {
//拆解PDF失败 重新设置状态
attachMapper.update(new LambdaUpdateWrapper<KnowledgeAttach>()
@@ -405,5 +405,5 @@ public class KnowledgeInfoServiceImpl implements IKnowledgeInfoService {
.set(KnowledgeAttach::getVectorStatus, DealStatus.STATUS_10)
.eq(KnowledgeAttach::getVectorStatus, DealStatus.STATUS_40));
}
@Scheduled(fixedDelay = 180000) // 3分钟执行一次
}