知识库召回跳过租户IgnoreTenant

This commit is contained in:
2026-01-20 17:50:47 +08:00
parent d04305c325
commit 9689ce9e23

View File

@@ -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()) {