weaviate-ruby 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 26b105dbf4536df10fd278d0aac2c437483440c87caf37ec9f6c06e34eeb690b
-  data.tar.gz: 24b3933c80ebd659f292c0785cdd311733337d1adf21e9ea9d04da602f274d43
+  metadata.gz: 52fb7ba7147345178fa220aa2bbed2ce66a7903a4df42b8cf5a016ddd329ff3f
+  data.tar.gz: c03e3a537651f2c9e5beb4febdf826f22eadbce6cc1c96ed983b8e0311140604
 SHA512:
-  metadata.gz: d1961b6e0123e21b60414b53055807ce83305a241c01b74747240cb0022e42483e2a2c2bb9305d2e35630ea2ab28517c311fa094de927c19246f199f91598ce4
-  data.tar.gz: 68ceaf84cded37b443cd7cf8879d5b947352b7be718f2b5a041571eacc584cfddfffd04f94f3b3daa6d1b712f842ac27a4ae105edb2711ebaef28a7cfe73541a
+  metadata.gz: f53cf142fa41ce9236e865c251c8e183669b0c3fea0b8b3fe0720a8e9b30dedfd0e37fea888a86b75e9e3d8c5a7118592e5d4b14b1dcfc36221c85a996e332c6
+  data.tar.gz: a71fa15a01d2d75c51de572b2f87d231862ae442a5841a2022ef25c7b20400cc3469217c1ca6d5af42a4995d5b4393f4f33795e846361cdfb663cdf6431edeed

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    weaviate-ruby (0.2.0)
+    weaviate-ruby (0.3.0)
       faraday (~> 2.7)
       graphlient (~> 0.7.0)
 

data/README.md

@@ -137,13 +137,21 @@ response.data
 ```
 
 ### Querying
+
+#### Get
 ```ruby
-near_text = { "concepts": ["biology"] }
+near_text = '{ concepts: ["biology"] }'
+sort_obj = '{ path: ["category"], order: desc }'
+where_obj = '{ path: ["id"], operator: Equal, valueString: "..." }'
 
 client.query.get(
     class_name: 'Question',
-    fields: ['question', 'answer', 'category'],
-    limit: 1,
+    fields: 'question answer category',
+    limit: "1",
+    offset: "1",
+    after: "id",
+    sort: sort_obj,
+    where_obj: where_obj,
 
     # To use this parameter you must have created your schema by setting the `vectorizer:` property to
     # either 'text2vec-transformers', 'text2vec-contextionary', 'text2vec-openai', 'multi2vec-clip', 'text2vec-huggingface' or 'text2vec-cohere'
@@ -158,6 +166,25 @@ client.query.get(
 
     # near_object: ...,
 )
+
+# Example queries:
+client.query.get class_name: 'Question', where: '{ operator: Like, valueText: "SCIENCE", path: ["category"] }', fields: 'answer question category', limit: "2"
+
+client.query.get class_name: 'Question', fields: 'answer question category _additional { id }', after: "3c5f7039-37f3-4244-b3e2-8f4a083e448d", limit: "1"
+
+
+
+```
+
+#### Aggs
+```ruby
+client.query.aggs(
+    class_name: "Question",
+    fields: 'meta { count }'
+    group_by: ["category"],
+    object_limit: "10",
+    near_text: "{ concepts: [\"knowledge\"] }"
+)
 ```
 
 ## Development

data/lib/weaviate/error.rb

@@ -1,6 +1,6 @@
 # frozen_string_literal: true
 
 module Weaviate
-  class Error
+  class Error < StandardError
   end
 end

data/lib/weaviate/query.rb

@@ -5,32 +5,119 @@ module Weaviate
     def get(
       class_name:,
       fields:,
+      after: nil,
       limit: nil,
-      near_text: nil
+      offset: nil,
+      sort: nil,
+      where: nil,
+      near_text: nil,
+      near_vector: nil,
+      near_object: nil
     )
-      params = {}
-      params["nearText"] = near_text unless near_text.nil?
-      params["limit"] = limit unless limit.nil?
-      # TODO implement the rest of the API params
-
-      response = client.graphql.execute(get_query(class_name, params, fields), near_text: near_text)
+      response = client.graphql.execute(
+        get_query(
+          class_name: class_name,
+          fields: fields,
+          sort: sort,
+          where: where,
+          near_text: near_text,
+          near_vector: near_vector,
+          near_object: near_object
+        ),
+        after: after,
+        limit: limit,
+        offset: offset
+      )
       response.data.get.send(class_name.downcase)
+    rescue Graphlient::Errors::ExecutionError => error
+      raise Weaviate::Error.new(error.response.data.get.errors.messages.to_h)
+    end
+
+    def aggs(
+      class_name:,
+      fields: nil,
+      object_limit: nil,
+      near_text: nil,
+      near_vector: nil,
+      near_object: nil,
+      group_by: nil
+    )
+      response = client.graphql.execute(
+        aggs_query(
+          class_name: class_name,
+          fields: fields,
+          near_text: near_text,
+          near_vector: near_vector,
+          near_object: near_object
+        ),
+        group_by: group_by,
+        object_limit: object_limit
+      )
+      response.data.aggregate.send(class_name.downcase)
+    rescue Graphlient::Errors::ExecutionError => error
+      raise Weaviate::Error.new(error.response.data.aggregate.errors.messages.to_h)
     end
 
     private
 
-    def get_query(class_name, params, fields)
-      client.graphql.parse do
-        query do
-          Get do
-            public_send(class_name, params) do
-              fields.map do |field|
-                public_send(field)
-              end
-            end
-          end
-        end
-      end
+    def get_query(
+      class_name:,
+      fields:,
+      where: nil,
+      near_text: nil,
+      near_vector: nil,
+      near_object: nil,
+      sort: nil
+    )
+      client.graphql.parse <<~GRAPHQL
+        query(
+          $after: String,
+          $limit: Int,
+          $offset: Int,
+        ) {
+          Get {
+            #{class_name}(
+              after: $after,
+              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 aggs_query(
+      class_name:,
+      fields:,
+      near_text: nil,
+      near_vector: nil,
+      near_object: nil
+    )
+      client.graphql.parse <<~GRAPHQL
+        query(
+          $group_by: [String],
+          $object_limit: Int,
+        ) {
+          Aggregate {
+            #{class_name}(
+              objectLimit: $object_limit,
+              groupBy: $group_by,
+              #{near_text.present? ? "nearText: #{near_text}" : ""},
+              #{near_vector.present? ? "nearVector: #{near_vector}" : ""},
+              #{near_object.present? ? "nearObject: #{near_object}" : ""}
+            ) {
+              #{fields}
+            }
+          }
+        }
+      GRAPHQL
     end
   end
 end

data/lib/weaviate/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Weaviate
-  VERSION = "0.2.0"
+  VERSION = "0.3.0"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: weaviate-ruby
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Andrei Bondarev
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-03-27 00:00:00.000000000 Z
+date: 2023-04-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday