pinecone 0.1.1 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0b99d69de7f1fb2cf2afbe928b8e48fd0d10c641e33515e4cdde22180d7d9d8b
-  data.tar.gz: f7688c5c74abbfedf560db22ae9f20980c2080e3b414639f99251676c8a3f323
+  metadata.gz: 5b17e749b07052d08df706ff8f60ca5a40c2bdb752a19662c15ebab0745cf813
+  data.tar.gz: f0789949de4ceab10e7a195798aa0ebc6d77dd9c32f049cab27ff82f13c182fc
 SHA512:
-  metadata.gz: bf5375603748aa96e744ba92110a5a3b4548f3d58c9388a1e76c4847e16c1fe3e069755df5dc8d87a7bcd1e2e8369589b5b3e37c2389251324489bd2e58b1a03
-  data.tar.gz: 88bc2b14f05e5e80380ebae0824667e4e9a0a4a35c482359e00bf58f9db2e623f00bacfd6b5123aaf738938dc48900379c2e615d6cc549904684c3d6e9bdec98
+  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

data/lib/pinecone.rb

@@ -3,6 +3,7 @@ require "httparty"
 require "pinecone/client"
 require "pinecone/index"
 require "pinecone/vector"
+require "pinecone/collection"
 require "pinecone/version"
 
 module Pinecone
@@ -47,47 +48,4 @@ module Pinecone
   def self.configure
     yield(configuration)
   end
-end
-
-# Vector Operations
-
-# # GET Describe Index Stats
-# https://index_name-project_id.svc.environment.pinecone.io/describe_index_stats
-
-# # POST Describe Index Stats
-# https://index_name-project_id.svc.environment.pinecone.io/describe_index_stats
-
-# # DELETE Delete Vectors
-# https://index_name-project_id.svc.environment.pinecone.io/vectors/delete
-
-# # POST Delete Vectors
-# # The Delete operation deletes vectors, by id, from a single namespace.
-# https://index_name-project_id.svc.environment.pinecone.io/vectors/delete
-
-# # GET Fetch
-# # The Fetch operation looks up and returns vectors, by ID, from a single namespace.
-# https://index_name-project_id.svc.environment.pinecone.io/vectors/fetch
-
-# # POST Update
-# # The Update operation updates vector in a namespace.
-# # If a value is included, it will overwrite the previous value.
-# https://index_name-project_id.svc.environment.pinecone.io/vectors/update
-
-# # GET list_collections
-# https://controller.unknown.pinecone.io/collections
-
-# POST create_collection
-
-# GET describe_collection
-
-# DELETE delete_collection
-
-# GET list_indexes
-
-# POST create_index
-
-# GET describe_index
-
-# DELETEdelete_index
-
-# Patch configure_index
+end

metadata

@@ -1,15 +1,127 @@
 --- !ruby/object:Gem::Specification
 name: pinecone
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.3
 platform: ruby
 authors:
 - Scott Carleton
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-03-21 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