Support where in aggregate functions

patterns-ai-core · Sep 11, 2023 · 6afa653 · 6afa653
1 parent 905dd3a
commit 6afa653
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/lib/weaviate/query.rb b/lib/weaviate/query.rb
@@ -48,6 +48,7 @@ def get(
     def aggs(
       class_name:,
       fields: nil,
+      where: nil,
       object_limit: nil,
       near_text: nil,
       near_vector: nil,
@@ -59,6 +60,7 @@ def aggs(
         aggs_query(
           class_name: class_name,
           fields: fields,
+          where: where,
           near_text: near_text,
           near_vector: near_vector,
           near_image: near_image,
@@ -183,6 +185,7 @@ def get_query(
     def aggs_query(
       class_name:,
       fields:,
+      where: nil,
       near_text: nil,
       near_vector: nil,
       near_image: nil,
@@ -200,7 +203,8 @@ def aggs_query(
               #{near_text.present? ? "nearText: #{near_text}" : ""},
               #{near_vector.present? ? "nearVector: #{near_vector}" : ""},
               #{near_image.present? ? "nearImage: #{near_image}" : ""},
-              #{near_object.present? ? "nearObject: #{near_object}" : ""}
+              #{near_object.present? ? "nearObject: #{near_object}" : ""},
+              #{where.present? ? "where: #{where}" : ""}
             ) {
               #{fields}
             }