ruby_llm 1.16.0 → 2.0.0.rc2

data/lib/generators/ruby_llm/install/templates/create_models_migration.rb.tt

@@ -1,39 +0,0 @@
-class <%= create_migration_class_name(model_table_name) %> < ActiveRecord::Migration<%= migration_version %>
-  def change
-    create_table :<%= model_table_name %> do |t|
-      t.string :model_id, null: false
-      t.string :name, null: false
-      t.string :provider, null: false
-      t.string :family
-      t.datetime :model_created_at
-      t.integer :context_window
-      t.integer :max_output_tokens
-      t.date :knowledge_cutoff
-<% if postgresql? %>
-      t.jsonb :modalities, default: {}
-      t.jsonb :capabilities, default: []
-      t.jsonb :pricing, default: {}
-      t.jsonb :metadata, default: {}
-<% elsif mysql? %>
-      t.json :modalities
-      t.json :capabilities
-      t.json :pricing
-      t.json :metadata
-<% else %>
-      t.json :modalities, default: {}
-      t.json :capabilities, default: []
-      t.json :pricing, default: {}
-      t.json :metadata, default: {}
-<% end %>
-      t.timestamps
-
-      t.index [ :provider, :model_id ], unique: true
-      t.index :provider
-      t.index :family
-<% if postgresql? -%>
-      t.index :capabilities, using: :gin
-      t.index :modalities, using: :gin
-<% end -%>
-    end
-  end
-end

data/lib/generators/ruby_llm/install/templates/create_tool_calls_migration.rb.tt

@@ -1,21 +0,0 @@
-<%#- # Migration for creating tool_calls table with database-specific JSON handling -%>
-class <%= create_migration_class_name(tool_call_table_name) %> < ActiveRecord::Migration<%= migration_version %>
-  def change
-    create_table :<%= tool_call_table_name %> do |t|
-      t.string :tool_call_id, null: false
-      t.string :name, null: false
-      t.text :thought_signature
-<% if postgresql? %>
-      t.jsonb :arguments, default: {}
-<% elsif mysql? %>
-      t.json :arguments
-<% else %>
-      t.json :arguments, default: {}
-<% end %>
-      t.timestamps
-    end
-
-    add_index :<%= tool_call_table_name %>, :tool_call_id, unique: true
-    add_index :<%= tool_call_table_name %>, :name
-  end
-end

data/lib/generators/ruby_llm/install/templates/model_model.rb.tt

@@ -1,3 +0,0 @@
-class <%= model_model_name %> < ApplicationRecord
-  <%= acts_as_model_declaration %>
-end

data/lib/generators/ruby_llm/install/templates/tool_call_model.rb.tt

@@ -1,3 +0,0 @@
-class <%= tool_call_model_name %> < ApplicationRecord
-  <%= acts_as_tool_call_declaration %>
-end

data/lib/generators/ruby_llm/upgrade_to_v1_10/templates/add_v1_10_message_columns.rb.tt

@@ -1,19 +0,0 @@
-class AddRubyLlmV110Columns < ActiveRecord::Migration<%= migration_version %>
-  def change
-    unless column_exists?(:<%= message_table_name %>, :thinking_text)
-      add_column :<%= message_table_name %>, :thinking_text, :text
-    end
-
-    unless column_exists?(:<%= message_table_name %>, :thinking_signature)
-      add_column :<%= message_table_name %>, :thinking_signature, :text
-    end
-
-    unless column_exists?(:<%= message_table_name %>, :thinking_tokens)
-      add_column :<%= message_table_name %>, :thinking_tokens, :integer
-    end
-
-    unless column_exists?(:<%= tool_call_table_name %>, :thought_signature)
-      add_column :<%= tool_call_table_name %>, :thought_signature, :string
-    end
-  end
-end

data/lib/generators/ruby_llm/upgrade_to_v1_10/upgrade_to_v1_10_generator.rb

@@ -1,50 +0,0 @@
-# frozen_string_literal: true
-
-require 'rails/generators'
-require 'rails/generators/active_record'
-require_relative '../generator_helpers'
-
-module RubyLLM
-  module Generators
-    # Generator to add v1.10 columns (thinking output + thinking tokens) to existing apps.
-    class UpgradeToV110Generator < Rails::Generators::Base
-      include Rails::Generators::Migration
-      include RubyLLM::Generators::GeneratorHelpers
-
-      namespace 'ruby_llm:upgrade_to_v1_10'
-      source_root File.expand_path('templates', __dir__)
-
-      argument :model_mappings, type: :array, default: [], banner: 'message:MessageName'
-
-      desc 'Adds thinking output columns and thinking token tracking introduced in v1.10.0'
-
-      def self.next_migration_number(dirname)
-        ::ActiveRecord::Generators::Base.next_migration_number(dirname)
-      end
-
-      def create_migration_file
-        parse_model_mappings
-
-        migration_template 'add_v1_10_message_columns.rb.tt',
-                           'db/migrate/add_ruby_llm_v1_10_columns.rb',
-                           migration_version: migration_version,
-                           message_table_name: message_table_name,
-                           tool_call_table_name: tool_call_table_name
-      end
-
-      def show_next_steps
-        say_status :success, 'Upgrade prepared!', :green
-        say <<~INSTRUCTIONS
-
-          Next steps:
-          1. Review the generated migration
-          2. Run: bin/rails db:migrate
-          3. Restart your application server
-
-          📚 See the v1.10.0 release notes for details on extended thinking support.
-
-        INSTRUCTIONS
-      end
-    end
-  end
-end

data/lib/generators/ruby_llm/upgrade_to_v1_14/templates/add_v1_14_tool_call_columns.rb.tt

@@ -1,7 +0,0 @@
-class AddRubyLlmV114Columns < ActiveRecord::Migration<%= migration_version %>
-  def change
-    if column_exists?(:<%= tool_call_table_name %>, :thought_signature, :string)
-      change_column :<%= tool_call_table_name %>, :thought_signature, :text
-    end
-  end
-end

data/lib/generators/ruby_llm/upgrade_to_v1_14/upgrade_to_v1_14_generator.rb

@@ -1,49 +0,0 @@
-# frozen_string_literal: true
-
-require 'rails/generators'
-require 'rails/generators/active_record'
-require_relative '../generator_helpers'
-
-module RubyLLM
-  module Generators
-    # Generator to fix tool call thought signature column type for MySQL safety.
-    class UpgradeToV114Generator < Rails::Generators::Base
-      include Rails::Generators::Migration
-      include RubyLLM::Generators::GeneratorHelpers
-
-      namespace 'ruby_llm:upgrade_to_v1_14'
-      source_root File.expand_path('templates', __dir__)
-
-      argument :model_mappings, type: :array, default: [], banner: 'tool_call:ToolCallName'
-
-      desc 'Updates tool call thought_signature column to text introduced in v1.14.0'
-
-      def self.next_migration_number(dirname)
-        ::ActiveRecord::Generators::Base.next_migration_number(dirname)
-      end
-
-      def create_migration_file
-        parse_model_mappings
-
-        migration_template 'add_v1_14_tool_call_columns.rb.tt',
-                           'db/migrate/add_ruby_llm_v1_14_columns.rb',
-                           migration_version: migration_version,
-                           tool_call_table_name: tool_call_table_name
-      end
-
-      def show_next_steps
-        say_status :success, 'Upgrade prepared!', :green
-        say <<~INSTRUCTIONS
-
-          Next steps:
-          1. Review the generated migration
-          2. Run: bin/rails db:migrate
-          3. Restart your application server
-
-          📚 See the v1.14.0 release notes for details on thought signature persistence.
-
-        INSTRUCTIONS
-      end
-    end
-  end
-end

data/lib/generators/ruby_llm/upgrade_to_v1_7/templates/migration.rb.tt

@@ -1,145 +0,0 @@
-class MigrateToRubyLlmModelReferences < ActiveRecord::Migration<%= migration_version %>
-  def up
-    model_class = <%= model_model_name %>
-    chat_class = <%= chat_model_name %>
-    message_class = <%= message_model_name %>
-<% if @model_table_already_existed %>
-    # Load models from models.json if Model table already existed
-    say_with_time "Loading models from models.json" do
-      RubyLLM.models.load_from_json!
-      model_class.save_to_database
-      "Loaded #{model_class.count} models"
-    end
-<% end %>
-
-    # Then check for any models in existing data that aren't in models.json
-    say_with_time "Checking for additional models in existing data" do
-      collect_and_create_models(chat_class, :<%= chat_table_name %>, model_class)
-      collect_and_create_models(message_class, :<%= message_table_name %>, model_class)
-      model_class.count
-    end
-
-    # Migrate foreign keys
-    migrate_foreign_key(:<%= chat_table_name %>, chat_class, model_class, :<%= model_table_name.singularize %>)
-    migrate_foreign_key(:<%= message_table_name %>, message_class, model_class, :<%= model_table_name.singularize %>)
-  end
-
-  def down
-    # Remove foreign key references
-    if column_exists?(:<%= message_table_name %>, :<%= model_table_name.singularize %>_id)
-      remove_reference :<%= message_table_name %>, :<%= model_table_name.singularize %>, foreign_key: true
-    end
-
-    if column_exists?(:<%= chat_table_name %>, :<%= model_table_name.singularize %>_id)
-      remove_reference :<%= chat_table_name %>, :<%= model_table_name.singularize %>, foreign_key: true
-    end
-
-    # Restore original model_id string columns
-    if column_exists?(:<%= message_table_name %>, :model_id_string)
-      rename_column :<%= message_table_name %>, :model_id_string, :model_id
-    end
-
-    if column_exists?(:<%= chat_table_name %>, :model_id_string)
-      rename_column :<%= chat_table_name %>, :model_id_string, :model_id
-    end
-  end
-
-  private
-
-  def collect_and_create_models(record_class, table_name, model_class)
-    return unless column_exists?(table_name, :model_id)
-
-    has_provider = column_exists?(table_name, :provider)
-
-    # Collect unique model/provider combinations using read_attribute to bypass overrides
-    models_set = Set.new
-
-    record_class.find_each do |record|
-      model_id = record.read_attribute(:model_id)
-      next if model_id.blank?
-
-      provider = has_provider ? record.read_attribute(:provider) : nil
-      models_set.add([ model_id, provider ])
-    end
-
-    models_set.each do |model_id, provider|
-      find_or_create_model(model_id, provider, model_class)
-    end
-  end
-
-  def find_or_create_model(model_id, provider, model_class)
-    return if model_id.blank?
-
-    begin
-      model_info, _provider = RubyLLM.models.resolve(model_id, provider: provider)
-
-      model_class.find_or_create_by!(
-        model_id: model_info.id,
-        provider: model_info.provider
-      ) do |m|
-        m.name = model_info.name || model_info.id
-        m.family = model_info.family
-        m.model_created_at = model_info.created_at
-        m.context_window = model_info.context_window
-        m.max_output_tokens = model_info.max_output_tokens
-        m.knowledge_cutoff = model_info.knowledge_cutoff
-        m.modalities = model_info.modalities.to_h
-        m.capabilities = model_info.capabilities
-        m.pricing = model_info.pricing.to_h
-        m.metadata = model_info.metadata
-      end
-    rescue => e
-      # Skip models that can't be resolved - they'll need manual fixing
-      Rails.logger.warn "Skipping unresolvable model: #{model_id} - will need manual update"
-      nil
-    end
-  end
-
-
-  def migrate_foreign_key(table_name, record_class, model_class, foreign_key_name)
-    return unless column_exists?(table_name, :model_id)
-
-    # Check if we need to rename the string column to avoid collision
-    if column_exists?(table_name, :model_id) && !foreign_key_exists?(table_name, :models)
-      # Temporarily rename the string column
-      rename_column table_name, :model_id, :model_id_string
-    end
-
-    # Add the foreign key reference
-    unless column_exists?(table_name, "#{foreign_key_name}_id")
-      add_reference table_name, foreign_key_name, foreign_key: true
-    end
-
-    say_with_time "Migrating #{table_name} model references" do
-      record_class.reset_column_information
-      has_provider = column_exists?(table_name, :provider)
-
-      # Determine which column to read from (renamed or original)
-      model_id_column = column_exists?(table_name, :model_id_string) ? :model_id_string : :model_id
-
-      record_class.find_each do |record|
-        model_id = record.read_attribute(model_id_column)
-        next if model_id.blank?
-
-        provider = has_provider ? record.read_attribute(:provider) : nil
-
-        model = if has_provider && provider.present?
-          model_class.find_by(model_id: model_id, provider: provider)
-        else
-          find_model_for_record(model_id, model_class)
-        end
-
-        record.update_column("#{foreign_key_name}_id", model.id) if model
-      end
-    end
-  end
-
-  def find_model_for_record(model_id, model_class)
-    begin
-      model_info, _provider = RubyLLM.models.resolve(model_id)
-      model_class.find_by(model_id: model_info.id, provider: model_info.provider)
-    rescue => e
-      model_class.find_by(model_id: model_id)
-    end
-  end
-end

data/lib/generators/ruby_llm/upgrade_to_v1_7/upgrade_to_v1_7_generator.rb

@@ -1,122 +0,0 @@
-# frozen_string_literal: true
-
-require 'rails/generators'
-require 'rails/generators/active_record'
-require_relative '../generator_helpers'
-
-module RubyLLM
-  module Generators
-    # Generator to upgrade existing RubyLLM apps to v1.7 with association-based Rails API
-    class UpgradeToV17Generator < Rails::Generators::Base
-      include Rails::Generators::Migration
-      include RubyLLM::Generators::GeneratorHelpers
-
-      namespace 'ruby_llm:upgrade_to_v1_7'
-      source_root File.expand_path('templates', __dir__)
-
-      # Override source_paths to include install templates
-      def self.source_paths
-        [
-          File.expand_path('templates', __dir__),
-          File.expand_path('../install/templates', __dir__)
-        ]
-      end
-
-      argument :model_mappings, type: :array, default: [], banner: 'chat:ChatName message:MessageName ...'
-
-      desc 'Upgrades existing RubyLLM apps to v1.7 with association-based Rails API\n' \
-           'Usage: bin/rails g ruby_llm:upgrade_to_v1_7 [chat:ChatName] [message:MessageName] ...'
-
-      def self.next_migration_number(dirname)
-        ::ActiveRecord::Generators::Base.next_migration_number(dirname)
-      end
-
-      def create_migration_file
-        @model_table_already_existed = table_exists?(table_name_for(model_model_name))
-
-        # First check if models table exists, if not create it
-        unless @model_table_already_existed
-          migration_template 'create_models_migration.rb.tt',
-                             "db/migrate/create_#{table_name_for(model_model_name)}.rb",
-                             migration_version: migration_version,
-                             model_model_name: model_model_name
-        end
-
-        migration_template 'migration.rb.tt',
-                           'db/migrate/migrate_to_ruby_llm_model_references.rb',
-                           migration_version: migration_version,
-                           chat_model_name: chat_model_name,
-                           message_model_name: message_model_name,
-                           tool_call_model_name: tool_call_model_name,
-                           model_model_name: model_model_name,
-                           model_table_already_existed: @model_table_already_existed
-      end
-
-      def create_model_file
-        create_namespace_modules
-
-        template 'model_model.rb.tt', "app/models/#{model_model_name.underscore}.rb"
-      end
-
-      def update_existing_models
-        update_model_acts_as(chat_model_name, 'acts_as_chat', acts_as_chat_declaration)
-        update_model_acts_as(message_model_name, 'acts_as_message', acts_as_message_declaration)
-        update_model_acts_as(tool_call_model_name, 'acts_as_tool_call', acts_as_tool_call_declaration)
-      end
-
-      def update_initializer
-        initializer_path = 'config/initializers/ruby_llm.rb'
-
-        unless File.exist?(initializer_path)
-          say_status :warning, 'No initializer found. Creating one...', :yellow
-          template 'initializer.rb.tt', initializer_path
-          return
-        end
-
-        initializer_content = File.read(initializer_path)
-
-        return if initializer_content.include?('config.use_new_acts_as')
-
-        inject_into_file initializer_path, before: /^end/ do
-          lines = ["\n  # Enable the association-based Rails API", '  config.use_new_acts_as = true']
-          lines << "  config.model_registry_class = \"#{model_model_name}\"" if model_model_name != 'Model'
-          lines << "\n"
-          lines.join("\n")
-        end
-      end
-
-      def show_next_steps
-        say_status :success, 'Upgrade prepared!', :green
-        say <<~INSTRUCTIONS
-
-          Next steps:
-          1. Review the generated migrations
-          2. Run: bin/rails db:migrate
-          3. Update your code to use the new API: #{chat_model_name}.create! now has the same signature as RubyLLM.chat
-
-          ⚠️  If you get "undefined method 'acts_as_model'" during migration:
-            Add this to config/application.rb BEFORE your Application class:
-
-            RubyLLM.configure do |config|
-              config.use_new_acts_as = true
-            end
-
-          📚 See the full migration guide: https://rubyllm.com/upgrading-to-1-7/
-
-        INSTRUCTIONS
-      end
-
-      private
-
-      def update_model_acts_as(model_name, old_acts_as, new_acts_as)
-        model_path = "app/models/#{model_name.underscore}.rb"
-        return unless File.exist?(Rails.root.join(model_path))
-
-        content = File.read(Rails.root.join(model_path))
-        return unless content.match?(/^\s*#{old_acts_as}/)
-
-        gsub_file model_path, /^\s*#{old_acts_as}.*$/, "  #{new_acts_as}"
-      end
-    end
-  end
-end

data/lib/generators/ruby_llm/upgrade_to_v1_9/templates/add_v1_9_message_columns.rb.tt

@@ -1,15 +0,0 @@
-class AddRubyLlmV19Columns < ActiveRecord::Migration<%= migration_version %>
-  def change
-    unless column_exists?(:<%= message_table_name %>, :cached_tokens)
-      add_column :<%= message_table_name %>, :cached_tokens, :integer
-    end
-
-    unless column_exists?(:<%= message_table_name %>, :cache_creation_tokens)
-      add_column :<%= message_table_name %>, :cache_creation_tokens, :integer
-    end
-
-    unless column_exists?(:<%= message_table_name %>, :content_raw)
-      add_column :<%= message_table_name %>, :content_raw, :json
-    end
-  end
-end

data/lib/generators/ruby_llm/upgrade_to_v1_9/upgrade_to_v1_9_generator.rb

@@ -1,49 +0,0 @@
-# frozen_string_literal: true
-
-require 'rails/generators'
-require 'rails/generators/active_record'
-require_relative '../generator_helpers'
-
-module RubyLLM
-  module Generators
-    # Generator to add v1.9 columns (cached tokens + raw content support) to existing apps.
-    class UpgradeToV19Generator < Rails::Generators::Base
-      include Rails::Generators::Migration
-      include RubyLLM::Generators::GeneratorHelpers
-
-      namespace 'ruby_llm:upgrade_to_v1_9'
-      source_root File.expand_path('templates', __dir__)
-
-      argument :model_mappings, type: :array, default: [], banner: 'message:MessageName'
-
-      desc 'Adds cached token columns and raw content storage fields introduced in v1.9.0'
-
-      def self.next_migration_number(dirname)
-        ::ActiveRecord::Generators::Base.next_migration_number(dirname)
-      end
-
-      def create_migration_file
-        parse_model_mappings
-
-        migration_template 'add_v1_9_message_columns.rb.tt',
-                           'db/migrate/add_ruby_llm_v1_9_columns.rb',
-                           migration_version: migration_version,
-                           message_table_name: message_table_name
-      end
-
-      def show_next_steps
-        say_status :success, 'Upgrade prepared!', :green
-        say <<~INSTRUCTIONS
-
-          Next steps:
-          1. Review the generated migration
-          2. Run: bin/rails db:migrate
-          3. Restart your application server
-
-          📚 See the v1.9.0 release notes for details on cached token tracking and raw content support.
-
-        INSTRUCTIONS
-      end
-    end
-  end
-end