malei05 1 bulan lalu
induk
melakukan
e7a40692ab

+ 3 - 1
fn-opensearch/src/main/java/net/yaoyi/pipeline/opensearch/OpenSearchHandler.java

@@ -219,7 +219,7 @@ public class OpenSearchHandler implements PojoRequestHandler<TaskImageMsg, List<
             SearchRequest request = new SearchRequest();
             request.setTableName(TABLE_NAME);
             request.setKnn(knn);
-            request.setOutputFields(List.of("taskId", "image"));
+            request.setOutputFields(List.of("taskId", "image", "taskFiledValue"));
 
             SearchResponse response = OpenSearchUtils.getClient().search(request);
             String responseBody = response.getBody();
@@ -251,6 +251,7 @@ public class OpenSearchHandler implements PojoRequestHandler<TaskImageMsg, List<
 
             multiResult.setTaskId(fields.get("taskId").asLong());
             multiResult.setImage(fields.get("image").asText());
+            multiResult.setTaskFiledValue(fields.get("taskFiledValue").asText());
 
             if (jsonNode.has("score")) {
                 multiResult.setScore(jsonNode.get("score").asDouble());
@@ -304,6 +305,7 @@ public class OpenSearchHandler implements PojoRequestHandler<TaskImageMsg, List<
                     ImageMultiMsg.MultiInfo info = new ImageMultiMsg.MultiInfo();
                     info.setTaskId(r.getTaskId());
                     info.setImage(r.getImage());
+                    info.setTaskFiledValue(r.getTaskFiledValue());
                     info.setScore(r.getScore());
                     multiInfoList.add(info);
                 }

+ 1 - 0
yyc-pipeline-model/src/main/java/net/yaoyi/pipeline/model/index/TaskImageIndex.java

@@ -23,6 +23,7 @@ public class TaskImageIndex {
     public static class MultiResult {
         private Long taskId;
         private String image;
+        private String taskFiledValue;
         private double score;
         private String type;
     }

+ 1 - 0
yyc-pipeline-model/src/main/java/net/yaoyi/pipeline/model/mq/ImageMultiMsg.java

@@ -15,6 +15,7 @@ public class ImageMultiMsg extends AlgoBaseInfo {
     @Data
     public static class MultiInfo {
         private Long taskId;
+        private String taskFiledValue;
         private String image;
         private double score;
     }