Merge pull request #217 from weaviate/hybrid_fusion_type_support

Adds hybrid fusion type support

Author: aliszka

src/main/java/io/weaviate/client/v1/graphql/query/argument/FusionType.java

@@ -0,0 +1,7 @@
+package io.weaviate.client.v1.graphql.query.argument;
+
+public interface FusionType {
+
+  String RANKED = "rankedFusion";
+  String RELATIVE_SCORE = "relativeScoreFusion";
+}

src/main/java/io/weaviate/client/v1/graphql/query/argument/HybridArgument.java

@@ -10,6 +10,7 @@
 import lombok.Getter;
 import lombok.ToString;
 import lombok.experimental.FieldDefaults;
+import org.apache.commons.lang3.StringUtils;
 
 @Getter
 @Builder
@@ -20,6 +21,7 @@ public class HybridArgument implements Argument {
   String query;
   Float alpha;
   Float[] vector;
+  String fusionType;
 
   @Override
   public String build() {
@@ -32,6 +34,9 @@ public String build() {
     if (alpha != null) {
       arg.add(String.format("alpha:%s", alpha));
     }
+    if (StringUtils.isNotBlank(fusionType)) {
+      arg.add(String.format("fusionType:%s", fusionType));
+    }
 
     return String.format("hybrid:{%s}", String.join(" ", arg));
   }

src/test/java/io/weaviate/client/v1/graphql/query/argument/HybridArgumentTest.java

@@ -42,4 +42,30 @@ public void shouldCreateArgumentWithChars() {
 
     assertThat(str).isEqualTo("hybrid:{query:\"\\\"I'm a complex string\\\" says the {'`:string:`'}\"}");
   }
+
+  @Test
+  public void shouldCreateArgumentWithFusionTypeRanked() {
+    HybridArgument hybrid = HybridArgument.builder()
+      .query("I'm a simple string")
+      .fusionType(FusionType.RANKED)
+      .build();
+
+    String str = hybrid.build();
+
+    assertThat(str).isEqualTo("hybrid:{query:\"I'm a simple string\" " +
+      "fusionType:rankedFusion}");
+  }
+
+  @Test
+  public void shouldCreateArgumentWithFusionTypeRelativeScore() {
+    HybridArgument hybrid = HybridArgument.builder()
+      .query("I'm a simple string")
+      .fusionType(FusionType.RELATIVE_SCORE)
+      .build();
+
+    String str = hybrid.build();
+
+    assertThat(str).isEqualTo("hybrid:{query:\"I'm a simple string\" " +
+      "fusionType:relativeScoreFusion}");
+  }
 }