pinecone 0.1.3 → 0.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5b17e749b07052d08df706ff8f60ca5a40c2bdb752a19662c15ebab0745cf813
-  data.tar.gz: f0789949de4ceab10e7a195798aa0ebc6d77dd9c32f049cab27ff82f13c182fc
+  metadata.gz: 1df7356e27643581b1acd0417647d0e83cb53d3268357b083c13188f96ee0377
+  data.tar.gz: 561f46ebf941d8f8652787b2034038bbab21303080f5a6f281463f9f1a2669aa
 SHA512:
-  metadata.gz: f41a0fddf6cd3376e4d4869b9695732db64bfcab2a0e487210927d0fa9e78d6ae3ab6e893c653afd9cbdba0e5bded804216e3b385a0882f1853eba518c2bc8b4
-  data.tar.gz: 314e5ae920fa3c02f1d46899c0bd9508fd6c5dee228e675d3a7f9d0fc742f10c2d43e24032a0a5a227ad32d7f1bb36e53b4221f228840b7357143c0708106bd1
+  metadata.gz: fa0470d86d105b2f33e16d0d5a5376cb5e09641ec825faddd9d9cc5b39426b15fcc0d382b3586625e26032d4a48593179f706e7e9276918fa666f5739c35a6c6
+  data.tar.gz: a06fddd61f295a1ba89c55180487ed89f9142fce8161f41c5aa5861190611c93303b581e971cfaae45f877060c2ab72b12e147eddc7d49329d2038fd7e1cb3bf

data/lib/pinecone/client.rb

@@ -18,6 +18,10 @@ module Pinecone
       Pinecone::Index.new.delete(index_name)
     end
 
+    def configure_index(index_name, body)
+      Pinecone::Index.new.configure(index_name, body)
+    end
+
     def list_collections
       Pinecone::Collection.new.list
     end

data/lib/pinecone/index.rb

@@ -28,6 +28,11 @@ module Pinecone
       self.class.delete("/databases/#{index_name}", options)
     end
 
+    def configure(index_name, body)
+      payload = options.merge(body: body.to_json)
+      self.class.patch("/databases/#{index_name}", payload)
+    end
+
     def options
       {
         headers: @headers,

data/lib/pinecone/vector.rb

@@ -1,3 +1,7 @@
+require "pinecone/vector/query"
+require "pinecone/vector/filter"
+require "pinecone/vector/sparse_vector"
+
 module Pinecone
   class Vector
     include HTTParty
@@ -22,10 +26,8 @@ module Pinecone
       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}"
+      query_string = URI.encode_www_form({ namespace: namespace, ids: ids})
       self.class.get("/vectors/fetch?#{query_string}", options)
     end
 
@@ -34,18 +36,35 @@ module Pinecone
       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)
+    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 describe_index_stats(filter: {})
+      payload = if filter.empty?
+        options
+      else
+        options.merge(body: {filter: filter}.to_json)
+      end
+      self.class.post('/describe_index_stats', payload)
+    end
+
+
     def options
       {
         headers: @headers,

data/lib/pinecone/version.rb

@@ -1,3 +1,3 @@
 module Pinecone
-  VERSION = "0.1.3".freeze
+  VERSION = "0.1.5".freeze
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: pinecone
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.5
 platform: ruby
 authors:
 - Scott Carleton
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2023-04-25 00:00:00.000000000 Z
+date: 2023-04-26 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: httparty
@@ -24,6 +24,34 @@ dependencies:
     - - "~>"
       - !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: 1.9.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 1.9.0
 - !ruby/object:Gem::Dependency
   name: dotenv
   requirement: !ruby/object:Gem::Requirement