ruby_llm 1.3.1 → 1.3.2beta1

data/lib/ruby_llm/models.rb

@@ -92,7 +92,8 @@ module RubyLLM
           f.response :json, parser_options: { symbolize_names: true }
         end
         response = connection.get 'https://api.parsera.org/v1/llm-specs'
-        response.body.map { |data| Model::Info.new(data) }
+        models = response.body.map { |data| Model::Info.new(data) }
+        models.reject { |model| model.provider.nil? || model.id.nil? }
       end
 
       def merge_models(provider_models, parsera_models)

data/lib/ruby_llm/provider.rb

@@ -34,7 +34,7 @@ module RubyLLM
       def embed(text, model:, connection:, dimensions:)
         payload = render_embedding_payload(text, model:, dimensions:)
         response = connection.post(embedding_url(model:), payload)
-        parse_embedding_response(response, model:)
+        parse_embedding_response(response, model:, text:)
       end
 
       def paint(prompt, model:, size:, connection:)

data/lib/ruby_llm/providers/bedrock/streaming/prelude_handling.rb

@@ -24,8 +24,8 @@ module RubyLLM
           end
 
           def read_prelude(chunk, offset)
-            total_length = chunk[offset...offset + 4].unpack1('N')
-            headers_length = chunk[offset + 4...offset + 8].unpack1('N')
+            total_length = chunk[offset...(offset + 4)].unpack1('N')
+            headers_length = chunk[(offset + 4)...(offset + 8)].unpack1('N')
             [total_length, headers_length]
           end
 
@@ -72,8 +72,8 @@ module RubyLLM
 
           def extract_potential_lengths(chunk, pos)
             [
-              chunk[pos...pos + 4].unpack1('N'),
-              chunk[pos + 4...pos + 8].unpack1('N')
+              chunk[pos...(pos + 4)].unpack1('N'),
+              chunk[(pos + 4)...(pos + 8)].unpack1('N')
             ]
           end
 

data/lib/ruby_llm/providers/gemini/embeddings.rb

@@ -15,9 +15,11 @@ module RubyLLM
           { requests: [text].flatten.map { |t| single_embedding_payload(t, model:, dimensions:) } }
         end
 
-        def parse_embedding_response(response, model:)
+        def parse_embedding_response(response, model:, text:)
           vectors = response.body['embeddings']&.map { |e| e['values'] }
-          vectors in [vectors]
+          # If we only got one embedding AND the input was a single string (not an array),
+          # return it as a single vector
+          vectors = vectors.first if vectors&.length == 1 && !text.is_a?(Array)
 
           Embedding.new(vectors:, model:, input_tokens: 0)
         end

data/lib/ruby_llm/providers/openai/embeddings.rb

@@ -19,13 +19,14 @@ module RubyLLM
           }.compact
         end
 
-        def parse_embedding_response(response, model:)
+        def parse_embedding_response(response, model:, text:)
           data = response.body
           input_tokens = data.dig('usage', 'prompt_tokens') || 0
           vectors = data['data'].map { |d| d['embedding'] }
 
-          # If we only got one embedding, return it as a single vector
-          vectors in [vectors]
+          # If we only got one embedding AND the input was a single string (not an array),
+          # return it as a single vector
+          vectors = vectors.first if vectors.length == 1 && !text.is_a?(Array)
 
           Embedding.new(vectors:, model:, input_tokens:)
         end

data/lib/ruby_llm/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module RubyLLM
-  VERSION = '1.3.1'
+  VERSION = '1.3.2beta1'
 end

data/lib/tasks/models_docs.rake

@@ -23,7 +23,7 @@ def generate_models_markdown
     layout: default
     title: Available Models
     parent: Guides
-    nav_order: 10
+    nav_order: 11
     permalink: /guides/available-models
     ---
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm
 version: !ruby/object:Gem::Version
-  version: 1.3.1
+  version: 1.3.2beta1
 platform: ruby
 authors:
 - Carmine Paolino
@@ -235,14 +235,14 @@ required_ruby_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
-      version: 3.1.0
+      version: 3.1.3
 required_rubygems_version: !ruby/object:Gem::Requirement
   requirements:
   - - ">="
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.7
+rubygems_version: 3.6.9
 specification_version: 4
 summary: A single delightful Ruby way to work with AI.
 test_files: []