pinecone 1.0.1 → 1.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ab3d946459844e4448d5d9ef3ec0856db62ce97c4e1201013ba33c046cc9d200
-  data.tar.gz: 419f33cf6d1c4876eb882b9109356bfc25f5d74fff52244f2b962c691d162301
+  metadata.gz: c9e5f89bdb821f056dde85c82168420d575599d116508b083fb987ea7b3b60da
+  data.tar.gz: b4d1da67d606505ca251db63cc63a9077d363e92425ddfeb4ea55263af30aac6
 SHA512:
-  metadata.gz: e552468414c4b38a77694e40b8cfd2782d0b48609b138e4e2559333b6ec1b31eef71274eb09db5bd15821a5e3419cfceb95bbdd67f5550066cebad14531e4fa3
-  data.tar.gz: fe81f10ec021bd54e9d5243e67446e69423d0e0845cfaa3f72cd06ab1a3c381136728b2eed9423fcb9305c9c8973101cef6588568ad7575c0be6dc40528e86f8
+  metadata.gz: c429ec85b521a797a73597a6cba93b77e4892e84176064079367605c8436b15709aa03fd7ab563240c96bc8406c6b805c3d43bb7cc7f91c8c7ad1b016724acfe
+  data.tar.gz: 749c157e0650086c2365023cf8a9626ef04fd4bce9b5eac886de5007480c64318aa0c3312ea1795713ca08475c5e2f5396130aaabe2bb84d5d853d52969c12d1

data/lib/pinecone/vector/filter.rb

@@ -57,7 +57,7 @@ module Pinecone
       def self.new(input)
         validation = FilterContract.new.call(input)
         if validation.success?
-          super(input)
+          super
         else
           raise ArgumentError.new(validation.errors.to_h.inspect)
         end

data/lib/pinecone/vector/query.rb

@@ -44,7 +44,7 @@ module Pinecone
       def self.new(input)
         validation = QueryContract.new.call(input)
         if validation.success?
-          super(input)
+          super
         else
           raise ArgumentError.new(validation.errors.to_h.inspect)
         end

data/lib/pinecone/vector/sparse_vector.rb

@@ -30,7 +30,7 @@ module Pinecone
       def self.new(input)
         validation = SparseVectorContract.new.call(input)
         if validation.success?
-          super(input)
+          super
         else
           raise ArgumentError.new(validation.errors.to_h.inspect)
         end

data/lib/pinecone/vector.rb

@@ -26,7 +26,6 @@ module Pinecone
         deleteAll: delete_all,
         filter: filter
       }
-
       inputs.delete(:filter) if delete_all || ids.any?
       payload = options.merge(body: inputs.to_json)
       self.class.post("#{@base_uri}/vectors/delete", payload)
@@ -37,6 +36,58 @@ module Pinecone
       self.class.get("#{@base_uri}/vectors/fetch?#{query_string}", options)
     end
 
+    def list(prefix: nil, limit: nil, namespace: nil, &block)
+      all_ids = []
+      pagination_token = nil
+      total_fetched = 0
+
+      loop do
+        current_limit = limit.nil? ? 100 : [limit - total_fetched, 100].min
+
+        response = list_paginated(
+          prefix: prefix,
+          limit: current_limit,
+          pagination_token: pagination_token,
+          namespace: namespace
+        )
+
+        break unless response.success?
+
+        results = response.parsed_response
+        ids = results["vectors"]&.map { |v| v["id"] } || []
+
+        if block
+          yield ids
+        else
+          all_ids.concat(ids)
+        end
+
+        total_fetched += ids.length
+        pagination_token = results.dig("pagination", "next")
+
+        break if ids.empty?
+        break if limit && total_fetched >= limit
+        break if pagination_token.nil? || pagination_token.empty?
+      end
+
+      if block
+        nil
+      else
+        limit ? all_ids.first(limit) : all_ids
+      end
+    end
+
+    def list_paginated(prefix: nil, limit: nil, pagination_token: nil, namespace: nil)
+      query_params = {}
+      query_params["prefix"] = prefix if prefix
+      query_params["limit"] = limit if limit
+      query_params["paginationToken"] = pagination_token if pagination_token
+      query_params["namespace"] = namespace if namespace
+
+      query_string = URI.encode_www_form(query_params)
+      self.class.get("#{@base_uri}/vectors/list?#{query_string}", options)
+    end
+
     def upsert(body)
       payload = options.merge(body: body.to_json)
       self.class.post("#{@base_uri}/vectors/upsert", payload)

data/lib/pinecone/version.rb

@@ -1,3 +1,3 @@
 module Pinecone
-  VERSION = "1.0.1".freeze
+  VERSION = "1.1.0".freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pinecone
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.1.0
 platform: ruby
 authors:
 - Scott Carleton
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-01-29 00:00:00.000000000 Z
+date: 2024-08-09 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: httparty
@@ -16,14 +16,14 @@ dependencies:
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.21.0
+        version: 0.22.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.21.0
+        version: 0.22.0
 - !ruby/object:Gem::Dependency
   name: dry-struct
   requirement: !ruby/object:Gem::Requirement
@@ -82,14 +82,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: '0'
+      version: '3'
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.19
+rubygems_version: 3.5.15
 signing_key:
 specification_version: 4
 summary: Ruby client library for Pinecone Vector DB