kennfalcon/entity-manage
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 实体召回支持模糊匹配

Browse Source
This commit is contained in:
杨丛聿
2025-02-11 17:12:22 +08:00
committed by haokai
parent 56a3570b3e
commit 5d9ac1a4b3
3 changed files with 16 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
.gitignore vendored
View File

@@ -20,6 +20,7 @@ spring-*/src/main/java/META-INF/MANIFEST.MF
*.ipr
*.iws
.idea
.smarttomcat
out
test-output
atlassian-ide-plugin.xml

2
entity-manage-service/src/main/java/com/shuwen/data/entity/manage/service/impl/GraphChatRetrievalService.java
View File

@@ -23,6 +23,7 @@ import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import java.util.*;
import static com.shuwen.data.entity.manage.common.entity.constant.GraphConstants.SearchMode.COMPLEX;
import static com.shuwen.data.entity.manage.service.chat.utils.RelationRephraseUtils.SNOWFLAKE_EXTEND_REL;
/**
@@ -75,6 +76,7 @@ public class GraphChatRetrievalService {
searchDto.setQueryString(queryString);
if (StringUtils.isNotEmpty(query)) {
searchDto.setQuery(query);
searchDto.setMode(COMPLEX);
}
searchDto.setSize(size);
Set<String> extFields = new HashSet<>();

21
entity-manage-service/src/main/java/com/shuwen/data/entity/manage/service/impl/GraphChatService.java
View File

@@ -722,15 +722,20 @@ public class GraphChatService implements IGraphChatService {
return;
}
//获取多个重名的
String query = "";
String query = slots.getSubject();
String queryString = "";
if (AwardUtils.isAward(slots.getSubject())) {
query = slots.getSubject();
} else {
queryString = "(name:" + slots.getSubject() + " OR alias:" + slots.getSubject() + ") AND summary:*";
if (slots.getSubject().contains(":") || slots.getSubject().contains("：")) {
queryString = "(name:" + "\"" + slots.getSubject() + "\"" + " OR alias:" + "\"" + slots.getSubject() + "\"" + ") AND summary:*";
}
// if (AwardUtils.isAward(slots.getSubject())) {
// query = slots.getSubject();
// } else {
// queryString = "(name:" + slots.getSubject() + " OR alias:" + slots.getSubject() + ") AND summary:*";
// if (slots.getSubject().contains(":") || slots.getSubject().contains("：")) {
// queryString = "(name:" + "\"" + slots.getSubject() + "\"" + " OR alias:" + "\"" + slots.getSubject() + "\"" + ") AND summary:*";
// }
// queryString = "summary:*";
// }
//奖项信息不在summary中
if (!AwardUtils.isAward(slots.getSubject())) {
queryString = "summary:*";
}
if (slots.getSubjectType().equals("creative_work")) {
restrict = "";