diff options
author | Arne Juul <arnej@vespa.ai> | 2023-10-25 12:48:50 +0000 |
---|---|---|
committer | Arne Juul <arnej@vespa.ai> | 2023-10-25 12:48:50 +0000 |
commit | c4a2fad59b0d38f74983b2c3e9552256c5202617 (patch) | |
tree | 50b7d913cb3661bd66498c6eacc09050aa63f4bb /config-model/src/main/java/com/yahoo/schema/processing/ReservedFunctionNames.java | |
parent | 2b7b0c032cf87883593a5e5d277a192eee5c5e4c (diff) |
add some more built-in function names
Diffstat (limited to 'config-model/src/main/java/com/yahoo/schema/processing/ReservedFunctionNames.java')
-rw-r--r-- | config-model/src/main/java/com/yahoo/schema/processing/ReservedFunctionNames.java | 17 |
1 files changed, 15 insertions, 2 deletions
diff --git a/config-model/src/main/java/com/yahoo/schema/processing/ReservedFunctionNames.java b/config-model/src/main/java/com/yahoo/schema/processing/ReservedFunctionNames.java index f4d2faf9444..83a23d73c6c 100644 --- a/config-model/src/main/java/com/yahoo/schema/processing/ReservedFunctionNames.java +++ b/config-model/src/main/java/com/yahoo/schema/processing/ReservedFunctionNames.java @@ -9,6 +9,7 @@ import com.yahoo.searchlib.rankingexpression.parser.RankingExpressionParserConst import com.yahoo.vespa.model.container.search.QueryProfiles; import java.util.Arrays; +import java.util.HashSet; import java.util.Set; import java.util.logging.Level; import java.util.stream.Collectors; @@ -46,8 +47,20 @@ public class ReservedFunctionNames extends Processor { } private static Set<String> getReservedNames() { - return Arrays.stream(RankingExpressionParserConstants.tokenImage) - .map(token -> token.substring(1, token.length()-1)).collect(Collectors.toUnmodifiableSet()); + Set<String> temp = new HashSet<>(); + Arrays.stream(RankingExpressionParserConstants.tokenImage) + .map(token -> token.substring(1, token.length()-1)).forEach(name -> temp.add(name)); + temp.add("customTokenInputIds"); + temp.add("normalize_linear"); + temp.add("onnx"); + temp.add("onnx_vespa"); + temp.add("reciprocal_rank"); + temp.add("reciprocal_rank_fusion"); + temp.add("tokenAttentionMask"); + temp.add("tokenInputIds"); + temp.add("tokenTypeIds"); + temp.add("xgboost"); + return Set.copyOf(temp); } } |