[improvement][headless-chat] filter schemaMapInfo !type =VALUE , Dat…

…aSetMatchResult.class Increase the number of computations 1.0, and compute similar totals, It is preferred to return data sets based on similar total values (tencentmusic#1614)
ylhh353 · Aug 30, 2024 · fca9ba1 · fca9ba1
1 parent 4fb212a
commit fca9ba1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...c/main/java/com/tencent/supersonic/headless/chat/parser/llm/HeuristicDataSetResolver.java b/...c/main/java/com/tencent/supersonic/headless/chat/parser/llm/HeuristicDataSetResolver.java
@@ -67,7 +67,7 @@ protected static Long selectDataSetBySchemaElementMatchScore(Map<Long, SemanticQ
                             return difference;
                         }
 
-                         difference = o2.getValue().getCount() - o1.getValue().getCount();
+                        difference = o2.getValue().getCount() - o1.getValue().getCount();
                         if (difference == 0) {
                             return (int) ((o2.getValue().getMaxSimilarity()
                                     - o1.getValue().getMaxSimilarity()) * 100);
@@ -120,7 +120,7 @@ public static Map<Long, DataSetMatchResult> getDataSetTypeMap(SchemaMapInfo sche
                 }
                 DataSetMatchResult dataSetMatchResult = dataSetCount.get(entry.getKey());
                 Set<SchemaElementType> schemaElementTypes = new HashSet<>();
-              /*  schemaElementMatches.stream()
+                /*schemaElementMatches.stream()
                         .forEach(schemaElementMatch -> schemaElementTypes.add(
                                 schemaElementMatch.getElement().getType()));*/
                 schemaElementMatches.forEach(schemaElementMatch -> {