ruby_llm-agents 1.0.0 → 1.2.0

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

@@ -1,44 +1,42 @@
 # frozen_string_literal: true
 
-module <%= @root_namespace %>
-  module Image
+module Images
 <%- if class_name.include?("::") -%>
 <%- class_name.split("::")[0..-2].each_with_index do |mod, i| -%>
-<%= "  " * (i + 2) %>module <%= mod %>
+<%= "  " * (i + 1) %>module <%= mod %>
 <%- end -%>
-<%= "  " * (class_name.split("::").length + 1) %>class <%= class_name.split("::").last %>BackgroundRemover < ApplicationBackgroundRemover
+<%= "  " * class_name.split("::").length %>class <%= class_name.split("::").last %>BackgroundRemover < ApplicationBackgroundRemover
 <%- else -%>
-    class <%= class_name %>BackgroundRemover < ApplicationBackgroundRemover
+  class <%= class_name %>BackgroundRemover < ApplicationBackgroundRemover
 <%- end -%>
-      # Model configuration
-      model "<%= options[:model] %>"
-      output_format :<%= options[:output_format] %>
+    # Model configuration
+    model "<%= options[:model] %>"
+    output_format :<%= options[:output_format] %>
 <% if options[:refine_edges] -%>
-      refine_edges true
+    refine_edges true
 <% end -%>
 <% if options[:alpha_matting] -%>
-      alpha_matting true
+    alpha_matting true
 <% end -%>
 <% if options[:return_mask] -%>
-      return_mask true
+    return_mask true
 <% end -%>
 <% if options[:cache] -%>
 
-      # Caching
-      cache_for <%= options[:cache] %>
+    # Caching
+    cache_for <%= options[:cache] %>
 <% end -%>
 
-      # Optional: Edge threshold
-      # edge_threshold 0.5
+    # Optional: Edge threshold
+    # edge_threshold 0.5
 
-      # Optional: Description
-      # description "Removes backgrounds from <%= class_name.downcase %> images"
+    # Optional: Description
+    # description "Removes backgrounds from <%= class_name.downcase %> images"
 <%- if class_name.include?("::") -%>
-<%- (class_name.split("::").length + 1).times do |i| -%>
-<%= "  " * (class_name.split("::").length + 1 - i) %>end
+<%- class_name.split("::").length.times do |i| -%>
+<%= "  " * (class_name.split("::").length - i) %>end
 <%- end -%>
 <%- else -%>
-    end
-<%- end -%>
   end
+<%- end -%>
 end

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

@@ -36,10 +36,21 @@ class CreateRubyLLMAgentsTenantBudgets < ActiveRecord::Migration<%= migration_ve
       # Whether to inherit from global config for unset limits
       t.boolean :inherit_global_defaults, default: true
 
+      # Polymorphic association to tenant model (e.g., Organization, Account)
+      # Allows direct association with ActiveRecord models using llm_tenant DSL
+      # Uses string type for tenant_record_id to support both integer and UUID primary keys
+      t.string :tenant_record_type
+      t.string :tenant_record_id
+
       t.timestamps
     end
 
     # Ensure unique tenant IDs
     add_index :ruby_llm_agents_tenant_budgets, :tenant_id, unique: true
+
+    # Index for polymorphic tenant record lookup
+    add_index :ruby_llm_agents_tenant_budgets,
+              [:tenant_record_type, :tenant_record_id],
+              name: "index_tenant_budgets_on_tenant_record"
   end
 end

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

@@ -0,0 +1,72 @@
+# frozen_string_literal: true
+
+# Migration to create the tenants table for multi-tenancy support
+#
+# This table stores tenant configuration including:
+# - Budget limits (cost, tokens, executions)
+# - Enforcement mode (none, soft, hard)
+# - Polymorphic link to user's tenant model
+#
+# Run with: rails db:migrate
+class CreateRubyLLMAgentsTenants < ActiveRecord::Migration<%= migration_version %>
+  def change
+    create_table :ruby_llm_agents_tenants do |t|
+      # Identity
+      t.string :tenant_id, null: false
+      t.string :name
+
+      # Polymorphic association to user's tenant model (e.g., Organization, Account)
+      # Uses string type for tenant_record_id to support both integer and UUID primary keys
+      t.string :tenant_record_type
+      t.string :tenant_record_id
+
+      # Cost limits (in USD, 6 decimal precision)
+      t.decimal :daily_limit, precision: 12, scale: 6
+      t.decimal :monthly_limit, precision: 12, scale: 6
+
+      # Token limits
+      t.bigint :daily_token_limit
+      t.bigint :monthly_token_limit
+
+      # Execution/call limits
+      t.bigint :daily_execution_limit
+      t.bigint :monthly_execution_limit
+
+      # Per-agent limits (JSON hash)
+      # Format: { "AgentName" => limit_value }
+      t.json :per_agent_daily, null: false, default: {}
+      t.json :per_agent_monthly, null: false, default: {}
+
+      # Enforcement mode: "none", "soft", or "hard"
+      # - none: no enforcement, only tracking
+      # - soft: log warnings when limits exceeded
+      # - hard: block execution when limits exceeded
+      t.string :enforcement, default: "soft"
+
+      # Whether to inherit from global config for unset limits
+      t.boolean :inherit_global_defaults, default: true
+
+      # Status (for soft-delete/disable)
+      t.boolean :active, default: true
+
+      # Extensible metadata (JSON)
+      t.json :metadata, null: false, default: {}
+
+      t.timestamps
+    end
+
+    # Ensure unique tenant IDs
+    add_index :ruby_llm_agents_tenants, :tenant_id, unique: true
+
+    # Index for name lookups
+    add_index :ruby_llm_agents_tenants, :name
+
+    # Index for active/inactive filtering
+    add_index :ruby_llm_agents_tenants, :active
+
+    # Index for polymorphic tenant record lookup
+    add_index :ruby_llm_agents_tenants,
+              [:tenant_record_type, :tenant_record_id],
+              name: "index_tenants_on_tenant_record"
+  end
+end

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

@@ -1,41 +1,39 @@
 # frozen_string_literal: true
 
-module <%= @root_namespace %>
-  module Text
+module Embedders
 <%- if class_name.include?("::") -%>
 <%- class_name.split("::")[0..-2].each_with_index do |mod, i| -%>
-<%= "  " * (i + 2) %>module <%= mod %>
+<%= "  " * (i + 1) %>module <%= mod %>
 <%- end -%>
-<%= "  " * (class_name.split("::").length + 1) %>class <%= class_name.split("::").last %>Embedder < ApplicationEmbedder
+<%= "  " * class_name.split("::").length %>class <%= class_name.split("::").last %>Embedder < ApplicationEmbedder
 <%- else -%>
-    class <%= class_name %>Embedder < ApplicationEmbedder
+  class <%= class_name %>Embedder < ApplicationEmbedder
 <%- end -%>
-      # Model configuration
-      model "<%= options[:model] %>"
+    # Model configuration
+    model "<%= options[:model] %>"
 <% if options[:dimensions] -%>
-      dimensions <%= options[:dimensions] %>
+    dimensions <%= options[:dimensions] %>
 <% end -%>
 <% if options[:batch_size] != 100 -%>
-      batch_size <%= options[:batch_size] %>
+    batch_size <%= options[:batch_size] %>
 <% end -%>
 <% if options[:cache] -%>
 
-      # Caching
-      cache_for <%= options[:cache] %>
+    # Caching
+    cache_for <%= options[:cache] %>
 <% end -%>
 
-      # Optional: Custom preprocessing
-      # Override this method to normalize/clean text before embedding
-      #
-      # def preprocess(text)
-      #   text.strip.downcase.gsub(/\s+/, ' ')
-      # end
+    # Optional: Custom preprocessing
+    # Override this method to normalize/clean text before embedding
+    #
+    # def preprocess(text)
+    #   text.strip.downcase.gsub(/\s+/, ' ')
+    # end
 <%- if class_name.include?("::") -%>
-<%- (class_name.split("::").length + 1).times do |i| -%>
-<%= "  " * (class_name.split("::").length + 1 - i) %>end
+<%- class_name.split("::").length.times do |i| -%>
+<%= "  " * (class_name.split("::").length - i) %>end
 <%- end -%>
 <%- else -%>
-    end
-<%- end -%>
   end
+<%- end -%>
 end

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

@@ -1,45 +1,43 @@
 # frozen_string_literal: true
 
-module <%= @root_namespace %>
-  module Image
+module Images
 <%- if class_name.include?("::") -%>
 <%- class_name.split("::")[0..-2].each_with_index do |mod, i| -%>
-<%= "  " * (i + 2) %>module <%= mod %>
+<%= "  " * (i + 1) %>module <%= mod %>
 <%- end -%>
-<%= "  " * (class_name.split("::").length + 1) %>class <%= class_name.split("::").last %>Analyzer < ApplicationImageAnalyzer
+<%= "  " * class_name.split("::").length %>class <%= class_name.split("::").last %>Analyzer < ApplicationImageAnalyzer
 <%- else -%>
-    class <%= class_name %>Analyzer < ApplicationImageAnalyzer
+  class <%= class_name %>Analyzer < ApplicationImageAnalyzer
 <%- end -%>
-      # Model configuration
-      model "<%= options[:model] %>"
-      analysis_type :<%= options[:analysis_type] %>
+    # Model configuration
+    model "<%= options[:model] %>"
+    analysis_type :<%= options[:analysis_type] %>
 <% if options[:extract_colors] -%>
-      extract_colors true
+    extract_colors true
 <% end -%>
 <% if options[:detect_objects] -%>
-      detect_objects true
+    detect_objects true
 <% end -%>
 <% if options[:extract_text] -%>
-      extract_text true
+    extract_text true
 <% end -%>
-      max_tags <%= options[:max_tags] %>
+    max_tags <%= options[:max_tags] %>
 <% if options[:cache] -%>
 
-      # Caching
-      cache_for <%= options[:cache] %>
+    # Caching
+    cache_for <%= options[:cache] %>
 <% end -%>
 
-      # Optional: Custom analysis prompt
-      # prompt "Analyze this image and describe..."
+    # Optional: Custom analysis prompt
+    # prompt "Analyze this image and describe..."
 
-      # Optional: Description
-      # description "Analyzes <%= class_name.downcase %> images"
+    # Optional: Description
+    # description "Analyzes <%= class_name.downcase %> images"
 <%- if class_name.include?("::") -%>
-<%- (class_name.split("::").length + 1).times do |i| -%>
-<%= "  " * (class_name.split("::").length + 1 - i) %>end
+<%- class_name.split("::").length.times do |i| -%>
+<%= "  " * (class_name.split("::").length - i) %>end
 <%- end -%>
 <%- else -%>
-    end
-<%- end -%>
   end
+<%- end -%>
 end

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

@@ -1,35 +1,33 @@
 # frozen_string_literal: true
 
-module <%= @root_namespace %>
-  module Image
+module Images
 <%- if class_name.include?("::") -%>
 <%- class_name.split("::")[0..-2].each_with_index do |mod, i| -%>
-<%= "  " * (i + 2) %>module <%= mod %>
+<%= "  " * (i + 1) %>module <%= mod %>
 <%- end -%>
-<%= "  " * (class_name.split("::").length + 1) %>class <%= class_name.split("::").last %>Editor < ApplicationImageEditor
+<%= "  " * class_name.split("::").length %>class <%= class_name.split("::").last %>Editor < ApplicationImageEditor
 <%- else -%>
-    class <%= class_name %>Editor < ApplicationImageEditor
+  class <%= class_name %>Editor < ApplicationImageEditor
 <%- end -%>
-      # Model configuration
-      model "<%= options[:model] %>"
-      size "<%= options[:size] %>"
+    # Model configuration
+    model "<%= options[:model] %>"
+    size "<%= options[:size] %>"
 <% if options[:content_policy] != "standard" -%>
-      content_policy :<%= options[:content_policy] %>
+    content_policy :<%= options[:content_policy] %>
 <% end -%>
 <% if options[:cache] -%>
 
-      # Caching
-      cache_for <%= options[:cache] %>
+    # Caching
+    cache_for <%= options[:cache] %>
 <% end -%>
 
-      # Optional: Description
-      # description "Edits <%= class_name.downcase %> images"
+    # Optional: Description
+    # description "Edits <%= class_name.downcase %> images"
 <%- if class_name.include?("::") -%>
-<%- (class_name.split("::").length + 1).times do |i| -%>
-<%= "  " * (class_name.split("::").length + 1 - i) %>end
+<%- class_name.split("::").length.times do |i| -%>
+<%= "  " * (class_name.split("::").length - i) %>end
 <%- end -%>
 <%- else -%>
-    end
-<%- end -%>
   end
+<%- end -%>
 end

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

@@ -1,47 +1,45 @@
 # frozen_string_literal: true
 
-module <%= @root_namespace %>
-  module Image
+module Images
 <%- if class_name.include?("::") -%>
 <%- class_name.split("::")[0..-2].each_with_index do |mod, i| -%>
-<%= "  " * (i + 2) %>module <%= mod %>
+<%= "  " * (i + 1) %>module <%= mod %>
 <%- end -%>
-<%= "  " * (class_name.split("::").length + 1) %>class <%= class_name.split("::").last %>Generator < ApplicationImageGenerator
+<%= "  " * class_name.split("::").length %>class <%= class_name.split("::").last %>Generator < ApplicationImageGenerator
 <%- else -%>
-    class <%= class_name %>Generator < ApplicationImageGenerator
+  class <%= class_name %>Generator < ApplicationImageGenerator
 <%- end -%>
-      # Model configuration
-      model "<%= options[:model] %>"
-      size "<%= options[:size] %>"
-      quality "<%= options[:quality] %>"
-      style "<%= options[:style] %>"
+    # Model configuration
+    model "<%= options[:model] %>"
+    size "<%= options[:size] %>"
+    quality "<%= options[:quality] %>"
+    style "<%= options[:style] %>"
 <% if options[:content_policy] != "standard" -%>
-      content_policy :<%= options[:content_policy] %>
+    content_policy :<%= options[:content_policy] %>
 <% end -%>
 <% if options[:cache] -%>
 
-      # Caching
-      cache_for <%= options[:cache] %>
+    # Caching
+    cache_for <%= options[:cache] %>
 <% end -%>
 
-      # Optional: Add a prompt template
-      # template "Professional {prompt}, high quality, detailed"
+    # Optional: Add a prompt template
+    # template "Professional {prompt}, high quality, detailed"
 
-      # Optional: Add negative prompts (for models that support it)
-      # negative_prompt "blurry, low quality, distorted"
+    # Optional: Add negative prompts (for models that support it)
+    # negative_prompt "blurry, low quality, distorted"
 
-      # Optional: Custom preprocessing
-      # Override this method to modify prompts before generation
-      #
-      # def preprocess_prompt(prompt)
-      #   "#{prompt}, professional quality"
-      # end
+    # Optional: Custom preprocessing
+    # Override this method to modify prompts before generation
+    #
+    # def preprocess_prompt(prompt)
+    #   "#{prompt}, professional quality"
+    # end
 <%- if class_name.include?("::") -%>
-<%- (class_name.split("::").length + 1).times do |i| -%>
-<%= "  " * (class_name.split("::").length + 1 - i) %>end
+<%- class_name.split("::").length.times do |i| -%>
+<%= "  " * (class_name.split("::").length - i) %>end
 <%- end -%>
 <%- else -%>
-    end
-<%- end -%>
   end
+<%- end -%>
 end

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

@@ -1,50 +1,48 @@
 # frozen_string_literal: true
 
-module <%= @root_namespace %>
-  module Image
+module Images
 <%- if class_name.include?("::") -%>
 <%- class_name.split("::")[0..-2].each_with_index do |mod, i| -%>
-<%= "  " * (i + 2) %>module <%= mod %>
+<%= "  " * (i + 1) %>module <%= mod %>
 <%- end -%>
-<%= "  " * (class_name.split("::").length + 1) %>class <%= class_name.split("::").last %>Pipeline < ApplicationImagePipeline
+<%= "  " * class_name.split("::").length %>class <%= class_name.split("::").last %>Pipeline < ApplicationImagePipeline
 <%- else -%>
-    class <%= class_name %>Pipeline < ApplicationImagePipeline
+  class <%= class_name %>Pipeline < ApplicationImagePipeline
 <%- end -%>
-      # Pipeline steps
+    # Pipeline steps
 <%- parsed_steps.each do |step| -%>
 <%- case step -%>
 <%- when :generate -%>
-      step :<%= step %>, generator: <%= class_name %>Generator
+    step :<%= step %>, generator: <%= class_name %>Generator
 <%- when :upscale -%>
-      step :<%= step %>, upscaler: <%= class_name %>Upscaler
+    step :<%= step %>, upscaler: <%= class_name %>Upscaler
 <%- when :transform -%>
-      step :<%= step %>, transformer: <%= class_name %>Transformer
+    step :<%= step %>, transformer: <%= class_name %>Transformer
 <%- when :analyze -%>
-      step :<%= step %>, analyzer: <%= class_name %>Analyzer
+    step :<%= step %>, analyzer: <%= class_name %>Analyzer
 <%- when :remove_background -%>
-      step :<%= step %>, remover: <%= class_name %>BackgroundRemover
+    step :<%= step %>, remover: <%= class_name %>BackgroundRemover
 <%- else -%>
-      # step :<%= step %>, ...: <%= class_name %><%= step.to_s.camelize %>
+    # step :<%= step %>, ...: <%= class_name %><%= step.to_s.camelize %>
 <%- end -%>
 <%- end -%>
 <%- if options[:stop_on_error] == false -%>
 
-      stop_on_error false
+    stop_on_error false
 <%- end -%>
 <% if options[:cache] -%>
 
-      # Caching
-      cache_for <%= options[:cache] %>
+    # Caching
+    cache_for <%= options[:cache] %>
 <% end -%>
 
-      description "<%= class_name %> image processing pipeline"
-      version "1.0"
+    description "<%= class_name %> image processing pipeline"
+    version "1.0"
 <%- if class_name.include?("::") -%>
-<%- (class_name.split("::").length + 1).times do |i| -%>
-<%= "  " * (class_name.split("::").length + 1 - i) %>end
+<%- class_name.split("::").length.times do |i| -%>
+<%= "  " * (class_name.split("::").length - i) %>end
 <%- end -%>
 <%- else -%>
-    end
-<%- end -%>
   end
+<%- end -%>
 end

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

@@ -1,44 +1,42 @@
 # frozen_string_literal: true
 
-module <%= @root_namespace %>
-  module Image
+module Images
 <%- if class_name.include?("::") -%>
 <%- class_name.split("::")[0..-2].each_with_index do |mod, i| -%>
-<%= "  " * (i + 2) %>module <%= mod %>
+<%= "  " * (i + 1) %>module <%= mod %>
 <%- end -%>
-<%= "  " * (class_name.split("::").length + 1) %>class <%= class_name.split("::").last %>Transformer < ApplicationImageTransformer
+<%= "  " * class_name.split("::").length %>class <%= class_name.split("::").last %>Transformer < ApplicationImageTransformer
 <%- else -%>
-    class <%= class_name %>Transformer < ApplicationImageTransformer
+  class <%= class_name %>Transformer < ApplicationImageTransformer
 <%- end -%>
-      # Model configuration
-      model "<%= options[:model] %>"
-      size "<%= options[:size] %>"
-      strength <%= options[:strength] %>
+    # Model configuration
+    model "<%= options[:model] %>"
+    size "<%= options[:size] %>"
+    strength <%= options[:strength] %>
 <% if options[:content_policy] != "standard" -%>
-      content_policy :<%= options[:content_policy] %>
+    content_policy :<%= options[:content_policy] %>
 <% end -%>
 <% if options[:template] -%>
 
-      # Prompt template
-      template "<%= options[:template] %>"
+    # Prompt template
+    template "<%= options[:template] %>"
 <% end -%>
 <% if options[:cache] -%>
 
-      # Caching
-      cache_for <%= options[:cache] %>
+    # Caching
+    cache_for <%= options[:cache] %>
 <% end -%>
 
-      # Optional: Negative prompts
-      # negative_prompt "blurry, low quality, distorted"
+    # Optional: Negative prompts
+    # negative_prompt "blurry, low quality, distorted"
 
-      # Optional: Description
-      # description "Transforms images into <%= class_name.downcase %> style"
+    # Optional: Description
+    # description "Transforms images into <%= class_name.downcase %> style"
 <%- if class_name.include?("::") -%>
-<%- (class_name.split("::").length + 1).times do |i| -%>
-<%= "  " * (class_name.split("::").length + 1 - i) %>end
+<%- class_name.split("::").length.times do |i| -%>
+<%= "  " * (class_name.split("::").length - i) %>end
 <%- end -%>
 <%- else -%>
-    end
-<%- end -%>
   end
+<%- end -%>
 end

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

@@ -1,38 +1,36 @@
 # frozen_string_literal: true
 
-module <%= @root_namespace %>
-  module Image
+module Images
 <%- if class_name.include?("::") -%>
 <%- class_name.split("::")[0..-2].each_with_index do |mod, i| -%>
-<%= "  " * (i + 2) %>module <%= mod %>
+<%= "  " * (i + 1) %>module <%= mod %>
 <%- end -%>
-<%= "  " * (class_name.split("::").length + 1) %>class <%= class_name.split("::").last %>Upscaler < ApplicationImageUpscaler
+<%= "  " * class_name.split("::").length %>class <%= class_name.split("::").last %>Upscaler < ApplicationImageUpscaler
 <%- else -%>
-    class <%= class_name %>Upscaler < ApplicationImageUpscaler
+  class <%= class_name %>Upscaler < ApplicationImageUpscaler
 <%- end -%>
-      # Model configuration
-      model "<%= options[:model] %>"
-      scale <%= options[:scale] %>
+    # Model configuration
+    model "<%= options[:model] %>"
+    scale <%= options[:scale] %>
 <% if options[:face_enhance] -%>
-      face_enhance true
+    face_enhance true
 <% end -%>
 <% if options[:cache] -%>
 
-      # Caching
-      cache_for <%= options[:cache] %>
+    # Caching
+    cache_for <%= options[:cache] %>
 <% end -%>
 
-      # Optional: Denoise strength (0.0-1.0)
-      # denoise_strength 0.5
+    # Optional: Denoise strength (0.0-1.0)
+    # denoise_strength 0.5
 
-      # Optional: Description
-      # description "Upscales <%= class_name.downcase %> images"
+    # Optional: Description
+    # description "Upscales <%= class_name.downcase %> images"
 <%- if class_name.include?("::") -%>
-<%- (class_name.split("::").length + 1).times do |i| -%>
-<%= "  " * (class_name.split("::").length + 1 - i) %>end
+<%- class_name.split("::").length.times do |i| -%>
+<%= "  " * (class_name.split("::").length - i) %>end
 <%- end -%>
 <%- else -%>
-    end
-<%- end -%>
   end
+<%- end -%>
 end

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

@@ -1,33 +1,31 @@
 # frozen_string_literal: true
 
-module <%= @root_namespace %>
-  module Image
+module Images
 <%- if class_name.include?("::") -%>
 <%- class_name.split("::")[0..-2].each_with_index do |mod, i| -%>
-<%= "  " * (i + 2) %>module <%= mod %>
+<%= "  " * (i + 1) %>module <%= mod %>
 <%- end -%>
-<%= "  " * (class_name.split("::").length + 1) %>class <%= class_name.split("::").last %>Variator < ApplicationImageVariator
+<%= "  " * class_name.split("::").length %>class <%= class_name.split("::").last %>Variator < ApplicationImageVariator
 <%- else -%>
-    class <%= class_name %>Variator < ApplicationImageVariator
+  class <%= class_name %>Variator < ApplicationImageVariator
 <%- end -%>
-      # Model configuration
-      model "<%= options[:model] %>"
-      size "<%= options[:size] %>"
-      variation_strength <%= options[:variation_strength] %>
+    # Model configuration
+    model "<%= options[:model] %>"
+    size "<%= options[:size] %>"
+    variation_strength <%= options[:variation_strength] %>
 <% if options[:cache] -%>
 
-      # Caching
-      cache_for <%= options[:cache] %>
+    # Caching
+    cache_for <%= options[:cache] %>
 <% end -%>
 
-      # Optional: Description
-      # description "Creates variations of <%= class_name.downcase %> images"
+    # Optional: Description
+    # description "Creates variations of <%= class_name.downcase %> images"
 <%- if class_name.include?("::") -%>
-<%- (class_name.split("::").length + 1).times do |i| -%>
-<%= "  " * (class_name.split("::").length + 1 - i) %>end
+<%- class_name.split("::").length.times do |i| -%>
+<%= "  " * (class_name.split("::").length - i) %>end
 <%- end -%>
 <%- else -%>
-    end
-<%- end -%>
   end
+<%- end -%>
 end