ruby_llm_community 0.0.6 → 1.1.0

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm_community
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 1.1.0
 platform: ruby
 authors:
 - Paul Shippy
@@ -136,16 +136,39 @@ 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/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/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
@@ -153,11 +176,13 @@ files:
 - lib/ruby_llm/chunk.rb
 - lib/ruby_llm/configuration.rb
 - lib/ruby_llm/connection.rb
+- lib/ruby_llm/connection_multipart.rb
 - lib/ruby_llm/content.rb
 - lib/ruby_llm/context.rb
 - lib/ruby_llm/embedding.rb
 - lib/ruby_llm/error.rb
 - lib/ruby_llm/image.rb
+- lib/ruby_llm/image_attachment.rb
 - lib/ruby_llm/message.rb
 - lib/ruby_llm/mime_type.rb
 - lib/ruby_llm/model.rb
@@ -204,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
@@ -213,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
@@ -231,6 +258,15 @@ 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/providers/xai.rb
+- lib/ruby_llm/providers/xai/capabilities.rb
+- lib/ruby_llm/providers/xai/chat.rb
+- lib/ruby_llm/providers/xai/models.rb
 - lib/ruby_llm/railtie.rb
 - lib/ruby_llm/stream_accumulator.rb
 - lib/ruby_llm/streaming.rb
@@ -240,10 +276,9 @@ files:
 - lib/ruby_llm/version.rb
 - lib/ruby_llm_community.rb
 - lib/shims/ruby_llm.rb
-- lib/tasks/aliases.rake
-- lib/tasks/models_docs.rake
-- lib/tasks/models_update.rake
+- lib/tasks/models.rake
 - lib/tasks/release.rake
+- lib/tasks/ruby_llm.rake
 - lib/tasks/vcr.rake
 homepage: https://rubyllm.com
 licenses:
@@ -255,6 +290,9 @@ metadata:
   documentation_uri: https://rubyllm.com
   bug_tracker_uri: https://github.com/tpaulshippy/ruby_llm_community/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

data/lib/generators/ruby_llm/install_generator.rb

@@ -1,121 +0,0 @@
-# frozen_string_literal: true
-
-require 'rails/generators'
-require 'rails/generators/active_record'
-
-module RubyLLM
-  # Generator for RubyLLM Rails models and migrations
-  class InstallGenerator < Rails::Generators::Base
-    include Rails::Generators::Migration
-
-    namespace 'ruby_llm:install'
-
-    source_root File.expand_path('install/templates', __dir__)
-
-    class_option :chat_model_name, type: :string, default: 'Chat',
-                                   desc: 'Name of the Chat model class'
-    class_option :message_model_name, type: :string, default: 'Message',
-                                      desc: 'Name of the Message model class'
-    class_option :tool_call_model_name, type: :string, default: 'ToolCall',
-                                        desc: 'Name of the ToolCall model class'
-
-    desc 'Creates model files for Chat, Message, and ToolCall, and creates migrations for RubyLLM Rails integration'
-
-    def self.next_migration_number(dirname)
-      ::ActiveRecord::Generators::Base.next_migration_number(dirname)
-    end
-
-    def migration_version
-      "[#{Rails::VERSION::MAJOR}.#{Rails::VERSION::MINOR}]"
-    end
-
-    def postgresql?
-      ::ActiveRecord::Base.connection.adapter_name.downcase.include?('postgresql')
-    rescue StandardError
-      false
-    end
-
-    def acts_as_chat_declaration
-      acts_as_chat_params = []
-      if options[:message_model_name] != 'Message'
-        acts_as_chat_params << "message_class: \"#{options[:message_model_name]}\""
-      end
-      if options[:tool_call_model_name] != 'ToolCall'
-        acts_as_chat_params << "tool_call_class: \"#{options[:tool_call_model_name]}\""
-      end
-      if acts_as_chat_params.any?
-        "acts_as_chat #{acts_as_chat_params.join(', ')}"
-      else
-        'acts_as_chat'
-      end
-    end
-
-    def acts_as_message_declaration
-      acts_as_message_params = []
-      acts_as_message_params << "chat_class: \"#{options[:chat_model_name]}\"" if options[:chat_model_name] != 'Chat'
-      if options[:tool_call_model_name] != 'ToolCall'
-        acts_as_message_params << "tool_call_class: \"#{options[:tool_call_model_name]}\""
-      end
-      if acts_as_message_params.any?
-        "acts_as_message #{acts_as_message_params.join(', ')}"
-      else
-        'acts_as_message'
-      end
-    end
-
-    def acts_as_tool_call_declaration
-      acts_as_tool_call_params = []
-      if options[:message_model_name] != 'Message'
-        acts_as_tool_call_params << "message_class: \"#{options[:message_model_name]}\""
-      end
-      if acts_as_tool_call_params.any?
-        "acts_as_tool_call #{acts_as_tool_call_params.join(', ')}"
-      else
-        'acts_as_tool_call'
-      end
-    end
-
-    def create_migration_files
-      # Create migrations with timestamps to ensure proper order
-      # First create chats table
-      migration_template 'create_chats_migration.rb.tt',
-                         "db/migrate/create_#{options[:chat_model_name].tableize}.rb"
-
-      # Then create messages table (must come before tool_calls due to foreign key)
-      sleep 1 # Ensure different timestamp
-      migration_template 'create_messages_migration.rb.tt',
-                         "db/migrate/create_#{options[:message_model_name].tableize}.rb"
-
-      # Finally create tool_calls table (references messages)
-      sleep 1 # Ensure different timestamp
-      migration_template 'create_tool_calls_migration.rb.tt',
-                         "db/migrate/create_#{options[:tool_call_model_name].tableize}.rb"
-    end
-
-    def create_model_files
-      template 'chat_model.rb.tt', "app/models/#{options[:chat_model_name].underscore}.rb"
-      template 'message_model.rb.tt', "app/models/#{options[:message_model_name].underscore}.rb"
-      template 'tool_call_model.rb.tt', "app/models/#{options[:tool_call_model_name].underscore}.rb"
-    end
-
-    def create_initializer
-      template 'initializer.rb.tt', 'config/initializers/ruby_llm.rb'
-    end
-
-    def show_install_info
-      say "\n  ✅ RubyLLM installed!", :green
-
-      say "\n  Next steps:", :yellow
-      say '     1. Run: rails db:migrate'
-      say '     2. Set your API keys in config/initializers/ruby_llm.rb'
-      say "     3. Start chatting: #{options[:chat_model_name]}.create!(model_id: 'gpt-4.1-nano').ask('Hello!')"
-
-      say "\n  📚 Full docs: https://rubyllm.com", :cyan
-
-      say "\n  ❤️  Love RubyLLM?", :magenta
-      say '     • ⭐ Star on GitHub: https://github.com/crmne/ruby_llm'
-      say '     • 💖 Sponsor: https://github.com/sponsors/crmne'
-      say "\n"
-    end
-  end
-end

data/lib/tasks/aliases.rake

@@ -1,235 +0,0 @@
-# frozen_string_literal: true
-
-require 'json'
-
-namespace :aliases do # rubocop:disable Metrics/BlockLength
-  desc 'Generate aliases.json from models in the registry'
-  task :generate do # rubocop:disable Metrics/BlockLength
-    require 'ruby_llm'
-
-    # Group models by provider
-    models = Hash.new { |h, k| h[k] = [] }
-
-    RubyLLM.models.all.each do |model|
-      models[model.provider] << model.id
-    end
-
-    aliases = {}
-
-    # OpenAI models
-    models['openai'].each do |model|
-      openrouter_model = "openai/#{model}"
-      next unless models['openrouter'].include?(openrouter_model)
-
-      alias_key = model.gsub('-latest', '')
-      aliases[alias_key] = {
-        'openai' => model,
-        'openrouter' => openrouter_model
-      }
-    end
-
-    # Anthropic models - group by base name and find latest
-    anthropic_latest = group_anthropic_models_by_base_name(models['anthropic'])
-
-    anthropic_latest.each do |base_name, latest_model|
-      # Check OpenRouter naming patterns for the BASE NAME (not the full dated model)
-      openrouter_variants = [
-        "anthropic/#{base_name}", # anthropic/claude-3-5-sonnet
-        "anthropic/#{base_name.gsub(/-(\d)/, '.\1')}", # anthropic/claude-3.5-sonnet
-        "anthropic/#{base_name.gsub(/claude-(\d+)-(\d+)/, 'claude-\1.\2')}", # claude-3-5 -> claude-3.5
-        "anthropic/#{base_name.gsub(/(\d+)-(\d+)/, '\1.\2')}" # any X-Y -> X.Y pattern
-      ]
-
-      openrouter_model = openrouter_variants.find { |v| models['openrouter'].include?(v) }
-
-      # Find corresponding Bedrock model
-      bedrock_model = find_best_bedrock_model(latest_model, models['bedrock'])
-
-      # Create alias if we have any match (OpenRouter OR Bedrock) OR if it's Anthropic-only
-      next unless openrouter_model || bedrock_model || models['anthropic'].include?(latest_model)
-
-      aliases[base_name] = {
-        'anthropic' => latest_model
-      }
-
-      aliases[base_name]['openrouter'] = openrouter_model if openrouter_model
-      aliases[base_name]['bedrock'] = bedrock_model if bedrock_model
-    end
-
-    # Also check if Bedrock has models that Anthropic doesn't
-    models['bedrock'].each do |bedrock_model|
-      next unless bedrock_model.start_with?('anthropic.')
-
-      # Extract the Claude model name
-      next unless bedrock_model =~ /anthropic\.(claude-[\d\.]+-[a-z]+)/
-
-      base_name = Regexp.last_match(1)
-      # Normalize to Anthropic naming convention
-      anthropic_name = base_name.tr('.', '-')
-
-      # Skip if we already have an alias for this
-      next if aliases[anthropic_name]
-
-      # Check if this model exists in OpenRouter
-      openrouter_variants = [
-        "anthropic/#{anthropic_name}",
-        "anthropic/#{base_name}" # Keep the dots
-      ]
-
-      openrouter_model = openrouter_variants.find { |v| models['openrouter'].include?(v) }
-
-      aliases[anthropic_name] = {
-        'bedrock' => bedrock_model
-      }
-
-      aliases[anthropic_name]['anthropic'] = anthropic_name if models['anthropic'].include?(anthropic_name)
-      aliases[anthropic_name]['openrouter'] = openrouter_model if openrouter_model
-    end
-
-    # Gemini models
-    models['gemini'].each do |model|
-      # OpenRouter uses "google/" prefix and sometimes different naming
-      openrouter_variants = [
-        "google/#{model}",
-        "google/#{model.gsub('gemini-', 'gemini-').tr('.', '-')}",
-        "google/#{model.gsub('gemini-', 'gemini-')}"
-      ]
-
-      openrouter_model = openrouter_variants.find { |v| models['openrouter'].include?(v) }
-      next unless openrouter_model
-
-      alias_key = model.gsub('-latest', '')
-      aliases[alias_key] = {
-        'gemini' => model,
-        'openrouter' => openrouter_model
-      }
-    end
-
-    # DeepSeek models
-    models['deepseek'].each do |model|
-      openrouter_model = "deepseek/#{model}"
-      next unless models['openrouter'].include?(openrouter_model)
-
-      alias_key = model.gsub('-latest', '')
-      aliases[alias_key] = {
-        'deepseek' => model,
-        'openrouter' => openrouter_model
-      }
-    end
-
-    # Write the result
-    sorted_aliases = aliases.sort.to_h
-    File.write('lib/ruby_llm/aliases.json', JSON.pretty_generate(sorted_aliases))
-
-    puts "Generated #{sorted_aliases.size} aliases"
-  end
-
-  def group_anthropic_models_by_base_name(anthropic_models) # rubocop:disable Rake/MethodDefinitionInTask
-    grouped = Hash.new { |h, k| h[k] = [] }
-
-    anthropic_models.each do |model|
-      base_name = extract_base_name(model)
-      grouped[base_name] << model
-    end
-
-    # Find the latest model for each base name
-    latest_models = {}
-    grouped.each do |base_name, model_list|
-      if model_list.size == 1
-        latest_models[base_name] = model_list.first
-      else
-        # Sort by date and take the latest
-        latest_model = model_list.max_by { |model| extract_date_from_model(model) }
-        latest_models[base_name] = latest_model
-      end
-    end
-
-    latest_models
-  end
-
-  def extract_base_name(model) # rubocop:disable Rake/MethodDefinitionInTask
-    # Remove date suffix (YYYYMMDD) from model name
-    if model =~ /^(.+)-(\d{8})$/
-      Regexp.last_match(1)
-    else
-      # Models without date suffix (like claude-2.0, claude-2.1)
-      model
-    end
-  end
-
-  def extract_date_from_model(model) # rubocop:disable Rake/MethodDefinitionInTask
-    # Extract date for comparison, return '00000000' for models without dates
-    if model =~ /-(\d{8})$/
-      Regexp.last_match(1)
-    else
-      '00000000' # Ensures models without dates sort before dated ones
-    end
-  end
-
-  def find_best_bedrock_model(anthropic_model, bedrock_models) # rubocop:disable Metrics/PerceivedComplexity,Rake/MethodDefinitionInTask
-    # Special mapping for Claude 2.x models
-    base_pattern = case anthropic_model
-                   when 'claude-2.0', 'claude-2'
-                     'claude-v2'
-                   when 'claude-2.1'
-                     'claude-v2:1'
-                   when 'claude-instant-v1', 'claude-instant'
-                     'claude-instant'
-                   else
-                     # For Claude 3+ models, extract base name
-                     extract_base_name(anthropic_model)
-                   end
-
-    # Find all matching Bedrock models by stripping provider prefix and comparing base name
-    matching_models = bedrock_models.select do |bedrock_model|
-      # Strip any provider prefix (anthropic. or us.anthropic.)
-      model_without_prefix = bedrock_model.sub(/^(?:us\.)?anthropic\./, '')
-      model_without_prefix.start_with?(base_pattern)
-    end
-
-    return nil if matching_models.empty?
-
-    # Get model info to check context window
-    begin
-      model_info = RubyLLM.models.find(anthropic_model)
-      target_context = model_info.context_window
-    rescue StandardError
-      target_context = nil
-    end
-
-    # If we have context window info, try to match it
-    if target_context
-      # Convert to k format (200000 -> 200k)
-      target_k = target_context / 1000
-
-      # Find models with this specific context window
-      with_context = matching_models.select do |m|
-        m.include?(":#{target_k}k") || m.include?(":0:#{target_k}k")
-      end
-
-      return with_context.first if with_context.any?
-    end
-
-    # Otherwise, pick the one with the highest context window or latest version
-    matching_models.min_by do |model|
-      # Extract context window if specified
-      context_priority = if model =~ /:(?:\d+:)?(\d+)k/
-                           -Regexp.last_match(1).to_i # Negative for descending sort
-                         else
-                           0 # No context specified
-                         end
-
-      # Extract version if present
-      version_priority = if model =~ /-v(\d+):/
-                           -Regexp.last_match(1).to_i # Negative for descending sort (latest version first)
-                         else
-                           0
-                         end
-
-      # Prefer models with explicit context windows
-      has_context_priority = model.include?('k') ? -1 : 0
-
-      [has_context_priority, context_priority, version_priority]
-    end
-  end
-end