weaviate-ruby 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4eb276e9b3dd49bfbdfbbe59abfba8f63c3e4d9f93f1942aede595c991300626
-  data.tar.gz: 7ed2ed88ae192a6b7cd52bdf01fd683cae274b78f61f1a19bf2703c63adc6dad
+  metadata.gz: 26b105dbf4536df10fd278d0aac2c437483440c87caf37ec9f6c06e34eeb690b
+  data.tar.gz: 24b3933c80ebd659f292c0785cdd311733337d1adf21e9ea9d04da602f274d43
 SHA512:
-  metadata.gz: cac32112b0c05d70ff0862c70b63567bd1688c8518617d263a3262711f5919a73e6a94eb5e0161e18a791249b89607a69a110f78caa7532088ab5329bede4d01
-  data.tar.gz: eb14f8b055ed26adb286d54232922d0bb883c11654b15261fc694391bc8ea36d21627d09416f6d81db8030cf956fd7c440db69bd6b33037e4a499f2f39654685
+  metadata.gz: d1961b6e0123e21b60414b53055807ce83305a241c01b74747240cb0022e42483e2a2c2bb9305d2e35630ea2ab28517c311fa094de927c19246f199f91598ce4
+  data.tar.gz: 68ceaf84cded37b443cd7cf8879d5b947352b7be718f2b5a041571eacc584cfddfffd04f94f3b3daa6d1b712f842ac27a4ae105edb2711ebaef28a7cfe73541a

data/Gemfile.lock

@@ -1,23 +1,40 @@
 PATH
   remote: .
   specs:
-    weaviate-ruby (0.1.0)
+    weaviate-ruby (0.2.0)
       faraday (~> 2.7)
+      graphlient (~> 0.7.0)
 
 GEM
   remote: https://rubygems.org/
   specs:
+    activesupport (7.0.4.3)
+      concurrent-ruby (~> 1.0, >= 1.0.2)
+      i18n (>= 1.6, < 2)
+      minitest (>= 5.1)
+      tzinfo (~> 2.0)
     ast (2.4.2)
     byebug (11.1.3)
     coderay (1.1.3)
+    concurrent-ruby (1.2.2)
     diff-lcs (1.5.0)
     faraday (2.7.4)
       faraday-net_http (>= 2.0, < 3.1)
       ruby2_keywords (>= 0.0.4)
     faraday-net_http (3.0.2)
+    graphlient (0.7.0)
+      faraday (~> 2.0)
+      graphql-client
+    graphql (2.0.19)
+    graphql-client (0.18.0)
+      activesupport (>= 3.0)
+      graphql
+    i18n (1.12.0)
+      concurrent-ruby (~> 1.0)
     json (2.6.3)
     language_server-protocol (3.17.0.3)
     method_source (1.0.0)
+    minitest (5.18.0)
     parallel (1.22.1)
     parser (3.2.1.1)
       ast (~> 2.4.1)
@@ -65,6 +82,8 @@ GEM
       language_server-protocol (~> 3.17.0.2)
       rubocop (~> 1.48.1)
       rubocop-performance (~> 1.16.0)
+    tzinfo (2.0.6)
+      concurrent-ruby (~> 1.0)
     unicode-display_width (2.4.2)
 
 PLATFORMS

data/README.md

@@ -48,7 +48,9 @@ client.schema.create(
             "description": "The category",
             "name": "category"
         }
-    ]
+    ],
+    # Possible values: 'text2vec-cohere', 'text2vec-openai', 'text2vec-huggingface', 'text2vec-transformers', 'text2vec-contextionary', 'img2vec-neural', 'multi2vec-clip', 'ref2vec-centroid'
+    vectorizer: "text2vec-openai"
 )
 
 # Get a single class from the schema
@@ -134,6 +136,30 @@ response = client.objects.batch_create(objects: [
 response.data
 ```
 
+### Querying
+```ruby
+near_text = { "concepts": ["biology"] }
+
+client.query.get(
+    class_name: 'Question',
+    fields: ['question', 'answer', 'category'],
+    limit: 1,
+
+    # 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'
+    near_text: near_text,
+
+    # To use this parameter you must have created your schema by setting the `vectorizer:` property to 'multi2vec-clip' or 'img2vec-neural'
+    # near_image: ...,
+
+    # hybrid: ...,
+
+    # bm25: ...,
+
+    # near_object: ...,
+)
+```
+
 ## 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/client.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
 require "faraday"
+require "graphlient"
 
 module Weaviate
   class Client
@@ -47,6 +48,26 @@ module Weaviate
       @objects ||= Weaviate::Objects.new(client: self)
     end
 
+    def query
+      @query ||= Weaviate::Query.new(client: self)
+    end
+
+    def graphql
+      headers = {}
+      if model_service && model_service_api_key
+        headers[API_KEY_HEADERS[model_service]] = model_service_api_key
+      end
+
+      @graphql ||= Graphlient::Client.new(
+        "#{scheme}://#{host}/#{API_VERSION}/graphql",
+        headers: headers,
+        http_options: {
+          read_timeout: 20,
+          write_timeout: 30
+        }
+      )
+    end
+
     def connection
       @connection ||= Faraday.new(url: "#{scheme}://#{host}/#{API_VERSION}/") do |faraday|
         faraday.request :json

data/lib/weaviate/objects.rb

@@ -23,7 +23,7 @@ module Weaviate
         req.params["sort"] = sort unless sort.nil?
         req.params["order"] = order unless order.nil?
       end
-      Response::Collection.from_response(response, key: "objects", type: Response::Object)
+      Response::Collection.from_response(response.body, key: "objects", type: Response::Object)
     end
 
     # Create a new data object. The provided meta-data and schema values are validated.
@@ -52,7 +52,7 @@ module Weaviate
       end
 
       if response.success?
-        Response::Collection.from_response(response, type: Response::Object)
+        Response::Collection.from_response(response.body, type: Response::Object)
       end
     end
 

data/lib/weaviate/query.rb

@@ -0,0 +1,36 @@
+# frozen_string_literal: true
+
+module Weaviate
+  class Query < Base
+    def get(
+      class_name:,
+      fields:,
+      limit: nil,
+      near_text: 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.data.get.send(class_name.downcase)
+    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
+    end
+  end
+end

data/lib/weaviate/response/collection.rb

@@ -5,8 +5,7 @@ module Weaviate
     class Collection
       attr_reader :data, :total_results
 
-      def self.from_response(response, type:, key: nil)
-        body = response.body
+      def self.from_response(body, type:, key: nil)
         new(
           data: (key.nil? ? body : body[key]).map { |attrs| type.new(attrs) }
           # TODO: Integrate and use the totalResults from the response.

data/lib/weaviate/schema.rb

@@ -7,7 +7,7 @@ module Weaviate
     # Dumps the current Weaviate schema. The result contains an array of objects.
     def list
       response = client.connection.get(PATH)
-      Response::Collection.from_response(response, key: "classes", type: Response::Class)
+      Response::Collection.from_response(response.body, key: "classes", type: Response::Class)
     end
 
     # Get a single class from the schema

data/lib/weaviate/version.rb

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

data/lib/weaviate.rb

@@ -10,6 +10,7 @@ module Weaviate
   autoload :Meta, "weaviate/meta"
   autoload :Objects, "weaviate/objects"
   autoload :OIDC, "weaviate/oidc"
+  autoload :Query, "weaviate/query"
 
   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.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - Andrei Bondarev
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2023-03-25 00:00:00.000000000 Z
+date: 2023-03-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday
@@ -24,6 +24,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '2.7'
+- !ruby/object:Gem::Dependency
+  name: graphlient
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.7.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.7.0
 - !ruby/object:Gem::Dependency
   name: pry-byebug
   requirement: !ruby/object:Gem::Requirement
@@ -60,6 +74,7 @@ files:
 - lib/weaviate/meta.rb
 - lib/weaviate/objects.rb
 - lib/weaviate/oidc.rb
+- lib/weaviate/query.rb
 - lib/weaviate/response/base.rb
 - lib/weaviate/response/class.rb
 - lib/weaviate/response/collection.rb