ruby_llm-agents 1.0.0 → 1.2.0

data/lib/generators/ruby_llm_agents/image_upscaler_generator.rb

@@ -9,10 +9,9 @@ module RubyLlmAgents
   #   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
+  #   - app/agents/images/photo_upscaler.rb
   #
   class ImageUpscalerGenerator < ::Rails::Generators::NamedBase
     source_root File.expand_path("templates", __dir__)
@@ -25,46 +24,34 @@ module RubyLlmAgents
                  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"
+      images_dir = "app/agents/images"
 
       # Create directory if needed
-      empty_directory upscalers_dir
+      empty_directory images_dir
 
       # Create base class if it doesn't exist
-      base_class_path = "#{upscalers_dir}/application_image_upscaler.rb"
+      base_class_path = "#{images_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"
+      skill_file_path = "#{images_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"
+      template "image_upscaler.rb.tt", "app/agents/images/#{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"
+      full_class_name = "Images::#{upscaler_class_name}Upscaler"
       say ""
       say "Image upscaler #{full_class_name} created!", :green
       say ""
@@ -82,22 +69,5 @@ module RubyLlmAgents
       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

@@ -9,10 +9,9 @@ module RubyLlmAgents
   #   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
+  #   - app/agents/images/logo_variator.rb
   #
   class ImageVariatorGenerator < ::Rails::Generators::NamedBase
     source_root File.expand_path("templates", __dir__)
@@ -25,46 +24,34 @@ module RubyLlmAgents
                  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"
+      images_dir = "app/agents/images"
 
       # Create directory if needed
-      empty_directory variators_dir
+      empty_directory images_dir
 
       # Create base class if it doesn't exist
-      base_class_path = "#{variators_dir}/application_image_variator.rb"
+      base_class_path = "#{images_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"
+      skill_file_path = "#{images_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"
+      template "image_variator.rb.tt", "app/agents/images/#{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"
+      full_class_name = "Images::#{variator_class_name}Variator"
       say ""
       say "Image variator #{full_class_name} created!", :green
       say ""
@@ -81,22 +68,5 @@ module RubyLlmAgents
       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,13 +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/{root}/agents/application_agent.rb base class
-  #   - Create app/{root}/text/embedders/application_embedder.rb base class
+  #   - Create app/agents/application_agent.rb base class
+  #   - Create app/agents/concerns/ directory
+  #   - Create app/workflows/application_workflow.rb base class
   #   - Optionally mount the dashboard engine in routes
   #
   class InstallGenerator < ::Rails::Generators::Base
@@ -28,14 +28,6 @@ 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]
@@ -53,43 +45,38 @@ module RubyLlmAgents
     end
 
     def create_directory_structure
-      say_status :create, "#{root_directory}/ directory structure", :green
+      say_status :create, "agents/ directory structure", :green
 
-      # Create agents directory
-      empty_directory "app/#{root_directory}/agents"
+      # Create agents directory and subdirectories
+      empty_directory "app/agents"
+      empty_directory "app/agents/concerns"
 
-      # Create text/embedders directory
-      empty_directory "app/#{root_directory}/text/embedders"
+      # Create workflows directory
+      empty_directory "app/workflows"
+
+      # Create tools directory
+      empty_directory "app/tools"
     end
 
     def create_application_agent
-      @root_namespace = root_namespace
-      template "application_agent.rb.tt", "app/#{root_directory}/agents/application_agent.rb"
+      template "application_agent.rb.tt", "app/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"
+    def create_application_workflow
+      template "application_workflow.rb.tt", "app/workflows/application_workflow.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"
+      template "skills/AGENTS.md.tt", "app/agents/AGENTS.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 skill file
+      template "skills/WORKFLOWS.md.tt", "app/workflows/WORKFLOWS.md"
 
-      # Create workflows directory and skill file
-      empty_directory "app/#{root_directory}/workflows"
-      template "skills/WORKFLOWS.md.tt", "app/#{root_directory}/workflows/WORKFLOWS.md"
+      # Create tools skill file
+      template "skills/TOOLS.md.tt", "app/tools/TOOLS.md"
     end
 
     def mount_dashboard_engine
@@ -112,28 +99,30 @@ module RubyLlmAgents
       say "RubyLLM::Agents has been installed!", :green
       say ""
       say "Directory structure created:"
-      say "  app/#{root_directory}/"
+      say "  app/"
       say "  ├── agents/"
       say "  │   ├── application_agent.rb"
+      say "  │   ├── concerns/"
       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 "  ├── workflows/"
+      say "  │   ├── application_workflow.rb"
+      say "  │   └── WORKFLOWS.md"
+      say "  └── tools/"
+      say "      └── TOOLS.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"
       say "  3. Access the dashboard at: /agents"
       say ""
+      say "Generator commands:"
+      say "  rails generate ruby_llm_agents:agent CustomerSupport query:required"
+      say "  rails generate ruby_llm_agents:image_generator Product"
+      say "  rails generate ruby_llm_agents:transcriber Meeting"
+      say "  rails generate ruby_llm_agents:embedder Semantic"
+      say ""
     end
 
     private
@@ -145,17 +134,5 @@ 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