rubycanusellm 0.2.0 → 0.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 4d8c900c65c552170f515c984fd822ee4a004dc45dbfd384106ee98e2f52a6b4
-  data.tar.gz: faad0d33ac6759890ba9aea14cdfd45c686f305f4e660dd1a99e25244fb7888a
+  metadata.gz: 76737cbccfa63deb73a8a13e57c7202d42eb0689595ddae1b1cedf122f262e91
+  data.tar.gz: 282fda500fb7f36bebcd7e4fc3683a25e264e483b63e264da5aeb21b5cf229ef
 SHA512:
-  metadata.gz: 8c56cf78abac614fccc2cb7595d1a356e015bbd4ad6ddc3af3afd6e2364cfa6317d4f45a676c93f869378f00e9a6651729541ae9304a3572aa1c29462040ecc3
-  data.tar.gz: cb98a023aa10da202850b5e7783030440384d9646b5ed9ffb2923377296bb07f550ddd057eadd04c96c7e60f4caaa0017c9e272bde08c345bca2cd637e257233
+  metadata.gz: 9ffb10c5216314c8ae733b3be09391197a8e785308c42555764e7bc0144e958638dfb8a253e5d818ed6ba77eb69ee07e1b9279f4a7aafd1aca6e7f0e26f8c1b1
+  data.tar.gz: de69c64ab2ee56df386974bd9f99ffbcbc374cd4661b6394ee3295338f02ecee6cab439fbcd0a6bd3b8d9e54b045ef065fda7a8e290d4f6c6a95cce69cdc7949

data/CHANGELOG.md

@@ -20,4 +20,12 @@
 - CLI with generators:
   - `generate:config` — scaffolds configuration file
   - `generate:completion` — scaffolds completion service object
-- Rails and plain Ruby project detection for generators
+- Rails and plain Ruby project detection for generators
+
+## [0.3.0] - 2025-04-03
+
+### Added
+
+- Embeddings support for OpenAI provider (text-embedding-3-small)
+- EmbeddingResponse object with cosine similarity helper
+- `RubyCanUseLLM.embed` method

data/lib/rubycanusellm/embedding_response.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+module RubyCanUseLLM
+  class EmbeddingResponse
+    attr_reader :embedding, :model, :tokens, :raw
+
+    def initialize(embedding:, model:, tokens:, raw:)
+      @embedding = embedding
+      @model = model
+      @tokens = tokens
+      @raw = raw
+    end
+
+    def cosine_similarity(other)
+      dot = embedding.zip(other).sum { |a, b| a * b }
+      mag_a = Math.sqrt(embedding.sum { |a| a**2 })
+      mag_b = Math.sqrt(other.sum { |b| b**2 })
+      return 0.0 if mag_a.zero? || mag_b.zero?
+
+      dot / (mag_a * mag_b)
+    end
+  end
+end

data/lib/rubycanusellm/providers/base.rb

@@ -10,6 +10,10 @@ module RubyCanUseLLM
       def chat(messages, **options)
         raise NotImplementedError, "#{self.class} must implement #chat"
       end
+      
+      def embed(text, **options)
+        raise NotImplementedError, "#{self.class} must implement #embed"
+      end
 
       private
 

data/lib/rubycanusellm/providers/openai.rb

@@ -20,6 +20,15 @@ module RubyCanUseLLM
         end
       end
 
+      def embed(text, **options)
+        body = {
+          model: options[:model] || "text-embedding-3-small",
+          input: text
+        }
+        response = embedding_request(body)
+        parse_embedding(response)
+      end
+
       private
 
       def build_body(messages, options)
@@ -118,6 +127,34 @@ module RubyCanUseLLM
           raw: data
         )
       end
+
+      def embedding_request(body)
+        uri = URI("https://api.openai.com/v1/embeddings")
+        http = Net::HTTP.new(uri.host, uri.port)
+        http.use_ssl = true
+        http.read_timeout = config.timeout
+
+        req = Net::HTTP::Post.new(uri)
+        req["Authorization"] = "Bearer #{config.api_key}"
+        req["Content-Type"] = "application/json"
+        req.body = body.to_json
+
+        handle_response(http.request(req))
+      rescue Net::ReadTimeout, Net::OpenTimeout
+        raise TimeoutError, "Request to OpenAI timed out after #{config.timeout}s"
+      end
+
+      def parse_embedding(data)
+        embedding = data.dig("data", 0, "embedding")
+        usage = data["usage"]
+
+        EmbeddingResponse.new(
+          embedding: embedding,
+          model: data["model"],
+          tokens: usage["total_tokens"],
+          raw: data
+        )
+      end
     end
   end
-end
+end

data/lib/rubycanusellm/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Rubycanusellm
-  VERSION = "0.2.0"
+  VERSION = "0.3.0"
 end

data/lib/rubycanusellm.rb

@@ -8,6 +8,7 @@ require_relative "rubycanusellm/chunk"
 require_relative "rubycanusellm/providers/base"
 require_relative "rubycanusellm/providers/openai"
 require_relative "rubycanusellm/providers/anthropic"
+require_relative "rubycanusellm/embedding_response"
 
 module RubyCanUseLLM
   PROVIDERS = {
@@ -39,5 +40,9 @@ module RubyCanUseLLM
     def chat(messages, **options, &block)
       client.chat(messages, **options, &block)
     end
+
+    def embed(text, **options)
+      client.embed(text, **options)
+    end
   end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rubycanusellm
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Juan Manuel Guzman Nava
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2026-04-01 00:00:00.000000000 Z
+date: 2026-04-03 00:00:00.000000000 Z
 dependencies: []
 description: One interface, every LLM. Rubycanusellm provides a unified client for
   OpenAI, Anthropic, and more, plus generators that scaffold the boilerplate so you
@@ -31,6 +31,7 @@ files:
 - lib/rubycanusellm/chunk.rb
 - lib/rubycanusellm/cli.rb
 - lib/rubycanusellm/configuration.rb
+- lib/rubycanusellm/embedding_response.rb
 - lib/rubycanusellm/errors.rb
 - lib/rubycanusellm/providers/anthropic.rb
 - lib/rubycanusellm/providers/base.rb