ruby_llm-agents 0.5.0 → 1.0.0

data/lib/generators/ruby_llm_agents/transcriber_generator.rb

@@ -0,0 +1,107 @@
+# frozen_string_literal: true
+
+require "rails/generators"
+
+module RubyLlmAgents
+  # Transcriber generator for creating new transcribers
+  #
+  # Usage:
+  #   rails generate ruby_llm_agents:transcriber Meeting
+  #   rails generate ruby_llm_agents:transcriber Meeting --model gpt-4o-transcribe
+  #   rails generate ruby_llm_agents:transcriber Meeting --language es
+  #   rails generate ruby_llm_agents:transcriber Meeting --root=ai
+  #
+  # This will create:
+  #   - app/{root}/audio/transcribers/meeting_transcriber.rb
+  #
+  class TranscriberGenerator < ::Rails::Generators::NamedBase
+    source_root File.expand_path("templates", __dir__)
+
+    class_option :model, type: :string, default: "whisper-1",
+                 desc: "The transcription model to use"
+    class_option :language, type: :string, default: nil,
+                 desc: "Language code (e.g., 'en', 'es')"
+    class_option :output_format, type: :string, default: "text",
+                 desc: "Output format (text, srt, vtt, json)"
+    class_option :cache, type: :string, default: nil,
+                 desc: "Cache TTL (e.g., '30.days')"
+    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
+      @audio_namespace = "#{root_namespace}::Audio"
+      transcribers_dir = "app/#{root_directory}/audio/transcribers"
+
+      # Create directory if needed
+      empty_directory transcribers_dir
+
+      # Create base class if it doesn't exist
+      base_class_path = "#{transcribers_dir}/application_transcriber.rb"
+      unless File.exist?(File.join(destination_root, base_class_path))
+        template "application_transcriber.rb.tt", base_class_path
+      end
+
+      # Create skill file if it doesn't exist
+      skill_file_path = "#{transcribers_dir}/TRANSCRIBERS.md"
+      unless File.exist?(File.join(destination_root, skill_file_path))
+        template "skills/TRANSCRIBERS.md.tt", skill_file_path
+      end
+    end
+
+    def create_transcriber_file
+      # Support nested paths: "interview/meeting" -> "app/{root}/audio/transcribers/interview/meeting_transcriber.rb"
+      @root_namespace = root_namespace
+      @audio_namespace = "#{root_namespace}::Audio"
+      transcriber_path = name.underscore
+      template "transcriber.rb.tt", "app/#{root_directory}/audio/transcribers/#{transcriber_path}_transcriber.rb"
+    end
+
+    def show_usage
+      # Build full class name from path
+      transcriber_class_name = name.split("/").map(&:camelize).join("::")
+      full_class_name = "#{root_namespace}::Audio::#{transcriber_class_name}Transcriber"
+      say ""
+      say "Transcriber #{full_class_name} created!", :green
+      say ""
+      say "Usage:"
+      say "  # From file path"
+      say "  #{full_class_name}.call(audio: \"recording.mp3\")"
+      say ""
+      say "  # From URL"
+      say "  #{full_class_name}.call(audio: \"https://example.com/audio.mp3\")"
+      say ""
+      say "  # Get subtitles"
+      say "  result = #{full_class_name}.call(audio: \"video.mp4\")"
+      say "  result.srt  # SRT format"
+      say "  result.vtt  # VTT format"
+      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)
+      # Handle special cases for common abbreviations
+      return "AI" if str.downcase == "ai"
+      return "ML" if str.downcase == "ml"
+      return "LLM" if str.downcase == "llm"
+
+      # Standard camelization
+      str.split(/[-_]/).map(&:capitalize).join
+    end
+  end
+end

data/lib/generators/ruby_llm_agents/upgrade_generator.rb

@@ -2,6 +2,7 @@
 
 require "rails/generators"
 require "rails/generators/active_record"
+require "fileutils"
 
 module RubyLlmAgents
   # Upgrade generator for ruby_llm-agents
@@ -133,12 +134,62 @@ module RubyLlmAgents
       )
     end
 
+    def create_add_execution_type_migration
+      # Check if columns already exist
+      if column_exists?(:ruby_llm_agents_executions, :execution_type)
+        say_status :skip, "execution_type column already exists", :yellow
+        return
+      end
+
+      migration_template(
+        "add_execution_type_migration.rb.tt",
+        File.join(db_migrate_path, "add_execution_type_to_ruby_llm_agents_executions.rb")
+      )
+    end
+
+    def migrate_agents_directory
+      root_dir = RubyLLM::Agents.configuration.root_directory
+      namespace = RubyLLM::Agents.configuration.root_namespace
+      migrate_directory("agents", "#{root_dir}/agents", namespace)
+    end
+
+    def migrate_tools_directory
+      root_dir = RubyLLM::Agents.configuration.root_directory
+      namespace = RubyLLM::Agents.configuration.root_namespace
+      migrate_directory("tools", "#{root_dir}/tools", namespace)
+    end
+
     def show_post_upgrade_message
       say ""
-      say "RubyLLM::Agents upgrade migration created!", :green
+      say "RubyLLM::Agents upgrade complete!", :green
       say ""
       say "Next steps:"
       say "  1. Run migrations: rails db:migrate"
+      say "  2. Update class references in your controllers, views, and tests"
+      say "  3. Run your test suite to find any broken references"
+      say ""
+    end
+
+    def show_migration_summary
+      namespace = RubyLLM::Agents.configuration.root_namespace
+      root_dir = RubyLLM::Agents.configuration.root_directory
+
+      return unless @agents_migrated || @tools_migrated
+
+      say ""
+      say "=" * 60
+      say "  File Migration Summary", :green
+      say "=" * 60
+      say ""
+      say "Your agents and tools have been migrated to the new structure:"
+      say ""
+      say "  app/agents/  →  app/#{root_dir}/agents/" if @agents_migrated
+      say "  app/tools/   →  app/#{root_dir}/tools/" if @tools_migrated
+      say ""
+      say "Classes are now namespaced under #{namespace}::"
+      say ""
+      say "  Before: GeneralAgent.call(...)"
+      say "  After:  #{namespace}::GeneralAgent.call(...)"
       say ""
     end
 
@@ -159,5 +210,105 @@ module RubyLlmAgents
     rescue StandardError
       false
     end
+
+    def migrate_directory(old_dir, new_dir, namespace)
+      source = Rails.root.join("app", old_dir)
+      destination = Rails.root.join("app", new_dir)
+
+      # Skip if source doesn't exist
+      unless File.directory?(source)
+        say_status :skip, "app/#{old_dir}/ does not exist", :yellow
+        return
+      end
+
+      # Skip if source and destination are the same
+      if source.to_s == destination.to_s
+        say_status :skip, "app/#{old_dir}/ is already at destination", :yellow
+        return
+      end
+
+      files_to_migrate = Dir.glob("#{source}/**/*.rb")
+      if files_to_migrate.empty?
+        say_status :skip, "app/#{old_dir}/ has no Ruby files to migrate", :yellow
+        return
+      end
+
+      if options[:pretend]
+        say_status :preview, "Would move #{files_to_migrate.size} files from app/#{old_dir}/ → app/#{new_dir}/", :yellow
+        files_to_migrate.each do |file|
+          relative = file.sub("#{source}/", "")
+          say_status :would_move, relative, :cyan
+        end
+        return
+      end
+
+      # Create destination directory
+      FileUtils.mkdir_p(destination)
+
+      # Track conflicts and migrated files
+      conflicts = []
+      migrated = []
+
+      files_to_migrate.each do |file|
+        relative_path = file.sub("#{source}/", "")
+        dest_file = File.join(destination, relative_path)
+
+        # Skip if destination file already exists (conflict)
+        if File.exist?(dest_file)
+          conflicts << relative_path
+          say_status :conflict, "app/#{new_dir}/#{relative_path} already exists, skipping", :red
+          next
+        end
+
+        FileUtils.mkdir_p(File.dirname(dest_file))
+        FileUtils.mv(file, dest_file)
+
+        wrap_in_namespace(dest_file, namespace)
+        say_status :migrated, "app/#{new_dir}/#{relative_path}", :green
+        migrated << relative_path
+      end
+
+      # Cleanup empty source directory
+      cleanup_empty_directory(source, old_dir)
+
+      # Track that migration happened for summary
+      instance_variable_set("@#{old_dir}_migrated", migrated.any?)
+
+      { migrated: migrated, conflicts: conflicts }
+    end
+
+    def wrap_in_namespace(file, namespace)
+      content = File.read(file)
+
+      # Skip if already namespaced
+      return if content.include?("module #{namespace}")
+
+      # Wrap content in namespace with proper indentation
+      indented = content.lines.map { |line| line.empty? || line.strip.empty? ? line : "  #{line}" }.join
+      wrapped = "module #{namespace}\n#{indented}end\n"
+
+      File.write(file, wrapped)
+    end
+
+    def cleanup_empty_directory(dir, dir_name)
+      return unless File.directory?(dir)
+
+      # Remove empty subdirectories first (deepest first)
+      Dir.glob("#{dir}/**/", File::FNM_DOTMATCH).sort_by(&:length).reverse.each do |subdir|
+        next if subdir.end_with?(".", "..")
+
+        FileUtils.rmdir(subdir) if File.directory?(subdir) && Dir.empty?(subdir)
+      rescue Errno::ENOTEMPTY, Errno::ENOENT
+        # Directory not empty or already removed
+      end
+
+      # Remove main directory if empty
+      if File.directory?(dir) && Dir.empty?(dir)
+        FileUtils.rmdir(dir)
+        say_status :removed, "app/#{dir_name}/ (empty)", :yellow
+      end
+    rescue Errno::ENOTEMPTY
+      say_status :kept, "app/#{dir_name}/ (not empty)", :yellow
+    end
   end
 end