pinecone 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: fbad389aa69ef30fb84c23a3f35097daf741984fd17248bff03ffb1fbc0b1e80
-  data.tar.gz: fb84e007e55cb7cb614de5d095878e6dfa310e4f17dc63b340d15dd058203ac0
+  metadata.gz: 5b17e749b07052d08df706ff8f60ca5a40c2bdb752a19662c15ebab0745cf813
+  data.tar.gz: f0789949de4ceab10e7a195798aa0ebc6d77dd9c32f049cab27ff82f13c182fc
 SHA512:
-  metadata.gz: f6560ed5b805e5e9de4e9149def0046318f6921cb162b4037993973596db4834432a6e6caba3be1ec3bc7d7df4919c309e2170a9f28950518af482934e1cd79e
-  data.tar.gz: aafb6ab8087f2d1c89278ec9d06880a270c7ea726773fd921e28170f839629eac9c3ad7eb43097c17323d91ce8996713b29d469d5a38f2bdb5f2f42647c34ef1
+  metadata.gz: f41a0fddf6cd3376e4d4869b9695732db64bfcab2a0e487210927d0fa9e78d6ae3ab6e893c653afd9cbdba0e5bded804216e3b385a0882f1853eba518c2bc8b4
+  data.tar.gz: 314e5ae920fa3c02f1d46899c0bd9508fd6c5dee228e675d3a7f9d0fc742f10c2d43e24032a0a5a227ad32d7f1bb36e53b4221f228840b7357143c0708106bd1

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,65 @@
+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
+
+    # This requires manually building the query string to unbundle ids
+    def fetch(namespace: "", ids: [])
+      ids_query_string = ids.map { |id| "ids=#{id}" }.join('&')
+      query_string = "namespace=#{namespace}&#{ids_query_string}"
+      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(namespace: "", vector:, top_k: 10, include_values: false, include_metadata: true)
+      inputs = {
+        "namespace": namespace,
+        "includeValues": include_values,
+        "includeMetadata": include_metadata,
+        "topK": top_k,
+        "vector": vector,
+      }.to_json
+      payload = options.merge(body: inputs)
+      self.class.post('/query', 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.3".freeze
+end

metadata

@@ -1,15 +1,127 @@
 --- !ruby/object:Gem::Specification
 name: pinecone
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - Scott Carleton
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-04-04 00:00:00.000000000 Z
-dependencies: []
+date: 2023-04-25 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: 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 +130,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