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

console_agent 0.2.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 (22) hide show

checksums.yaml +4 -4
data/README.md +117 -231
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 +8 -4
data/lib/console_agent/console_methods.rb +125 -5
data/lib/console_agent/context_builder.rb +12 -132
data/lib/console_agent/engine.rb +5 -0
data/lib/console_agent/executor.rb +19 -1
data/lib/console_agent/repl.rb +299 -34
data/lib/console_agent/session_logger.rb +79 -0
data/lib/console_agent/tools/registry.rb +156 -2
data/lib/console_agent/version.rb +1 -1
data/lib/console_agent.rb +125 -3
data/lib/generators/console_agent/templates/initializer.rb +14 -6
metadata +11 -1

data/app/views/console_agent/sessions/show.html.erb ADDED Viewed

@@ -0,0 +1,56 @@
+<a href="<%= console_agent.sessions_path %>" class="back-link">&larr; Back to sessions</a>
+<div class="meta-card">
+  <div class="meta-grid">
+    <div class="meta-item">
+      <label>Query</label>
+      <span><%= @session.query %></span>
+    </div>
+    <% if @session.name.present? %>
+      <div class="meta-item">
+        <label>Name</label>
+        <span><%= @session.name %></span>
+      </div>
+    <% end %>
+    <div class="meta-item">
+      <label>Mode</label>
+      <span class="badge badge-<%= @session.mode %>"><%= @session.mode %></span>
+    </div>
+    <div class="meta-item">
+      <label>User</label>
+      <span><%= @session.user_name || '-' %></span>
+    </div>
+    <div class="meta-item">
+      <label>Provider / Model</label>
+      <span><%= @session.provider %> / <%= @session.model %></span>
+    </div>
+    <div class="meta-item">
+      <label>Tokens (in / out)</label>
+      <span class="mono"><%= @session.input_tokens %> / <%= @session.output_tokens %></span>
+    </div>
+    <div class="meta-item">
+      <label>Duration</label>
+      <span class="mono"><%= @session.duration_ms ? "#{@session.duration_ms}ms" : '-' %></span>
+    </div>
+    <div class="meta-item">
+      <label>Executed?</label>
+      <span><%= @session.executed? ? '<span class="check">Yes</span>'.html_safe : '<span class="cross">No</span>'.html_safe %></span>
+    </div>
+    <div class="meta-item">
+      <label>Time</label>
+      <span><%= @session.created_at.strftime('%Y-%m-%d %H:%M:%S') %></span>
+    </div>
+  </div>
+</div>
+<h3 style="font-size: 16px; margin-bottom: 12px;">Console Output</h3>
+<% if @session.console_output.present? %>
+  <div class="terminal">
+    <pre><%= ansi_to_html(@session.console_output) %></pre>
+  </div>
+<% else %>
+  <div class="meta-card">
+    <p class="text-muted">No console output recorded for this session.</p>
+  </div>
+<% end %>

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,12 +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,
+                  :auto_execute, :temperature,
                   :timeout, :debug, :max_tool_rounds,
-                  :storage_adapter, :memories_enabled
+                  :storage_adapter, :memories_enabled,
+                  :session_logging, :connection_class,
+                  :admin_username, :admin_password
     def initialize
       @provider     = :anthropic
@@ -14,13 +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 = 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

@@ -46,17 +46,99 @@ module ConsoleAgent
       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_memories - show recent memories (ai_memories(n) for last n)\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'
@@ -70,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'
@@ -93,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'
@@ -107,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,25 +5,10 @@ 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
@@ -60,10 +45,18 @@ module ConsoleAgent
         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.
@@ -74,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}"
@@ -112,93 +88,6 @@ 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
-        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
-      lines.join("\n")
-    rescue => e
-      ConsoleAgent.logger.debug("ConsoleAgent: route introspection failed: #{e.message}")
-      nil
-    end
     def memory_context
       return nil unless @config.memories_enabled
@@ -216,14 +105,5 @@ module ConsoleAgent
       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

@@ -3,6 +3,8 @@ require 'stringio'
 module ConsoleAgent
   # Writes to two IO streams simultaneously
   class TeeIO
+    attr_reader :secondary
     def initialize(primary, secondary)
       @primary = primary
       @secondary = secondary
@@ -42,6 +44,7 @@ module ConsoleAgent
     CODE_REGEX = /```ruby\s*\n(.*?)```/m
     attr_reader :binding_context
+    attr_accessor :on_prompt
     def initialize(binding_context)
       @binding_context = binding_context
@@ -101,6 +104,10 @@ module ConsoleAgent
       @last_output
     end
+    def last_answer
+      @last_answer
+    end
     def last_cancelled?
       @last_cancelled
     end
@@ -109,8 +116,12 @@ module ConsoleAgent
       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'
@@ -121,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)
@@ -139,6 +152,11 @@ module ConsoleAgent
     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'