summaryrefslogtreecommitdiffstats
path: root/container-search/src/main/java/com/yahoo/search/query/properties
diff options
context:
space:
mode:
authorLester Solbakken <lesters@oath.com>2022-03-21 14:16:42 +0100
committerLester Solbakken <lesters@oath.com>2022-03-21 14:16:42 +0100
commit8a64a50ac9f0cbea18f0c1a8e1ef482d3311e873 (patch)
tree19853dccb4c68714885e7cb73d32e1f191ef306a /container-search/src/main/java/com/yahoo/search/query/properties
parentc5e464f1a6da3a74113d775805187a547074a2da (diff)
Add embedder selection argument to query parameter transformation
Diffstat (limited to 'container-search/src/main/java/com/yahoo/search/query/properties')
-rw-r--r--container-search/src/main/java/com/yahoo/search/query/properties/QueryProperties.java10
1 files changed, 7 insertions, 3 deletions
diff --git a/container-search/src/main/java/com/yahoo/search/query/properties/QueryProperties.java b/container-search/src/main/java/com/yahoo/search/query/properties/QueryProperties.java
index 243915662d2..dc901589cde 100644
--- a/container-search/src/main/java/com/yahoo/search/query/properties/QueryProperties.java
+++ b/container-search/src/main/java/com/yahoo/search/query/properties/QueryProperties.java
@@ -34,12 +34,16 @@ public class QueryProperties extends Properties {
private Query query;
private final CompiledQueryProfileRegistry profileRegistry;
- private final Embedder embedder;
+ private final Map<String, Embedder> embedders;
public QueryProperties(Query query, CompiledQueryProfileRegistry profileRegistry, Embedder embedder) {
+ this(query, profileRegistry, Map.of(Embedder.defaultEmbedderName, embedder));
+ }
+
+ public QueryProperties(Query query, CompiledQueryProfileRegistry profileRegistry, Map<String, Embedder> embedders) {
this.query = query;
this.profileRegistry = profileRegistry;
- this.embedder = embedder;
+ this.embedders = embedders;
}
public void setParentQuery(Query query) {
@@ -394,7 +398,7 @@ public class QueryProperties extends Properties {
if (type == null) return value; // no type info -> keep as string
FieldDescription field = type.getField(key);
if (field == null) return value; // ditto
- return field.getType().convertFrom(value, new ConversionContext(key, profileRegistry, embedder, context));
+ return field.getType().convertFrom(value, new ConversionContext(key, profileRegistry, embedders, context));
}
private void throwIllegalParameter(String key,String namespace) {