emb 0.1.0.pre.fix

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA256:
+  metadata.gz: a8986240f45d756adc3ca4ca7ac920b9f2f4e9c9f2203c62d703f4ceb30df7cb
+  data.tar.gz: 0d5873987e49ef44ea775d26bd7b13f3110e38b0ee92838ef704caa5cc245f37
+SHA512:
+  metadata.gz: 60b3d4a9ab3a57c6b101bb4093447ed0516f7850c7a83af5810a4289e76871bdb26c57cc93c3e2ff4565c0de341e02755ac5478cd152677dabcfea8e8f5669c1
+  data.tar.gz: 2e9a0eb3cc8f1f9ab61b10c6c91eb17f0d6f732048a233b788d920938a9a036923c0c2075b3bb2378bb32f3c1fc82c591672273e6f768036ebd334410016ca59

data/Gemfile

@@ -0,0 +1,11 @@
+# frozen_string_literal: true
+
+source 'https://rubygems.org'
+
+gemspec
+
+gem 'connection_pool'
+gem 'redis-client'
+
+gem 'rake', require: false
+gem 'rspec', require: false

data/README.md

@@ -0,0 +1,45 @@
+# emb — Ruby client
+
+Thin Ruby wrapper for [emb](https://github.com/elcuervo/emb), a Redis-compatible embedding server.
+
+## Usage
+
+```ruby
+require "emb"
+
+Emb.setup(host: "localhost", port: 6379, pool: 5)
+
+Emb[:minilm]["hello world"]
+# → raw float32 bytes
+
+Emb.models
+# → [{name: "minilm", dim: 384, status: "ready"}, ...]
+
+Emb.info(:minilm)
+# → {dim: 384, workers: 10, ...}
+
+Emb.multi do |m|
+  m[:minilm]["hello"]
+  m[:bge]["world"]
+end
+# → EMB.MULTI minilm "hello" bge "world"
+```
+
+## Testing end to end
+
+### Prerequisites
+
+- An `emb` binary built from the repo root (`just build` or `go build`)
+- Ruby 3.3+ with bundler
+
+### Running tests
+
+```bash
+# From the repo root, start the emb server:
+./bin/emb -config test-two-models.yaml &
+
+# From gems/emb/, run the test suite:
+bundle exec rake
+```
+
+This starts the full test suite against a real running `emb` server. Tests cover all commands: `EMB`, `EMB.MODELS`, `EMB.INFO`, `EMB.HELP`, `PING`, and `EMB.MULTI`.

data/lib/emb/client.rb

@@ -0,0 +1,35 @@
+# frozen_string_literal: true
+
+require "connection_pool"
+require "redis_client"
+
+module Emb
+  @pool = nil
+
+  DEFAULTS = {host: "localhost", port: 6379, pool: 5}.freeze
+
+  class << self
+    def setup(host: DEFAULTS[:host], port: DEFAULTS[:port], pool: DEFAULTS[:pool])
+      @pool = ConnectionPool.new(size: pool) do
+        RedisClient.new(host: host, port: port, protocol: 2, reconnect_attempts: 3)
+      end
+    end
+    alias_method :config, :setup
+
+    def send_command(*args)
+      pool.with { |r| r.call(*args) }
+    end
+
+    private
+
+    def pool
+      @pool ||= default_pool
+    end
+
+    def default_pool
+      ConnectionPool.new(size: DEFAULTS[:pool]) do
+        RedisClient.new(host: DEFAULTS[:host], port: DEFAULTS[:port], protocol: 2, reconnect_attempts: 3)
+      end
+    end
+  end
+end

data/lib/emb/multi.rb

@@ -0,0 +1,37 @@
+# frozen_string_literal: true
+
+module Emb
+  class MultiProxy
+    def initialize
+      @pairs = []
+    end
+
+    def [](name)
+      PairCollector.new(@pairs, name.to_sym)
+    end
+
+    def run
+      args = @pairs.flat_map { |pair| [pair[:model].to_s, pair[:text]] }
+      Emb.send_command("EMB.MULTI", *args)
+    end
+
+    class PairCollector
+      def initialize(pairs, model)
+        @pairs = pairs
+        @model = model
+      end
+
+      def [](text)
+        @pairs << {model: @model, text: text}
+      end
+    end
+  end
+
+  class << self
+    def multi
+      mp = MultiProxy.new
+      yield mp
+      mp.run
+    end
+  end
+end

data/lib/emb/proxy.rb

@@ -0,0 +1,31 @@
+# frozen_string_literal: true
+
+module Emb
+  @registry = {}
+
+  class << self
+    def [](name)
+      @registry[name] ||= Proxy.new(name.to_sym)
+    end
+
+    def reset_registry!
+      @registry.clear
+    end
+  end
+
+  class Proxy
+    attr_reader :name
+
+    def initialize(name)
+      @name = name
+    end
+
+    def [](text, *texts)
+      Emb.send_command("EMB", @name.to_s, text, *texts)
+    end
+
+    def inspect
+      "#<Emb::Proxy #{@name}>"
+    end
+  end
+end

data/lib/emb/version.rb

@@ -0,0 +1,10 @@
+# frozen_string_literal: true
+
+module Emb
+  VERSION =
+    begin
+      File.read(File.expand_path('../../../../VERSION', __dir__)).strip
+    rescue Errno::ENOENT
+      Gem::Specification.find_by_name("emb").version.to_s
+    end
+end

data/lib/emb.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+require_relative "emb/version"
+require_relative "emb/client"
+require_relative "emb/proxy"
+require_relative "emb/multi"
+
+module Emb
+  class << self
+    def models
+      raw = send_command("EMB.MODELS")
+      return [] if raw.nil?
+      raw.map { |name, dim, status| {name: name, dim: dim.to_i, status: status} }
+    end
+
+    def info(name)
+      raw = send_command("EMB.INFO", name.to_s)
+
+      return {} if raw.nil?
+
+      raw.each_slice(2).map { |k, v| [k.to_sym, v] }.to_h
+    end
+
+    def stats = send_command("EMB.STATS")
+
+    def help = send_command("EMB.HELP")
+
+    def ping = send_command("PING")
+  end
+end

metadata

@@ -0,0 +1,93 @@
+--- !ruby/object:Gem::Specification
+name: emb
+version: !ruby/object:Gem::Version
+  version: 0.1.0.pre.fix
+platform: ruby
+authors:
+- elcuervo
+autorequire:
+bindir: bin
+cert_chain: []
+date: 2026-07-01 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: connection_pool
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.5'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.5'
+- !ruby/object:Gem::Dependency
+  name: redis-client
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.24'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '0.24'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.13'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.13'
+description:
+email:
+- elcuervo@elcuervo.net
+executables: []
+extensions: []
+extra_rdoc_files: []
+files:
+- Gemfile
+- README.md
+- lib/emb.rb
+- lib/emb/client.rb
+- lib/emb/multi.rb
+- lib/emb/proxy.rb
+- lib/emb/version.rb
+homepage: https://github.com/elcuervo/emb
+licenses:
+- MIT
+metadata:
+  rubygems_mfa_required: 'true'
+post_install_message:
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '3.3'
+required_rubygems_version: !ruby/object:Gem::Requirement
+  requirements:
+  - - ">="
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubygems_version: 3.5.22
+signing_key:
+specification_version: 4
+summary: 'Client for emb: a Redis-compatible embedding server'
+test_files: []