Merge pull request #9 from andreibondarev/explore-endpoint

Explore{} endpoint support

Author: andreibondarev

README.md

@@ -198,6 +198,15 @@ client.query.aggs(
 )
 ```
 
+#### Explore
+```ruby
+client.query.explore(
+    fields: 'className',
+    near_text: "{ concepts: [\"science\"] }",
+    limit: "1"
+)
+```
+
 ### Classification
 ```ruby
 client.classifications.create(

lib/weaviate/query.rb

@@ -58,8 +58,63 @@ def aggs(
       raise Weaviate::Error.new(error.response.data.aggregate.errors.messages.to_h)
     end
 
+    def explore(
+      fields:,
+      after: nil,
+      limit: nil,
+      offset: nil,
+      sort: nil,
+      where: nil,
+      near_text: nil,
+      near_vector: nil,
+      near_object: nil
+    )
+      response = client.graphql.execute(
+        explore_query(
+          fields: fields,
+          near_text: near_text,
+          near_vector: near_vector,
+          near_object: near_object
+        ),
+        after: after,
+        limit: limit,
+        offset: offset
+      )
+      response.data.explore
+    rescue Graphlient::Errors::ExecutionError => error
+      raise Weaviate::Error.new(error.to_s)
+    end
+
     private
 
+    def explore_query(
+      fields:,
+      where: nil,
+      near_text: nil,
+      near_vector: nil,
+      near_object: nil,
+      sort: nil
+    )
+      client.graphql.parse <<~GRAPHQL
+        query(
+          $limit: Int,
+          $offset: Int
+        ) {
+          Explore (
+            limit: $limit,
+            offset: $offset,
+            #{near_text.present? ? "nearText: #{near_text}" : ""},
+            #{near_vector.present? ? "nearVector: #{near_vector}" : ""},
+            #{near_object.present? ? "nearObject: #{near_object}" : ""}
+            #{where.present? ? "where: #{where}" : ""},
+            #{sort.present? ? "sort: #{sort}" : ""}
+          ) {
+            #{fields}
+          }
+        }
+      GRAPHQL
+    end
+
     def get_query(
       class_name:,
       fields:,

spec/weaviate/query_spec.rb

@@ -105,4 +105,48 @@
       expect(data.first.question).to eq("In 1953 Watson & Crick built a model of the molecular structure of this, the gene-carrying substance")
     end
   end
+
+  describe "#explore" do
+    let(:response) {
+      double(
+        data: double(
+          explore: [double(certainty: "0.9999", class_name: "Question")]
+        )
+      )
+    }
+
+    let(:graphql_query) {
+      <<-GRAPHQL
+        query {
+          Explore(
+            limit: 1
+            nearText: { concepts: ["biology"] }
+          ) {
+            certainty
+            className
+          }
+        }
+      GRAPHQL
+    }
+
+    before do
+      allow_any_instance_of(Graphlient::Client).to receive(:parse)
+        .and_return(graphql_query)
+
+      allow_any_instance_of(Graphlient::Client).to receive(:execute)
+        .and_return(response)
+    end
+
+    it "returns the query" do
+      response = query.explore(
+        fields: "certainty className",
+        near_text: "{ concepts: [\"biology\"] }",
+        limit: "1"
+      )
+
+      expect(response.count).to eq(1)
+      expect(response.first.certainty).to eq("0.9999")
+      expect(response.first.class_name).to eq("Question")
+    end
+  end
 end