ruby_llm 1.3.1 → 1.4.0

data/lib/ruby_llm/streaming.rb

@@ -11,7 +11,7 @@ module RubyLLM
     def stream_response(connection, payload, &block)
       accumulator = StreamAccumulator.new
 
-      connection.post stream_url, payload do |req|
+      response = connection.post stream_url, payload do |req|
         if req.options.respond_to?(:on_data)
           # Handle Faraday 2.x streaming with on_data method
           req.options.on_data = handle_stream do |chunk|
@@ -27,7 +27,7 @@ module RubyLLM
         end
       end
 
-      accumulator.to_message
+      accumulator.to_message(response)
     end
 
     def handle_stream(&block)
@@ -55,13 +55,13 @@ module RubyLLM
       end
     end
 
-    def process_stream_chunk(chunk, parser, _env, &)
+    def process_stream_chunk(chunk, parser, env, &)
       RubyLLM.logger.debug "Received chunk: #{chunk}"
 
       if error_chunk?(chunk)
-        handle_error_chunk(chunk, nil)
+        handle_error_chunk(chunk, env)
       else
-        yield handle_sse(chunk, parser, nil, &)
+        yield handle_sse(chunk, parser, env, &)
       end
     end
 
@@ -88,7 +88,16 @@ module RubyLLM
     def handle_error_chunk(chunk, env)
       error_data = chunk.split("\n")[1].delete_prefix('data: ')
       status, _message = parse_streaming_error(error_data)
-      error_response = env.merge(body: JSON.parse(error_data), status: status)
+      parsed_data = JSON.parse(error_data)
+
+      # Create a response-like object that works for both Faraday v1 and v2
+      error_response = if env
+                         env.merge(body: parsed_data, status: status)
+                       else
+                         # For Faraday v1, create a simple object that responds to .status and .body
+                         Struct.new(:body, :status).new(parsed_data, status)
+                       end
+
       ErrorMiddleware.parse_error(provider: self, response: error_response)
     rescue JSON::ParserError => e
       RubyLLM.logger.debug "Failed to parse error chunk: #{e.message}"
@@ -122,7 +131,16 @@ module RubyLLM
 
     def handle_error_event(data, env)
       status, _message = parse_streaming_error(data)
-      error_response = env.merge(body: JSON.parse(data), status: status)
+      parsed_data = JSON.parse(data)
+
+      # Create a response-like object that works for both Faraday v1 and v2
+      error_response = if env
+                         env.merge(body: parsed_data, status: status)
+                       else
+                         # For Faraday v1, create a simple object that responds to .status and .body
+                         Struct.new(:body, :status).new(parsed_data, status)
+                       end
+
       ErrorMiddleware.parse_error(provider: self, response: error_response)
     rescue JSON::ParserError => e
       RubyLLM.logger.debug "Failed to parse error event: #{e.message}"

data/lib/ruby_llm/utils.rb

@@ -23,5 +23,15 @@ module RubyLLM
         Array(item)
       end
     end
+
+    def deep_merge(params, payload)
+      params.merge(payload) do |_key, params_value, payload_value|
+        if params_value.is_a?(Hash) && payload_value.is_a?(Hash)
+          deep_merge(params_value, payload_value)
+        else
+          payload_value
+        end
+      end
+    end
   end
 end

data/lib/ruby_llm/version.rb

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

data/lib/ruby_llm.rb

@@ -18,11 +18,13 @@ loader.inflector.inflect(
   'deepseek' => 'DeepSeek',
   'bedrock' => 'Bedrock',
   'openrouter' => 'OpenRouter',
+  'gpustack' => 'GPUStack',
   'pdf' => 'PDF'
 )
 loader.ignore("#{__dir__}/tasks")
 loader.ignore("#{__dir__}/ruby_llm/railtie")
 loader.ignore("#{__dir__}/ruby_llm/active_record")
+loader.ignore("#{__dir__}/generators")
 loader.setup
 
 # A delightful Ruby interface to modern AI language models.
@@ -83,6 +85,7 @@ RubyLLM::Provider.register :deepseek, RubyLLM::Providers::DeepSeek
 RubyLLM::Provider.register :bedrock, RubyLLM::Providers::Bedrock
 RubyLLM::Provider.register :openrouter, RubyLLM::Providers::OpenRouter
 RubyLLM::Provider.register :ollama, RubyLLM::Providers::Ollama
+RubyLLM::Provider.register :gpustack, RubyLLM::Providers::GPUStack
 
 if defined?(Rails::Railtie)
   require 'ruby_llm/railtie'

data/lib/tasks/models_docs.rake

@@ -23,14 +23,15 @@ def generate_models_markdown
     layout: default
     title: Available Models
     parent: Guides
-    nav_order: 10
+    nav_order: 11
     permalink: /guides/available-models
+    description: Browse hundreds of AI models from every major provider. Always up-to-date, automatically generated.
     ---
 
     # Available Models
     {: .no_toc }
 
-    This guide lists all models available in RubyLLM, automatically generated from the current [model registry](https://github.com/crmne/ruby_llm/blob/main/lib/ruby_llm/models.json).
+    Every model, every provider, always current. Your complete AI model reference.
     {: .fs-6 .fw-300 }
 
     ## Table of contents

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm
 version: !ruby/object:Gem::Version
-  version: 1.3.1
+  version: 1.4.0
 platform: ruby
 authors:
 - Carmine Paolino
@@ -135,6 +135,15 @@ extra_rdoc_files: []
 files:
 - LICENSE
 - README.md
+- lib/generators/ruby_llm/install/templates/INSTALL_INFO.md.tt
+- lib/generators/ruby_llm/install/templates/chat_model.rb.tt
+- lib/generators/ruby_llm/install/templates/create_chats_migration.rb.tt
+- lib/generators/ruby_llm/install/templates/create_messages_migration.rb.tt
+- lib/generators/ruby_llm/install/templates/create_tool_calls_migration.rb.tt
+- lib/generators/ruby_llm/install/templates/initializer.rb.tt
+- lib/generators/ruby_llm/install/templates/message_model.rb.tt
+- lib/generators/ruby_llm/install/templates/tool_call_model.rb.tt
+- lib/generators/ruby_llm/install_generator.rb
 - lib/ruby_llm.rb
 - lib/ruby_llm/active_record/acts_as.rb
 - lib/ruby_llm/aliases.json
@@ -192,6 +201,9 @@ files:
 - lib/ruby_llm/providers/gemini/models.rb
 - lib/ruby_llm/providers/gemini/streaming.rb
 - lib/ruby_llm/providers/gemini/tools.rb
+- lib/ruby_llm/providers/gpustack.rb
+- lib/ruby_llm/providers/gpustack/chat.rb
+- lib/ruby_llm/providers/gpustack/models.rb
 - lib/ruby_llm/providers/ollama.rb
 - lib/ruby_llm/providers/ollama/chat.rb
 - lib/ruby_llm/providers/ollama/media.rb
@@ -235,14 +247,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: []