pinecone 0.0.0 → 0.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9de142f1852fd8049b7a1c35466e51b45564a3a3a9dcfb102c9347407f53e1e9
-  data.tar.gz: 4cce7e39f6b2c16885059341f19ff5bafcd5745b76660131d4a35d380cd22cb9
+  metadata.gz: 0b99d69de7f1fb2cf2afbe928b8e48fd0d10c641e33515e4cdde22180d7d9d8b
+  data.tar.gz: f7688c5c74abbfedf560db22ae9f20980c2080e3b414639f99251676c8a3f323
 SHA512:
-  metadata.gz: 19aac1527777be26bfec6e35f056b50029d1656bdc24f71912c8dfe5d9de5d19b0298c9a58767dacab39f6df4dd33ac736f1223151b309681850723122271a72
-  data.tar.gz: 26d1506aa12318845f02b2a60fcf2d10113b53459356ffc2239ede9b8ed53a1a6fcdd6c46ed10652ed674e8ab96d217e49f0c30c51c74602d297dcae12fe4bf4
+  metadata.gz: bf5375603748aa96e744ba92110a5a3b4548f3d58c9388a1e76c4847e16c1fe3e069755df5dc8d87a7bcd1e2e8369589b5b3e37c2389251324489bd2e58b1a03
+  data.tar.gz: 88bc2b14f05e5e80380ebae0824667e4e9a0a4a35c482359e00bf58f9db2e623f00bacfd6b5123aaf738938dc48900379c2e615d6cc549904684c3d6e9bdec98

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.1
 platform: ruby
 authors:
 - Scott Carleton
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-02-03 00:00:00.000000000 Z
+date: 2023-03-21 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: []