RubyGems - console_agent - Versions diffs - 0.1.0 → 0.3.0 - Mend

console_agent 0.1.0 → 0.3.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (25) hide show

checksums.yaml +4 -4
data/README.md +129 -143
data/app/controllers/console_agent/application_controller.rb +21 -0
data/app/controllers/console_agent/sessions_controller.rb +16 -0
data/app/helpers/console_agent/sessions_helper.rb +42 -0
data/app/models/console_agent/session.rb +23 -0
data/app/views/console_agent/sessions/index.html.erb +57 -0
data/app/views/console_agent/sessions/show.html.erb +56 -0
data/app/views/layouts/console_agent/application.html.erb +83 -0
data/config/routes.rb +4 -0
data/lib/console_agent/configuration.rb +12 -5
data/lib/console_agent/console_methods.rb +167 -4
data/lib/console_agent/context_builder.rb +34 -125
data/lib/console_agent/engine.rb +5 -0
data/lib/console_agent/executor.rb +81 -1
data/lib/console_agent/repl.rb +363 -42
data/lib/console_agent/session_logger.rb +79 -0
data/lib/console_agent/storage/base.rb +27 -0
data/lib/console_agent/storage/file_storage.rb +63 -0
data/lib/console_agent/tools/memory_tools.rb +136 -0
data/lib/console_agent/tools/registry.rb +228 -2
data/lib/console_agent/version.rb +1 -1
data/lib/console_agent.rb +143 -3
data/lib/generators/console_agent/templates/initializer.rb +14 -6
metadata +14 -1

data/app/views/layouts/console_agent/application.html.erb ADDED Viewed

@@ -0,0 +1,83 @@
+<!DOCTYPE html>
+<html>
+<head>
+  <title>ConsoleAgent Admin</title>
+  <meta name="viewport" content="width=device-width, initial-scale=1">
+  <style>
+    * { margin: 0; padding: 0; box-sizing: border-box; }
+    body {
+      font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Roboto, Helvetica, Arial, sans-serif;
+      background: #f5f5f5;
+      color: #333;
+      line-height: 1.6;
+    }
+    .header {
+      background: #1a1a2e;
+      color: #fff;
+      padding: 16px 24px;
+      display: flex;
+      align-items: center;
+      justify-content: space-between;
+    }
+    .header h1 { font-size: 18px; font-weight: 600; }
+    .header a { color: #8be9fd; text-decoration: none; font-size: 14px; }
+    .container { max-width: 1200px; margin: 24px auto; padding: 0 24px; }
+    table { width: 100%; border-collapse: collapse; background: #fff; border-radius: 8px; overflow: hidden; box-shadow: 0 1px 3px rgba(0,0,0,0.1); }
+    th { background: #f8f9fa; text-align: left; padding: 12px 16px; font-size: 13px; font-weight: 600; color: #555; border-bottom: 2px solid #e9ecef; }
+    td { padding: 10px 16px; border-bottom: 1px solid #f0f0f0; font-size: 14px; }
+    tr:hover td { background: #f8f9fa; }
+    a { color: #4a6fa5; text-decoration: none; }
+    a:hover { text-decoration: underline; }
+    .badge {
+      display: inline-block; padding: 2px 8px; border-radius: 12px;
+      font-size: 11px; font-weight: 600; text-transform: uppercase;
+    }
+    .badge-one_shot { background: #d4edda; color: #155724; }
+    .badge-interactive { background: #cce5ff; color: #004085; }
+    .badge-explain { background: #fff3cd; color: #856404; }
+    .meta-card {
+      background: #fff; border-radius: 8px; padding: 20px;
+      box-shadow: 0 1px 3px rgba(0,0,0,0.1); margin-bottom: 24px;
+    }
+    .meta-grid { display: grid; grid-template-columns: repeat(auto-fill, minmax(200px, 1fr)); gap: 12px; }
+    .meta-item label { font-size: 11px; font-weight: 600; color: #888; text-transform: uppercase; display: block; }
+    .meta-item span { font-size: 14px; color: #333; }
+    .terminal {
+      background: #1e1e1e; border-radius: 8px; overflow: hidden;
+      box-shadow: 0 2px 8px rgba(0,0,0,0.2);
+    }
+    .terminal pre {
+      background: transparent; color: #d4d4d4;
+      padding: 16px 20px; margin: 0;
+      font-family: "SF Mono", "Monaco", "Menlo", "Consolas", monospace;
+      font-size: 13px; line-height: 1.5;
+      overflow-x: auto; white-space: pre-wrap; word-wrap: break-word;
+    }
+    .mono { font-family: "SF Mono", "Monaco", "Menlo", "Consolas", monospace; font-size: 13px; }
+    .text-muted { color: #888; }
+    .back-link { margin-bottom: 16px; display: inline-block; }
+    .check { color: #28a745; }
+    .cross { color: #dc3545; }
+    .pagination {
+      display: flex; justify-content: space-between; align-items: center;
+      margin-top: 16px; padding: 12px 0;
+    }
+    .pagination a {
+      padding: 6px 14px; background: #fff; border: 1px solid #ddd; border-radius: 4px;
+      font-size: 13px; color: #4a6fa5;
+    }
+    .pagination a:hover { background: #f8f9fa; text-decoration: none; }
+    .pagination .disabled { padding: 6px 14px; font-size: 13px; color: #ccc; }
+    .pagination .page-info { font-size: 13px; color: #888; }
+  </style>
+</head>
+<body>
+  <div class="header">
+    <h1>ConsoleAgent Admin</h1>
+    <a href="<%= console_agent.root_path %>">Sessions</a>
+  </div>
+  <div class="container">
+    <%= yield %>
+  </div>
+</body>
+</html>

data/config/routes.rb ADDED Viewed

@@ -0,0 +1,4 @@
+ConsoleAgent::Engine.routes.draw do
+  root to: 'sessions#index'
+  resources :sessions, only: [:index, :show]
+end

data/lib/console_agent/configuration.rb CHANGED Viewed

@@ -1,11 +1,13 @@
 module ConsoleAgent
   class Configuration
     PROVIDERS = %i[anthropic openai].freeze
-    CONTEXT_MODES = %i[full smart].freeze
     attr_accessor :provider, :api_key, :model, :max_tokens,
-                  :auto_execute, :context_mode, :temperature,
-                  :timeout, :debug, :max_tool_rounds
+                  :auto_execute, :temperature,
+                  :timeout, :debug, :max_tool_rounds,
+                  :storage_adapter, :memories_enabled,
+                  :session_logging, :connection_class,
+                  :admin_username, :admin_password
     def initialize
       @provider     = :anthropic
@@ -13,11 +15,16 @@ module ConsoleAgent
       @model        = nil
       @max_tokens   = 4096
       @auto_execute = false
-      @context_mode = :smart
       @temperature  = 0.2
       @timeout      = 30
       @debug        = false
-      @max_tool_rounds = 10
+      @max_tool_rounds = 100
+      @storage_adapter  = nil
+      @memories_enabled = true
+      @session_logging  = true
+      @connection_class = nil
+      @admin_username   = nil
+      @admin_password   = nil
     end
     def resolved_api_key

data/lib/console_agent/console_methods.rb CHANGED Viewed

@@ -4,16 +4,141 @@ module ConsoleAgent
       ConsoleAgent.status
     end
+    def ai_memories(n = nil)
+      require 'yaml'
+      require 'console_agent/tools/memory_tools'
+      storage = ConsoleAgent.storage
+      keys = storage.list('memories/*.md').sort
+      if keys.empty?
+        $stdout.puts "\e[2mNo memories stored yet.\e[0m"
+        return nil
+      end
+      memories = keys.filter_map do |key|
+        content = storage.read(key)
+        next if content.nil? || content.strip.empty?
+        next unless content =~ /\A---\s*\n(.*?\n)---\s*\n(.*)/m
+        fm = YAML.safe_load($1, permitted_classes: [Time, Date]) || {}
+        fm.merge('description' => $2.strip, 'file' => key)
+      end
+      if memories.empty?
+        $stdout.puts "\e[2mNo memories stored yet.\e[0m"
+        return nil
+      end
+      shown = n ? memories.last(n) : memories.last(5)
+      total = memories.length
+      $stdout.puts "\e[36m[Memories — showing last #{shown.length} of #{total}]\e[0m"
+      shown.each do |m|
+        $stdout.puts "\e[33m  #{m['name']}\e[0m"
+        $stdout.puts "\e[2m  #{m['description']}\e[0m"
+        tags = Array(m['tags'])
+        $stdout.puts "\e[2m  tags: #{tags.join(', ')}\e[0m" unless tags.empty?
+        $stdout.puts
+      end
+      path = storage.respond_to?(:root_path) ? File.join(storage.root_path, 'memories') : 'memories/'
+      $stdout.puts "\e[2mStored in: #{path}/\e[0m"
+      $stdout.puts "\e[2mUse ai_memories(n) to show last n.\e[0m"
+      nil
+    end
+    def ai_sessions(n = 10, search: nil)
+      require 'console_agent/session_logger'
+      session_class = Object.const_get('ConsoleAgent::Session')
+      scope = session_class.recent
+      scope = scope.search(search) if search
+      sessions = scope.limit(n)
+      if sessions.empty?
+        $stdout.puts "\e[2mNo sessions found.\e[0m"
+        return nil
+      end
+      $stdout.puts "\e[36m[Sessions — showing #{sessions.length}#{search ? " matching \"#{search}\"" : ''}]\e[0m"
+      $stdout.puts
+      sessions.each do |s|
+        id_str = "\e[2m##{s.id}\e[0m"
+        name_str = s.name ? "\e[33m#{s.name}\e[0m " : ""
+        query_str = s.name ? "\e[2m#{truncate_str(s.query, 50)}\e[0m" : truncate_str(s.query, 50)
+        mode_str = "\e[2m[#{s.mode}]\e[0m"
+        time_str = "\e[2m#{time_ago(s.created_at)}\e[0m"
+        tokens = (s.input_tokens || 0) + (s.output_tokens || 0)
+        token_str = tokens > 0 ? "\e[2m#{tokens} tokens\e[0m" : ""
+        $stdout.puts "  #{id_str} #{name_str}#{query_str}"
+        $stdout.puts "     #{mode_str} #{time_str} #{token_str}"
+        $stdout.puts
+      end
+      $stdout.puts "\e[2mUse ai_resume(id_or_name) to resume a session.\e[0m"
+      $stdout.puts "\e[2mUse ai_sessions(n, search: \"term\") to filter.\e[0m"
+      nil
+    rescue => e
+      $stderr.puts "\e[31mConsoleAgent error: #{e.message}\e[0m"
+      nil
+    end
+    def ai_resume(identifier)
+      __ensure_console_agent_user
+      require 'console_agent/context_builder'
+      require 'console_agent/providers/base'
+      require 'console_agent/executor'
+      require 'console_agent/repl'
+      require 'console_agent/session_logger'
+      session = __find_session(identifier)
+      unless session
+        $stderr.puts "\e[31mSession not found: #{identifier}\e[0m"
+        return nil
+      end
+      repl = Repl.new(__console_agent_binding)
+      repl.resume(session)
+    rescue => e
+      $stderr.puts "\e[31mConsoleAgent error: #{e.message}\e[0m"
+      nil
+    end
+    def ai_name(identifier, new_name)
+      require 'console_agent/session_logger'
+      session = __find_session(identifier)
+      unless session
+        $stderr.puts "\e[31mSession not found: #{identifier}\e[0m"
+        return nil
+      end
+      ConsoleAgent::SessionLogger.update(session.id, name: new_name)
+      $stdout.puts "\e[36mSession ##{session.id} named: #{new_name}\e[0m"
+      nil
+    rescue => e
+      $stderr.puts "\e[31mConsoleAgent error: #{e.message}\e[0m"
+      nil
+    end
     def ai(query = nil)
       if query.nil?
         $stderr.puts "\e[33mUsage: ai \"your question here\"\e[0m"
-        $stderr.puts "\e[33m  ai  \"query\" - ask + confirm execution\e[0m"
-        $stderr.puts "\e[33m  ai! \"query\" - enter interactive mode (or ai! with no args)\e[0m"
-        $stderr.puts "\e[33m  ai? \"query\" - explain only, no execution\e[0m"
-        $stderr.puts "\e[33m  ai_status   - show current configuration\e[0m"
+        $stderr.puts "\e[33m  ai  \"query\"  - ask + confirm execution\e[0m"
+        $stderr.puts "\e[33m  ai! \"query\"  - enter interactive mode (or ai! with no args)\e[0m"
+        $stderr.puts "\e[33m  ai? \"query\"  - explain only, no execution\e[0m"
+        $stderr.puts "\e[33m  ai_sessions  - list recent sessions\e[0m"
+        $stderr.puts "\e[33m  ai_resume    - resume a session by name or id\e[0m"
+        $stderr.puts "\e[33m  ai_name      - name a session: ai_name 42, \"my_label\"\e[0m"
+        $stderr.puts "\e[33m  ai_status    - show current configuration\e[0m"
+        $stderr.puts "\e[33m  ai_memories  - show recent memories (ai_memories(n) for last n)\e[0m"
         return nil
       end
+      __ensure_console_agent_user
       require 'console_agent/context_builder'
       require 'console_agent/providers/base'
       require 'console_agent/executor'
@@ -27,6 +152,8 @@ module ConsoleAgent
     end
     def ai!(query = nil)
+      __ensure_console_agent_user
       require 'console_agent/context_builder'
       require 'console_agent/providers/base'
       require 'console_agent/executor'
@@ -50,6 +177,8 @@ module ConsoleAgent
         return nil
       end
+      __ensure_console_agent_user
       require 'console_agent/context_builder'
       require 'console_agent/providers/base'
       require 'console_agent/executor'
@@ -64,6 +193,40 @@ module ConsoleAgent
     private
+    def __find_session(identifier)
+      session_class = Object.const_get('ConsoleAgent::Session')
+      if identifier.is_a?(Integer)
+        session_class.find_by(id: identifier)
+      else
+        session_class.where(name: identifier.to_s).recent.first ||
+          session_class.find_by(id: identifier.to_i)
+      end
+    end
+    def truncate_str(str, max)
+      return '' if str.nil?
+      str.length > max ? str[0...max] + '...' : str
+    end
+    def time_ago(time)
+      return '' unless time
+      seconds = Time.now - time
+      case seconds
+      when 0...60       then "just now"
+      when 60...3600    then "#{(seconds / 60).to_i}m ago"
+      when 3600...86400 then "#{(seconds / 3600).to_i}h ago"
+      else                   "#{(seconds / 86400).to_i}d ago"
+      end
+    end
+    def __ensure_console_agent_user
+      return if ConsoleAgent.current_user
+      $stdout.puts "\e[36mConsoleAgent logs all AI sessions for audit purposes.\e[0m"
+      $stdout.print "\e[36mPlease enter your name: \e[0m"
+      name = $stdin.gets.to_s.strip
+      ConsoleAgent.current_user = name.empty? ? ENV['USER'] : name
+    end
     def __console_agent_binding
       # Try IRB workspace binding
       if defined?(IRB) && IRB.respond_to?(:CurrentContext)

data/lib/console_agent/context_builder.rb CHANGED Viewed

@@ -5,31 +5,17 @@ module ConsoleAgent
     end
     def build
-      case @config.context_mode
-      when :smart
-        build_smart
-      else
-        build_full
-      end
+      build_smart
     rescue => e
       ConsoleAgent.logger.warn("ConsoleAgent: context build error: #{e.message}")
-      system_instructions + "\n\n" + environment_context
-    end
-    def build_full
-      parts = []
-      parts << system_instructions
-      parts << environment_context
-      parts << schema_context
-      parts << models_context
-      parts << routes_context
-      parts.compact.join("\n\n")
+      smart_system_instructions + "\n\n" + environment_context
     end
     def build_smart
       parts = []
       parts << smart_system_instructions
       parts << environment_context
+      parts << memory_context
       parts.compact.join("\n\n")
     end
@@ -48,10 +34,29 @@ module ConsoleAgent
         you need specific information to write accurate code — such as which user they are, which
         record to target, or what value to use.
+        You have memory tools to persist what you learn across sessions:
+        - save_memory: persist facts or procedures you learn about this codebase.
+          If a memory with the same name already exists, it will be updated in place.
+        - delete_memory: remove a memory by name
+        - recall_memories: search your saved memories for details
+        IMPORTANT: Check the Memories section below BEFORE answering. If a memory is relevant,
+        use recall_memories to get full details and apply that knowledge to your answer.
+        When you use a memory, mention it briefly (e.g. "Based on what I know about sharding...").
+        When you discover important patterns about this app, save them as memories.
+        You have an execute_plan tool to run multi-step code. When a task requires multiple
+        sequential operations, use execute_plan with an array of steps (each with a description
+        and Ruby code). The plan is shown to the user for review before execution begins.
+        After each step runs, its return value is stored as step1, step2, etc. — use these
+        variables in later steps to reference earlier results (e.g. `api = SalesforceApi.new(step1)`).
+        For simple single-expression answers, you may respond with a ```ruby code block instead.
         RULES:
         - Give ONE concise answer. Do not offer multiple alternatives or variations.
-        - Respond with a single ```ruby code block that directly answers the question.
-        - Include a brief one-line explanation before the code block.
+        - For multi-step tasks, use execute_plan to break the work into small, clear steps.
+        - For simple queries, respond with a single ```ruby code block.
+        - Include a brief one-line explanation before any code block.
         - Use the app's actual model names, associations, and schema.
         - Prefer ActiveRecord query interface over raw SQL.
         - For destructive operations, add a comment warning.
@@ -62,23 +67,6 @@ module ConsoleAgent
       PROMPT
     end
-    def system_instructions
-      <<~PROMPT.strip
-        You are a Ruby on Rails console assistant. The user is in a `rails console` session.
-        You help them query data, debug issues, and understand their application.
-        RULES:
-        - Give ONE concise answer. Do not offer multiple alternatives or variations.
-        - Respond with a single ```ruby code block that directly answers the question.
-        - Include a brief one-line explanation before the code block.
-        - Use the app's actual model names, associations, and schema.
-        - Prefer ActiveRecord query interface over raw SQL.
-        - For destructive operations, add a comment warning.
-        - If the request is ambiguous, ask a clarifying question instead of guessing.
-        - Keep code concise and idiomatic.
-      PROMPT
-    end
     def environment_context
       lines = ["## Environment"]
       lines << "- Ruby #{RUBY_VERSION}"
@@ -100,101 +88,22 @@ module ConsoleAgent
       lines.join("\n")
     end
-    def schema_context
-      return nil unless defined?(ActiveRecord::Base) && ActiveRecord::Base.connected?
-      conn = ActiveRecord::Base.connection
-      tables = conn.tables.sort
-      return nil if tables.empty?
-      lines = ["## Database Schema"]
-      tables.each do |table|
-        next if table == 'schema_migrations' || table == 'ar_internal_metadata'
-        cols = conn.columns(table).map { |c| "#{c.name}:#{c.type}" }
-        lines << "- #{table} (#{cols.join(', ')})"
-      end
-      lines.join("\n")
-    rescue => e
-      ConsoleAgent.logger.debug("ConsoleAgent: schema introspection failed: #{e.message}")
-      nil
-    end
-    def models_context
-      return nil unless defined?(ActiveRecord::Base)
-      eager_load_app!
-      base_class = defined?(ApplicationRecord) ? ApplicationRecord : ActiveRecord::Base
-      models = ObjectSpace.each_object(Class).select { |c|
-        c < base_class && !c.abstract_class? && c.name && !c.name.start_with?('HABTM_')
-      }.sort_by(&:name)
-      return nil if models.empty?
-      lines = ["## Models"]
-      models.each do |model|
-        parts = [model.name]
-        assocs = model.reflect_on_all_associations.map { |a| "#{a.macro} :#{a.name}" }
-        parts << "  associations: #{assocs.join(', ')}" unless assocs.empty?
-        scopes = model.methods.grep(/^_scope_/).map { |m| m.to_s.sub('_scope_', '') } rescue []
-        if scopes.empty?
-          # Alternative: check singleton methods that return ActiveRecord::Relation
-          scope_names = (model.singleton_methods - base_class.singleton_methods).select { |m|
-            m != :all && !m.to_s.start_with?('_') && !m.to_s.start_with?('find')
-          }.first(10)
-          # We won't list these to avoid noise — scopes are hard to detect reliably
-        end
+    def memory_context
+      return nil unless @config.memories_enabled
-        validations = model.validators.map { |v|
-          attrs = v.attributes.join(', ')
-          "#{v.class.name.demodulize.underscore.sub('_validator', '')} on #{attrs}"
-        }.uniq.first(5) rescue []
-        parts << "  validations: #{validations.join('; ')}" unless validations.empty?
-        lines << "- #{parts.join("\n")}"
-      end
-      lines.join("\n")
-    rescue => e
-      ConsoleAgent.logger.debug("ConsoleAgent: model introspection failed: #{e.message}")
-      nil
-    end
-    def routes_context
-      return nil unless defined?(Rails) && Rails.respond_to?(:application)
-      routes = Rails.application.routes.routes
-      return nil if routes.empty?
-      lines = ["## Routes (summary)"]
-      count = 0
-      routes.each do |route|
-        next if route.internal?
-        path = route.path.spec.to_s.sub('(.:format)', '')
-        verb = route.verb.to_s
-        next if verb.empty?
-        action = route.defaults[:controller].to_s + '#' + route.defaults[:action].to_s
-        lines << "- #{verb} #{path} -> #{action}"
-        count += 1
-        break if count >= 50
-      end
-      lines << "- ... (#{routes.size - count} more routes)" if routes.size > count
+      require 'console_agent/tools/memory_tools'
+      summaries = Tools::MemoryTools.new.memory_summaries
+      return nil if summaries.nil? || summaries.empty?
+      lines = ["## Memories"]
+      lines.concat(summaries)
+      lines << ""
+      lines << "Call recall_memories to get details before answering. Do NOT guess from the name alone."
       lines.join("\n")
     rescue => e
-      ConsoleAgent.logger.debug("ConsoleAgent: route introspection failed: #{e.message}")
+      ConsoleAgent.logger.debug("ConsoleAgent: memory context failed: #{e.message}")
       nil
     end
-    def eager_load_app!
-      return unless defined?(Rails) && Rails.respond_to?(:application)
-      if Rails.application.respond_to?(:eager_load!)
-        Rails.application.eager_load!
-      end
-    rescue => e
-      ConsoleAgent.logger.debug("ConsoleAgent: eager_load failed: #{e.message}")
-    end
   end
 end

data/lib/console_agent/engine.rb ADDED Viewed

@@ -0,0 +1,5 @@
+module ConsoleAgent
+  class Engine < ::Rails::Engine
+    isolate_namespace ConsoleAgent
+  end
+end

data/lib/console_agent/executor.rb CHANGED Viewed

@@ -1,8 +1,50 @@
+require 'stringio'
 module ConsoleAgent
+  # Writes to two IO streams simultaneously
+  class TeeIO
+    attr_reader :secondary
+    def initialize(primary, secondary)
+      @primary = primary
+      @secondary = secondary
+    end
+    def write(str)
+      @primary.write(str)
+      @secondary.write(str)
+    end
+    def puts(*args)
+      @primary.puts(*args)
+      # Capture what puts would output
+      args.each { |a| @secondary.write("#{a}\n") }
+      @secondary.write("\n") if args.empty?
+    end
+    def print(*args)
+      @primary.print(*args)
+      args.each { |a| @secondary.write(a.to_s) }
+    end
+    def flush
+      @primary.flush if @primary.respond_to?(:flush)
+    end
+    def respond_to_missing?(method, include_private = false)
+      @primary.respond_to?(method, include_private) || super
+    end
+    def method_missing(method, *args, &block)
+      @primary.send(method, *args, &block)
+    end
+  end
   class Executor
     CODE_REGEX = /```ruby\s*\n(.*?)```/m
     attr_reader :binding_context
+    attr_accessor :on_prompt
     def initialize(binding_context)
       @binding_context = binding_context
@@ -33,23 +75,53 @@ module ConsoleAgent
     def execute(code)
       return nil if code.nil? || code.strip.empty?
+      captured_output = StringIO.new
+      old_stdout = $stdout
+      # Tee output: capture it and also print to the real stdout
+      $stdout = TeeIO.new(old_stdout, captured_output)
       result = binding_context.eval(code, "(console_agent)", 1)
+      $stdout = old_stdout
       $stdout.puts colorize("=> #{result.inspect}", :green)
+      @last_output = captured_output.string
       result
     rescue SyntaxError => e
+      $stdout = old_stdout if old_stdout
       $stderr.puts colorize("SyntaxError: #{e.message}", :red)
+      @last_output = nil
       nil
     rescue => e
+      $stdout = old_stdout if old_stdout
       $stderr.puts colorize("Error: #{e.class}: #{e.message}", :red)
       e.backtrace.first(3).each { |line| $stderr.puts colorize("  #{line}", :red) }
+      @last_output = captured_output&.string
       nil
     end
+    def last_output
+      @last_output
+    end
+    def last_answer
+      @last_answer
+    end
+    def last_cancelled?
+      @last_cancelled
+    end
     def confirm_and_execute(code)
       return nil if code.nil? || code.strip.empty?
+      @last_cancelled = false
+      @last_answer = nil
       $stdout.print colorize("Execute? [y/N/edit] ", :yellow)
+      @on_prompt&.call
       answer = $stdin.gets.to_s.strip.downcase
+      @last_answer = answer
+      echo_stdin(answer)
       case answer
       when 'y', 'yes'
@@ -60,7 +132,9 @@ module ConsoleAgent
           $stdout.puts colorize("# Edited code:", :yellow)
           $stdout.puts highlight_code(edited)
           $stdout.print colorize("Execute edited code? [y/N] ", :yellow)
-          if $stdin.gets.to_s.strip.downcase == 'y'
+          edit_answer = $stdin.gets.to_s.strip.downcase
+          echo_stdin(edit_answer)
+          if edit_answer == 'y'
             execute(edited)
           else
             $stdout.puts colorize("Cancelled.", :yellow)
@@ -71,12 +145,18 @@ module ConsoleAgent
         end
       else
         $stdout.puts colorize("Cancelled.", :yellow)
+        @last_cancelled = true
         nil
       end
     end
     private
+    # Write stdin input to the capture IO only (avoids double-echo on terminal)
+    def echo_stdin(text)
+      $stdout.secondary.write("#{text}\n") if $stdout.respond_to?(:secondary)
+    end
     def open_in_editor(code)
       require 'tempfile'
       editor = ENV['EDITOR'] || 'vi'