emb 0.4.0.pre3 → 0.4.0.pre5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7f3eb540be54e104585249da101c6dd0a77c22c5f7ace642be44b11927d71643
-  data.tar.gz: c963843c42d593640a11201abbb64b698e353c3d674df53d027bbce363ee2df1
+  metadata.gz: a9b606e95ab03ea635c8ec7acf1aaa0024a4646cb1a8f41d2e3509c8acefedfa
+  data.tar.gz: 78c102fe8b5f547104e5b0e124223f34b4694b19de7c5bccdf2881423fedbbd7
 SHA512:
-  metadata.gz: cd19e8b5017cfa0d377a8940a6ed44ecdc78bb5ac5e0905cff9e364ff02d49f68714b4542d2af0fb5f5b89254c8736294c7a62e0ba1506138fa04c5b5ac6f645
-  data.tar.gz: 3a66bb58df4664a86ccaddf991d584de31b7b7ffce1aacf5c4d1f8f9151e252e4409719fc20af54d2f52aabae6f66df9092546e8c91d19eda52b4804a9017dd4
+  metadata.gz: c8cd7137ff7332f70537789a6bd9ddac1cf3d591156c3bf51676e5242236e8aabd247312e1a166a2aa584ad3041d45a3016979b383b640ffc8dbc6ecc2cf8b64
+  data.tar.gz: ef7a6c243c471b0026c4b35b73120e10b38a2384e12de142e5ebef1b3a9c3200a8180bf1b109eb8282114f7546060b43eed1a3a76adc6e8ad14b09d4da3a4b4e

data/Gemfile

@@ -9,6 +9,11 @@ gem 'redis-client'
 gem 'rake', require: false
 gem 'rspec', require: false
 
+# Development-only: the Railtie integration suite boots a real Rails app so the
+# real middleware-stack objects and boot order are exercised. Rails is never a
+# runtime dependency of the gem. Select a line with RAILS_VERSION.
+gem 'rails', ENV.fetch('RAILS_VERSION', '~> 8.0'), require: false
+
 gem 'rubocop', require: false
 gem 'rubocop-rake', require: false
 gem 'rubocop-rspec', require: false

data/README.md

@@ -533,13 +533,22 @@ If your Gemfile loads `emb` before Rails is required (non-standard boot order),
 guarded railtie require in `emb.rb` is skipped — add `require "emb/railtie"` in
 `config/application.rb` right after `require "rails/all"` (or in an initializer).
 
-The middleware is also safe to mount manually in any Rack app (the Railtie
-skips insertion when it is already present):
+The middleware is also safe to mount manually in any Rack app. In a Rails app,
+set the opt-out as well so the Railtie does not add a second copy:
 
 ```ruby
+# config/application.rb
+config.emb.middleware = false
+```
+
+```ruby
+# wherever you build the stack
 use Emb::Middleware
 ```
 
+A duplicate mount is harmless — clearing the per-thread scope is idempotent — but
+the opt-out keeps the stack clean and makes the intended wiring explicit.
+
 The scope is cleared even when the app raises, and a fresh scope starts
 automatically with the next request. Loaders created but never used within a
 request are dropped — the create-then-consume contract applies per request.

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)"

data/lib/emb/railtie.rb

@@ -13,9 +13,13 @@ if defined?(Rails::Railtie)
       config.emb.middleware = true
       config.emb.job_middleware = true
 
+      # Insertion is unconditional (opt-out excepted): during initialization the
+      # stack is a Rails::Configuration::MiddlewareStackProxy, which records
+      # operations but cannot be inspected. Duplicate insertion is safe because
+      # Emb::BatchScope clearing is idempotent, so manual mounts use the opt-out
+      # instead of relying on detection.
       initializer 'emb.middleware' do |app|
         next if config.emb.middleware == false
-        next if app.middleware.include?(Emb::Middleware)
 
         app.middleware.use Emb::Middleware
       end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: emb
 version: !ruby/object:Gem::Version
-  version: 0.4.0.pre3
+  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