weaviate-ruby 0.3.0 → 0.5.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 52fb7ba7147345178fa220aa2bbed2ce66a7903a4df42b8cf5a016ddd329ff3f
-  data.tar.gz: c03e3a537651f2c9e5beb4febdf826f22eadbce6cc1c96ed983b8e0311140604
+  metadata.gz: 7188f7ab1cb378b6ffecf86e173bcfd2d6f513cd004559d18cfa011d8c2f9407
+  data.tar.gz: fd07382e798c763a27adbce3543d224bfed059a2035625569a5247e83d11f598
 SHA512:
-  metadata.gz: f53cf142fa41ce9236e865c251c8e183669b0c3fea0b8b3fe0720a8e9b30dedfd0e37fea888a86b75e9e3d8c5a7118592e5d4b14b1dcfc36221c85a996e332c6
-  data.tar.gz: a71fa15a01d2d75c51de572b2f87d231862ae442a5841a2022ef25c7b20400cc3469217c1ca6d5af42a4995d5b4393f4f33795e846361cdfb663cdf6431edeed
+  metadata.gz: 5cb63f9f341edb1acdc580f3fab0e27190221400183c9191590d4ed7e2d08f3d48ce01167381885e3cbdda894cd0cb32c1c418a2827e2625a2a16d26306395ee
+  data.tar.gz: 17aa43b9a7f42b063a3e5f148a697481ea3b73bfc6dc62c7b6ab101cc24d30d8d78e4e501b3e414e6bf6ef6f73d94914df211f0025aac0f1c8d5b0fe599a1a6e

data/Gemfile.lock

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

data/README.md

@@ -134,6 +134,16 @@ response = client.objects.batch_create(objects: [
     }
 ])
 response.data
+
+# Batch delete objects
+client.objects.batch_delete(
+    class_name: "Question",
+    where: {
+        valueString: "1",
+        operator: "Equal",
+        path: ["id"]
+    }
+)
 ```
 
 ### Querying
@@ -187,6 +197,55 @@ client.query.aggs(
 )
 ```
 
+### Classification
+```ruby
+client.classifications.create(
+    type: "zeroshot",
+    class_name: "Posts",
+    classify_properties: ["hasColor"],
+    based_on_properties: ["text"]
+)
+
+client.classifications.get(
+    id: ""
+)
+```
+
+### Backups
+```ruby
+client.backups.create(
+    backend: "filesystem",
+    id: "my-first-backup",
+    include: ["Question"]
+)
+
+client.backups.get(
+    backend: "filesystem",
+    id: "my-first-backup"
+)
+
+client.backups.restore(
+    backend: "filesystem",
+    id: "my-first-backup"
+)
+
+client.backups.restore_status(
+    backend: "filesystem",
+    id: "my-first-backup"
+)
+```
+
+### Nodes
+```ruby
+client.nodes
+```
+
+### Health
+```ruby
+client.live?
+client.ready?
+```
+
 ## Development
 
 After checking out the repo, run `bin/setup` to install dependencies. Then, run `rake spec` to run the tests. You can also run `bin/console` for an interactive prompt that will allow you to experiment.

data/lib/weaviate/backups.rb

@@ -0,0 +1,54 @@
+# frozen_string_literal: true
+
+module Weaviate
+  class Backups < Base
+    PATH = "backups"
+
+    def create(
+      backend:,
+      id:,
+      include: nil,
+      exclude: nil
+    )
+      response = client.connection.post("#{PATH}/#{backend}") do |req|
+        req.body = {}
+        req.body["id"] = id
+        req.body["include"] = include if include
+        req.body["exclude"] = exclude if exclude
+      end
+
+      response.body
+    end
+
+    def get(
+      backend:,
+      id:
+    )
+      response = client.connection.get("#{PATH}/#{backend}/#{id}")
+      response.body
+    end
+
+    def restore(
+      backend:,
+      id:,
+      include: nil,
+      exclude: nil
+    )
+      response = client.connection.post("#{PATH}/#{backend}/#{id}/restore") do |req|
+        req.body = {}
+        req.body["include"] = include if include
+        req.body["exclude"] = exclude if exclude
+      end
+
+      response.body
+    end
+
+    def restore_status(
+      backend:,
+      id:
+    )
+      response = client.connection.get("#{PATH}/#{backend}/#{id}/restore")
+      response.body
+    end
+  end
+end

data/lib/weaviate/classifications.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+module Weaviate
+  class Classifications < Base
+    PATH = "classifications"
+
+    def get(id:)
+      response = client.connection.get("#{PATH}/#{id}")
+      response.body
+    end
+
+    def create(
+      class_name:,
+      type:,
+      classify_properties: nil,
+      based_on_properties: nil,
+      settings: nil,
+      filters: nil
+    )
+      response = client.connection.post(PATH) do |req|
+        req.body = {}
+        req.body["class"] = class_name
+        req.body["type"] = type
+        req.body["classifyProperties"] = classify_properties if classify_properties
+        req.body["basedOnProperties"] = based_on_properties if based_on_properties
+        req.body["settings"] = settings if settings
+        req.body["filters"] = filters if filters
+      end
+
+      if response.success?
+        response.body
+      end
+    end
+  end
+end

data/lib/weaviate/client.rb

@@ -44,6 +44,29 @@ module Weaviate
       @meta.get
     end
 
+    def nodes
+      @nodes ||= Weaviate::Nodes.new(client: self)
+      @nodes.list
+    end
+
+    def live?
+      @health ||= Weaviate::Health.new(client: self)
+      @health.live?
+    end
+
+    def ready?
+      @health ||= Weaviate::Health.new(client: self)
+      @health.ready?
+    end
+
+    def backups
+      @backups ||= Weaviate::Backups.new(client: self)
+    end
+
+    def classifications
+      @classifications ||= Weaviate::Classifications.new(client: self)
+    end
+
     def objects
       @objects ||= Weaviate::Objects.new(client: self)
     end

data/lib/weaviate/health.rb

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+
+module Weaviate
+  class Health < Base
+    PATH = ".well-known"
+
+    def live?
+      response = client.connection.get("#{PATH}/live")
+      response.status == 200
+    end
+
+    def ready?
+      response = client.connection.get("#{PATH}/ready")
+      response.status == 200
+    end
+  end
+end

data/lib/weaviate/nodes.rb

@@ -0,0 +1,12 @@
+# frozen_string_literal: true
+
+module Weaviate
+  class Nodes < Base
+    PATH = "nodes"
+
+    def list
+      response = client.connection.get(PATH)
+      response.body
+    end
+  end
+end

data/lib/weaviate/objects.rb

@@ -40,8 +40,11 @@ module Weaviate
         req.body["id"] = id unless id.nil?
         req.body["vector"] = vector unless vector.nil?
       end
+
       if response.success?
         Weaviate::Response::Object.new(response.body)
+      else
+        response.body
       end
     end
 
@@ -105,6 +108,30 @@ module Weaviate
       response.success? && response.body.empty?
     end
 
+    def batch_delete(
+      class_name:,
+      where:,
+      consistency_level: nil,
+      output: nil,
+      dry_run: nil
+    )
+      path = "batch/#{PATH}"
+      path << "?consistency_level=#{consistency_level}" unless consistency_level.nil?
+
+      response = client.connection.delete(path) do |req|
+        req.body = {
+          match: {
+            class: class_name,
+            where: where
+          }
+        }
+        req.body["output"] = output unless output.nil?
+        req.body["dryRun"] = dry_run unless dry_run.nil?
+      end
+
+      response.body
+    end
+
     # Validate a data object
     # def validate
     # end

data/lib/weaviate/version.rb

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

data/lib/weaviate.rb

@@ -11,6 +11,10 @@ module Weaviate
   autoload :Objects, "weaviate/objects"
   autoload :OIDC, "weaviate/oidc"
   autoload :Query, "weaviate/query"
+  autoload :Nodes, "weaviate/nodes"
+  autoload :Health, "weaviate/health"
+  autoload :Classifications, "weaviate/classifications"
+  autoload :Backups, "weaviate/backups"
 
   module Response
     autoload :Base, "weaviate/response/base"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: weaviate-ruby
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.5.0
 platform: ruby
 authors:
 - Andrei Bondarev
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-04-02 00:00:00.000000000 Z
+date: 2023-04-04 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday
@@ -68,10 +68,14 @@ files:
 - README.md
 - Rakefile
 - lib/weaviate.rb
+- lib/weaviate/backups.rb
 - lib/weaviate/base.rb
+- lib/weaviate/classifications.rb
 - lib/weaviate/client.rb
 - lib/weaviate/error.rb
+- lib/weaviate/health.rb
 - lib/weaviate/meta.rb
+- lib/weaviate/nodes.rb
 - lib/weaviate/objects.rb
 - lib/weaviate/oidc.rb
 - lib/weaviate/query.rb
@@ -82,13 +86,13 @@ files:
 - lib/weaviate/schema.rb
 - lib/weaviate/version.rb
 - sig/weaviate.rbs
-homepage: https://github.com/andreibondarev/weaviate
+homepage: https://github.com/andreibondarev/weaviate-ruby
 licenses:
 - MIT
 metadata:
-  homepage_uri: https://github.com/andreibondarev/weaviate
-  source_code_uri: https://github.com/andreibondarev/weaviate
-  changelog_uri: https://github.com/andreibondarev/weaviate/CHANGELOG.md
+  homepage_uri: https://github.com/andreibondarev/weaviate-ruby
+  source_code_uri: https://github.com/andreibondarev/weaviate-ruby
+  changelog_uri: https://github.com/andreibondarev/weaviate-ruby/CHANGELOG.md
 post_install_message:
 rdoc_options: []
 require_paths: