ZF/SmartJavaAI
1
0
Fork 0
mirror of https://github.com/geekwenjie/SmartJavaAI.git synced 2026-09-13 13:18:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

- 新增机器翻译模块：支持 200+ 种语言之间的相互翻译

Browse Source 
- 人脸识别模块：修复批量删除人脸数据时的异常问题
- 人脸识别模块：修复人脸检索 Top大于 1 时报异常问题
This commit is contained in:
dengwenjie
2025-06-18 20:05:40 +08:00
parent f7a59cc3aa
commit a6b3655907
24 changed files with 90 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
smartjavaai-face/src/main/java/cn/smartjavaai/face/vector/core/MilvusClient.java
View File

@@ -397,11 +397,18 @@ public class MilvusClient implements VectorDBClient {
// 构建IN表达式: id in ["id1", "id2", ...]
StringBuilder expr = new StringBuilder(VectorDBConstants.FieldNames.ID_FIELD + " in [");
for (int i = 0; i < ids.size(); i++) {
expr.append(ids.get(i)); // 不加引号
if(config.getIdStrategy() == IdStrategy.AUTO){
expr.append(ids.get(i)); // 不加引号
}else{
expr.append("'" + ids.get(i) + "'"); // 不加引号
}
if (i < ids.size() - 1) {
expr.append(", ");
}
}
expr.append("]");
DeleteParam deleteParam = DeleteParam.newBuilder()
.withCollectionName(collectionName)
@@ -453,7 +460,6 @@ public class MilvusClient implements VectorDBClient {
// 3. 获取字段数据（FieldData）
List<SearchResultsWrapper.IDScore> scores = wrapper.getIDScore(0); // 默认只有一条 query 向量
List<FaceSearchResult> finalResults = new ArrayList<>();
for (int i = 0; i < scores.size(); i++) {
SearchResultsWrapper.IDScore score = scores.get(i);
@@ -464,9 +470,9 @@ public class MilvusClient implements VectorDBClient {
}
if (similarity >= faceSearchParams.getThreshold()) {
// 获取 Metadata
String metadata = wrapper.getFieldData(VectorDBConstants.FieldNames.METADATA_FIELD, i).get(0).toString();
String metadata = wrapper.getFieldData(VectorDBConstants.FieldNames.METADATA_FIELD, 0).get(i).toString();
// 获取 ID
String id = wrapper.getFieldData(VectorDBConstants.FieldNames.ID_FIELD, i).get(0).toString();
String id = wrapper.getFieldData(VectorDBConstants.FieldNames.ID_FIELD, 0).get(i).toString();
finalResults.add(new FaceSearchResult(id, similarity, metadata));
}
}