ruby_llm 1.6.4 → 1.7.1

data/lib/ruby_llm/providers/vertexai/models.rb

@@ -0,0 +1,130 @@
+# frozen_string_literal: true
+
+module RubyLLM
+  module Providers
+    class VertexAI
+      # Models methods for the Vertex AI integration
+      module Models
+        # Gemini and other Google models that aren't returned by the API
+        KNOWN_GOOGLE_MODELS = %w[
+          gemini-2.5-flash-lite
+          gemini-2.5-pro
+          gemini-2.5-flash
+          gemini-2.0-flash-lite-001
+          gemini-2.0-flash-001
+          gemini-2.0-flash
+          gemini-2.0-flash-exp
+          gemini-1.5-pro-002
+          gemini-1.5-pro
+          gemini-1.5-flash-002
+          gemini-1.5-flash
+          gemini-1.5-flash-8b
+          gemini-pro
+          gemini-pro-vision
+          gemini-exp-1206
+          gemini-exp-1121
+          gemini-embedding-001
+          text-embedding-005
+          text-embedding-004
+          text-multilingual-embedding-002
+        ].freeze
+
+        def list_models
+          all_models = []
+          page_token = nil
+
+          all_models.concat(build_known_models)
+
+          loop do
+            response = @connection.get('publishers/google/models') do |req|
+              req.headers['x-goog-user-project'] = @config.vertexai_project_id
+              req.params = { pageSize: 100 }
+              req.params[:pageToken] = page_token if page_token
+            end
+
+            publisher_models = response.body['publisherModels'] || []
+            publisher_models.each do |model_data|
+              next if model_data['launchStage'] == 'DEPRECATED'
+
+              model_id = extract_model_id_from_path(model_data['name'])
+              all_models << build_model_from_api_data(model_data, model_id)
+            end
+
+            page_token = response.body['nextPageToken']
+            break unless page_token
+          end
+
+          all_models
+        rescue StandardError => e
+          RubyLLM.logger.debug "Error fetching Vertex AI models: #{e.message}"
+          build_known_models
+        end
+
+        private
+
+        def build_known_models
+          KNOWN_GOOGLE_MODELS.map do |model_id|
+            Model::Info.new(
+              id: model_id,
+              name: model_id,
+              provider: slug,
+              family: determine_model_family(model_id),
+              created_at: nil,
+              context_window: nil,
+              max_output_tokens: nil,
+              modalities: nil,
+              capabilities: %w[streaming function_calling],
+              pricing: nil,
+              metadata: {
+                source: 'known_models'
+              }
+            )
+          end
+        end
+
+        def build_model_from_api_data(model_data, model_id)
+          Model::Info.new(
+            id: model_id,
+            name: model_id,
+            provider: slug,
+            family: determine_model_family(model_id),
+            created_at: nil,
+            context_window: nil,
+            max_output_tokens: nil,
+            modalities: nil,
+            capabilities: extract_capabilities(model_data),
+            pricing: nil,
+            metadata: {
+              version_id: model_data['versionId'],
+              open_source_category: model_data['openSourceCategory'],
+              launch_stage: model_data['launchStage'],
+              supported_actions: model_data['supportedActions'],
+              publisher_model_template: model_data['publisherModelTemplate']
+            }
+          )
+        end
+
+        def extract_model_id_from_path(path)
+          path.split('/').last
+        end
+
+        def determine_model_family(model_id)
+          case model_id
+          when /^gemini-2\.\d+/ then 'gemini-2'
+          when /^gemini-1\.\d+/ then 'gemini-1.5'
+          when /^text-embedding/ then 'text-embedding'
+          when /bison/ then 'palm'
+          else 'gemini'
+          end
+        end
+
+        def extract_capabilities(model_data)
+          capabilities = ['streaming']
+          model_name = model_data['name']
+          capabilities << 'function_calling' if model_name.include?('gemini')
+          capabilities.uniq
+        end
+      end
+    end
+  end
+end

data/lib/ruby_llm/providers/vertexai/streaming.rb

@@ -0,0 +1,14 @@
+# frozen_string_literal: true
+
+module RubyLLM
+  module Providers
+    class VertexAI
+      # Streaming methods for the Vertex AI implementation
+      module Streaming
+        def stream_url
+          "projects/#{@config.vertexai_project_id}/locations/#{@config.vertexai_location}/publishers/google/models/#{@model}:streamGenerateContent?alt=sse" # rubocop:disable Layout/LineLength
+        end
+      end
+    end
+  end
+end

data/lib/ruby_llm/providers/vertexai.rb

@@ -0,0 +1,55 @@
+# frozen_string_literal: true
+
+module RubyLLM
+  module Providers
+    # Google Vertex AI implementation
+    class VertexAI < Gemini
+      include VertexAI::Chat
+      include VertexAI::Streaming
+      include VertexAI::Embeddings
+      include VertexAI::Models
+
+      def initialize(config)
+        super
+        @authorizer = nil
+      end
+
+      def api_base
+        "https://#{@config.vertexai_location}-aiplatform.googleapis.com/v1beta1"
+      end
+
+      def headers
+        {
+          'Authorization' => "Bearer #{access_token}"
+        }
+      end
+
+      class << self
+        def configuration_requirements
+          %i[vertexai_project_id vertexai_location]
+        end
+      end
+
+      private
+
+      def access_token
+        return 'test-token' if defined?(VCR) && !VCR.current_cassette.recording?
+
+        initialize_authorizer unless @authorizer
+        @authorizer.fetch_access_token!['access_token']
+      end
+
+      def initialize_authorizer
+        require 'googleauth'
+        @authorizer = ::Google::Auth.get_application_default(
+          scope: [
+            'https://www.googleapis.com/auth/cloud-platform',
+            'https://www.googleapis.com/auth/generative-language.retriever'
+          ]
+        )
+      rescue LoadError
+        raise Error, 'The googleauth gem is required for Vertex AI. Please add it to your Gemfile: gem "googleauth"'
+      end
+    end
+  end
+end

data/lib/ruby_llm/railtie.rb

@@ -3,14 +3,31 @@
 module RubyLLM
   # Rails integration for RubyLLM
   class Railtie < Rails::Railtie
+    initializer 'ruby_llm.inflections' do
+      ActiveSupport::Inflector.inflections(:en) do |inflect|
+        inflect.acronym 'LLM'
+      end
+    end
+
     initializer 'ruby_llm.active_record' do
       ActiveSupport.on_load :active_record do
-        include RubyLLM::ActiveRecord::ActsAs
+        if RubyLLM.config.use_new_acts_as
+          require 'ruby_llm/active_record/acts_as'
+          ::ActiveRecord::Base.include RubyLLM::ActiveRecord::ActsAs
+        else
+          require 'ruby_llm/active_record/acts_as_legacy'
+          ::ActiveRecord::Base.include RubyLLM::ActiveRecord::ActsAsLegacy
+
+          Rails.logger.warn(
+            "\n!!! RubyLLM's legacy acts_as API is deprecated and will be removed in RubyLLM 2.0.0. " \
+            "Please consult the migration guide at https://rubyllm.com/upgrading-to-1-7/\n"
+          )
+        end
       end
     end
 
-    generators do
-      require 'generators/ruby_llm/install_generator'
+    rake_tasks do
+      load 'tasks/ruby_llm.rake'
     end
   end
 end

data/lib/ruby_llm/streaming.rb

@@ -24,7 +24,7 @@ module RubyLLM
       end
 
       message = accumulator.to_message(response)
-      RubyLLM.logger.debug "Stream completed: #{message.inspect}"
+      RubyLLM.logger.debug "Stream completed: #{message.content}"
       message
     end
 

data/lib/ruby_llm/utils.rb

@@ -5,10 +5,6 @@ module RubyLLM
   module Utils
     module_function
 
-    def format_text_file_for_llm(text_file)
-      "<file name='#{text_file.filename}' mime_type='#{text_file.mime_type}'>#{text_file.content}</file>"
-    end
-
     def hash_get(hash, key)
       hash[key.to_sym] || hash[key.to_s]
     end
@@ -36,12 +32,12 @@ module RubyLLM
       value.is_a?(Date) ? value : Date.parse(value.to_s)
     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)
+    def deep_merge(original, overrides)
+      original.merge(overrides) do |_key, original_value, overrides_value|
+        if original_value.is_a?(Hash) && overrides_value.is_a?(Hash)
+          deep_merge(original_value, overrides_value)
         else
-          payload_value
+          overrides_value
         end
       end
     end

data/lib/ruby_llm/version.rb

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

data/lib/ruby_llm.rb

@@ -21,11 +21,11 @@ loader.inflector.inflect(
   'openrouter' => 'OpenRouter',
   'gpustack' => 'GPUStack',
   'mistral' => 'Mistral',
-  'pdf' => 'PDF'
+  'vertexai' => 'VertexAI',
+  'pdf' => 'PDF',
+  'UI' => 'UI'
 )
 loader.ignore("#{__dir__}/tasks")
-loader.ignore("#{__dir__}/ruby_llm/railtie")
-loader.ignore("#{__dir__}/ruby_llm/active_record")
 loader.ignore("#{__dir__}/generators")
 loader.setup
 
@@ -88,6 +88,7 @@ RubyLLM::Provider.register :ollama, RubyLLM::Providers::Ollama
 RubyLLM::Provider.register :openai, RubyLLM::Providers::OpenAI
 RubyLLM::Provider.register :openrouter, RubyLLM::Providers::OpenRouter
 RubyLLM::Provider.register :perplexity, RubyLLM::Providers::Perplexity
+RubyLLM::Provider.register :vertexai, RubyLLM::Providers::VertexAI
 
 if defined?(Rails::Railtie)
   require 'ruby_llm/railtie'

data/lib/tasks/models.rake

@@ -46,6 +46,8 @@ def configure_from_env
     config.perplexity_api_key = ENV.fetch('PERPLEXITY_API_KEY', nil)
     config.openrouter_api_key = ENV.fetch('OPENROUTER_API_KEY', nil)
     config.mistral_api_key = ENV.fetch('MISTRAL_API_KEY', nil)
+    config.vertexai_location = ENV.fetch('GOOGLE_CLOUD_LOCATION', nil)
+    config.vertexai_project_id = ENV.fetch('GOOGLE_CLOUD_PROJECT', nil)
     configure_bedrock(config)
     config.request_timeout = 30
   end
@@ -74,7 +76,7 @@ def refresh_models
     validate_models!(models)
 
     puts "Saving models.json (#{models.all.size} models)"
-    models.save_models
+    models.save_to_json
   end
 
   @models = models
@@ -152,28 +154,11 @@ def generate_models_markdown
 
     ---
 
-    After reading this guide, you will know:
+    ## Model Data Sources
 
-    * How RubyLLM's model registry works and where data comes from
-    * How to find models by provider, capability, or purpose
-    * What information is available for each model
-    * How to use model aliases for simpler configuration
-
-    ## How Model Data Works
-
-    RubyLLM's model registry combines data from multiple sources:
-
-    - **OpenAI, Anthropic, DeepSeek, Gemini**: Data from [Parsera](https://api.parsera.org/v1/llm-specs)
-    - **OpenRouter**: Direct from OpenRouter's API
-    - **Other providers**: Defined in `capabilities.rb` files
-
-    ## Contributing Model Updates
-
-    **For major providers** (OpenAI, Anthropic, DeepSeek, Gemini): File issues with [Parsera](https://github.com/parsera-labs/api-llm-specs/issues) for public model data corrections.
-
-    **For other providers**: Edit `lib/ruby_llm/providers/<provider>/capabilities.rb` then run `rake models:update`.
-
-    See the [Contributing Guide](https://github.com/crmne/ruby_llm/blob/main/CONTRIBUTING.md) for details.
+    - **OpenAI, Anthropic, DeepSeek, Gemini, VertexAI**: Enriched by [🚀 Parsera](https://parsera.org/) *([free LLM metadata API](https://api.parsera.org/v1/llm-specs) - [go say thanks!](https://github.com/parsera-labs/api-llm-specs))*
+    - **OpenRouter**: Direct API
+    - **Others**: Local capabilities files
 
     ## Last Updated
     {: .d-inline-block }
@@ -369,7 +354,7 @@ def generate_aliases # rubocop:disable Metrics/PerceivedComplexity
 
   models['bedrock'].each do |bedrock_model|
     next unless bedrock_model.start_with?('anthropic.')
-    next unless bedrock_model =~ /anthropic\.(claude-[\d\.]+-[a-z]+)/
+    next unless bedrock_model =~ /anthropic\.(claude-[\d.]+-[a-z]+)/
 
     base_name = Regexp.last_match(1)
     anthropic_name = base_name.tr('.', '-')
@@ -388,6 +373,7 @@ def generate_aliases # rubocop:disable Metrics/PerceivedComplexity
     aliases[anthropic_name]['openrouter'] = openrouter_model if openrouter_model
   end
 
+  # Gemini models (also map to vertexai)
   models['gemini'].each do |model|
     openrouter_variants = [
       "google/#{model}",
@@ -396,13 +382,38 @@ def generate_aliases # rubocop:disable Metrics/PerceivedComplexity
     ]
 
     openrouter_model = openrouter_variants.find { |v| models['openrouter'].include?(v) }
-    next unless openrouter_model
+    vertexai_model = models['vertexai'].include?(model) ? model : nil
+
+    next unless openrouter_model || vertexai_model
 
     alias_key = model.gsub('-latest', '')
-    aliases[alias_key] = {
-      'gemini' => model,
-      'openrouter' => openrouter_model
-    }
+    aliases[alias_key] = { 'gemini' => model }
+    aliases[alias_key]['openrouter'] = openrouter_model if openrouter_model
+    aliases[alias_key]['vertexai'] = vertexai_model if vertexai_model
+  end
+
+  # VertexAI models that aren't in Gemini (e.g. older models like text-bison)
+  models['vertexai'].each do |model|
+    # Skip if already handled above
+    next if models['gemini'].include?(model)
+
+    # Check if OpenRouter has this Google model
+    openrouter_variants = [
+      "google/#{model}",
+      "google/#{model.tr('.', '-')}"
+    ]
+
+    openrouter_model = openrouter_variants.find { |v| models['openrouter'].include?(v) }
+    gemini_model = models['gemini'].include?(model) ? model : nil
+
+    next unless openrouter_model || gemini_model
+
+    alias_key = model.gsub('-latest', '')
+    next if aliases[alias_key] # Skip if already created
+
+    aliases[alias_key] = { 'vertexai' => model }
+    aliases[alias_key]['openrouter'] = openrouter_model if openrouter_model
+    aliases[alias_key]['gemini'] = gemini_model if gemini_model
   end
 
   models['deepseek'].each do |model|

data/lib/tasks/ruby_llm.rake

@@ -0,0 +1,15 @@
+# frozen_string_literal: true
+
+namespace :ruby_llm do
+  desc 'Load models from models.json into the database'
+  task load_models: :environment do
+    if RubyLLM.config.model_registry_class
+      RubyLLM.models.load_from_json!
+      model_class = RubyLLM.config.model_registry_class.constantize
+      model_class.save_to_database
+      puts "✅ Loaded #{model_class.count} models into database"
+    else
+      puts 'Model registry not configured. Run rails generate ruby_llm:install'
+    end
+  end
+end

data/lib/tasks/vcr.rake

@@ -75,11 +75,11 @@ end
 
 namespace :vcr do
   desc 'Record VCR cassettes (rake vcr:record[all] or vcr:record[openai,anthropic])'
-  task :record, [:providers] do |_, args|
+  task :record, :providers do |_, args|
     require 'fileutils'
     require 'ruby_llm'
 
-    providers = (args[:providers] || '').downcase.split(',')
+    providers = args.extras.unshift(args[:providers]).compact.map(&:downcase)
     cassette_dir = 'spec/fixtures/vcr_cassettes'
     FileUtils.mkdir_p(cassette_dir)
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm
 version: !ruby/object:Gem::Version
-  version: 1.6.4
+  version: 1.7.1
 platform: ruby
 authors:
 - Carmine Paolino
@@ -136,17 +136,41 @@ extra_rdoc_files: []
 files:
 - LICENSE
 - README.md
-- lib/generators/ruby_llm/install/templates/INSTALL_INFO.md.tt
+- lib/generators/ruby_llm/chat_ui/chat_ui_generator.rb
+- lib/generators/ruby_llm/chat_ui/templates/controllers/chats_controller.rb.tt
+- lib/generators/ruby_llm/chat_ui/templates/controllers/messages_controller.rb.tt
+- lib/generators/ruby_llm/chat_ui/templates/controllers/models_controller.rb.tt
+- lib/generators/ruby_llm/chat_ui/templates/jobs/chat_response_job.rb.tt
+- lib/generators/ruby_llm/chat_ui/templates/views/chats/_chat.html.erb.tt
+- lib/generators/ruby_llm/chat_ui/templates/views/chats/_form.html.erb.tt
+- lib/generators/ruby_llm/chat_ui/templates/views/chats/index.html.erb.tt
+- lib/generators/ruby_llm/chat_ui/templates/views/chats/new.html.erb.tt
+- lib/generators/ruby_llm/chat_ui/templates/views/chats/show.html.erb.tt
+- lib/generators/ruby_llm/chat_ui/templates/views/messages/_form.html.erb.tt
+- lib/generators/ruby_llm/chat_ui/templates/views/messages/_message.html.erb.tt
+- lib/generators/ruby_llm/chat_ui/templates/views/messages/create.turbo_stream.erb.tt
+- lib/generators/ruby_llm/chat_ui/templates/views/models/_model.html.erb.tt
+- lib/generators/ruby_llm/chat_ui/templates/views/models/index.html.erb.tt
+- lib/generators/ruby_llm/chat_ui/templates/views/models/show.html.erb.tt
+- lib/generators/ruby_llm/generator_helpers.rb
+- lib/generators/ruby_llm/install/install_generator.rb
 - 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_models_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/model_model.rb.tt
 - lib/generators/ruby_llm/install/templates/tool_call_model.rb.tt
-- lib/generators/ruby_llm/install_generator.rb
+- lib/generators/ruby_llm/upgrade_to_v1_7/templates/migration.rb.tt
+- lib/generators/ruby_llm/upgrade_to_v1_7/upgrade_to_v1_7_generator.rb
 - lib/ruby_llm.rb
 - lib/ruby_llm/active_record/acts_as.rb
+- lib/ruby_llm/active_record/acts_as_legacy.rb
+- lib/ruby_llm/active_record/chat_methods.rb
+- lib/ruby_llm/active_record/message_methods.rb
+- lib/ruby_llm/active_record/model_methods.rb
 - lib/ruby_llm/aliases.json
 - lib/ruby_llm/aliases.rb
 - lib/ruby_llm/attachment.rb
@@ -205,6 +229,7 @@ files:
 - 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/media.rb
 - lib/ruby_llm/providers/gpustack/models.rb
 - lib/ruby_llm/providers/mistral.rb
 - lib/ruby_llm/providers/mistral/capabilities.rb
@@ -214,6 +239,7 @@ files:
 - lib/ruby_llm/providers/ollama.rb
 - lib/ruby_llm/providers/ollama/chat.rb
 - lib/ruby_llm/providers/ollama/media.rb
+- lib/ruby_llm/providers/ollama/models.rb
 - lib/ruby_llm/providers/openai.rb
 - lib/ruby_llm/providers/openai/capabilities.rb
 - lib/ruby_llm/providers/openai/chat.rb
@@ -229,6 +255,11 @@ files:
 - lib/ruby_llm/providers/perplexity/capabilities.rb
 - lib/ruby_llm/providers/perplexity/chat.rb
 - lib/ruby_llm/providers/perplexity/models.rb
+- lib/ruby_llm/providers/vertexai.rb
+- lib/ruby_llm/providers/vertexai/chat.rb
+- lib/ruby_llm/providers/vertexai/embeddings.rb
+- lib/ruby_llm/providers/vertexai/models.rb
+- lib/ruby_llm/providers/vertexai/streaming.rb
 - lib/ruby_llm/railtie.rb
 - lib/ruby_llm/stream_accumulator.rb
 - lib/ruby_llm/streaming.rb
@@ -238,6 +269,7 @@ files:
 - lib/ruby_llm/version.rb
 - lib/tasks/models.rake
 - lib/tasks/release.rake
+- lib/tasks/ruby_llm.rake
 - lib/tasks/vcr.rake
 homepage: https://rubyllm.com
 licenses:
@@ -249,6 +281,9 @@ metadata:
   documentation_uri: https://rubyllm.com
   bug_tracker_uri: https://github.com/crmne/ruby_llm/issues
   rubygems_mfa_required: 'true'
+post_install_message: |
+  Upgrading from RubyLLM <= 1.6.x? Check the upgrade guide for new features and migration instructions
+  --> https://rubyllm.com/upgrading-to-1-7/
 rdoc_options: []
 require_paths:
 - lib

data/lib/generators/ruby_llm/install/templates/INSTALL_INFO.md.tt

@@ -1,108 +0,0 @@
-# RubyLLM Rails Setup Complete!
-
-Thanks for installing RubyLLM in your Rails application. Here's what was created:
-
-## Models
-
-- `<%= options[:chat_model_name] %>` - Stores chat sessions and their associated model ID
-- `<%= options[:message_model_name] %>` - Stores individual messages in a chat
-- `<%= options[:tool_call_model_name] %>` - Stores tool calls made by language models
-
-**Note:** Do not add `validates :content, presence: true` to your Message model - RubyLLM creates empty assistant messages before API calls for streaming support.
-
-## Configuration Options
-
-The generator supports the following options to customize model names:
-
-```bash
-rails generate ruby_llm:install \
-  --chat-model-name=Conversation \
-  --message-model-name=ChatMessage \
-  --tool-call-model-name=FunctionCall
-```
-
-This is useful when you need to avoid namespace collisions with existing models in your application. Table names will be automatically derived from the model names following Rails conventions.
-
-## Next Steps
-
-1. **Run migrations:**
-   ```bash
-   rails db:migrate
-   ```
-
-   **Database Note:** The migrations use `jsonb` for PostgreSQL and `json` for MySQL/SQLite automatically.
-
-2. **Set your API keys** in `config/initializers/ruby_llm.rb` or using environment variables:
-   ```ruby
-   # config/initializers/ruby_llm.rb
-   RubyLLM.configure do |config|
-     config.openai_api_key = ENV['OPENAI_API_KEY']
-     config.anthropic_api_key = ENV['ANTHROPIC_API_KEY']
-     config.gemini_api_key = ENV['GEMINI_API_KEY']
-     # ... add other providers as needed
-   end
-   ```
-
-3. **Start using RubyLLM in your code:**
-   ```ruby
-   # Basic usage
-   chat = <%= options[:chat_model_name] %>.create!(model_id: 'gpt-4.1-nano')
-   response = chat.ask("What is Ruby on Rails?")
-
-   # With file attachments (requires ActiveStorage setup)
-   chat.ask("What's in this file?", with: "report.pdf")
-   chat.ask("Analyze these files", with: ["image.jpg", "data.csv", "notes.txt"])
-   ```
-
-4. **For streaming responses** with Hotwire/Turbo:
-   ```ruby
-   # app/models/<%= options[:message_model_name].underscore %>.rb
-   class <%= options[:message_model_name] %> < ApplicationRecord
-     acts_as_message
-
-     # Helper to broadcast chunks during streaming
-     def broadcast_append_chunk(chunk_content)
-       broadcast_append_to [ chat, "messages" ],
-         target: dom_id(self, "content"),
-         html: chunk_content
-     end
-   end
-
-   # app/jobs/chat_stream_job.rb
-   class ChatStreamJob < ApplicationJob
-     def perform(chat_id, user_content)
-       chat = <%= options[:chat_model_name] %>.find(chat_id)
-       chat.ask(user_content) do |chunk|
-         assistant_message = chat.messages.last
-         if chunk.content && assistant_message
-           assistant_message.broadcast_append_chunk(chunk.content)
-         end
-       end
-     end
-   end
-
-   # In your controller
-   ChatStreamJob.perform_later(@chat.id, params[:content])
-   ```
-
-## Optional: ActiveStorage for Attachments
-
-If you want to use file attachments (PDFs, images, etc.), set up ActiveStorage:
-
-```bash
-rails active_storage:install
-rails db:migrate
-```
-
-Then add to your Message model:
-```ruby
-class <%= options[:message_model_name] %> < ApplicationRecord
-  acts_as_message
-  has_many_attached :attachments
-end
-```
-
-## Learn More
-
-- See the [Rails Integration Guide](https://rubyllm.com/guides/rails) for detailed examples
-- Visit the [RubyLLM Documentation](https://rubyllm.com) for full API reference