console_agent 0.2.0 → 0.3.0

data/lib/console_agent/tools/registry.rb

@@ -6,9 +6,10 @@ module ConsoleAgent
       attr_reader :definitions
 
       # Tools that should never be cached (side effects or user interaction)
-      NO_CACHE = %w[ask_user save_memory delete_memory].freeze
+      NO_CACHE = %w[ask_user save_memory delete_memory execute_plan].freeze
 
-      def initialize
+      def initialize(executor: nil)
+        @executor = executor
         @definitions = []
         @handlers = {}
         @cache = {}
@@ -180,6 +181,7 @@ module ConsoleAgent
         )
 
         register_memory_tools
+        register_execute_plan
       end
 
       def register_memory_tools
@@ -232,6 +234,158 @@ module ConsoleAgent
         )
       end
 
+      def register_execute_plan
+        return unless @executor
+
+        register(
+          name: 'execute_plan',
+          description: 'Execute a multi-step plan. Each step has a description and Ruby code. The plan is shown to the user for approval, then each step is executed in order. After each step executes, its return value is stored as step1, step2, etc. Use these variables in later steps to reference earlier results (e.g. `token = step1`).',
+          parameters: {
+            'type' => 'object',
+            'properties' => {
+              'steps' => {
+                'type' => 'array',
+                'description' => 'Ordered list of steps to execute',
+                'items' => {
+                  'type' => 'object',
+                  'properties' => {
+                    'description' => { 'type' => 'string', 'description' => 'What this step does' },
+                    'code' => { 'type' => 'string', 'description' => 'Ruby code to execute' }
+                  },
+                  'required' => %w[description code]
+                }
+              }
+            },
+            'required' => ['steps']
+          },
+          handler: ->(args) { execute_plan(args['steps'] || []) }
+        )
+      end
+
+      def execute_plan(steps)
+        return 'No steps provided.' if steps.nil? || steps.empty?
+
+        auto = ConsoleAgent.configuration.auto_execute
+
+        # Display full plan
+        $stdout.puts
+        $stdout.puts "\e[36m  Plan (#{steps.length} steps):\e[0m"
+        steps.each_with_index do |step, i|
+          $stdout.puts "\e[36m  #{i + 1}. #{step['description']}\e[0m"
+          $stdout.puts highlight_plan_code(step['code'])
+        end
+        $stdout.puts
+
+        # Ask for plan approval (unless auto-execute)
+        skip_confirmations = auto
+        unless auto
+          $stdout.print "\e[33m  Accept plan? [y/N/a(uto)] \e[0m"
+          answer = $stdin.gets.to_s.strip.downcase
+          case answer
+          when 'a', 'auto'
+            skip_confirmations = true
+          when 'y', 'yes'
+            # proceed with per-step confirmation
+          else
+            $stdout.puts "\e[33m  Plan declined.\e[0m"
+            feedback = ask_feedback("What would you like changed?")
+            return "User declined the plan. Feedback: #{feedback}"
+          end
+        end
+
+        # Execute steps one by one
+        results = []
+        steps.each_with_index do |step, i|
+          $stdout.puts
+          $stdout.puts "\e[36m  Step #{i + 1}/#{steps.length}: #{step['description']}\e[0m"
+          $stdout.puts "\e[33m  # Code:\e[0m"
+          $stdout.puts highlight_plan_code(step['code'])
+
+          # Per-step confirmation (unless auto-execute or plan-level auto)
+          unless skip_confirmations
+            $stdout.print "\e[33m  Run? [y/N/edit] \e[0m"
+            step_answer = $stdin.gets.to_s.strip.downcase
+
+            case step_answer
+            when 'e', 'edit'
+              edited = edit_step_code(step['code'])
+              if edited && edited != step['code']
+                $stdout.puts "\e[33m  # Edited code:\e[0m"
+                $stdout.puts highlight_plan_code(edited)
+                $stdout.print "\e[33m  Run edited code? [y/N] \e[0m"
+                confirm = $stdin.gets.to_s.strip.downcase
+                unless confirm == 'y' || confirm == 'yes'
+                  feedback = ask_feedback("What would you like changed?")
+                  results << "Step #{i + 1}: User declined after edit. Feedback: #{feedback}"
+                  break
+                end
+                step['code'] = edited
+              end
+            when 'y', 'yes'
+              # proceed
+            else
+              feedback = ask_feedback("What would you like changed?")
+              results << "Step #{i + 1}: User declined. Feedback: #{feedback}"
+              break
+            end
+          end
+
+          exec_result = @executor.execute(step['code'])
+          # Make result available as step1, step2, etc. for subsequent steps
+          @executor.binding_context.local_variable_set(:"step#{i + 1}", exec_result)
+          output = @executor.last_output
+
+          step_report = "Step #{i + 1} (#{step['description']}):\n"
+          if output && !output.strip.empty?
+            step_report += "Output: #{output.strip}\n"
+          end
+          step_report += "Return value: #{exec_result.inspect}"
+          results << step_report
+        end
+
+        results.join("\n\n")
+      end
+
+      def highlight_plan_code(code)
+        if coderay_available?
+          CodeRay.scan(code, :ruby).terminal.gsub(/^/, '     ')
+        else
+          code.split("\n").map { |l| "     \e[37m#{l}\e[0m" }.join("\n")
+        end
+      end
+
+      def edit_step_code(code)
+        require 'tempfile'
+        editor = ENV['EDITOR'] || 'vi'
+        tmpfile = Tempfile.new(['console_agent_step', '.rb'])
+        tmpfile.write(code)
+        tmpfile.flush
+        system("#{editor} #{tmpfile.path}")
+        File.read(tmpfile.path).strip
+      rescue => e
+        $stderr.puts "\e[31m  Editor error: #{e.message}\e[0m"
+        code
+      ensure
+        tmpfile.close! if tmpfile
+      end
+
+      def coderay_available?
+        return @coderay_available unless @coderay_available.nil?
+        @coderay_available = begin
+          require 'coderay'
+          true
+        rescue LoadError
+          false
+        end
+      end
+
+      def ask_feedback(prompt)
+        $stdout.print "\e[36m  #{prompt} > \e[0m"
+        feedback = $stdin.gets
+        return '(no feedback provided)' if feedback.nil?
+        feedback.strip.empty? ? '(no feedback provided)' : feedback.strip
+      end
+
       def ask_user(question)
         $stdout.puts "\e[36m  ? #{question}\e[0m"
         $stdout.print "\e[36m  > \e[0m"

data/lib/console_agent/version.rb

@@ -1,3 +1,3 @@
 module ConsoleAgent
-  VERSION = '0.2.0'.freeze
+  VERSION = '0.3.0'.freeze
 end

data/lib/console_agent.rb

@@ -45,6 +45,14 @@ module ConsoleAgent
       @logger = log
     end
 
+    def current_user
+      @current_user
+    end
+
+    def current_user=(name)
+      @current_user = name
+    end
+
     def status
       c = configuration
       key = c.resolved_api_key
@@ -59,18 +67,132 @@ module ConsoleAgent
       lines << "  Provider:       #{c.provider}"
       lines << "  Model:          #{c.resolved_model}"
       lines << "  API key:        #{masked_key}"
-      lines << "  Context mode:   #{c.context_mode}"
       lines << "  Max tokens:     #{c.max_tokens}"
       lines << "  Temperature:    #{c.temperature}"
       lines << "  Timeout:        #{c.timeout}s"
-      lines << "  Max tool rounds:#{c.max_tool_rounds}" if c.context_mode == :smart
+      lines << "  Max tool rounds:#{c.max_tool_rounds}"
       lines << "  Auto-execute:   #{c.auto_execute}"
       lines << "  Memories:       #{c.memories_enabled}"
+      lines << "  Session logging:#{session_table_status}"
       lines << "  Debug:          #{c.debug}"
       $stdout.puts lines.join("\n")
       nil
     end
+
+    def setup!
+      conn = session_connection
+      table = 'console_agent_sessions'
+
+      if conn.table_exists?(table)
+        $stdout.puts "\e[32mConsoleAgent: #{table} already exists. Run ConsoleAgent.teardown! first to recreate.\e[0m"
+      else
+        conn.create_table(table) do |t|
+          t.text    :query,         null: false
+          t.text    :conversation,  null: false
+          t.integer :input_tokens,  default: 0
+          t.integer :output_tokens, default: 0
+          t.string  :user_name,     limit: 255
+          t.string  :mode,          limit: 20, null: false
+          t.text    :code_executed
+          t.text    :code_output
+          t.text    :code_result
+          t.text    :console_output
+          t.boolean :executed,      default: false
+          t.string  :provider,      limit: 50
+          t.string  :model,         limit: 100
+          t.string  :name,          limit: 255
+          t.integer :duration_ms
+          t.datetime :created_at,   null: false
+        end
+
+        conn.add_index(table, :created_at)
+        conn.add_index(table, :user_name)
+        conn.add_index(table, :name)
+
+        $stdout.puts "\e[32mConsoleAgent: created #{table} table.\e[0m"
+      end
+    rescue => e
+      $stderr.puts "\e[31mConsoleAgent setup failed: #{e.class}: #{e.message}\e[0m"
+    end
+
+    def migrate!
+      conn = session_connection
+      table = 'console_agent_sessions'
+
+      unless conn.table_exists?(table)
+        $stderr.puts "\e[33mConsoleAgent: #{table} does not exist. Run ConsoleAgent.setup! first.\e[0m"
+        return
+      end
+
+      migrations = []
+
+      unless conn.column_exists?(table, :name)
+        conn.add_column(table, :name, :string, limit: 255)
+        conn.add_index(table, :name) unless conn.index_exists?(table, :name)
+        migrations << 'name'
+      end
+
+      if migrations.empty?
+        $stdout.puts "\e[32mConsoleAgent: #{table} is up to date.\e[0m"
+      else
+        $stdout.puts "\e[32mConsoleAgent: added columns: #{migrations.join(', ')}.\e[0m"
+      end
+    rescue => e
+      $stderr.puts "\e[31mConsoleAgent migrate failed: #{e.class}: #{e.message}\e[0m"
+    end
+
+    def teardown!
+      conn = session_connection
+      table = 'console_agent_sessions'
+
+      unless conn.table_exists?(table)
+        $stdout.puts "\e[33mConsoleAgent: #{table} does not exist, nothing to remove.\e[0m"
+        return
+      end
+
+      count = conn.select_value("SELECT COUNT(*) FROM #{conn.quote_table_name(table)}")
+      $stdout.print "\e[33mDrop #{table} (#{count} sessions)? [y/N] \e[0m"
+      answer = $stdin.gets.to_s.strip.downcase
+
+      unless answer == 'y' || answer == 'yes'
+        $stdout.puts "\e[33mCancelled.\e[0m"
+        return
+      end
+
+      conn.drop_table(table)
+      $stdout.puts "\e[32mConsoleAgent: dropped #{table}.\e[0m"
+    rescue => e
+      $stderr.puts "\e[31mConsoleAgent teardown failed: #{e.class}: #{e.message}\e[0m"
+    end
+
+    private
+
+    def session_table_status
+      return 'disabled' unless configuration.session_logging
+      conn = session_connection
+      if conn.table_exists?('console_agent_sessions')
+        count = conn.select_value("SELECT COUNT(*) FROM #{conn.quote_table_name('console_agent_sessions')}")
+        "\e[32m#{count} sessions\e[0m"
+      else
+        "\e[33mtable missing (run ConsoleAgent.setup!)\e[0m"
+      end
+    rescue
+      "\e[33munavailable\e[0m"
+    end
+
+    def session_connection
+      klass = configuration.connection_class
+      if klass
+        klass = Object.const_get(klass) if klass.is_a?(String)
+        klass.connection
+      else
+        ActiveRecord::Base.connection
+      end
+    end
   end
 end
 
-require 'console_agent/railtie' if defined?(Rails::Railtie)
+if defined?(Rails::Railtie)
+  require 'console_agent/railtie'
+  require 'console_agent/engine'
+end

data/lib/generators/console_agent/templates/initializer.rb

@@ -17,12 +17,7 @@ ConsoleAgent.configure do |config|
   # Auto-execute generated code without confirmation (use with caution!)
   config.auto_execute = false
 
-  # Context mode:
-  #   :smart - (default) minimal system prompt, LLM uses tools to fetch schema/model/code details on demand
-  #   :full  - sends all schema, models, and routes in the system prompt every time
-  config.context_mode = :smart
-
-  # Max tool-use rounds per query in :smart mode (safety cap)
+  # Max tool-use rounds per query (safety cap)
   config.max_tool_rounds = 10
 
   # HTTP timeout in seconds
@@ -30,4 +25,17 @@ ConsoleAgent.configure do |config|
 
   # Debug mode: prints full API requests/responses and tool calls to stderr
   # config.debug = true
+
+  # Session logging: persist AI sessions to the database
+  # Run ConsoleAgent.setup! in the Rails console to create the table
+  config.session_logging = true
+
+  # Database connection for ConsoleAgent tables (default: ActiveRecord::Base)
+  # Set to a class that responds to .connection if tables live on a different DB
+  # config.connection_class = Sharding::CentralizedModel
+
+  # Admin UI credentials (mount ConsoleAgent::Engine => '/console_agent' in routes.rb)
+  # When nil, no authentication is required (convenient for development)
+  # config.admin_username = 'admin'
+  # config.admin_password = 'changeme'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: console_agent
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.3.0
 platform: ruby
 authors:
 - Cortfr
@@ -88,16 +88,26 @@ extra_rdoc_files: []
 files:
 - LICENSE
 - README.md
+- app/controllers/console_agent/application_controller.rb
+- app/controllers/console_agent/sessions_controller.rb
+- app/helpers/console_agent/sessions_helper.rb
+- app/models/console_agent/session.rb
+- app/views/console_agent/sessions/index.html.erb
+- app/views/console_agent/sessions/show.html.erb
+- app/views/layouts/console_agent/application.html.erb
+- config/routes.rb
 - lib/console_agent.rb
 - lib/console_agent/configuration.rb
 - lib/console_agent/console_methods.rb
 - lib/console_agent/context_builder.rb
+- lib/console_agent/engine.rb
 - lib/console_agent/executor.rb
 - lib/console_agent/providers/anthropic.rb
 - lib/console_agent/providers/base.rb
 - lib/console_agent/providers/openai.rb
 - lib/console_agent/railtie.rb
 - lib/console_agent/repl.rb
+- lib/console_agent/session_logger.rb
 - lib/console_agent/storage/base.rb
 - lib/console_agent/storage/file_storage.rb
 - lib/console_agent/tools/code_tools.rb