Add near_image: option to querying

Author: andreibondarev

lib/weaviate/clusters.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+module Weaviate
+  class Clusters < Base
+    PATH = "clusters"
+
+    # Create the cluster and return The Weaviate server URL.
+    def create(
+      modules:, cluster_name: nil,
+      cluster_type: "sandbox",
+      with_auth: false
+    )
+      response = client.wsc.post(PATH) do |req|
+        req.body = {}
+        req.body["id"] = cluster_name if cluster_name
+        req.body["configuration"] = {
+          tier: cluster_type,
+          requiresAuthentication: with_auth,
+          modules: modules
+        }
+      end
+    end
+
+    def get(
+      cluster_name:
+    )
+      response = client.wsc.get("#{PATH}/#{cluster_name.downcase}")
+      response.body
+    end
+  end
+end

lib/weaviate/query.rb

@@ -12,6 +12,7 @@ def get(
       where: nil,
       near_text: nil,
       near_vector: nil,
+      near_image: nil,
       near_object: nil,
       ask: nil
     )
@@ -23,6 +24,7 @@ def get(
           where: where,
           near_text: near_text,
           near_vector: near_vector,
+          near_image: near_image,
           near_object: near_object,
           ask: ask
         ),
@@ -41,6 +43,7 @@ def aggs(
       object_limit: nil,
       near_text: nil,
       near_vector: nil,
+      near_image: nil,
       near_object: nil,
       group_by: nil
     )
@@ -50,6 +53,7 @@ def aggs(
           fields: fields,
           near_text: near_text,
           near_vector: near_vector,
+          near_image: near_image,
           near_object: near_object
         ),
         group_by: group_by,
@@ -69,13 +73,15 @@ def explore(
       where: nil,
       near_text: nil,
       near_vector: nil,
+      near_image: nil,
       near_object: nil
     )
       response = client.graphql.execute(
         explore_query(
           fields: fields,
           near_text: near_text,
           near_vector: near_vector,
+          near_image: near_image,
           near_object: near_object
         ),
         after: after,
@@ -94,6 +100,7 @@ def explore_query(
       where: nil,
       near_text: nil,
       near_vector: nil,
+      near_image: nil,
       near_object: nil,
       sort: nil
     )
@@ -107,6 +114,7 @@ def explore_query(
             offset: $offset,
             #{near_text.present? ? "nearText: #{near_text}" : ""},
             #{near_vector.present? ? "nearVector: #{near_vector}" : ""},
+            #{near_image.present? ? "nearImage: #{near_image}" : ""},
             #{near_object.present? ? "nearObject: #{near_object}" : ""}
             #{where.present? ? "where: #{where}" : ""},
             #{sort.present? ? "sort: #{sort}" : ""}
@@ -123,6 +131,7 @@ def get_query(
       where: nil,
       near_text: nil,
       near_vector: nil,
+      near_image: nil,
       near_object: nil,
       ask: nil,
       sort: nil
@@ -140,6 +149,7 @@ def get_query(
               offset: $offset,
               #{near_text.present? ? "nearText: #{near_text}" : ""},
               #{near_vector.present? ? "nearVector: #{near_vector}" : ""},
+              #{near_image.present? ? "nearImage: #{near_image}" : ""},
               #{near_object.present? ? "nearObject: #{near_object}" : ""},
               #{ask.present? ? "ask: #{ask}" : ""},
               #{where.present? ? "where: #{where}" : ""},
@@ -157,6 +167,7 @@ def aggs_query(
       fields:,
       near_text: nil,
       near_vector: nil,
+      near_image: nil,
       near_object: nil
     )
       client.graphql.parse <<~GRAPHQL
@@ -170,6 +181,7 @@ def aggs_query(
               groupBy: $group_by,
               #{near_text.present? ? "nearText: #{near_text}" : ""},
               #{near_vector.present? ? "nearVector: #{near_vector}" : ""},
+              #{near_image.present? ? "nearImage: #{near_image}" : ""},
               #{near_object.present? ? "nearObject: #{near_object}" : ""}
             ) {
               #{fields}