ruby_llm-agents 1.0.0 → 1.2.0

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

@@ -1,12 +1,13 @@
 # frozen_string_literal: true
 
-# Migration to add tenant_id column to executions for multi-tenancy support
+# Migration to add tenant columns to executions for multi-tenancy support
 #
-# This migration adds a tenant_id column to track which tenant each execution
+# This migration adds tenant columns to track which tenant each execution
 # belongs to, enabling:
 # - Filtering executions by tenant
 # - Tenant-scoped analytics and reporting
 # - Per-tenant budget tracking and circuit breakers
+# - Polymorphic association with tenant models (e.g., Organization, Account)
 #
 # Run with: rails db:migrate
 class AddTenantIdToRubyLLMAgentsExecutions < ActiveRecord::Migration<%= migration_version %>
@@ -14,10 +15,20 @@ class AddTenantIdToRubyLLMAgentsExecutions < ActiveRecord::Migration<%= migratio
     # Add tenant_id column (nullable for backward compatibility)
     add_column :ruby_llm_agents_executions, :tenant_id, :string
 
+    # Polymorphic association to tenant model (for llm_tenant DSL)
+    # Uses string type for tenant_record_id to support both integer and UUID primary keys
+    add_column :ruby_llm_agents_executions, :tenant_record_type, :string
+    add_column :ruby_llm_agents_executions, :tenant_record_id, :string
+
     # Add indexes for efficient tenant-scoped queries
     add_index :ruby_llm_agents_executions, :tenant_id
     add_index :ruby_llm_agents_executions, [:tenant_id, :created_at]
     add_index :ruby_llm_agents_executions, [:tenant_id, :agent_type]
     add_index :ruby_llm_agents_executions, [:tenant_id, :status]
+
+    # Index for polymorphic tenant record lookup
+    add_index :ruby_llm_agents_executions,
+              [:tenant_record_type, :tenant_record_id],
+              name: "index_executions_on_tenant_record"
   end
 end

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

@@ -1,13 +1,12 @@
 # frozen_string_literal: true
 
-module <%= @root_namespace %>
 <%- if class_name.include?("::") -%>
 <%- class_name.split("::")[0..-2].each_with_index do |mod, i| -%>
-<%= "  " * (i + 1) %>module <%= mod %>
+<%= "  " * i %>module <%= mod %>
 <%- end -%>
-<%= "  " * class_name.split("::").length %>class <%= class_name.split("::").last %>Agent < ApplicationAgent
+<%= "  " * (class_name.split("::").length - 1) %>class <%= class_name.split("::").last %>Agent < ApplicationAgent
 <%- else -%>
-  class <%= class_name %>Agent < ApplicationAgent
+class <%= class_name %>Agent < ApplicationAgent
 <%- end -%>
     # ============================================
     # Model Configuration
@@ -114,10 +113,8 @@ module <%= @root_namespace %>
     #   { query: params[:query], locale: I18n.locale }
     # end
 <%- if class_name.include?("::") -%>
-<%- (class_name.split("::").length).times do |i| -%>
-<%= "  " * (class_name.split("::").length - i) %>end
+<%- (class_name.split("::").length - 1).times do |i| -%>
+<%= "  " * (class_name.split("::").length - 1 - i) %>end
 <%- end -%>
-<%- else -%>
-  end
 <%- end -%>
 end

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

@@ -6,7 +6,7 @@
 # that apply to all agents, or override them per-agent as needed.
 #
 # Example:
-#   class <%= @root_namespace %>::MyAgent < ApplicationAgent
+#   class MyAgent < ApplicationAgent
 #     param :query, required: true
 #
 #     def user_prompt
@@ -15,57 +15,55 @@
 #   end
 #
 # Usage:
-#   <%= @root_namespace %>::MyAgent.call(query: "hello")
-#   <%= @root_namespace %>::MyAgent.call(query: "hello", dry_run: true)    # Debug mode
-#   <%= @root_namespace %>::MyAgent.call(query: "hello", skip_cache: true) # Bypass cache
+#   MyAgent.call(query: "hello")
+#   MyAgent.call(query: "hello", dry_run: true)    # Debug mode
+#   MyAgent.call(query: "hello", skip_cache: true) # Bypass cache
 #
-module <%= @root_namespace %>
-  class ApplicationAgent < RubyLLM::Agents::Base
-    # ============================================
-    # Shared Model Configuration
-    # ============================================
-    # These settings are inherited by all agents
+class ApplicationAgent < RubyLLM::Agents::Base
+  # ============================================
+  # Shared Model Configuration
+  # ============================================
+  # These settings are inherited by all agents
 
-    # model "gemini-2.0-flash"   # Default model for all agents
-    # temperature 0.0            # Default temperature (0.0 = deterministic)
-    # timeout 60                 # Default timeout in seconds
+  # model "gemini-2.0-flash"   # Default model for all agents
+  # temperature 0.0            # Default temperature (0.0 = deterministic)
+  # timeout 60                 # Default timeout in seconds
 
-    # ============================================
-    # Shared Caching
-    # ============================================
+  # ============================================
+  # Shared Caching
+  # ============================================
 
-    # cache 1.hour  # Enable caching for all agents (override per-agent if needed)
+  # cache 1.hour  # Enable caching for all agents (override per-agent if needed)
 
-    # ============================================
-    # Shared Reliability Settings
-    # ============================================
-    # Configure once here, all agents inherit these settings
+  # ============================================
+  # Shared Reliability Settings
+  # ============================================
+  # Configure once here, all agents inherit these settings
 
-    # Automatic retries for all agents
-    # retries max: 2, backoff: :exponential, base: 0.4, max_delay: 3.0
+  # Automatic retries for all agents
+  # retries max: 2, backoff: :exponential, base: 0.4, max_delay: 3.0
 
-    # Shared fallback models
-    # fallback_models ["gpt-4o-mini", "claude-3-haiku"]
+  # Shared fallback models
+  # fallback_models ["gpt-4o-mini", "claude-3-haiku"]
 
-    # Total timeout across retries/fallbacks
-    # total_timeout 30
+  # Total timeout across retries/fallbacks
+  # total_timeout 30
 
-    # Circuit breaker (per agent-model pair)
-    # circuit_breaker errors: 5, within: 60, cooldown: 300
+  # Circuit breaker (per agent-model pair)
+  # circuit_breaker errors: 5, within: 60, cooldown: 300
 
-    # ============================================
-    # Shared Helper Methods
-    # ============================================
-    # Define methods here that can be used by all agents
+  # ============================================
+  # Shared Helper Methods
+  # ============================================
+  # Define methods here that can be used by all agents
 
-    # Example: Common system prompt prefix
-    # def system_prompt_prefix
-    #   "You are an AI assistant for #{Rails.application.class.module_parent_name}."
-    # end
+  # Example: Common system prompt prefix
+  # def system_prompt_prefix
+  #   "You are an AI assistant for #{Rails.application.class.module_parent_name}."
+  # end
 
-    # Example: Common metadata
-    # def execution_metadata
-    #   { app_version: Rails.application.config.version }
-    # end
-  end
+  # Example: Common metadata
+  # def execution_metadata
+  #   { app_version: Rails.application.config.version }
+  # end
 end

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

@@ -1,26 +1,24 @@
 # frozen_string_literal: true
 
-module <%= @root_namespace %>
-  module Image
-    # Base class for all background removers in the application
-    #
-    # Inherits from RubyLLM::Agents::BackgroundRemover and provides a common
-    # foundation for custom background removal classes.
-    #
-    # @example Creating a custom remover
-    #   class ProductBackgroundRemover < ApplicationBackgroundRemover
-    #     model "rembg"
-    #     output_format :png
-    #     refine_edges true
-    #     alpha_matting true
-    #   end
-    #
-    class ApplicationBackgroundRemover < RubyLLM::Agents::BackgroundRemover
-      # Default configuration (can be overridden in subclasses)
-      # model "rembg"
-      # output_format :png
-      # refine_edges false
-      # alpha_matting false
-    end
+module Images
+  # Base class for all background removers in the application
+  #
+  # Inherits from RubyLLM::Agents::BackgroundRemover and provides a common
+  # foundation for custom background removal classes.
+  #
+  # @example Creating a custom remover
+  #   class ProductBackgroundRemover < ApplicationBackgroundRemover
+  #     model "rembg"
+  #     output_format :png
+  #     refine_edges true
+  #     alpha_matting true
+  #   end
+  #
+  class ApplicationBackgroundRemover < RubyLLM::Agents::BackgroundRemover
+    # Default configuration (can be overridden in subclasses)
+    # model "rembg"
+    # output_format :png
+    # refine_edges false
+    # alpha_matting false
   end
 end

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

@@ -6,7 +6,7 @@
 # that apply to all embedders, or override them per-embedder as needed.
 #
 # Example:
-#   class <%= @text_namespace %>::DocumentEmbedder < ApplicationEmbedder
+#   class Embedders::DocumentEmbedder < ApplicationEmbedder
 #     dimensions 512  # Reduce dimensions for storage efficiency
 #
 #     def preprocess(text)
@@ -15,36 +15,34 @@
 #   end
 #
 # Usage:
-#   <%= @text_namespace %>::DocumentEmbedder.call(text: "hello world")
-#   <%= @text_namespace %>::DocumentEmbedder.call(texts: ["hello", "world"])  # Batch
+#   Embedders::DocumentEmbedder.call(text: "hello world")
+#   Embedders::DocumentEmbedder.call(texts: ["hello", "world"])  # Batch
 #
-module <%= @root_namespace %>
-  module Text
-    class ApplicationEmbedder < RubyLLM::Agents::Embedder
-      # ============================================
-      # Shared Model Configuration
-      # ============================================
-      # These settings are inherited by all embedders
+module Embedders
+  class ApplicationEmbedder < RubyLLM::Agents::Embedder
+    # ============================================
+    # Shared Model Configuration
+    # ============================================
+    # These settings are inherited by all embedders
 
-      # model "text-embedding-3-small"  # Default embedding model
-      # dimensions 1536                  # Vector dimensions (nil = model default)
-      # batch_size 100                   # Texts per API call for batch processing
+    # model "text-embedding-3-small"  # Default embedding model
+    # dimensions 1536                  # Vector dimensions (nil = model default)
+    # batch_size 100                   # Texts per API call for batch processing
 
-      # ============================================
-      # Shared Caching
-      # ============================================
-      # Same text always produces same embedding, so caching is very effective
+    # ============================================
+    # Shared Caching
+    # ============================================
+    # Same text always produces same embedding, so caching is very effective
 
-      # cache_for 1.week  # Cache embeddings for a week
+    # cache_for 1.week  # Cache embeddings for a week
 
-      # ============================================
-      # Shared Preprocessing
-      # ============================================
-      # Override in subclass for custom preprocessing
+    # ============================================
+    # Shared Preprocessing
+    # ============================================
+    # Override in subclass for custom preprocessing
 
-      # def preprocess(text)
-      #   text.strip.gsub(/\s+/, ' ').truncate(8000)
-      # end
-    end
+    # def preprocess(text)
+    #   text.strip.gsub(/\s+/, ' ').truncate(8000)
+    # end
   end
 end

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

@@ -1,26 +1,24 @@
 # frozen_string_literal: true
 
-module <%= @root_namespace %>
-  module Image
-    # Base class for all image analyzers in the application
-    #
-    # Inherits from RubyLLM::Agents::ImageAnalyzer and provides a common
-    # foundation for custom image analysis classes.
-    #
-    # @example Creating a custom analyzer
-    #   class ProductAnalyzer < ApplicationImageAnalyzer
-    #     model "gpt-4o"
-    #     analysis_type :detailed
-    #     extract_colors true
-    #     detect_objects true
-    #   end
-    #
-    class ApplicationImageAnalyzer < RubyLLM::Agents::ImageAnalyzer
-      # Default configuration (can be overridden in subclasses)
-      # model "gpt-4o"
-      # analysis_type :detailed
-      # extract_colors false
-      # detect_objects false
-    end
+module Images
+  # Base class for all image analyzers in the application
+  #
+  # Inherits from RubyLLM::Agents::ImageAnalyzer and provides a common
+  # foundation for custom image analysis classes.
+  #
+  # @example Creating a custom analyzer
+  #   class ProductAnalyzer < ApplicationImageAnalyzer
+  #     model "gpt-4o"
+  #     analysis_type :detailed
+  #     extract_colors true
+  #     detect_objects true
+  #   end
+  #
+  class ApplicationImageAnalyzer < RubyLLM::Agents::ImageAnalyzer
+    # Default configuration (can be overridden in subclasses)
+    # model "gpt-4o"
+    # analysis_type :detailed
+    # extract_colors false
+    # detect_objects false
   end
 end

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

@@ -1,20 +1,22 @@
 # frozen_string_literal: true
 
-# Base class for all image editors in the application
-#
-# Inherits from RubyLLM::Agents::ImageEditor and provides a common
-# foundation for custom image editing classes.
-#
-# @example Creating a custom editor
-#   class BackgroundEditor < ApplicationImageEditor
-#     model "gpt-image-1"
-#     size "1024x1024"
-#     content_policy :strict
-#   end
-#
-class ApplicationImageEditor < RubyLLM::Agents::ImageEditor
-  # Default configuration (can be overridden in subclasses)
-  # model "gpt-image-1"
-  # size "1024x1024"
-  # content_policy :standard
+module Images
+  # Base class for all image editors in the application
+  #
+  # Inherits from RubyLLM::Agents::ImageEditor and provides a common
+  # foundation for custom image editing classes.
+  #
+  # @example Creating a custom editor
+  #   class BackgroundEditor < ApplicationImageEditor
+  #     model "gpt-image-1"
+  #     size "1024x1024"
+  #     content_policy :strict
+  #   end
+  #
+  class ApplicationImageEditor < RubyLLM::Agents::ImageEditor
+    # Default configuration (can be overridden in subclasses)
+    # model "gpt-image-1"
+    # size "1024x1024"
+    # content_policy :standard
+  end
 end

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

@@ -1,38 +1,36 @@
 # frozen_string_literal: true
 
-module <%= @root_namespace %>
-  module Image
-    # Base class for all image generators in the application
-    #
-    # Inherits from RubyLLM::Agents::ImageGenerator and provides a common
-    # foundation for custom image generation classes.
-    #
-    # @example Creating a custom generator
-    #   class LogoGenerator < ApplicationImageGenerator
-    #     model "gpt-image-1"
-    #     size "1024x1024"
-    #     quality "hd"
-    #     style "vivid"
-    #     content_policy :strict
-    #
-    #     template "Professional logo design for {prompt}, minimalist, vector style"
-    #   end
-    #
-    # Usage:
-    #   <%= @root_namespace %>::Image::LogoGenerator.call(prompt: "Company logo")
-    #
-    class ApplicationImageGenerator < RubyLLM::Agents::ImageGenerator
-      # Include ActiveStorage support if available
-      if defined?(ActiveStorage)
-        include RubyLLM::Agents::ImageGenerator::ActiveStorageSupport
-      end
-
-      # Default configuration (can be overridden in subclasses)
-      # model "gpt-image-1"
-      # size "1024x1024"
-      # quality "standard"
-      # style "vivid"
-      # content_policy :standard
+module Images
+  # Base class for all image generators in the application
+  #
+  # Inherits from RubyLLM::Agents::ImageGenerator and provides a common
+  # foundation for custom image generation classes.
+  #
+  # @example Creating a custom generator
+  #   class LogoGenerator < ApplicationImageGenerator
+  #     model "gpt-image-1"
+  #     size "1024x1024"
+  #     quality "hd"
+  #     style "vivid"
+  #     content_policy :strict
+  #
+  #     template "Professional logo design for {prompt}, minimalist, vector style"
+  #   end
+  #
+  # Usage:
+  #   Images::LogoGenerator.call(prompt: "Company logo")
+  #
+  class ApplicationImageGenerator < RubyLLM::Agents::ImageGenerator
+    # Include ActiveStorage support if available
+    if defined?(ActiveStorage)
+      include RubyLLM::Agents::ImageGenerator::ActiveStorageSupport
     end
+
+    # Default configuration (can be overridden in subclasses)
+    # model "gpt-image-1"
+    # size "1024x1024"
+    # quality "standard"
+    # style "vivid"
+    # content_policy :standard
   end
 end