Skip to content

Commit 899db5b

Browse files
committed
feat: add overloaded methods for single-value input
1 parent 0dea546 commit 899db5b

2 files changed

Lines changed: 22 additions & 4 deletions

File tree

  • src
    • main/java/io/weaviate/client6/v1/api/collections/query
    • test/java/io/weaviate/client6/v1/api/collections/query

src/main/java/io/weaviate/client6/v1/api/collections/query/Target.java

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -278,22 +278,42 @@ static Target combine(List<String> query, CombinationMethod combinationMethod, S
278278
return combine(query, combinationMethod, vectorWeights);
279279
}
280280

281+
static Target sum(String query, String... targetVectors) {
282+
return sum(List.of(query), targetVectors);
283+
}
284+
281285
static Target sum(List<String> query, String... targetVectors) {
282286
return combine(query, CombinationMethod.SUM, targetVectors);
283287
}
284288

289+
static Target min(String query, String... targetVectors) {
290+
return min(List.of(query), targetVectors);
291+
}
292+
285293
static Target min(List<String> query, String... targetVectors) {
286294
return combine(query, CombinationMethod.MIN, targetVectors);
287295
}
288296

297+
static Target average(String query, String... targetVectors) {
298+
return average(List.of(query), targetVectors);
299+
}
300+
289301
static Target average(List<String> query, String... targetVectors) {
290302
return combine(query, CombinationMethod.AVERAGE, targetVectors);
291303
}
292304

305+
static Target relativeScore(String query, VectorWeight... weights) {
306+
return relativeScore(List.of(query), weights);
307+
}
308+
293309
static Target relativeScore(List<String> query, VectorWeight... weights) {
294310
return combine(query, CombinationMethod.RELATIVE_SCORE, weights);
295311
}
296312

313+
static Target manualWeights(String query, VectorWeight... weights) {
314+
return manualWeights(List.of(query), weights);
315+
}
316+
297317
static Target manualWeights(List<String> query, VectorWeight... weights) {
298318
return combine(query, CombinationMethod.MANUAL_WEIGHTS, weights);
299319
}

src/test/java/io/weaviate/client6/v1/api/collections/query/TargetTest.java

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,5 @@
11
package io.weaviate.client6.v1.api.collections.query;
22

3-
import java.util.List;
4-
53
import org.assertj.core.api.Assertions;
64
import org.junit.Test;
75
import org.junit.runner.RunWith;
@@ -68,7 +66,7 @@ public static Object[][] appendTargetsTestCases() {
6866
},
6967
{
7068
Target.min(
71-
List.of("day", "night"),
69+
"in the summertime",
7270
"title_vec", "body_vec"),
7371
"""
7472
{
@@ -80,7 +78,7 @@ public static Object[][] appendTargetsTestCases() {
8078
},
8179
{
8280
Target.relativeScore(
83-
List.of("one", "two", "three"),
81+
"one and two and three and four",
8482
Target.weight("title_vec", 2),
8583
Target.weight("body_vec", 3)),
8684
"""

0 commit comments

Comments
 (0)