pinecone 1.0.1 → 1.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: ab3d946459844e4448d5d9ef3ec0856db62ce97c4e1201013ba33c046cc9d200
-  data.tar.gz: 419f33cf6d1c4876eb882b9109356bfc25f5d74fff52244f2b962c691d162301
+  metadata.gz: bd51f81f28189542ff84a77216c9d2dae42d1946733415ef804a955eb7bcf7e0
+  data.tar.gz: 548e9a37125b33ed6d19a3e63f18365608810f55a557ba6d95d16a1cc3a31f93
 SHA512:
-  metadata.gz: e552468414c4b38a77694e40b8cfd2782d0b48609b138e4e2559333b6ec1b31eef71274eb09db5bd15821a5e3419cfceb95bbdd67f5550066cebad14531e4fa3
-  data.tar.gz: fe81f10ec021bd54e9d5243e67446e69423d0e0845cfaa3f72cd06ab1a3c381136728b2eed9423fcb9305c9c8973101cef6588568ad7575c0be6dc40528e86f8
+  metadata.gz: 2c13dad3d6c0f20f092369fc92c87fb26d2aa0345d02355b06945565959303e5717bb72893e5307d270cc6a4b6dca7b3a2241cbdff1aad2cecaba8ff0f0155cf
+  data.tar.gz: f75288c1975136acd8125d5bd74b46d80e8b756a6c3e8a9f2e329d5e973cf7af2af076993142e4153e46a0e8299dab0c4f68e2e07c372b3fe53b63525c7b357f

data/lib/pinecone/client.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module Pinecone
   class Client
     include HTTParty
@@ -40,8 +42,24 @@ module Pinecone
 
     # This is a very confusing naming convention
     # Pinecone's reference now delineates between 'control plane' and 'data plane' which we'll reflect eventually
-    def index(index_name)
-      Pinecone::Vector.new(index_name)
+    def index(index_name = nil, host: nil)
+      # Direct host provided
+      return Pinecone::Vector.new(host: host) if host
+
+      # Use global host if configured
+      return Pinecone::Vector.new(host: Pinecone.configuration.host) if Pinecone.configuration.host
+
+      # Legacy: index name provided
+      if index_name
+        unless Pinecone.configuration.silence_deprecation_warnings?
+          warn "[DEPRECATED] client.index('name') is deprecated. Use client.index(host: 'host') for better performance."
+        end
+        return Pinecone::Vector.new(index_name)
+      end
+
+      # No host available
+      raise ArgumentError,
+        "No host provided. Set via Pinecone.configure { |c| c.host = 'host' } or client.index(host: 'host')"
     end
   end
 end

data/lib/pinecone/collection.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module Pinecone
   class Collection
     include HTTParty

data/lib/pinecone/index.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module Pinecone
   class Index
     include HTTParty

data/lib/pinecone/response_parser.rb

@@ -1,10 +1,12 @@
+# frozen_string_literal: true
+
 module Pinecone
   class ResponseParser < HTTParty::Parser
     # standard:disable Naming/ConstantName
     SupportedFormats = {
       "application/json" => :json,
       "text/plain" => :json
-    }
+    }.freeze
     # standard:enable Naming/ConstantName
 
     def json

data/lib/pinecone/vector/filter.rb

@@ -33,7 +33,9 @@ module Pinecone
             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)
+              unless v.is_a?(Filter) || to_filter(v).is_a?(Filter)
+                key(:$and).failure("'$any' must be an array of filters")
+              end
             end
           end
         end
@@ -43,24 +45,25 @@ module Pinecone
             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)
+              unless v.is_a?(Filter) || to_filter(v).is_a?(Filter)
+                key(:$or).failure("'$or' must be an array of filters")
+              end
             end
           end
         end
 
         def to_filter(input)
           return false unless input.is_a?(Hash)
+
           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
+        raise ArgumentError, validation.errors.to_h.inspect unless validation.success?
+
+        super
       end
 
       def self.default?

data/lib/pinecone/vector/query.rb

@@ -43,19 +43,17 @@ module Pinecone
 
       def self.new(input)
         validation = QueryContract.new.call(input)
-        if validation.success?
-          super(input)
-        else
-          raise ArgumentError.new(validation.errors.to_h.inspect)
-        end
+        raise ArgumentError, validation.errors.to_h.inspect unless validation.success?
+
+        super
       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
+      def to_json(*_args)
+        to_h.transform_keys do |key|
+          key.to_s.split("_").map.with_index do |word, index|
+            index.zero? ? word : word.capitalize
+          end.join.to_sym
+        end.to_json
       end
     end
   end

data/lib/pinecone/vector/sparse_vector.rb

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

data/lib/pinecone/vector.rb

@@ -1,6 +1,9 @@
+# frozen_string_literal: true
+
 require "pinecone/vector/query"
 require "pinecone/vector/filter"
 require "pinecone/vector/sparse_vector"
+require "json"
 
 module Pinecone
   class Vector
@@ -9,8 +12,22 @@ module Pinecone
 
     attr_reader :base_uri
 
-    def initialize(index_name)
-      @base_uri = set_base_uri(index_name)
+    def initialize(index_name = nil, host: nil)
+      if host
+        # Direct host targeting (preferred)
+        @base_uri = if host.start_with?("http://", "https://")
+          host
+        elsif host.start_with?("localhost")
+          "http://#{host}" # Use HTTP for localhost
+        else
+          "https://#{host}" # Use HTTPS for production hosts
+        end
+      elsif index_name
+        # Legacy path: call describe_index
+        @base_uri = set_fallback_base_uri(index_name)
+      else
+        raise ArgumentError, "Must provide either index_name or host: parameter"
+      end
 
       @headers = {
         "Content-Type" => "application/json",
@@ -26,7 +43,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 +53,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)
@@ -63,7 +131,7 @@ module Pinecone
 
     def describe_index_stats(filter: {})
       payload = if filter.empty?
-        options
+        options.merge(body: {}.to_json)
       else
         options.merge(body: {filter: filter}.to_json)
       end
@@ -79,9 +147,10 @@ module Pinecone
     private
 
     # https://index_name-project_id.svc.environment.pinecone.io
-    def set_base_uri(index_name)
+    def set_fallback_base_uri(index_name)
       index_description = Pinecone::Index.new.describe(index_name)
       raise Pinecone::IndexNotFoundError, "Index #{index_name} does not exist" if index_description.code != 200
+
       uri = index_description.parsed_response["host"]
       "https://#{uri}"
     end

data/lib/pinecone/version.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module Pinecone
-  VERSION = "1.0.1".freeze
+  VERSION = "1.2.1"
 end

data/lib/pinecone.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 require "httparty"
 
 require "pinecone/response_parser"
@@ -13,12 +15,15 @@ module Pinecone
   class IndexNotFoundError < StandardError; end
 
   class Configuration
-    attr_writer :api_key, :base_uri, :environment
+    attr_writer :api_key, :base_uri, :environment, :silence_deprecation_warnings
+    attr_accessor :host
 
     def initialize
       @api_key = nil
       @environment = nil
       @base_uri = nil
+      @host = nil
+      @silence_deprecation_warnings = false
     end
 
     def api_key
@@ -38,6 +43,10 @@ module Pinecone
 
       raise ConfigurationError, "Pinecone environment not set"
     end
+
+    def silence_deprecation_warnings?
+      @silence_deprecation_warnings
+    end
   end
 
   class << self

metadata

@@ -1,57 +1,56 @@
 --- !ruby/object:Gem::Specification
 name: pinecone
 version: !ruby/object:Gem::Version
-  version: 1.0.1
+  version: 1.2.1
 platform: ruby
 authors:
 - Scott Carleton
-autorequire:
 bindir: bin
 cert_chain: []
-date: 2024-01-29 00:00:00.000000000 Z
+date: 2025-06-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: httparty
+  name: dry-struct
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.21.0
+        version: '1.6'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 0.21.0
+        version: '1.6'
 - !ruby/object:Gem::Dependency
-  name: dry-struct
+  name: dry-validation
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.6.0
+        version: '1.10'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.6.0
+        version: '1.10'
 - !ruby/object:Gem::Dependency
-  name: dry-validation
+  name: httparty
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.10.0
+        version: 0.22.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - "~>"
       - !ruby/object:Gem::Version
-        version: 1.10.0
+        version: 0.22.0
 description: Ruby client library which includes index and vector operations to upload
   embeddings into Pinecone and do similarity searches on them.
 email: scott@extrayarn.com
@@ -74,7 +73,6 @@ licenses:
 - MIT
 metadata:
   source_code_uri: https://github.com/ScotterC/pinecone
-post_install_message:
 rdoc_options: []
 require_paths:
 - lib
@@ -82,15 +80,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
-signing_key:
+rubygems_version: 3.6.2
 specification_version: 4
 summary: Ruby client library for Pinecone Vector DB
 test_files: []