llm.rb 0.16.1 → 0.16.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c46802e2152430f164990a91499be669e928201e7793b162f8f62152349078af
-  data.tar.gz: 58269b584d08e9d236a3d85ac5728f4e4453a845ace9ebbf8c1ac22890609a1f
+  metadata.gz: 01f2984f551757482fbf590485875bef064060d27eb84d3d4325e307db1029c5
+  data.tar.gz: a492d121a5dc1916412296269cf0055ce14512855fecb94e2b018dcefb9b404d
 SHA512:
-  metadata.gz: 9003c9cac451081ce0589f54acb2d72813e53a8708426e320e30732aa5842d99187af14f1c54bab69cfcc3bcc401529375377547a090f530aa1da140fd0a4b3f
-  data.tar.gz: 691e4ebfd80fcfa7af22f1ca26393256a69ddf6957be8960590eebddc5e20a1a30db3ba8cfc1ae65dba10491df6f2844fd72d3ca2444420cb23db357a0225e25
+  metadata.gz: 341d074d9d732056599d6f27865f8fcf3fecd0f6062de73ec5dffcd056031e727c43f8058c472d69b598ae3fa4232bb47b960b331b20b8f7cc9323a86e9beeb7
+  data.tar.gz: aa9c19a4b51b8c379f6d8a8f486172e332d6ca1fb4a3f65fa2b32ff7685a3f48fd6d5db0092bcc3a97fca7f36b4f4214df1dfd675ad1215a172e5ae9ce76cc00

data/lib/llm/buffer.rb

@@ -65,12 +65,20 @@ module LLM
     alias_method :push, :<<
 
     ##
-    # @param [Integer, #to_i] index
+    # @param [Integer, Range, #to_i] index
     #  The message index
     # @return [LLM::Message, nil]
     #  Returns a message, or nil
     def [](index)
-      @completed[index.to_i] || to_a[index.to_i]
+      if index.respond_to?(:to_i)
+        @completed[index.to_i] || to_a[index.to_i]
+      elsif Range === index
+        slice = @completed[index]
+        invalidate = slice.nil? || slice.size < index.size
+        invalidate ? to_a[index] : slice
+      else
+        raise TypeError, "index must be an Integer or Range"
+      end
     end
 
     ##

data/lib/llm/error.rb

@@ -35,6 +35,10 @@ module LLM
   # HTTPServerError
   ServerError = Class.new(ResponseError)
 
+  ##
+  # When no images are found in a response
+  NoImageError = Class.new(ResponseError)
+
   ##
   # When an given an input object that is not understood
   FormatError = Class.new(Error)

data/lib/llm/providers/gemini/images.rb

@@ -36,12 +36,13 @@ class LLM::Gemini
     # @param [String] prompt The prompt
     # @param [Hash] params Other parameters (see Gemini docs)
     # @raise (see LLM::Provider#request)
+    # @raise [LLM::NoImageError] when no images are returned
     # @note
     #  The prompt should make it clear you want to generate an image, or you
     #  might unexpectedly receive a purely textual response. This is due to how
     #  Gemini implements image generation under the hood.
     # @return [LLM::Response]
-    def create(prompt:, model: "gemini-2.0-flash-exp-image-generation", **params)
+    def create(prompt:, model: "gemini-2.5-flash-image-preview", **params)
       req  = Net::HTTP::Post.new("/v1beta/models/#{model}:generateContent?key=#{key}", headers)
       body = JSON.dump({
         contents: [{parts: [{text: create_prompt}, {text: prompt}]}],
@@ -49,7 +50,7 @@ class LLM::Gemini
       }.merge!(params))
       req.body = body
       res = execute(request: req)
-      LLM::Response.new(res).extend(LLM::Gemini::Response::Image)
+      validate LLM::Response.new(res).extend(LLM::Gemini::Response::Image)
     end
 
     ##
@@ -63,9 +64,10 @@ class LLM::Gemini
     # @param [String] prompt The prompt
     # @param [Hash] params Other parameters (see Gemini docs)
     # @raise (see LLM::Provider#request)
+    # @raise [LLM::NoImageError] when no images are returned
     # @note (see LLM::Gemini::Images#create)
     # @return [LLM::Response]
-    def edit(image:, prompt:, model: "gemini-2.0-flash-exp-image-generation", **params)
+    def edit(image:, prompt:, model: "gemini-2.5-flash-image-preview", **params)
       req   = Net::HTTP::Post.new("/v1beta/models/#{model}:generateContent?key=#{key}", headers)
       image = LLM.File(image)
       body  = JSON.dump({
@@ -74,7 +76,7 @@ class LLM::Gemini
       }.merge!(params)).b
       set_body_stream(req, StringIO.new(body))
       res = execute(request: req)
-      LLM::Response.new(res).extend(LLM::Gemini::Response::Image)
+      validate LLM::Response.new(res).extend(LLM::Gemini::Response::Image)
     end
 
     ##
@@ -119,6 +121,11 @@ class LLM::Gemini
       PROMPT
     end
 
+    def validate(res)
+      return res unless res.images.empty?
+      raise LLM::NoImageError.new { _1.response = res.res }, "no images found in response"
+    end
+
     [:headers, :execute, :set_body_stream].each do |m|
       define_method(m) { |*args, **kwargs, &b| @provider.send(m, *args, **kwargs, &b) }
     end

data/lib/llm/providers/gemini/response/image.rb

@@ -22,5 +22,10 @@ module LLM::Gemini::Response
     #  will always return an empty array.
     # @return [Array<String>]
     def urls = []
+
+    ##
+    # Returns one or more candidates, or an empty array
+    # @return [Array<Hash>]
+    def candidates = body.candidates || []
   end
 end

data/lib/llm/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module LLM
-  VERSION = "0.16.1"
+  VERSION = "0.16.2"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: llm.rb
 version: !ruby/object:Gem::Version
-  version: 0.16.1
+  version: 0.16.2
 platform: ruby
 authors:
 - Antar Azri