pinecone 0.1.2 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fbad389aa69ef30fb84c23a3f35097daf741984fd17248bff03ffb1fbc0b1e80
-  data.tar.gz: fb84e007e55cb7cb614de5d095878e6dfa310e4f17dc63b340d15dd058203ac0
+  metadata.gz: 6ba29ca8e87743d1fb9641ecdd722ac1c18cc55f06d0314dd688d0474da48257
+  data.tar.gz: 29660db1e94027e9701999487c022f8b7ff281881e4166db05d972e55d054630
 SHA512:
-  metadata.gz: f6560ed5b805e5e9de4e9149def0046318f6921cb162b4037993973596db4834432a6e6caba3be1ec3bc7d7df4919c309e2170a9f28950518af482934e1cd79e
-  data.tar.gz: aafb6ab8087f2d1c89278ec9d06880a270c7ea726773fd921e28170f839629eac9c3ad7eb43097c17323d91ce8996713b29d469d5a38f2bdb5f2f42647c34ef1
+  metadata.gz: ad82b573c181757f03ea1df9f7f3343561d50899e884177ea6924941c361dce337e32288a66c2f47828fe39ac7ca14753550e005c7ae7c47448f31aa4c684da0
+  data.tar.gz: beee5be5eff743e68e65d8df4ea51b4f9dceeec1f761ff08506b8357787f0dc1c65e68da0ca369ed51ae6ea3ed838fd9134d567b5223b4cf6e44cc071a79a0f5

data/lib/pinecone/client.rb

@@ -0,0 +1,42 @@
+module Pinecone
+  class Client
+    include HTTParty
+
+    def list_indexes
+      Pinecone::Index.new.list
+    end
+
+    def describe_index(index_name)
+      Pinecone::Index.new.describe(index_name)
+    end
+
+    def create_index(body)
+      Pinecone::Index.new.create(body)
+    end
+
+    def delete_index(index_name)
+      Pinecone::Index.new.delete(index_name)
+    end
+
+    def list_collections
+      Pinecone::Collection.new.list
+    end
+
+    def describe_collection(collection_name)
+      Pinecone::Collection.new.describe(collection_name)
+    end
+
+    def create_collection(body)
+      Pinecone::Collection.new.create(body)
+    end
+
+    def delete_collection(collection_name)
+      Pinecone::Collection.new.delete(collection_name)
+    end
+
+    # This is a very confusing naming convention
+    def index(index_name)
+      @index ||= Pinecone::Vector.new(index_name)
+    end
+  end
+end

data/lib/pinecone/collection.rb

@@ -0,0 +1,38 @@
+module Pinecone
+  class Collection
+    include HTTParty
+
+    def initialize
+      self.class.base_uri "https://controller.#{Pinecone.configuration.environment}.pinecone.io"
+
+      @headers = {
+        "Content-Type" => "application/json",
+        "Accept" => "application/json",
+        "Api-Key" => Pinecone.configuration.api_key,
+      }
+    end
+
+    def list
+      self.class.get('/collections', options)
+    end
+
+    def describe(collection_name)
+      self.class.get("/collections/#{collection_name}", options)
+    end
+
+    def create(body)
+      payload = options.merge(body: body.to_json)
+      self.class.post('/collections', payload)
+    end
+
+    def delete(collection_name)
+      self.class.delete("/collections/#{collection_name}", options)
+    end
+
+    def options
+      {
+        headers: @headers,
+      }
+    end
+  end
+end

data/lib/pinecone/index.rb

@@ -0,0 +1,37 @@
+module Pinecone
+  class Index
+    include HTTParty
+
+    def initialize
+      self.class.base_uri "https://controller.#{Pinecone.configuration.environment}.pinecone.io"
+      @headers = {
+        "Content-Type" => "application/json",
+        "Accept" => "application/json",
+        "Api-Key" => Pinecone.configuration.api_key,
+      }
+    end
+
+    def list
+      self.class.get('/databases', options)
+    end
+
+    def describe(index_name)
+      self.class.get("/databases/#{index_name}", options)
+    end
+    
+    def create(body)
+      payload = options.merge(body: body.to_json)
+      self.class.post('/databases', payload)
+    end
+
+    def delete(index_name)
+      self.class.delete("/databases/#{index_name}", options)
+    end
+
+    def options
+      {
+        headers: @headers,
+      }
+    end
+  end
+end

data/lib/pinecone/vector.rb

@@ -0,0 +1,75 @@
+require "pinecone/vector/query"
+require "pinecone/vector/filter"
+require "pinecone/vector/sparse_vector"
+
+module Pinecone
+  class Vector
+    include HTTParty
+
+    def initialize(index_name)
+      self.class.base_uri set_base_uri(index_name)
+
+      @headers = {
+        "Content-Type" => "application/json",
+        "Accept" => "application/json",
+        "Api-Key" => Pinecone.configuration.api_key,
+      }
+    end
+
+    def delete(namespace: "", ids: [], delete_all: false)
+      inputs = {
+        "namespace": namespace,
+        "ids": ids,
+        "deleteAll": delete_all,
+      }.to_json
+      payload = options.merge(body: inputs)
+      self.class.post('/vectors/delete', payload)
+    end
+
+    def fetch(namespace: "", ids: [])
+      query_string = URI.encode_www_form({ namespace: namespace, ids: ids})
+      self.class.get("/vectors/fetch?#{query_string}", options)
+    end
+
+    def upsert(body)
+      payload = options.merge(body: body.to_json)
+      self.class.post('/vectors/upsert', payload)
+    end
+
+    def query(query)
+      object = query.is_a?(Pinecone::Vector::Query) ? query : Pinecone::Vector::Query.new(query)
+      payload = options.merge(body: object.to_json)
+      self.class.post('/query', payload)
+    end
+
+    def update(id:, values: [], sparse_values: {indices: [], values: []}, set_metadata: {}, namespace: "")
+      inputs = {
+        "id": id
+      }
+      inputs["values"] = values unless values.empty?
+      inputs["sparseValues"] = sparse_values unless sparse_values[:indices].empty? || sparse_values[:values].empty?
+      inputs["setMetadata"] = set_metadata unless set_metadata.empty?
+      inputs["namespace"] = namespace unless namespace.empty?
+
+      payload = options.merge(body: inputs.to_json)
+      self.class.post('/vectors/update', payload)
+    end
+
+
+    def options
+      {
+        headers: @headers,
+      }
+    end
+
+    private
+
+    # https://index_name-project_id.svc.environment.pinecone.io
+    def set_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["status"]["host"]
+      "https://#{uri}"
+    end
+  end
+end

data/lib/pinecone/version.rb

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

metadata

@@ -1,15 +1,155 @@
 --- !ruby/object:Gem::Specification
 name: pinecone
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.4
 platform: ruby
 authors:
 - Scott Carleton
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-04-04 00:00:00.000000000 Z
-dependencies: []
+date: 2023-04-26 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: httparty
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.21.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.21.0
+- !ruby/object:Gem::Dependency
+  name: dry-struct
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.5.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.5.0
+- !ruby/object:Gem::Dependency
+  name: dry-validation
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: dotenv
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.8'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.8'
+- !ruby/object:Gem::Dependency
+  name: awesome_print
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.9'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.9'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '13.0'
+- !ruby/object:Gem::Dependency
+  name: debug
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1.7'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.12'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.12'
+- !ruby/object:Gem::Dependency
+  name: webmock
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.14'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.14'
+- !ruby/object:Gem::Dependency
+  name: vcr
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.1'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '6.1'
 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
@@ -18,6 +158,11 @@ extensions: []
 extra_rdoc_files: []
 files:
 - lib/pinecone.rb
+- lib/pinecone/client.rb
+- lib/pinecone/collection.rb
+- lib/pinecone/index.rb
+- lib/pinecone/vector.rb
+- lib/pinecone/version.rb
 homepage: https://rubygems.org/gems/pinecone
 licenses:
 - MIT