pinecone 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5b17e749b07052d08df706ff8f60ca5a40c2bdb752a19662c15ebab0745cf813
-  data.tar.gz: f0789949de4ceab10e7a195798aa0ebc6d77dd9c32f049cab27ff82f13c182fc
+  metadata.gz: 6ba29ca8e87743d1fb9641ecdd722ac1c18cc55f06d0314dd688d0474da48257
+  data.tar.gz: 29660db1e94027e9701999487c022f8b7ff281881e4166db05d972e55d054630
 SHA512:
-  metadata.gz: f41a0fddf6cd3376e4d4869b9695732db64bfcab2a0e487210927d0fa9e78d6ae3ab6e893c653afd9cbdba0e5bded804216e3b385a0882f1853eba518c2bc8b4
-  data.tar.gz: 314e5ae920fa3c02f1d46899c0bd9508fd6c5dee228e675d3a7f9d0fc742f10c2d43e24032a0a5a227ad32d7f1bb36e53b4221f228840b7357143c0708106bd1
+  metadata.gz: ad82b573c181757f03ea1df9f7f3343561d50899e884177ea6924941c361dce337e32288a66c2f47828fe39ac7ca14753550e005c7ae7c47448f31aa4c684da0
+  data.tar.gz: beee5be5eff743e68e65d8df4ea51b4f9dceeec1f761ff08506b8357787f0dc1c65e68da0ca369ed51ae6ea3ed838fd9134d567b5223b4cf6e44cc071a79a0f5

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,26 @@ 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 options
       {
         headers: @headers,

data/lib/pinecone/version.rb

@@ -1,3 +1,3 @@
 module Pinecone
-  VERSION = "0.1.3".freeze
+  VERSION = "0.1.4".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.4
 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: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: dotenv
   requirement: !ruby/object:Gem::Requirement