weaviate-ruby 0.8.3 → 0.8.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8e63b3f0149e87dc2aec7c38970e91866e45dfccfc9e46c1378c27b9135f4a4d
-  data.tar.gz: f22c4a1c4eb8cdd2acd9d70b95404c46afb11c19e3821be35fb5b49048563dcb
+  metadata.gz: ffbe97178df453d77bcd0071e405f77cdad6bea3278103ec23cd04250dbd5735
+  data.tar.gz: '038c9ef137b6c3222ec1eabbd18b571dc3db71b29e734f0045241790a94f96c2'
 SHA512:
-  metadata.gz: 843a49164ce05ed851dcf80e228e75839f05d7ee73c65087da21f72ab689f2c5388ac14ba06a1782c3aeda62ff1b01104bf52eb230313df145b9afafcb856c25
-  data.tar.gz: b1fd3ee9aa230da2499cc9ce1e8c24f69bb29a3199b07e16cf0f1095f53da9094a83ae849c9d266ff27a8292cb6d7b4166c7e257cb02a698b7d53702bbdb1f20
+  metadata.gz: 48689c5b2005f63fa6b55b975778aa5c0da34a82c1caf62abe8224f4ad1d2722481177ebb7f684cf12fa3f7485f058ab433d1ac3b8a8b0a818ea666a9b8eef2b
+  data.tar.gz: 3d06817dcab514c2e16ec68f8a870272635f0cfc19b5a6f6a12cc263f32fb99d18b7472c863c8f9144182dbec5550a3a1bbab4a3d3c925d42af120ab16f3db30

data/CHANGELOG.md

@@ -1,5 +1,11 @@
 ## [Unreleased]
 
+## [0.8.5] - 2023-07-19
+- Add multi-tenancy support
+
+## [0.8.4] - 2023-06-30
+- Adding yard
+
 ## [0.8.3] - 2023-06-25
 - Add Google PaLM support
 

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    weaviate-ruby (0.8.3)
+    weaviate-ruby (0.8.5)
       faraday (~> 1)
       faraday_middleware (~> 1)
       graphlient (~> 0.6.0)
@@ -50,6 +50,7 @@ GEM
       pry (>= 0.13, < 0.15)
     rainbow (3.1.1)
     rake (13.0.6)
+    rdiscount (2.2.7.1)
     regexp_parser (2.7.0)
     rexml (3.2.5)
     rspec (3.11.0)
@@ -89,17 +90,21 @@ GEM
     tzinfo (2.0.6)
       concurrent-ruby (~> 1.0)
     unicode-display_width (2.4.2)
+    yard (0.9.34)
 
 PLATFORMS
+  arm64-darwin-21
   x86_64-darwin-19
   x86_64-linux
 
 DEPENDENCIES
   pry-byebug (~> 3.9)
   rake (~> 13.0)
+  rdiscount
   rspec (~> 3.0)
   standard (~> 1.25.0)
   weaviate-ruby!
+  yard
 
 BUNDLED WITH
    2.4.0

data/README.md

@@ -6,9 +6,15 @@
     <img alt='Ruby logo' src='https://user-images.githubusercontent.com/541665/230231593-43861278-4550-421d-a543-fd3553aac4f6.png' height='40' />
 </p>
 
-Ruby wrapper for the Weaviate.io API
+Ruby wrapper for the Weaviate.io API.
 
-![Tests status](https://github.com/andreibondarev/weaviate-ruby/actions/workflows/ci.yml/badge.svg) [![Gem Version](https://badge.fury.io/rb/weaviate-ruby.svg)](https://badge.fury.io/rb/weaviate-ruby)
+Part of the [Langchain.rb](https://github.com/andreibondarev/langchainrb) stack.
+
+![Tests status](https://github.com/andreibondarev/weaviate-ruby/actions/workflows/ci.yml/badge.svg)
+[![Gem Version](https://badge.fury.io/rb/weaviate-ruby.svg)](https://badge.fury.io/rb/weaviate-ruby)
+[![Docs](http://img.shields.io/badge/yard-docs-blue.svg)](http://rubydoc.info/gems/weaviate-ruby)
+[![License](https://img.shields.io/badge/license-MIT-green.svg)](https://github.com/andreibondarev/weaviate-ruby/blob/main/LICENSE.txt)
+[![](https://dcbadge.vercel.app/api/server/WDARp7J2n8?compact=true&style=flat)](https://discord.gg/WDARp7J2n8)
 
 ## Installation
 

data/Rakefile

@@ -2,7 +2,12 @@
 
 require "bundler/gem_tasks"
 require "rspec/core/rake_task"
+require "yard"
 
 RSpec::Core::RakeTask.new(:spec)
 
 task default: :spec
+
+YARD::Rake::YardocTask.new do |t|
+  t.options = ["--fail-on-warning"]
+end

data/lib/weaviate/client.rb

@@ -1,7 +1,6 @@
 # frozen_string_literal: true
 
 require "faraday"
-require "faraday_middleware"
 require "graphlient"
 
 module Weaviate

data/lib/weaviate/objects.rb

@@ -8,6 +8,7 @@ module Weaviate
     def list(
       class_name: nil,
       limit: nil,
+      tenant: nil,
       offset: nil,
       after: nil,
       include: nil,
@@ -16,6 +17,7 @@ module Weaviate
     )
       response = client.connection.get(PATH) do |req|
         req.params["class"] = class_name unless class_name.nil?
+        req.params["tenant"] = tenant unless tenant.nil?
         req.params["limit"] = limit unless limit.nil?
         req.params["offset"] = offset unless offset.nil?
         req.params["after"] = after unless after.nil?
@@ -31,6 +33,7 @@ module Weaviate
     def create(
       class_name:,
       properties:,
+      tenant: nil,
       consistency_level: nil,
       id: nil,
       vector: nil
@@ -47,6 +50,7 @@ module Weaviate
         req.body = {}
         req.body["class"] = class_name
         req.body["properties"] = properties
+        req.body["tenant"] = tenant unless tenant.blank?
         req.body["id"] = id unless id.nil?
         req.body["vector"] = vector unless vector.nil?
       end

data/lib/weaviate/query.rb

@@ -6,6 +6,7 @@ module Weaviate
       class_name:,
       fields:,
       after: nil,
+      tenant: nil,
       limit: nil,
       offset: nil,
       sort: nil,
@@ -21,6 +22,7 @@ module Weaviate
       response = client.graphql.execute(
         get_query(
           class_name: class_name,
+          tenant: tenant,
           fields: fields,
           sort: sort,
           where: where,
@@ -131,7 +133,7 @@ module Weaviate
 
     def get_query(
       class_name:,
-      fields:,
+      fields:, tenant: nil,
       where: nil,
       near_text: nil,
       near_vector: nil,
@@ -153,6 +155,7 @@ module Weaviate
               after: $after,
               limit: $limit,
               offset: $offset,
+              #{tenant.present? ? "tenant: #{tenant}" : ""}
               #{near_text.present? ? "nearText: #{near_text}" : ""},
               #{near_vector.present? ? "nearVector: #{near_vector}" : ""},
               #{near_image.present? ? "nearImage: #{near_image}" : ""},

data/lib/weaviate/schema.rb

@@ -26,6 +26,7 @@ module Weaviate
       class_name:,
       description: nil,
       properties: nil,
+      multi_tenant: nil,
       vector_index_type: nil,
       vector_index_config: nil,
       vectorizer: nil,
@@ -42,6 +43,7 @@ module Weaviate
         req.body["vectorizer"] = vectorizer unless vectorizer.nil?
         req.body["moduleConfig"] = module_config unless module_config.nil?
         req.body["properties"] = properties unless properties.nil?
+        req.body["multiTenancyConfig"] = {enabled: true} unless multi_tenant.nil?
         req.body["invertedIndexConfig"] = inverted_index_config unless inverted_index_config.nil?
         req.body["replicationConfig"] = replication_config unless replication_config.nil?
       end
@@ -93,6 +95,39 @@ module Weaviate
       response.body
     end
 
+    # Adds one or more tenants to a class.
+    def add_tenants(
+      class_name:,
+      tenants:
+    )
+      response = client.connection.post("#{PATH}/#{class_name}/tenants") do |req|
+        tenants_str = tenants.map { |t| %({"name": "#{t}"}) }.join(", ")
+        req.body = "[#{tenants_str}]"
+      end
+      response.body
+    end
+
+    # List tenants of a class.
+    def list_tenants(class_name:)
+      response = client.connection.get("#{PATH}/#{class_name}/tenants")
+      response.body
+    end
+
+    # Remove one or more tenants from a class.
+    def remove_tenants(
+      class_name:,
+      tenants:
+    )
+      response = client.connection.delete("#{PATH}/#{class_name}/tenants") do |req|
+        req.body = tenants
+      end
+
+      if response.success?
+      end
+
+      response.body
+    end
+
     # Add a property to an existing schema class.
     def add_property(
       class_name:,

data/lib/weaviate/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: weaviate-ruby
 version: !ruby/object:Gem::Version
-  version: 0.8.3
+  version: 0.8.5
 platform: ruby
 authors:
 - Andrei Bondarev
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-06-26 00:00:00.000000000 Z
+date: 2023-07-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday
@@ -66,6 +66,34 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '3.9'
+- !ruby/object:Gem::Dependency
+  name: yard
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rdiscount
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: Ruby wrapper for the Weaviate.io API
 email:
 - andrei@sourcelabs.io