pinecone 0.0.0 → 0.1.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9de142f1852fd8049b7a1c35466e51b45564a3a3a9dcfb102c9347407f53e1e9
-  data.tar.gz: 4cce7e39f6b2c16885059341f19ff5bafcd5745b76660131d4a35d380cd22cb9
+  metadata.gz: 6c4641b066c5db948312f69332a370dd4e7d3f73505f84ebcbaf0e31706a8d93
+  data.tar.gz: 98f401fb73d77984f92dc833f1e8d99d10cd3fe208983cf5f0adb48eb998a11f
 SHA512:
-  metadata.gz: 19aac1527777be26bfec6e35f056b50029d1656bdc24f71912c8dfe5d9de5d19b0298c9a58767dacab39f6df4dd33ac736f1223151b309681850723122271a72
-  data.tar.gz: 26d1506aa12318845f02b2a60fcf2d10113b53459356ffc2239ede9b8ed53a1a6fcdd6c46ed10652ed674e8ab96d217e49f0c30c51c74602d297dcae12fe4bf4
+  metadata.gz: 5afd17e4ec5e36c78f83ad75833654bab6b049d6196fac37fd87a082154c15badfb26687b7acfed9b67b1c6f4a6033244fef7c17fdd0bb086fbe90f50cd4afcd
+  data.tar.gz: 62674b0d1b9d4aea529eaff9d2156f1dcaaec9bf97f04ad7200345bf82369364173732d6ae631fe5782d55e3dc873dc0691d8ef1debf3eacd5c8cd84a7a37e38

data/lib/pinecone.rb

@@ -1,19 +1,20 @@
-require 'httparty'
+require "httparty"
 
-# Usage
-# 
-# pinecone = Pinecone.new
-# pinecone.upsert(Vector Hash)
-# pinecone.query(Vector Array)
+require "pinecone/client"
+require "pinecone/index"
+require "pinecone/vector"
+require "pinecone/version"
 
-class Pinecone
+module Pinecone
   class ConfigurationError < StandardError; end
+  class IndexNotFoundError < StandardError; end
   class Configuration
-    attr_writer :api_key, :base_uri
+    attr_writer :api_key, :base_uri, :environment
 
     def initialize
-      @api_key  = nil
-      @base_uri = nil
+      @api_key     = nil
+      @environment = nil
+      @base_uri    = nil
     end
 
     def api_key
@@ -27,9 +28,13 @@ class Pinecone
 
       raise ConfigurationError, "Pinecone domain not set"
     end
-  end
 
-  include HTTParty
+    def environment
+      return @environment if @environment
+
+      raise ConfigurationError, "Pinecone environment not set"
+    end
+  end
 
   class << self
     attr_writer :configuration
@@ -42,43 +47,6 @@ class Pinecone
   def self.configure
     yield(configuration)
   end
-
-  def initialize
-    self.class.base_uri Pinecone.configuration.base_uri
-    @headers = {
-      "Content-Type" => "application/json",
-      "Accept" => "application/json",
-      "Api-Key" => Pinecone.configuration.api_key,
-    }
-  end
-
-  # # Post Upsert
-  # # The Upsert operation inserts or updates vectors in a namespace.
-  # https://index_name-project_id.svc.environment.pinecone.io/vectors/upsert
-  def upsert(body)
-    payload = options.merge(body: body.to_json)
-    self.class.post('/vectors/upsert', payload)
-  end
-
-  # # POST Query
-  # https://index_name-project_id.svc.environment.pinecone.io/query
-  # vector is an array of floats
-  def query(vector)
-    defaults = {
-      "includeValues": false,
-      "includeMetadata": true,
-      "topK": 5,
-      "vector": vector
-    }.to_json
-    payload = options.merge(body: defaults)
-    self.class.post('/query', payload)
-  end
-
-  def options
-    {
-      headers: @headers,
-    }
-  end
 end
 
 # Vector Operations

metadata

@@ -1,16 +1,17 @@
 --- !ruby/object:Gem::Specification
 name: pinecone
 version: !ruby/object:Gem::Version
-  version: 0.0.0
+  version: 0.1.0
 platform: ruby
 authors:
 - Scott Carleton
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-02-03 00:00:00.000000000 Z
+date: 2023-02-24 00:00:00.000000000 Z
 dependencies: []
-description: Enables ruby access to upsert and query vectors in Pinecone
+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
 executables: []
 extensions: []