ZF/weiyu
1
0
Fork 0
mirror of https://gitee.com/270580156/weiyu.git synced 2026-05-18 21:27:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
jack ning
2025-05-14 10:30:52 +08:00
parent 44b08d827e
commit 7aa3f10f0e
1 changed files with 17 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
modules/kbase/src/main/java/com/bytedesk/kbase/llm_text/TextElasticService.java
View File

@@ -17,6 +17,7 @@ import java.time.LocalDateTime;
import java.time.format.DateTimeFormatter;
import java.util.ArrayList;
import java.util.List;
import java.util.regex.Pattern;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.elasticsearch.client.elc.NativeQuery;
@@ -157,6 +158,9 @@ public class TextElasticService {
}
try {
// 转义正则表达式特殊字符
query = Pattern.quote(query);
// 构建查询条件
BoolQuery.Builder boolQueryBuilder = new BoolQuery.Builder();
@@ -238,9 +242,10 @@ public class TextElasticService {
int end = Math.min(content.length(), index + query.length() + 50);
String snippet = content.substring(start, end);
// 添加高亮标记
// 添加高亮标记 - 对特殊正则表达式字符进行转义
String queryEscaped = Pattern.quote(query);
String highlighted = snippet.replaceAll(
"(?i)" + query,
"(?i)" + queryEscaped,
"<em>" + query + "</em>"
);
@@ -257,8 +262,9 @@ public class TextElasticService {
if (title != null && !title.trim().isEmpty() && query != null && !query.isEmpty()) {
// 在包含查询词的部分手动添加高亮标签
if (title.toLowerCase().contains(query.toLowerCase())) {
String queryEscaped = Pattern.quote(query);
String highlighted = title.replaceAll(
"(?i)" + query,
"(?i)" + queryEscaped,
"<em>" + query + "</em>"
);
result.setHighlightedName(highlighted);
@@ -295,6 +301,9 @@ public class TextElasticService {
}
try {
// 转义正则表达式特殊字符
query = Pattern.quote(query);
// 构建查询条件
BoolQuery.Builder boolQueryBuilder = new BoolQuery.Builder();
@@ -383,9 +392,10 @@ public class TextElasticService {
int end = Math.min(content.length(), index + query.length() + 50);
String snippet = content.substring(start, end);
// 添加高亮标记
// 添加高亮标记 - 对特殊正则表达式字符进行转义
String queryEscaped = Pattern.quote(query);
String highlighted = snippet.replaceAll(
"(?i)" + query,
"(?i)" + queryEscaped,
"<em>" + query + "</em>"
);
@@ -402,8 +412,9 @@ public class TextElasticService {
if (title != null && !title.trim().isEmpty() && query != null && !query.isEmpty()) {
// 在包含查询词的部分手动添加高亮标签
if (title.toLowerCase().contains(query.toLowerCase())) {
String queryEscaped = Pattern.quote(query);
String highlighted = title.replaceAll(
"(?i)" + query,
"(?i)" + queryEscaped,
"<em>" + query + "</em>"
);
result.setHighlightedName(highlighted);