pinecone 0.1.4 → 0.1.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6ba29ca8e87743d1fb9641ecdd722ac1c18cc55f06d0314dd688d0474da48257
-  data.tar.gz: 29660db1e94027e9701999487c022f8b7ff281881e4166db05d972e55d054630
+  metadata.gz: e278d71fc695cd9e6f0297cc0dd5e8fb6ec144cd1e599d394bb34f25dbcb4783
+  data.tar.gz: ecb5b591b5a8b3564c5b2790fbadf99f253925d60e25934f9fbcfec67a09b919
 SHA512:
-  metadata.gz: ad82b573c181757f03ea1df9f7f3343561d50899e884177ea6924941c361dce337e32288a66c2f47828fe39ac7ca14753550e005c7ae7c47448f31aa4c684da0
-  data.tar.gz: beee5be5eff743e68e65d8df4ea51b4f9dceeec1f761ff08506b8357787f0dc1c65e68da0ca369ed51ae6ea3ed838fd9134d567b5223b4cf6e44cc071a79a0f5
+  metadata.gz: 9b09cd21a242d82794ed5f6af15f9bd0f190d872f48a9ddce66e80eea9c9bdce191c09185d6e9d62058ca01d4ee32a98b2c4ee10460a4e726261eec1093d9da6
+  data.tar.gz: c3254d2c223308f0a0c56bb0c4b4f08fae0a807682f1ab502b7a622ebdea63a2cd719a691a1589d61888485938e5750fd4ea7d9b4e003d95d2115e9543e97a6d

data/lib/pinecone/client.rb

@@ -18,6 +18,10 @@ module Pinecone
       Pinecone::Index.new.delete(index_name)
     end
 
+    def configure_index(index_name, body)
+      Pinecone::Index.new.configure(index_name, body)
+    end
+
     def list_collections
       Pinecone::Collection.new.list
     end

data/lib/pinecone/index.rb

@@ -28,6 +28,11 @@ module Pinecone
       self.class.delete("/databases/#{index_name}", options)
     end
 
+    def configure(index_name, body)
+      payload = options.merge(body: body.to_json)
+      self.class.patch("/databases/#{index_name}", payload)
+    end
+
     def options
       {
         headers: @headers,

data/lib/pinecone/vector/filter.rb

@@ -0,0 +1,71 @@
+# frozen_string_literal: true
+
+require "dry-struct"
+require "dry-validation"
+
+module Types
+  include Dry.Types()
+
+  StringOrNumberOrBoolean = Dry::Types['string'] | Dry::Types['integer'] | Dry::Types['float'] | Dry::Types['bool']
+  StringOrNumber = Dry::Types['string'] | Dry::Types['integer'] | Dry::Types['float']
+  Number = Dry::Types['integer'] | Dry::Types['float']
+end
+
+module Pinecone
+  class Vector
+    class Filter < Dry::Struct
+      class FilterContract < Dry::Validation::Contract
+        schema do
+          optional(:$and).filled(:array)
+          optional(:$or).filled(:array)
+          optional(:$eq).filled(Types::StringOrNumberOrBoolean)
+          optional(:$ne).filled(Types::StringOrNumberOrBoolean)
+          optional(:$gt).filled(Types::Number)
+          optional(:$gte).filled(Types::Number)
+          optional(:$lt).filled(Types::Number)
+          optional(:$lte).filled(Types::Number)
+          optional(:$in).filled(:array).each(Types::StringOrNumber)
+          optional(:$nin).filled(:array).each(Types::StringOrNumber)
+        end
+
+        rule(:$and) do
+          if key?
+            key(:$and).failure("'$any' must be an array") unless value.is_a?(Array)
+
+            value.each do |v|
+              key(:$and).failure("'$any' must be an array of filters") unless v.is_a?(Filter) || to_filter(v).is_a?(Filter)
+            end
+          end
+        end
+
+        rule(:$or) do
+          if key?
+            key(:$or).failure("'$or' must be an array") unless value.is_a?(Array)
+
+            value.each do |v|
+              key(:$or).failure("'$or' must be an array of filters") unless v.is_a?(Filter) || to_filter(v).is_a?(Filter)
+            end
+          end
+        end
+
+        def to_filter(input)
+          return false unless input.is_a?(Hash)
+          return Filter.new(input)
+        end
+      end
+
+      def self.new(input)
+        validation = FilterContract.new.call(input)
+        if validation.success?
+          super(input)
+        else
+          raise ArgumentError.new(validation.errors.to_h.inspect)
+        end
+      end
+
+      def self.default?
+        nil
+      end
+    end
+  end
+end

data/lib/pinecone/vector/query.rb

@@ -0,0 +1,62 @@
+# frozen_string_literal: true
+
+require "dry-struct"
+require "dry-validation"
+
+require "pinecone/vector/filter"
+require "pinecone/vector/sparse_vector"
+
+module Types
+  include Dry.Types()
+end
+
+module Pinecone
+  class Vector
+    class Query < Dry::Struct
+      class QueryContract < Dry::Validation::Contract
+        params do
+          optional(:vector).filled(:array)
+          optional(:id).filled(:string)
+        end
+
+        rule(:vector, :id) do
+          if !values[:vector].nil? && !values[:id].nil?
+            key(:vector).failure("Only one of vector or id can be specified")
+            key(:id).failure("Only one of vector or id can be specified")
+          end
+        end
+      end
+
+      schema schema.strict
+
+      attribute :namespace, Dry::Types['string'].default("")
+      attribute :include_values, Dry::Types['bool'].default(false)
+      attribute :include_metadata, Dry::Types['bool'].default(true)
+      attribute :top_k, Dry::Types['integer'].default(10)
+      attribute? :vector, Dry::Types['array'].of(Dry::Types['float'] | Dry::Types['integer'])
+      # Disabled contract since it wasn't carrying forward attributes to to_json
+      # See failing test in query_spec.rb
+      # attribute? :filter, Filter 
+      attribute? :filter, Dry::Types['hash']
+      attribute? :sparse_vector, SparseVector
+      attribute? :id, Dry::Types['string']
+
+      def self.new(input)
+        validation = QueryContract.new.call(input)
+        if validation.success?
+          super(input)
+        else
+          raise ArgumentError.new(validation.errors.to_h.inspect)
+        end
+      end
+
+      def to_json
+        to_h.map do |key, value|
+          [key.to_s.split("_").map.with_index do |word, index|
+            index == 0 ? word : word.capitalize
+          end.join.to_sym, value]
+        end.to_h.to_json
+      end
+    end
+  end
+end

data/lib/pinecone/vector/sparse_vector.rb

@@ -0,0 +1,40 @@
+# frozen_string_literal: true
+
+require "dry-struct"
+require "dry-validation"
+
+module Types
+  include Dry.Types()
+end
+
+module Pinecone
+  class Vector
+    class SparseVector < Dry::Struct
+      class SparseVectorContract < Dry::Validation::Contract
+        params do
+          required(:indices).filled(:array)
+          required(:values).filled(:array)
+        end
+
+        rule(:indices, :values) do
+          unless values[:indices].size === values[:values].size
+            key(:indices).failure("Indices and values must be the same size")
+            key(:values).failure("Indices and values must be the same size")
+          end
+        end
+      end
+
+      attribute :indices, Dry::Types['array'].of(Dry::Types['integer'])
+      attribute :values, Dry::Types['array'].of(Dry::Types['float'] | Dry::Types['integer'])
+
+      def self.new(input)
+        validation = SparseVectorContract.new.call(input)
+        if validation.success?
+          super(input)
+        else
+          raise ArgumentError.new(validation.errors.to_h.inspect)
+        end
+      end
+    end
+  end
+end

data/lib/pinecone/vector.rb

@@ -55,6 +55,15 @@ module Pinecone
       self.class.post('/vectors/update', payload)
     end
 
+    def describe_index_stats(filter: {})
+      payload = if filter.empty?
+        options
+      else
+        options.merge(body: {filter: filter}.to_json)
+      end
+      self.class.post('/describe_index_stats', payload)
+    end
+
 
     def options
       {

data/lib/pinecone/version.rb

@@ -1,3 +1,3 @@
 module Pinecone
-  VERSION = "0.1.4".freeze
+  VERSION = "0.1.6".freeze
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: pinecone
 version: !ruby/object:Gem::Version
-  version: 0.1.4
+  version: 0.1.6
 platform: ruby
 authors:
 - Scott Carleton
@@ -42,16 +42,16 @@ dependencies:
   name: dry-validation
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 1.9.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 1.9.0
 - !ruby/object:Gem::Dependency
   name: dotenv
   requirement: !ruby/object:Gem::Requirement
@@ -162,6 +162,9 @@ files:
 - lib/pinecone/collection.rb
 - lib/pinecone/index.rb
 - lib/pinecone/vector.rb
+- lib/pinecone/vector/filter.rb
+- lib/pinecone/vector/query.rb
+- lib/pinecone/vector/sparse_vector.rb
 - lib/pinecone/version.rb
 homepage: https://rubygems.org/gems/pinecone
 licenses: