emb 0.4.0.pre4 → 0.4.0.pre5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0c6f451b27da29d343a893e88a8ae949620e33213941a785088d0e66f1cbe8ab
-  data.tar.gz: 7eb7bc8fa4069bedc853d1580c5504a1a1fbbdff432ff458f61152487199de78
+  metadata.gz: a9b606e95ab03ea635c8ec7acf1aaa0024a4646cb1a8f41d2e3509c8acefedfa
+  data.tar.gz: 78c102fe8b5f547104e5b0e124223f34b4694b19de7c5bccdf2881423fedbbd7
 SHA512:
-  metadata.gz: 94e720e10c56ccca0655c7dab9e28ac71d1731a4da5206664338be49de3dd2756f2acc56aa938a0b949f22dbe38927e2231abfee44cdf5339646079d8bf01fd0
-  data.tar.gz: 115a9d1814d895c856f39f6a557625cdb19abb59628dca39e314f35ace1c97c171c5a44c3c06c53500a6dee4332602f43bc4da04f6dabf0da053524ca43b6030
+  metadata.gz: c8cd7137ff7332f70537789a6bd9ddac1cf3d591156c3bf51676e5242236e8aabd247312e1a166a2aa584ad3041d45a3016979b383b640ffc8dbc6ecc2cf8b64
+  data.tar.gz: ef7a6c243c471b0026c4b35b73120e10b38a2384e12de142e5ebef1b3a9c3200a8180bf1b109eb8282114f7546060b43eed1a3a76adc6e8ad14b09d4da3a4b4e

data/lib/emb/proxy.rb

@@ -38,8 +38,51 @@ module Emb
       "#<Emb::Proxy #{@name}>"
     end
 
+    # Embeds one or more raw images (EMB.IMG). Each argument is the encoded
+    # bytes of an image (JPEG/PNG/GIF/WebP) and is sent unchanged as an
+    # ASCII-8BIT RESP bulk, so no base64/URL encoding is needed:
+    #
+    #   Emb[:clip].image(File.binread('cat.jpg'))
+    #   Emb[:clip].image(io1, io2)                 # IO objects are read
+    #
+    # format: :binary (default) returns the float32-unpacked vectors from the
+    # compact BLOB wire (nil for a failed/truncated slot); format: :values
+    # returns the decoded dtype/shape/values envelope. The float32 reply decode
+    # is the same `unpack('e*')` path text embeddings use.
+    def image(image_bytes, *more, format: :binary)
+      format = normalize_format(format)
+      images = [image_bytes, *more].map { |bytes| self.class.coerce_image_bytes(bytes) }
+
+      if format == :values
+        reply = @client.send_command('EMB.IMG', @name.to_s, 'VALUES', *images)
+        envelope = Emb::ValuesReply.parse(reply)
+        return envelope if images.size == 1
+
+        return envelope.merge(values: Emb::ValuesReply.rows(envelope[:shape], envelope[:values]))
+      end
+
+      reply = @client.send_command('EMB.IMG', @name.to_s, *images)
+      return unpack_embedding(reply) if images.size == 1
+
+      Array(reply).map { |entry| entry.nil? ? nil : unpack_embedding(entry) }
+    end
+
+    # coerce_image_bytes normalizes an image argument to a binary byte string
+    # without transcoding: a String's bytes are preserved exactly (only the
+    # encoding label is set to ASCII-8BIT), and an IO is read fully.
+    def self.coerce_image_bytes(bytes)
+      data = bytes.respond_to?(:read) ? bytes.read : bytes
+      data = data.to_s
+      data = data.dup if data.frozen?
+      data.force_encoding(Encoding::BINARY)
+    end
+
     private
 
+    def unpack_embedding(entry)
+      entry.nil? ? nil : entry.unpack('e*')
+    end
+
     def normalize_format(format)
       unless %i[binary values].include?(format)
         raise ArgumentError, "unknown format #{format.inspect} (expected :binary or :values)"

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: emb
 version: !ruby/object:Gem::Version
-  version: 0.4.0.pre4
+  version: 0.4.0.pre5
 platform: ruby
 authors:
 - elcuervo
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2026-09-10 00:00:00.000000000 Z
+date: 2026-09-15 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: batch-loader