ruby_llm-agents 3.0.0 → 3.2.0

data/lib/generators/ruby_llm_agents/embedder_generator.rb

@@ -17,13 +17,13 @@ module RubyLlmAgents
     source_root File.expand_path("templates", __dir__)
 
     class_option :model, type: :string, default: "text-embedding-3-small",
-                 desc: "The embedding model to use"
+      desc: "The embedding model to use"
     class_option :dimensions, type: :numeric, default: nil,
-                 desc: "Vector dimensions (nil for model default)"
+      desc: "Vector dimensions (nil for model default)"
     class_option :batch_size, type: :numeric, default: 100,
-                 desc: "Texts per API call for batch processing"
+      desc: "Texts per API call for batch processing"
     class_option :cache, type: :string, default: nil,
-                 desc: "Cache TTL (e.g., '1.week', '1.day')"
+      desc: "Cache TTL (e.g., '1.week', '1.day')"
 
     def ensure_base_class_and_skill_file
       embedders_dir = "app/agents/embedders"

data/lib/generators/ruby_llm_agents/image_analyzer_generator.rb

@@ -17,19 +17,19 @@ module RubyLlmAgents
     source_root File.expand_path("templates", __dir__)
 
     class_option :model, type: :string, default: "gpt-4o",
-                 desc: "The vision model to use"
+      desc: "The vision model to use"
     class_option :analysis_type, type: :string, default: "detailed",
-                 desc: "Analysis type (caption, detailed, tags, objects, colors, all)"
+      desc: "Analysis type (caption, detailed, tags, objects, colors, all)"
     class_option :extract_colors, type: :boolean, default: false,
-                 desc: "Enable color extraction"
+      desc: "Enable color extraction"
     class_option :detect_objects, type: :boolean, default: false,
-                 desc: "Enable object detection"
+      desc: "Enable object detection"
     class_option :extract_text, type: :boolean, default: false,
-                 desc: "Enable text extraction (OCR)"
+      desc: "Enable text extraction (OCR)"
     class_option :max_tags, type: :string, default: "10",
-                 desc: "Maximum number of tags to return"
+      desc: "Maximum number of tags to return"
     class_option :cache, type: :string, default: nil,
-                 desc: "Cache TTL (e.g., '1.hour', '1.day')"
+      desc: "Cache TTL (e.g., '1.hour', '1.day')"
 
     def ensure_base_class_and_skill_file
       images_dir = "app/agents/images"

data/lib/generators/ruby_llm_agents/image_editor_generator.rb

@@ -17,13 +17,13 @@ module RubyLlmAgents
     source_root File.expand_path("templates", __dir__)
 
     class_option :model, type: :string, default: "gpt-image-1",
-                 desc: "The image model to use"
+      desc: "The image model to use"
     class_option :size, type: :string, default: "1024x1024",
-                 desc: "Output image size (e.g., 1024x1024)"
+      desc: "Output image size (e.g., 1024x1024)"
     class_option :content_policy, type: :string, default: "standard",
-                 desc: "Content policy level (none, standard, moderate, strict)"
+      desc: "Content policy level (none, standard, moderate, strict)"
     class_option :cache, type: :string, default: nil,
-                 desc: "Cache TTL (e.g., '1.hour', '1.day')"
+      desc: "Cache TTL (e.g., '1.hour', '1.day')"
 
     def ensure_base_class_and_skill_file
       images_dir = "app/agents/images"

data/lib/generators/ruby_llm_agents/image_generator_generator.rb

@@ -17,17 +17,17 @@ module RubyLlmAgents
     source_root File.expand_path("templates", __dir__)
 
     class_option :model, type: :string, default: "gpt-image-1",
-                 desc: "The image generation model to use"
+      desc: "The image generation model to use"
     class_option :size, type: :string, default: "1024x1024",
-                 desc: "Image size (e.g., 1024x1024, 1792x1024)"
+      desc: "Image size (e.g., 1024x1024, 1792x1024)"
     class_option :quality, type: :string, default: "standard",
-                 desc: "Image quality (standard, hd)"
+      desc: "Image quality (standard, hd)"
     class_option :style, type: :string, default: "vivid",
-                 desc: "Image style (vivid, natural)"
+      desc: "Image style (vivid, natural)"
     class_option :content_policy, type: :string, default: "standard",
-                 desc: "Content policy level (none, standard, moderate, strict)"
+      desc: "Content policy level (none, standard, moderate, strict)"
     class_option :cache, type: :string, default: nil,
-                 desc: "Cache TTL (e.g., '1.hour', '1.day')"
+      desc: "Cache TTL (e.g., '1.hour', '1.day')"
 
     def ensure_base_class_and_skill_file
       images_dir = "app/agents/images"

data/lib/generators/ruby_llm_agents/image_pipeline_generator.rb

@@ -17,11 +17,11 @@ module RubyLlmAgents
     source_root File.expand_path("templates", __dir__)
 
     class_option :steps, type: :string, default: "generate,upscale",
-                 desc: "Pipeline steps (comma-separated: generate,upscale,transform,analyze,remove_background)"
+      desc: "Pipeline steps (comma-separated: generate,upscale,transform,analyze,remove_background)"
     class_option :stop_on_error, type: :boolean, default: true,
-                 desc: "Stop pipeline on first error"
+      desc: "Stop pipeline on first error"
     class_option :cache, type: :string, default: nil,
-                 desc: "Cache TTL (e.g., '1.hour', '1.day')"
+      desc: "Cache TTL (e.g., '1.hour', '1.day')"
 
     def ensure_base_class_and_skill_file
       images_dir = "app/agents/images"
@@ -132,17 +132,17 @@ module RubyLlmAgents
         class_base = name.split("/").map(&:camelize).join("::")
         case step
         when :generate
-          { step: step, type: :generator, class_name: "Images::#{class_base}Generator" }
+          {step: step, type: :generator, class_name: "Images::#{class_base}Generator"}
         when :upscale
-          { step: step, type: :upscaler, class_name: "Images::#{class_base}Upscaler" }
+          {step: step, type: :upscaler, class_name: "Images::#{class_base}Upscaler"}
         when :transform
-          { step: step, type: :transformer, class_name: "Images::#{class_base}Transformer" }
+          {step: step, type: :transformer, class_name: "Images::#{class_base}Transformer"}
         when :analyze
-          { step: step, type: :analyzer, class_name: "Images::#{class_base}Analyzer" }
+          {step: step, type: :analyzer, class_name: "Images::#{class_base}Analyzer"}
         when :remove_background
-          { step: step, type: :remover, class_name: "Images::#{class_base}BackgroundRemover" }
+          {step: step, type: :remover, class_name: "Images::#{class_base}BackgroundRemover"}
         else
-          { step: step, type: step, class_name: "Images::#{class_base}#{step.to_s.camelize}" }
+          {step: step, type: step, class_name: "Images::#{class_base}#{step.to_s.camelize}"}
         end
       end
     end

data/lib/generators/ruby_llm_agents/image_transformer_generator.rb

@@ -17,17 +17,17 @@ module RubyLlmAgents
     source_root File.expand_path("templates", __dir__)
 
     class_option :model, type: :string, default: "sdxl",
-                 desc: "The image model to use"
+      desc: "The image model to use"
     class_option :size, type: :string, default: "1024x1024",
-                 desc: "Output image size (e.g., 1024x1024)"
+      desc: "Output image size (e.g., 1024x1024)"
     class_option :strength, type: :string, default: "0.75",
-                 desc: "Transformation strength (0.0-1.0)"
+      desc: "Transformation strength (0.0-1.0)"
     class_option :template, type: :string, default: nil,
-                 desc: "Prompt template (use {prompt} as placeholder)"
+      desc: "Prompt template (use {prompt} as placeholder)"
     class_option :content_policy, type: :string, default: "standard",
-                 desc: "Content policy level (none, standard, moderate, strict)"
+      desc: "Content policy level (none, standard, moderate, strict)"
     class_option :cache, type: :string, default: nil,
-                 desc: "Cache TTL (e.g., '1.hour', '1.day')"
+      desc: "Cache TTL (e.g., '1.hour', '1.day')"
 
     def ensure_base_class_and_skill_file
       images_dir = "app/agents/images"

data/lib/generators/ruby_llm_agents/image_upscaler_generator.rb

@@ -17,13 +17,13 @@ module RubyLlmAgents
     source_root File.expand_path("templates", __dir__)
 
     class_option :model, type: :string, default: "real-esrgan",
-                 desc: "The upscaling model to use"
+      desc: "The upscaling model to use"
     class_option :scale, type: :string, default: "4",
-                 desc: "Upscale factor (2, 4, or 8)"
+      desc: "Upscale factor (2, 4, or 8)"
     class_option :face_enhance, type: :boolean, default: false,
-                 desc: "Enable face enhancement"
+      desc: "Enable face enhancement"
     class_option :cache, type: :string, default: nil,
-                 desc: "Cache TTL (e.g., '1.hour', '1.day')"
+      desc: "Cache TTL (e.g., '1.hour', '1.day')"
 
     def ensure_base_class_and_skill_file
       images_dir = "app/agents/images"

data/lib/generators/ruby_llm_agents/image_variator_generator.rb

@@ -17,13 +17,13 @@ module RubyLlmAgents
     source_root File.expand_path("templates", __dir__)
 
     class_option :model, type: :string, default: "gpt-image-1",
-                 desc: "The image model to use"
+      desc: "The image model to use"
     class_option :size, type: :string, default: "1024x1024",
-                 desc: "Output image size (e.g., 1024x1024)"
+      desc: "Output image size (e.g., 1024x1024)"
     class_option :variation_strength, type: :string, default: "0.5",
-                 desc: "Variation strength (0.0-1.0)"
+      desc: "Variation strength (0.0-1.0)"
     class_option :cache, type: :string, default: nil,
-                 desc: "Cache TTL (e.g., '1.hour', '1.day')"
+      desc: "Cache TTL (e.g., '1.hour', '1.day')"
 
     def ensure_base_class_and_skill_file
       images_dir = "app/agents/images"

data/lib/generators/ruby_llm_agents/install_generator.rb

@@ -22,11 +22,11 @@ module RubyLlmAgents
     source_root File.expand_path("templates", __dir__)
 
     class_option :skip_migration, type: :boolean, default: false,
-                 desc: "Skip generating the migration file"
+      desc: "Skip generating the migration file"
     class_option :skip_initializer, type: :boolean, default: false,
-                 desc: "Skip generating the initializer file"
+      desc: "Skip generating the initializer file"
     class_option :mount_dashboard, type: :boolean, default: true,
-                 desc: "Mount the dashboard engine in routes"
+      desc: "Mount the dashboard engine in routes"
 
     def create_migration_file
       return if options[:skip_migration]

data/lib/generators/ruby_llm_agents/migrate_structure_generator.rb

@@ -158,7 +158,7 @@ module RubyLlmAgents
 
       directories.each do |dir|
         if options[:dry_run]
-          say_status :dry_run, "Would create #{dir.sub(Rails.root.to_s + '/', '')}", :yellow
+          say_status :dry_run, "Would create #{dir.sub(Rails.root.to_s + "/", "")}", :yellow
         else
           FileUtils.mkdir_p(dir)
           say_status :mkdir, dir.sub(Rails.root.to_s + "/", ""), :green
@@ -393,7 +393,7 @@ module RubyLlmAgents
 
       if content != original_content
         File.write(file_path, content)
-        say_status :updated, "base classes in #{file_path.sub(Rails.root.to_s + '/', '')}", :blue
+        say_status :updated, "base classes in #{file_path.sub(Rails.root.to_s + "/", "")}", :blue
       end
     end
 
@@ -441,7 +441,7 @@ module RubyLlmAgents
         next unless File.directory?(subdir) && Dir.empty?(subdir)
 
         if options[:dry_run]
-          say_status :dry_run, "Would remove empty #{subdir.sub(Rails.root.to_s + '/', '')}", :yellow
+          say_status :dry_run, "Would remove empty #{subdir.sub(Rails.root.to_s + "/", "")}", :yellow
         else
           FileUtils.rmdir(subdir)
           say_status :removed, subdir.sub(Rails.root.to_s + "/", ""), :red
@@ -452,7 +452,7 @@ module RubyLlmAgents
       return unless File.directory?(dir) && Dir.empty?(dir)
 
       if options[:dry_run]
-        say_status :dry_run, "Would remove empty #{dir.sub(Rails.root.to_s + '/', '')}", :yellow
+        say_status :dry_run, "Would remove empty #{dir.sub(Rails.root.to_s + "/", "")}", :yellow
       else
         FileUtils.rmdir(dir)
         say_status :removed, dir.sub(Rails.root.to_s + "/", ""), :red

data/lib/generators/ruby_llm_agents/multi_tenancy_generator.rb

@@ -102,7 +102,7 @@ module RubyLlmAgents
 
     def table_exists?(table)
       ActiveRecord::Base.connection.table_exists?(table)
-    rescue StandardError
+    rescue
       false
     end
 
@@ -110,7 +110,7 @@ module RubyLlmAgents
       return false unless ActiveRecord::Base.connection.table_exists?(table)
 
       ActiveRecord::Base.connection.column_exists?(table, column)
-    rescue StandardError
+    rescue
       false
     end
   end

data/lib/generators/ruby_llm_agents/restructure_generator.rb

@@ -37,25 +37,25 @@ module RubyLlmAgents
     # Maps old directory -> { category:, type: }
     DIRECTORY_MAPPING = {
       # Top-level under llm/
-      "agents" => { category: nil, type: "agents" },
-      "tools" => { category: nil, type: "tools" },
+      "agents" => {category: nil, type: "agents"},
+      "tools" => {category: nil, type: "tools"},
 
       # Audio group
-      "speakers" => { category: :audio, type: "speakers" },
-      "transcribers" => { category: :audio, type: "transcribers" },
+      "speakers" => {category: :audio, type: "speakers"},
+      "transcribers" => {category: :audio, type: "transcribers"},
 
       # Image group
-      "image_generators" => { category: :image, type: "generators" },
-      "image_editors" => { category: :image, type: "editors" },
-      "image_analyzers" => { category: :image, type: "analyzers" },
-      "image_transformers" => { category: :image, type: "transformers" },
-      "image_upscalers" => { category: :image, type: "upscalers" },
-      "image_variators" => { category: :image, type: "variators" },
-      "background_removers" => { category: :image, type: "background_removers" },
+      "image_generators" => {category: :image, type: "generators"},
+      "image_editors" => {category: :image, type: "editors"},
+      "image_analyzers" => {category: :image, type: "analyzers"},
+      "image_transformers" => {category: :image, type: "transformers"},
+      "image_upscalers" => {category: :image, type: "upscalers"},
+      "image_variators" => {category: :image, type: "variators"},
+      "background_removers" => {category: :image, type: "background_removers"},
 
       # Text group
-      "embedders" => { category: :text, type: "embedders" },
-      "moderators" => { category: :text, type: "moderators" }
+      "embedders" => {category: :text, type: "embedders"},
+      "moderators" => {category: :text, type: "moderators"}
     }.freeze
 
     def validate_root_directory

data/lib/generators/ruby_llm_agents/speaker_generator.rb

@@ -17,17 +17,17 @@ module RubyLlmAgents
     source_root File.expand_path("templates", __dir__)
 
     class_option :provider, type: :string, default: "openai",
-                 desc: "The TTS provider to use (openai, elevenlabs)"
+      desc: "The TTS provider to use (openai, elevenlabs)"
     class_option :model, type: :string, default: nil,
-                 desc: "The TTS model to use"
+      desc: "The TTS model to use"
     class_option :voice, type: :string, default: "nova",
-                 desc: "The voice to use"
+      desc: "The voice to use"
     class_option :speed, type: :numeric, default: 1.0,
-                 desc: "Speech speed (0.25-4.0 for OpenAI)"
+      desc: "Speech speed (0.25-4.0 for OpenAI)"
     class_option :format, type: :string, default: "mp3",
-                 desc: "Output format (mp3, wav, ogg, flac)"
+      desc: "Output format (mp3, wav, ogg, flac)"
     class_option :cache, type: :string, default: nil,
-                 desc: "Cache TTL (e.g., '7.days')"
+      desc: "Cache TTL (e.g., '7.days')"
 
     def ensure_base_class_and_skill_file
       audio_dir = "app/agents/audio"

data/lib/generators/ruby_llm_agents/templates/add_assistant_prompt_migration.rb.tt

@@ -0,0 +1,9 @@
+# frozen_string_literal: true
+
+class AddAssistantPromptToExecutionDetails < ActiveRecord::Migration<%= migration_version %>
+  def change
+    unless column_exists?(:ruby_llm_agents_execution_details, :assistant_prompt)
+      add_column :ruby_llm_agents_execution_details, :assistant_prompt, :text
+    end
+  end
+end

data/lib/generators/ruby_llm_agents/templates/split_execution_details_migration.rb.tt

@@ -9,7 +9,7 @@
 class SplitExecutionDetailsFromExecutions < ActiveRecord::Migration<%= migration_version %>
   # Columns that belong on execution_details, not executions
   DETAIL_COLUMNS = %i[
-    error_message system_prompt user_prompt response messages_summary
+    error_message system_prompt user_prompt assistant_prompt response messages_summary
     tool_calls attempts fallback_chain parameters routed_to
     classification_result cached_at cache_creation_tokens
   ].freeze
@@ -50,6 +50,7 @@ class SplitExecutionDetailsFromExecutions < ActiveRecord::Migration<%= migration
       t.text     :error_message
       t.text     :system_prompt
       t.text     :user_prompt
+      t.text     :assistant_prompt
       t.json     :response,             default: {}
       t.json     :messages_summary,     default: {}, null: false
       t.json     :tool_calls,           default: [], null: false

data/lib/generators/ruby_llm_agents/transcriber_generator.rb

@@ -17,13 +17,13 @@ module RubyLlmAgents
     source_root File.expand_path("templates", __dir__)
 
     class_option :model, type: :string, default: "whisper-1",
-                 desc: "The transcription model to use"
+      desc: "The transcription model to use"
     class_option :language, type: :string, default: nil,
-                 desc: "Language code (e.g., 'en', 'es')"
+      desc: "Language code (e.g., 'en', 'es')"
     class_option :output_format, type: :string, default: "text",
-                 desc: "Output format (text, srt, vtt, json)"
+      desc: "Output format (text, srt, vtt, json)"
     class_option :cache, type: :string, default: nil,
-                 desc: "Cache TTL (e.g., '30.days')"
+      desc: "Cache TTL (e.g., '30.days')"
 
     def ensure_base_class_and_skill_file
       audio_dir = "app/agents/audio"

data/lib/generators/ruby_llm_agents/upgrade_generator.rb

@@ -60,6 +60,25 @@ module RubyLlmAgents
       )
     end
 
+    # Add assistant_prompt column to execution_details (v3.0 -> v3.1 upgrade)
+    def create_add_assistant_prompt_migration
+      if column_exists?(:ruby_llm_agents_execution_details, :assistant_prompt)
+        say_status :skip, "assistant_prompt column already exists on execution_details", :yellow
+        return
+      end
+
+      unless table_exists?(:ruby_llm_agents_execution_details)
+        say_status :skip, "execution_details table does not exist yet", :yellow
+        return
+      end
+
+      say_status :upgrade, "Adding assistant_prompt to execution_details", :blue
+      migration_template(
+        "add_assistant_prompt_migration.rb.tt",
+        File.join(db_migrate_path, "add_assistant_prompt_to_execution_details.rb")
+      )
+    end
+
     def suggest_config_consolidation
       ruby_llm_initializer = File.join(destination_root, "config/initializers/ruby_llm.rb")
       agents_initializer = File.join(destination_root, "config/initializers/ruby_llm_agents.rb")
@@ -116,7 +135,7 @@ module RubyLlmAgents
 
     # Detail columns that should only exist on execution_details, not executions
     DETAIL_COLUMNS = %i[
-      error_message system_prompt user_prompt response messages_summary
+      error_message system_prompt user_prompt assistant_prompt response messages_summary
       tool_calls attempts fallback_chain parameters routed_to
       classification_result cached_at cache_creation_tokens
     ].freeze
@@ -145,13 +164,13 @@ module RubyLlmAgents
       return false unless ActiveRecord::Base.connection.table_exists?(table)
 
       ActiveRecord::Base.connection.column_exists?(table, column)
-    rescue StandardError
+    rescue
       false
     end
 
     def table_exists?(table)
       ActiveRecord::Base.connection.table_exists?(table)
-    rescue StandardError
+    rescue
       false
     end
   end

data/lib/ruby_llm/agents/audio/speaker.rb

@@ -2,6 +2,8 @@
 
 require "digest"
 require_relative "../results/speech_result"
+require_relative "speech_client"
+require_relative "speech_pricing"
 
 module RubyLLM
   module Agents
@@ -194,19 +196,19 @@ module RubyLLM
 
         def default_tts_provider
           RubyLLM::Agents.configuration.default_tts_provider
-        rescue StandardError
+        rescue
           :openai
         end
 
         def default_tts_model
           RubyLLM::Agents.configuration.default_tts_model
-        rescue StandardError
+        rescue
           "tts-1"
         end
 
         def default_tts_voice
           RubyLLM::Agents.configuration.default_tts_voice
-        rescue StandardError
+        rescue
           "nova"
         end
       end
@@ -410,7 +412,15 @@ module RubyLLM
 
       # Executes standard (non-streaming) speech synthesis
       def execute_standard_speech(text, options)
-        response = RubyLLM.speak(text, **options)
+        response = speech_client.speak(
+          text,
+          model: options[:model],
+          voice: options[:voice],
+          voice_id: resolved_voice_id,
+          speed: options[:speed],
+          response_format: options[:response_format] || "mp3",
+          voice_settings: options[:voice_settings]
+        )
 
         {
           audio: response.audio,
@@ -428,9 +438,17 @@ module RubyLLM
       def execute_streaming_speech(text, options)
         audio_chunks = []
 
-        RubyLLM.speak(text, **options.merge(stream: true)) do |chunk|
+        speech_client.speak_streaming(
+          text,
+          model: options[:model],
+          voice: options[:voice],
+          voice_id: resolved_voice_id,
+          speed: options[:speed],
+          response_format: options[:response_format] || "mp3",
+          voice_settings: options[:voice_settings]
+        ) do |chunk|
           audio_chunks << chunk.audio if chunk.respond_to?(:audio)
-          @streaming_block.call(chunk) if @streaming_block
+          @streaming_block&.call(chunk)
         end
 
         {
@@ -445,7 +463,7 @@ module RubyLLM
         }
       end
 
-      # Builds options for RubyLLM.speak
+      # Builds options for SpeechClient
       def build_speak_options
         options = {
           model: resolved_model,
@@ -453,13 +471,11 @@ module RubyLLM
         }
 
         speed = resolved_speed
-        options[:speed] = speed if speed && speed != 1.0
+        options[:speed] = speed if speed && (speed - 1.0).abs > Float::EPSILON
         options[:response_format] = resolved_output_format.to_s
 
-        if resolved_provider == :elevenlabs
-          voice_settings = self.class.voice_settings_config
-          options[:voice_settings] = voice_settings.to_h if voice_settings
-        end
+        voice_settings = self.class.voice_settings_config
+        options[:voice_settings] = voice_settings.to_h if voice_settings
 
         options
       end
@@ -488,29 +504,17 @@ module RubyLLM
 
       # Calculates cost for speech synthesis
       def calculate_cost(raw_result)
-        characters = raw_result[:characters] || 0
-
-        if raw_result[:raw_response].respond_to?(:cost) && raw_result[:raw_response].cost
+        if raw_result[:raw_response].respond_to?(:cost) && raw_result[:raw_response]&.cost
           return raw_result[:raw_response].cost
         end
 
-        provider = raw_result[:provider]
-        model_name = raw_result[:model].to_s
-
-        price_per_1k_chars = case provider
-                            when :openai
-                              model_name.include?("hd") ? 0.030 : 0.015
-                            when :elevenlabs
-                              0.30
-                            when :google
-                              0.016
-                            when :polly
-                              0.016
-                            else
-                              0.015
-                            end
+        characters = raw_result[:characters] || 0
 
-        (characters / 1000.0) * price_per_1k_chars
+        Audio::SpeechPricing.calculate_cost(
+          provider: raw_result[:provider],
+          model_id: raw_result[:model].to_s,
+          characters: characters
+        )
       end
 
       # Resolves the provider to use
@@ -547,6 +551,11 @@ module RubyLLM
       def streaming_enabled?
         @runtime_streaming || self.class.streaming?
       end
+
+      # Returns a SpeechClient for the resolved provider
+      def speech_client
+        @speech_client ||= Audio::SpeechClient.new(provider: resolved_provider)
+      end
     end
   end
 end