ruby_llm-agents 0.4.0 → 1.0.0.beta.1

data/lib/generators/ruby_llm_agents/image_pipeline_generator.rb

@@ -0,0 +1,178 @@
+# frozen_string_literal: true
+
+require "rails/generators"
+
+module RubyLlmAgents
+  # ImagePipeline generator for creating new image pipelines
+  #
+  # Usage:
+  #   rails generate ruby_llm_agents:image_pipeline Product
+  #   rails generate ruby_llm_agents:image_pipeline Ecommerce --steps generate,upscale,remove_background
+  #   rails generate ruby_llm_agents:image_pipeline Content --steps generate,analyze
+  #   rails generate ruby_llm_agents:image_pipeline Product --root=ai
+  #
+  # This will create:
+  #   - app/{root}/image/pipelines/product_pipeline.rb
+  #
+  class ImagePipelineGenerator < ::Rails::Generators::NamedBase
+    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)"
+    class_option :stop_on_error, type: :boolean, default: true,
+                 desc: "Stop pipeline on first error"
+    class_option :cache, type: :string, default: nil,
+                 desc: "Cache TTL (e.g., '1.hour', '1.day')"
+    class_option :root,
+                 type: :string,
+                 default: nil,
+                 desc: "Root directory name (default: uses config or 'llm')"
+    class_option :namespace,
+                 type: :string,
+                 default: nil,
+                 desc: "Root namespace (default: camelized root or config)"
+
+    def ensure_base_class_and_skill_file
+      @root_namespace = root_namespace
+      @image_namespace = "#{root_namespace}::Image"
+      pipelines_dir = "app/#{root_directory}/image/pipelines"
+
+      # Create directory if needed
+      empty_directory pipelines_dir
+
+      # Create base class if it doesn't exist
+      base_class_path = "#{pipelines_dir}/application_image_pipeline.rb"
+      unless File.exist?(File.join(destination_root, base_class_path))
+        template "application_image_pipeline.rb.tt", base_class_path
+      end
+
+      # Create skill file if it doesn't exist
+      skill_file_path = "#{pipelines_dir}/IMAGE_PIPELINES.md"
+      unless File.exist?(File.join(destination_root, skill_file_path))
+        template "skills/IMAGE_PIPELINES.md.tt", skill_file_path
+      end
+    end
+
+    def create_image_pipeline_file
+      @root_namespace = root_namespace
+      @image_namespace = "#{root_namespace}::Image"
+      pipeline_path = name.underscore
+      template "image_pipeline.rb.tt", "app/#{root_directory}/image/pipelines/#{pipeline_path}_pipeline.rb"
+    end
+
+    def create_step_classes
+      # Create stub classes for referenced steps if they don't exist
+      parsed_steps.each do |step|
+        create_step_stub(step) if should_create_stub?(step)
+      end
+    end
+
+    def show_usage
+      pipeline_class_name = name.split("/").map(&:camelize).join("::")
+      full_class_name = "#{root_namespace}::Image::#{pipeline_class_name}Pipeline"
+      say ""
+      say "Image pipeline #{full_class_name} created!", :green
+      say ""
+      say "Usage:"
+      say "  # Run the pipeline"
+      say "  result = #{full_class_name}.call(prompt: 'Product photo')"
+      say "  result.final_image   # => The processed image"
+      say "  result.total_cost    # => Combined cost of all steps"
+      say "  result.step_count    # => Number of steps executed"
+      say ""
+      say "  # Access individual step results"
+      say "  result.step(:generate)  # => ImageGenerationResult"
+      say "  result.step(:upscale)   # => ImageUpscaleResult"
+      say "  result.analysis         # => ImageAnalysisResult (if analyzer step)"
+      say ""
+      say "  # Save the final image"
+      say "  result.save('output.png')"
+      say ""
+      say "  # With tenant tracking"
+      say "  result = #{full_class_name}.call("
+      say "    prompt: 'Product photo',"
+      say "    tenant: current_organization"
+      say "  )"
+      say ""
+    end
+
+    private
+
+    def root_directory
+      @root_directory ||= options[:root] || RubyLLM::Agents.configuration.root_directory
+    end
+
+    def root_namespace
+      @root_namespace ||= options[:namespace] || camelize(root_directory)
+    end
+
+    def camelize(str)
+      return "AI" if str.downcase == "ai"
+      return "ML" if str.downcase == "ml"
+      return "LLM" if str.downcase == "llm"
+      str.split(/[-_]/).map(&:capitalize).join
+    end
+
+    def parsed_steps
+      options[:steps].to_s.split(",").map(&:strip).map(&:to_sym)
+    end
+
+    def should_create_stub?(step)
+      case step
+      when :generate
+        !File.exist?("app/#{root_directory}/image/generators/#{name.underscore}_generator.rb")
+      when :upscale
+        !File.exist?("app/#{root_directory}/image/upscalers/#{name.underscore}_upscaler.rb")
+      when :transform
+        !File.exist?("app/#{root_directory}/image/transformers/#{name.underscore}_transformer.rb")
+      when :analyze
+        !File.exist?("app/#{root_directory}/image/analyzers/#{name.underscore}_analyzer.rb")
+      when :remove_background
+        !File.exist?("app/#{root_directory}/image/background_removers/#{name.underscore}_background_remover.rb")
+      else
+        false
+      end
+    end
+
+    def create_step_stub(step)
+      # Just show a note - don't auto-create stubs
+      case step
+      when :generate
+        say "  Note: You may want to create #{name}Generator with:", :yellow
+        say "    rails generate ruby_llm_agents:image_generator #{name}"
+      when :upscale
+        say "  Note: You may want to create #{name}Upscaler with:", :yellow
+        say "    rails generate ruby_llm_agents:image_upscaler #{name}"
+      when :transform
+        say "  Note: You may want to create #{name}Transformer with:", :yellow
+        say "    rails generate ruby_llm_agents:image_transformer #{name}"
+      when :analyze
+        say "  Note: You may want to create #{name}Analyzer with:", :yellow
+        say "    rails generate ruby_llm_agents:image_analyzer #{name}"
+      when :remove_background
+        say "  Note: You may want to create #{name}BackgroundRemover with:", :yellow
+        say "    rails generate ruby_llm_agents:background_remover #{name}"
+      end
+    end
+
+    def step_classes
+      @step_classes ||= parsed_steps.map do |step|
+        class_base = name.split("/").map(&:camelize).join("::")
+        case step
+        when :generate
+          { step: step, type: :generator, class_name: "#{@image_namespace}::#{class_base}Generator" }
+        when :upscale
+          { step: step, type: :upscaler, class_name: "#{@image_namespace}::#{class_base}Upscaler" }
+        when :transform
+          { step: step, type: :transformer, class_name: "#{@image_namespace}::#{class_base}Transformer" }
+        when :analyze
+          { step: step, type: :analyzer, class_name: "#{@image_namespace}::#{class_base}Analyzer" }
+        when :remove_background
+          { step: step, type: :remover, class_name: "#{@image_namespace}::#{class_base}BackgroundRemover" }
+        else
+          { step: step, type: step, class_name: "#{@image_namespace}::#{class_base}#{step.to_s.camelize}" }
+        end
+      end
+    end
+  end
+end

data/lib/generators/ruby_llm_agents/image_transformer_generator.rb

@@ -0,0 +1,109 @@
+# frozen_string_literal: true
+
+require "rails/generators"
+
+module RubyLlmAgents
+  # ImageTransformer generator for creating new image transformers
+  #
+  # Usage:
+  #   rails generate ruby_llm_agents:image_transformer Anime
+  #   rails generate ruby_llm_agents:image_transformer Watercolor --model sdxl --strength 0.8
+  #   rails generate ruby_llm_agents:image_transformer Oil --template "oil painting, {prompt}"
+  #   rails generate ruby_llm_agents:image_transformer Anime --root=ai
+  #
+  # This will create:
+  #   - app/{root}/image/transformers/anime_transformer.rb
+  #
+  class ImageTransformerGenerator < ::Rails::Generators::NamedBase
+    source_root File.expand_path("templates", __dir__)
+
+    class_option :model, type: :string, default: "sdxl",
+                 desc: "The image model to use"
+    class_option :size, type: :string, default: "1024x1024",
+                 desc: "Output image size (e.g., 1024x1024)"
+    class_option :strength, type: :string, default: "0.75",
+                 desc: "Transformation strength (0.0-1.0)"
+    class_option :template, type: :string, default: nil,
+                 desc: "Prompt template (use {prompt} as placeholder)"
+    class_option :content_policy, type: :string, default: "standard",
+                 desc: "Content policy level (none, standard, moderate, strict)"
+    class_option :cache, type: :string, default: nil,
+                 desc: "Cache TTL (e.g., '1.hour', '1.day')"
+    class_option :root,
+                 type: :string,
+                 default: nil,
+                 desc: "Root directory name (default: uses config or 'llm')"
+    class_option :namespace,
+                 type: :string,
+                 default: nil,
+                 desc: "Root namespace (default: camelized root or config)"
+
+    def ensure_base_class_and_skill_file
+      @root_namespace = root_namespace
+      @image_namespace = "#{root_namespace}::Image"
+      transformers_dir = "app/#{root_directory}/image/transformers"
+
+      # Create directory if needed
+      empty_directory transformers_dir
+
+      # Create base class if it doesn't exist
+      base_class_path = "#{transformers_dir}/application_image_transformer.rb"
+      unless File.exist?(File.join(destination_root, base_class_path))
+        template "application_image_transformer.rb.tt", base_class_path
+      end
+
+      # Create skill file if it doesn't exist
+      skill_file_path = "#{transformers_dir}/IMAGE_TRANSFORMERS.md"
+      unless File.exist?(File.join(destination_root, skill_file_path))
+        template "skills/IMAGE_TRANSFORMERS.md.tt", skill_file_path
+      end
+    end
+
+    def create_image_transformer_file
+      @root_namespace = root_namespace
+      @image_namespace = "#{root_namespace}::Image"
+      transformer_path = name.underscore
+      template "image_transformer.rb.tt", "app/#{root_directory}/image/transformers/#{transformer_path}_transformer.rb"
+    end
+
+    def show_usage
+      transformer_class_name = name.split("/").map(&:camelize).join("::")
+      full_class_name = "#{root_namespace}::Image::#{transformer_class_name}Transformer"
+      say ""
+      say "Image transformer #{full_class_name} created!", :green
+      say ""
+      say "Usage:"
+      say "  # Transform an image with a style"
+      say "  result = #{full_class_name}.call("
+      say "    image: 'photo.jpg',"
+      say "    prompt: 'portrait of a person'"
+      say "  )"
+      say "  result.url  # => 'https://...'"
+      say ""
+      say "  # Override strength at runtime"
+      say "  result = #{full_class_name}.call("
+      say "    image: 'photo.jpg',"
+      say "    prompt: 'detailed portrait',"
+      say "    strength: 0.9"
+      say "  )"
+      say ""
+    end
+
+    private
+
+    def root_directory
+      @root_directory ||= options[:root] || RubyLLM::Agents.configuration.root_directory
+    end
+
+    def root_namespace
+      @root_namespace ||= options[:namespace] || camelize(root_directory)
+    end
+
+    def camelize(str)
+      return "AI" if str.downcase == "ai"
+      return "ML" if str.downcase == "ml"
+      return "LLM" if str.downcase == "llm"
+      str.split(/[-_]/).map(&:capitalize).join
+    end
+  end
+end

data/lib/generators/ruby_llm_agents/image_upscaler_generator.rb

@@ -0,0 +1,103 @@
+# frozen_string_literal: true
+
+require "rails/generators"
+
+module RubyLlmAgents
+  # ImageUpscaler generator for creating new image upscalers
+  #
+  # Usage:
+  #   rails generate ruby_llm_agents:image_upscaler Photo
+  #   rails generate ruby_llm_agents:image_upscaler Portrait --model real-esrgan --scale 4
+  #   rails generate ruby_llm_agents:image_upscaler Face --face_enhance
+  #   rails generate ruby_llm_agents:image_upscaler Photo --root=ai
+  #
+  # This will create:
+  #   - app/{root}/image/upscalers/photo_upscaler.rb
+  #
+  class ImageUpscalerGenerator < ::Rails::Generators::NamedBase
+    source_root File.expand_path("templates", __dir__)
+
+    class_option :model, type: :string, default: "real-esrgan",
+                 desc: "The upscaling model to use"
+    class_option :scale, type: :string, default: "4",
+                 desc: "Upscale factor (2, 4, or 8)"
+    class_option :face_enhance, type: :boolean, default: false,
+                 desc: "Enable face enhancement"
+    class_option :cache, type: :string, default: nil,
+                 desc: "Cache TTL (e.g., '1.hour', '1.day')"
+    class_option :root,
+                 type: :string,
+                 default: nil,
+                 desc: "Root directory name (default: uses config or 'llm')"
+    class_option :namespace,
+                 type: :string,
+                 default: nil,
+                 desc: "Root namespace (default: camelized root or config)"
+
+    def ensure_base_class_and_skill_file
+      @root_namespace = root_namespace
+      @image_namespace = "#{root_namespace}::Image"
+      upscalers_dir = "app/#{root_directory}/image/upscalers"
+
+      # Create directory if needed
+      empty_directory upscalers_dir
+
+      # Create base class if it doesn't exist
+      base_class_path = "#{upscalers_dir}/application_image_upscaler.rb"
+      unless File.exist?(File.join(destination_root, base_class_path))
+        template "application_image_upscaler.rb.tt", base_class_path
+      end
+
+      # Create skill file if it doesn't exist
+      skill_file_path = "#{upscalers_dir}/IMAGE_UPSCALERS.md"
+      unless File.exist?(File.join(destination_root, skill_file_path))
+        template "skills/IMAGE_UPSCALERS.md.tt", skill_file_path
+      end
+    end
+
+    def create_image_upscaler_file
+      @root_namespace = root_namespace
+      @image_namespace = "#{root_namespace}::Image"
+      upscaler_path = name.underscore
+      template "image_upscaler.rb.tt", "app/#{root_directory}/image/upscalers/#{upscaler_path}_upscaler.rb"
+    end
+
+    def show_usage
+      upscaler_class_name = name.split("/").map(&:camelize).join("::")
+      full_class_name = "#{root_namespace}::Image::#{upscaler_class_name}Upscaler"
+      say ""
+      say "Image upscaler #{full_class_name} created!", :green
+      say ""
+      say "Usage:"
+      say "  # Upscale an image"
+      say "  result = #{full_class_name}.call(image: 'low_res.jpg')"
+      say "  result.url          # => 'https://...'"
+      say "  result.output_size  # => '4096x4096'"
+      say ""
+      say "  # Override settings at runtime"
+      say "  result = #{full_class_name}.call("
+      say "    image: 'photo.jpg',"
+      say "    scale: 8,"
+      say "    face_enhance: true"
+      say "  )"
+      say ""
+    end
+
+    private
+
+    def root_directory
+      @root_directory ||= options[:root] || RubyLLM::Agents.configuration.root_directory
+    end
+
+    def root_namespace
+      @root_namespace ||= options[:namespace] || camelize(root_directory)
+    end
+
+    def camelize(str)
+      return "AI" if str.downcase == "ai"
+      return "ML" if str.downcase == "ml"
+      return "LLM" if str.downcase == "llm"
+      str.split(/[-_]/).map(&:capitalize).join
+    end
+  end
+end

data/lib/generators/ruby_llm_agents/image_variator_generator.rb

@@ -0,0 +1,102 @@
+# frozen_string_literal: true
+
+require "rails/generators"
+
+module RubyLlmAgents
+  # ImageVariator generator for creating new image variators
+  #
+  # Usage:
+  #   rails generate ruby_llm_agents:image_variator Logo
+  #   rails generate ruby_llm_agents:image_variator Product --model gpt-image-1 --size 1024x1024
+  #   rails generate ruby_llm_agents:image_variator Avatar --variation_strength 0.3
+  #   rails generate ruby_llm_agents:image_variator Logo --root=ai
+  #
+  # This will create:
+  #   - app/{root}/image/variators/logo_variator.rb
+  #
+  class ImageVariatorGenerator < ::Rails::Generators::NamedBase
+    source_root File.expand_path("templates", __dir__)
+
+    class_option :model, type: :string, default: "gpt-image-1",
+                 desc: "The image model to use"
+    class_option :size, type: :string, default: "1024x1024",
+                 desc: "Output image size (e.g., 1024x1024)"
+    class_option :variation_strength, type: :string, default: "0.5",
+                 desc: "Variation strength (0.0-1.0)"
+    class_option :cache, type: :string, default: nil,
+                 desc: "Cache TTL (e.g., '1.hour', '1.day')"
+    class_option :root,
+                 type: :string,
+                 default: nil,
+                 desc: "Root directory name (default: uses config or 'llm')"
+    class_option :namespace,
+                 type: :string,
+                 default: nil,
+                 desc: "Root namespace (default: camelized root or config)"
+
+    def ensure_base_class_and_skill_file
+      @root_namespace = root_namespace
+      @image_namespace = "#{root_namespace}::Image"
+      variators_dir = "app/#{root_directory}/image/variators"
+
+      # Create directory if needed
+      empty_directory variators_dir
+
+      # Create base class if it doesn't exist
+      base_class_path = "#{variators_dir}/application_image_variator.rb"
+      unless File.exist?(File.join(destination_root, base_class_path))
+        template "application_image_variator.rb.tt", base_class_path
+      end
+
+      # Create skill file if it doesn't exist
+      skill_file_path = "#{variators_dir}/IMAGE_VARIATORS.md"
+      unless File.exist?(File.join(destination_root, skill_file_path))
+        template "skills/IMAGE_VARIATORS.md.tt", skill_file_path
+      end
+    end
+
+    def create_image_variator_file
+      @root_namespace = root_namespace
+      @image_namespace = "#{root_namespace}::Image"
+      variator_path = name.underscore
+      template "image_variator.rb.tt", "app/#{root_directory}/image/variators/#{variator_path}_variator.rb"
+    end
+
+    def show_usage
+      variator_class_name = name.split("/").map(&:camelize).join("::")
+      full_class_name = "#{root_namespace}::Image::#{variator_class_name}Variator"
+      say ""
+      say "Image variator #{full_class_name} created!", :green
+      say ""
+      say "Usage:"
+      say "  # Generate variations of an image"
+      say "  result = #{full_class_name}.call(image: 'logo.png', count: 4)"
+      say "  result.urls  # => ['https://...', ...]"
+      say ""
+      say "  # Override settings at runtime"
+      say "  result = #{full_class_name}.call("
+      say "    image: 'logo.png',"
+      say "    variation_strength: 0.7,"
+      say "    count: 3"
+      say "  )"
+      say ""
+    end
+
+    private
+
+    def root_directory
+      @root_directory ||= options[:root] || RubyLLM::Agents.configuration.root_directory
+    end
+
+    def root_namespace
+      @root_namespace ||= options[:namespace] || camelize(root_directory)
+    end
+
+    def camelize(str)
+      return "AI" if str.downcase == "ai"
+      return "ML" if str.downcase == "ml"
+      return "LLM" if str.downcase == "llm"
+      str.split(/[-_]/).map(&:capitalize).join
+    end
+  end
+end

data/lib/generators/ruby_llm_agents/install_generator.rb

@@ -8,11 +8,13 @@ module RubyLlmAgents
   #
   # Usage:
   #   rails generate ruby_llm_agents:install
+  #   rails generate ruby_llm_agents:install --root=ai
   #
   # This will:
   #   - Create the migration for ruby_llm_agents_executions table
   #   - Create the initializer at config/initializers/ruby_llm_agents.rb
-  #   - Create app/agents/application_agent.rb base class
+  #   - Create app/{root}/agents/application_agent.rb base class
+  #   - Create app/{root}/text/embedders/application_embedder.rb base class
   #   - Optionally mount the dashboard engine in routes
   #
   class InstallGenerator < ::Rails::Generators::Base
@@ -26,6 +28,14 @@ module RubyLlmAgents
                  desc: "Skip generating the initializer file"
     class_option :mount_dashboard, type: :boolean, default: true,
                  desc: "Mount the dashboard engine in routes"
+    class_option :root,
+                 type: :string,
+                 default: nil,
+                 desc: "Root directory name (default: uses config or 'llm')"
+    class_option :namespace,
+                 type: :string,
+                 default: nil,
+                 desc: "Root namespace (default: camelized root or config)"
 
     def create_migration_file
       return if options[:skip_migration]
@@ -42,12 +52,44 @@ module RubyLlmAgents
       template "initializer.rb.tt", "config/initializers/ruby_llm_agents.rb"
     end
 
-    def create_agents_directory
-      empty_directory "app/agents"
+    def create_directory_structure
+      say_status :create, "#{root_directory}/ directory structure", :green
+
+      # Create agents directory
+      empty_directory "app/#{root_directory}/agents"
+
+      # Create text/embedders directory
+      empty_directory "app/#{root_directory}/text/embedders"
     end
 
     def create_application_agent
-      template "application_agent.rb.tt", "app/agents/application_agent.rb"
+      @root_namespace = root_namespace
+      template "application_agent.rb.tt", "app/#{root_directory}/agents/application_agent.rb"
+    end
+
+    def create_application_embedder
+      @root_namespace = root_namespace
+      @text_namespace = "#{root_namespace}::Text"
+      template "application_embedder.rb.tt", "app/#{root_directory}/text/embedders/application_embedder.rb"
+    end
+
+    def create_skill_files
+      @root_namespace = root_namespace
+      say_status :create, "skill documentation files", :green
+
+      # Create agents skill file
+      template "skills/AGENTS.md.tt", "app/#{root_directory}/agents/AGENTS.md"
+
+      # Create embedders skill file
+      template "skills/EMBEDDERS.md.tt", "app/#{root_directory}/text/embedders/EMBEDDERS.md"
+
+      # Create tools directory and skill file
+      empty_directory "app/#{root_directory}/tools"
+      template "skills/TOOLS.md.tt", "app/#{root_directory}/tools/TOOLS.md"
+
+      # Create workflows directory and skill file
+      empty_directory "app/#{root_directory}/workflows"
+      template "skills/WORKFLOWS.md.tt", "app/#{root_directory}/workflows/WORKFLOWS.md"
     end
 
     def mount_dashboard_engine
@@ -69,6 +111,24 @@ module RubyLlmAgents
       say ""
       say "RubyLLM::Agents has been installed!", :green
       say ""
+      say "Directory structure created:"
+      say "  app/#{root_directory}/"
+      say "  ├── agents/"
+      say "  │   ├── application_agent.rb"
+      say "  │   └── AGENTS.md"
+      say "  ├── text/"
+      say "  │   └── embedders/"
+      say "  │       ├── application_embedder.rb"
+      say "  │       └── EMBEDDERS.md"
+      say "  ├── tools/"
+      say "  │   └── TOOLS.md"
+      say "  └── workflows/"
+      say "      └── WORKFLOWS.md"
+      say ""
+      say "Skill files (*.md) help AI coding assistants understand how to use this gem."
+      say ""
+      say "Namespace: #{root_namespace}::"
+      say ""
       say "Next steps:"
       say "  1. Run migrations: rails db:migrate"
       say "  2. Generate an agent: rails generate ruby_llm_agents:agent MyAgent query:required"
@@ -85,5 +145,17 @@ module RubyLlmAgents
     def db_migrate_path
       "db/migrate"
     end
+
+    def root_directory
+      @root_directory ||= options[:root] || RubyLLM::Agents.configuration.root_directory
+    end
+
+    def root_namespace
+      @root_namespace ||= options[:namespace] || camelize(root_directory)
+    end
+
+    def camelize(str)
+      str.split(/[-_]/).map(&:capitalize).join
+    end
   end
 end