知识库召回跳过租户IgnoreTenant
This commit is contained in:
@@ -15,6 +15,7 @@ import com.gxwebsoft.ai.entity.AiCloudFile;
|
|||||||
import com.gxwebsoft.ai.factory.KnowledgeBaseClientFactory;
|
import com.gxwebsoft.ai.factory.KnowledgeBaseClientFactory;
|
||||||
import com.gxwebsoft.ai.service.AiCloudFileService;
|
import com.gxwebsoft.ai.service.AiCloudFileService;
|
||||||
import com.gxwebsoft.ai.util.KnowledgeBaseUtil;
|
import com.gxwebsoft.ai.util.KnowledgeBaseUtil;
|
||||||
|
import com.gxwebsoft.common.core.context.TenantContext;
|
||||||
|
|
||||||
import cn.hutool.core.util.StrUtil;
|
import cn.hutool.core.util.StrUtil;
|
||||||
import cn.hutool.http.HttpUtil;
|
import cn.hutool.http.HttpUtil;
|
||||||
@@ -123,7 +124,7 @@ public abstract class AbstractAuditContentService {
|
|||||||
}
|
}
|
||||||
|
|
||||||
// 2. 批量查询文件URL
|
// 2. 批量查询文件URL
|
||||||
Map<String, String> fileUrlMap = batchQueryFileUrls(allNodes);
|
Map<String, String> fileUrlMap = TenantContext.callIgnoreTenant(() -> batchQueryFileUrls(allNodes));
|
||||||
|
|
||||||
// 3. 处理节点生成结果
|
// 3. 处理节点生成结果
|
||||||
return processNodesToResults(allNodes, fileUrlMap);
|
return processNodesToResults(allNodes, fileUrlMap);
|
||||||
@@ -168,7 +169,7 @@ public abstract class AbstractAuditContentService {
|
|||||||
/**
|
/**
|
||||||
* 批量查询文件URL
|
* 批量查询文件URL
|
||||||
*/
|
*/
|
||||||
private Map<String, String> batchQueryFileUrls(List<RetrieveResponseBodyDataNodes> nodes) {
|
protected Map<String, String> batchQueryFileUrls(List<RetrieveResponseBodyDataNodes> nodes) {
|
||||||
// 收集所有文档ID
|
// 收集所有文档ID
|
||||||
Set<String> docIds = nodes.stream().map(this::extractDocumentId).filter(StrUtil::isNotBlank).collect(Collectors.toSet());
|
Set<String> docIds = nodes.stream().map(this::extractDocumentId).filter(StrUtil::isNotBlank).collect(Collectors.toSet());
|
||||||
if (docIds.isEmpty()) {
|
if (docIds.isEmpty()) {
|
||||||
|
|||||||
Reference in New Issue
Block a user