diff options
author | Arne H Juul <arnej27959@users.noreply.github.com> | 2024-02-27 15:41:09 +0100 |
---|---|---|
committer | GitHub <noreply@github.com> | 2024-02-27 15:41:09 +0100 |
commit | ce4669f88b5b5993b64c017401c92044cdb1e49d (patch) | |
tree | 487d043cc6369b1247faf875855e5bad8dd190f2 /container-search | |
parent | dfd761f33616840741d18efc76b84ac9e0d7ce09 (diff) | |
parent | 8a32b84caaa9219a67210c389fd643154ef10ec6 (diff) |
Merge pull request #30407 from vespa-engine/bratseth/resolve-aliases
Resolve alias references
Diffstat (limited to 'container-search')
2 files changed, 17 insertions, 3 deletions
diff --git a/container-search/src/main/java/com/yahoo/search/query/properties/RankProfileInputProperties.java b/container-search/src/main/java/com/yahoo/search/query/properties/RankProfileInputProperties.java index 537ffd05463..3ac4178be5f 100644 --- a/container-search/src/main/java/com/yahoo/search/query/properties/RankProfileInputProperties.java +++ b/container-search/src/main/java/com/yahoo/search/query/properties/RankProfileInputProperties.java @@ -48,7 +48,7 @@ public class RankProfileInputProperties extends Properties { value, query.getModel().getLanguage(), context, - this); + query.properties()); } } catch (IllegalArgumentException e) { diff --git a/container-search/src/test/java/com/yahoo/search/searchers/ValidateNearestNeighborTestCase.java b/container-search/src/test/java/com/yahoo/search/searchers/ValidateNearestNeighborTestCase.java index 8e7c7276de1..7bbc1801204 100644 --- a/container-search/src/test/java/com/yahoo/search/searchers/ValidateNearestNeighborTestCase.java +++ b/container-search/src/test/java/com/yahoo/search/searchers/ValidateNearestNeighborTestCase.java @@ -223,19 +223,33 @@ public class ValidateNearestNeighborTestCase { @Test void testWithQueryProfileArgument() { + testWithQueryProfileArgument("foo"); + } + + @Test + void testWithQueryProfileArgumentFromBuiltInProperties() { + testWithQueryProfileArgument("model.queryTree"); + } + + @Test + void testWithQueryProfileArgumentFromBuiltInPropertyAlias() { + testWithQueryProfileArgument("query"); + } + + private void testWithQueryProfileArgument(String argument) { var embedder = new MockEmbedder("test text", Language.UNKNOWN, Tensor.from("tensor<float>(x[3]):[1.0, 2.0, 3.0]")); var registry = new QueryProfileRegistry(); var profile = new QueryProfile("test"); - profile.set("ranking.features.query(qvector)", "embed(@foo)", registry); + profile.set("ranking.features.query(qvector)", "embed(@" + argument + ")", registry); registry.register(profile); var queryString = makeQuery("fvector", "qvector"); var query = new Query.Builder() .setSchemaInfo(createSchemaInfo()) .setQueryProfile(registry.compile().findQueryProfile("test")) .setEmbedder(embedder) - .setRequestMap(Map.of("foo", "test text")) + .setRequestMap(Map.of(argument, "test text")) .build(); setYqlQuery(query, queryString); var result = doSearch(searcher, query); |