Add support for Group

weaviate · Nov 26, 2024 · 5a04c7b · 5a04c7b
1 parent 8f7ae87
commit 5a04c7b
Showing 1 changed file with 31 additions and 0 deletions.
diff --git a/src/main/java/io/weaviate/client/v1/graphql/model/GraphQLGetBaseObject.java b/src/main/java/io/weaviate/client/v1/graphql/model/GraphQLGetBaseObject.java
@@ -1,6 +1,7 @@
 package io.weaviate.client.v1.graphql.model;
 
 import com.google.gson.annotations.SerializedName;
+import java.util.List;
 import java.util.Map;
 import lombok.Getter;
 
@@ -21,17 +22,47 @@ public static class Additional {
     Float[] vector;
     Map<String, Float[]> vectors;
     Generate generate;
+    Group group;
 
     @Getter
     public static class Generate {
       String singleResult;
       String groupedResult;
       String error;
       Debug debug;
+
       @Getter
       public static class Debug {
         String prompt;
       }
     }
+
+    @Getter
+    public static class Group {
+      public String id;
+      public GroupedBy groupedBy;
+      public Integer count;
+      public Float maxDistance;
+      public Float minDistance;
+      public List<GroupHit> hits;
+
+      @Getter
+      public static class GroupedBy {
+        public String value;
+        public String[] path;
+      }
+
+      @Getter
+      public static class GroupHit {
+        // Properties go here
+        AdditionalGroupHit _additional;
+
+        @Getter
+        public static class AdditionalGroupHit {
+          String id;
+          Float distance;
+        }
+      }
+    }
   }
 }