rails_console_ai 0.15.0 → 0.17.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f82dc8df360173a57a1152e60a47e7bf59c31b3721658ee590b7d5cb7c765fbd
-  data.tar.gz: 942c2df0dda0048784e7cce6911a20685480de5a4c39ae5b1ca0d875d2f6627d
+  metadata.gz: 98abad8ed9e3d5d510ffc055352687156ffd4a5c02fbde2610f8ecd46cf2c263
+  data.tar.gz: 2979eeb2d6f7e3f8df58a4cd25666307cd4e108dfec379020e3ad6422bbba911
 SHA512:
-  metadata.gz: 32edecb4a5c29703ce2ff309fd922c11de57240f98df92072c86f99f3373c3eaeaf689e912960cc60a71cc40e37b2dacf2cea73dd6385cb101f36dd61d913f99
-  data.tar.gz: e36e885ad95efcf976ee249c2524348e84b11aa61d447d9680ffeb049b7a6f4628c0e34b5027127e04ac83979259b4e16fea3f7208be2c6645d9bf25cf52bdcf
+  metadata.gz: bc6695f1321d2c5b39b14e08e676b354f89eb62f29718244a0a25580a5d13250c7581ccfdd52552489f7a53c252433be465fb6b95c9de981825317680f575f5e
+  data.tar.gz: 35ff574a1c8404cc43c308f5c5bd13bcc1acda339c9543026a7ac216e4756988629088c436a38ab9dac2a34dc2e4a994168a4f8c70a57fbf7262135db8ac8b7d

data/CHANGELOG.md

@@ -2,6 +2,22 @@
 
 All notable changes to this project will be documented in this file.
 
+## [0.17.0]
+
+- Add `/retry` command
+- Print provider information when `ai!` starts
+- Keep Slack bot alive during long-running sessions
+- Improve Slack bot log prefixes for production log search
+- Catch safety errors even when swallowed by executed code
+- Fix Bedrock handling of multiple tool results
+
+## [0.16.0]
+
+- Run migrations during setup
+- Clarify LLM activity messages in Slack channel
+- Clean up error messages
+- Rotate thinking messages in Slack bot
+
 ## [0.15.0]
 
 - Add `config.code_search_paths` to configure searchable code directories

data/lib/rails_console_ai/channel/console.rb

@@ -158,6 +158,8 @@ module RailsConsoleAi
         guards = RailsConsoleAi.configuration.safety_guards
         name_display = @engine.session_name ? " (#{@engine.session_name})" : ""
         @real_stdout.puts "\e[36mRailsConsoleAi interactive mode#{name_display}. Type 'exit' or 'quit' to leave.\e[0m"
+        config = RailsConsoleAi.configuration
+        @real_stdout.puts "\e[2m  Provider: #{config.provider} | Model: #{config.resolved_model}\e[0m"
         safe_info = guards.empty? ? '' : " | Safe mode: #{guards.enabled? ? 'ON' : 'OFF'} (/danger to toggle)"
         @real_stdout.puts "\e[2m  Auto-execute: #{auto ? 'ON' : 'OFF'} (Shift-Tab or /auto to toggle)#{safe_info} | > code | /usage | /cost | /compact | /think | /name <label>\e[0m"
 
@@ -259,6 +261,8 @@ module RailsConsoleAi
           else
             @real_stdout.puts "\e[33mNo omitted output with id #{expand_id}\e[0m"
           end
+        when '/retry'
+          retry_last_code
         when /\A\/name/
           handle_name_command(input)
         else
@@ -267,6 +271,10 @@ module RailsConsoleAi
         true
       end
 
+      def retry_last_code
+        @engine.retry_last_code
+      end
+
       def handle_direct_execution(input)
         raw_code = input.sub(/\A>\s?/, '')
         Readline::HISTORY.push(input) unless input == Readline::HISTORY.to_a.last
@@ -341,6 +349,7 @@ module RailsConsoleAi
         @real_stdout.puts "\e[2m    /system      Show the system prompt\e[0m"
         @real_stdout.puts "\e[2m    /expand <id> Show full omitted output\e[0m"
         @real_stdout.puts "\e[2m    /debug       Toggle debug summaries (context stats, cost per call)\e[0m"
+        @real_stdout.puts "\e[2m    /retry       Re-execute the last code block\e[0m"
         @real_stdout.puts "\e[2m    > code       Execute Ruby directly (skip LLM)\e[0m"
         @real_stdout.puts "\e[2m    exit/quit    Leave interactive mode\e[0m"
       end

data/lib/rails_console_ai/channel/slack.rb

@@ -5,24 +5,6 @@ module RailsConsoleAi
     class Slack < Base
       ANSI_REGEX = /\e\[[0-9;]*m/
 
-      THINKING_MESSAGES = [
-        "Thinking...",
-        "Reticulating splines...",
-        "Scrubbing encryption bits...",
-        "Consulting the oracle...",
-        "Rummaging through the database...",
-        "Warming up the hamster wheel...",
-        "Polishing the pixels...",
-        "Untangling the spaghetti code...",
-        "Asking the magic 8-ball...",
-        "Counting all the things...",
-        "Herding the electrons...",
-        "Dusting off the old records...",
-        "Feeding the algorithms...",
-        "Shaking the data tree...",
-        "Bribing the servers...",
-      ].freeze
-
       def initialize(slack_bot:, channel_id:, thread_ts:, user_name: nil)
         @slack_bot = slack_bot
         @channel_id = channel_id
@@ -47,15 +29,22 @@ module RailsConsoleAi
       end
 
       def display_dim(text)
-        stripped = strip_ansi(text).strip
-        if stripped =~ /\AThinking\.\.\.|\ACalling LLM/
-          post(random_thinking_message)
-        elsif stripped =~ /\AAttempting to fix|\ACancelled|\A_session:/
+        raw = strip_ansi(text)
+        stripped = raw.strip
+
+        if stripped =~ /\AThinking\.\.\.|\AAttempting to fix|\ACancelled|\A_session:/
+          post(stripped)
+        elsif stripped =~ /\ACalling LLM/
+          # Technical LLM round status — suppress in Slack
+          @output_log.write("#{stripped}\n")
+          STDOUT.puts "#{@log_prefix} (dim) #{stripped}"
+        elsif raw =~ /\A {2,4}\S/ && stripped.length > 10
+          # LLM thinking text (2-space indent from conversation engine) — show as status
           post(stripped)
         else
-          # Log for engineers but don't post to Slack
+          # Tool result previews (5+ space indent) and other technical noise — log only
           @output_log.write("#{stripped}\n")
-          $stdout.puts "#{@log_prefix} (dim) #{stripped}"
+          STDOUT.puts "#{@log_prefix} (dim) #{stripped}"
         end
       end
 
@@ -135,6 +124,7 @@ module RailsConsoleAi
           - The output of `puts` in your code is automatically shown to the user. Do NOT
             repeat or re-display data that your code already printed via `puts`.
             Just add a brief summary after (e.g. "10 events found" or "Let me know if you need more detail").
+          - Do not offer to make changes or take actions on behalf of the user. Only report findings.
           - This is a live production database — other processes, users, and background jobs are
             constantly changing data. Never assume results will be the same as a previous query.
             Always re-run queries when asked, even if you just ran the same one.
@@ -160,7 +150,7 @@ module RailsConsoleAi
       def post(text)
         return if text.nil? || text.strip.empty?
         @output_log.write("#{text}\n")
-        $stdout.puts "#{@log_prefix} >> #{text}"
+        STDOUT.puts "#{@log_prefix} >> #{text}"
         @slack_bot.send(:post_message,
           channel: @channel_id,
           thread_ts: @thread_ts,
@@ -170,10 +160,6 @@ module RailsConsoleAi
         RailsConsoleAi.logger.error("Slack post failed: #{e.message}")
       end
 
-      def random_thinking_message
-        THINKING_MESSAGES.sample
-      end
-
       def strip_ansi(text)
         text.to_s.gsub(ANSI_REGEX, '')
       end

data/lib/rails_console_ai/conversation_engine.rb

@@ -221,6 +221,16 @@ module RailsConsoleAi
       end
     end
 
+    def retry_last_code
+      code = @last_interactive_code
+      unless code
+        @channel.display_warning("No code to retry.")
+        return
+      end
+      @channel.display_dim("  Retrying last code...")
+      execute_direct(code)
+    end
+
     def execute_direct(raw_code)
       exec_result = @executor.execute(raw_code)
 

data/lib/rails_console_ai/executor.rb

@@ -1,4 +1,5 @@
 require 'stringio'
+require_relative 'safety_guards'
 
 module RailsConsoleAi
   # Writes to two IO streams simultaneously
@@ -97,12 +98,25 @@ module RailsConsoleAi
       # Tee output: capture it and also print to the real stdout
       $stdout = TeeIO.new(old_stdout, captured_output)
 
+      RailsConsoleAi::SafetyError.clear!
+
       result = with_safety_guards do
         binding_context.eval(code, "(rails_console_ai)", 1)
       end
 
       $stdout = old_stdout
 
+      # Check if a SafetyError was raised but swallowed by a rescue inside the eval'd code
+      if (swallowed = RailsConsoleAi::SafetyError.last_raised)
+        RailsConsoleAi::SafetyError.clear!
+        @last_error = "SafetyError: #{swallowed.message}"
+        @last_safety_error = true
+        @last_safety_exception = swallowed
+        display_error("Blocked: #{swallowed.message}")
+        @last_output = captured_output&.string
+        return nil
+      end
+
       # Send captured puts output through channel before the return value
       if @channel && !captured_output.string.empty?
         @channel.display_result_output(captured_output.string)
@@ -114,6 +128,7 @@ module RailsConsoleAi
       result
     rescue RailsConsoleAi::SafetyError => e
       $stdout = old_stdout if old_stdout
+      RailsConsoleAi::SafetyError.clear!
       @last_error = "SafetyError: #{e.message}"
       @last_safety_error = true
       @last_safety_exception = e
@@ -140,8 +155,8 @@ module RailsConsoleAi
         return nil
       end
       @last_error = "#{e.class}: #{e.message}"
-      display_error("Error: #{@last_error}")
-      e.backtrace.first(3).each { |line| display_error("  #{line}") }
+      backtrace = e.backtrace.first(3).map { |line| "  #{line}" }.join("\n")
+      display_error("Error: #{@last_error}\n#{backtrace}")
       @last_output = captured_output&.string
       nil
     end

data/lib/rails_console_ai/prefixed_io.rb

@@ -0,0 +1,50 @@
+module RailsConsoleAi
+  class PrefixedIO
+    def initialize(io)
+      @io = io
+    end
+
+    def write(str)
+      prefix = Thread.current[:log_prefix]
+      if prefix && str.is_a?(String) && !str.strip.empty?
+        prefixed = str.gsub(/^(?=.)/, "#{prefix} ")
+        @io.write(prefixed)
+      else
+        @io.write(str)
+      end
+    end
+
+    def puts(*args)
+      prefix = Thread.current[:log_prefix]
+      if prefix
+        args = [""] if args.empty?
+        args.each do |a|
+          line = a.to_s
+          if line.strip.empty?
+            @io.write("\n")
+          else
+            @io.write("#{prefix} #{line}\n")
+          end
+        end
+      else
+        @io.puts(*args)
+      end
+    end
+
+    def print(*args)
+      @io.print(*args)
+    end
+
+    def flush
+      @io.flush
+    end
+
+    def respond_to_missing?(method, include_private = false)
+      @io.respond_to?(method, include_private) || super
+    end
+
+    def method_missing(method, *args, &block)
+      @io.send(method, *args, &block)
+    end
+  end
+end

data/lib/rails_console_ai/providers/bedrock.rb

@@ -115,7 +115,7 @@ module RailsConsoleAi
       end
 
       def format_messages(messages)
-        messages.map do |msg|
+        formatted = messages.map do |msg|
           content = if msg[:content].is_a?(Array)
                       msg[:content]
                     else
@@ -123,6 +123,18 @@ module RailsConsoleAi
                     end
           { role: msg[:role].to_s, content: content }
         end
+
+        # Bedrock requires all tool_result blocks for a single assistant turn
+        # to be in one user message. Merge consecutive same-role messages.
+        merged = []
+        formatted.each do |msg|
+          if merged.last && merged.last[:role] == msg[:role]
+            merged.last[:content].concat(msg[:content])
+          else
+            merged << msg
+          end
+        end
+        merged
       end
 
       def extract_text(response)

data/lib/rails_console_ai/safety_guards.rb

@@ -5,10 +5,21 @@ module RailsConsoleAi
   class SafetyError < StandardError
     attr_reader :guard, :blocked_key
 
+    # Thread-local tracking so the executor can detect safety errors
+    # even when swallowed by a rescue inside eval'd code.
+    def self.last_raised
+      Thread.current[:rails_console_ai_last_safety_error]
+    end
+
+    def self.clear!
+      Thread.current[:rails_console_ai_last_safety_error] = nil
+    end
+
     def initialize(message, guard: nil, blocked_key: nil)
       super(message)
       @guard = guard
       @blocked_key = blocked_key
+      Thread.current[:rails_console_ai_last_safety_error] = self
     end
   end
 

data/lib/rails_console_ai/slack_bot.rb

@@ -2,6 +2,7 @@ require 'json'
 require 'uri'
 require 'net/http'
 require 'openssl'
+require 'rails_console_ai/prefixed_io'
 require 'rails_console_ai/channel/slack'
 require 'rails_console_ai/conversation_engine'
 require 'rails_console_ai/context_builder'
@@ -10,6 +11,9 @@ require 'rails_console_ai/executor'
 
 module RailsConsoleAi
   class SlackBot
+    PING_INTERVAL = 30  # seconds — send ping if no data received
+    PONG_TIMEOUT  = 60  # seconds — reconnect if no pong after ping
+
     def initialize
       @bot_token = RailsConsoleAi.configuration.slack_bot_token || ENV['SLACK_BOT_TOKEN']
       @app_token = RailsConsoleAi.configuration.slack_app_token || ENV['SLACK_APP_TOKEN']
@@ -26,6 +30,11 @@ module RailsConsoleAi
     end
 
     def start
+      $stdout.sync = true
+      $stderr.sync = true
+      $stdout = RailsConsoleAi::PrefixedIO.new($stdout) unless $stdout.is_a?(RailsConsoleAi::PrefixedIO)
+      $stderr = RailsConsoleAi::PrefixedIO.new($stderr) unless $stderr.is_a?(RailsConsoleAi::PrefixedIO)
+
       @bot_user_id = slack_api("auth.test", token: @bot_token).dig("user_id")
       log_startup
 
@@ -78,9 +87,27 @@ module RailsConsoleAi
 
       puts "Connected to Slack Socket Mode."
 
-      # Main read loop
+      # Main read loop with keepalive
+      last_activity = Time.now
+      ping_sent = false
+
       loop do
+        ready = IO.select([ssl.to_io], nil, nil, PING_INTERVAL)
+
+        if ready.nil?
+          # Timeout — no data received
+          if ping_sent && (Time.now - last_activity) > PONG_TIMEOUT
+            puts "Slack connection timed out (no pong received). Reconnecting..."
+            break
+          end
+          send_ws_ping(ssl)
+          ping_sent = true
+          next
+        end
+
         data = read_ws_frame(ssl)
+        last_activity = Time.now
+        ping_sent = false
         next unless data
 
         begin
@@ -130,6 +157,11 @@ module RailsConsoleAi
         send_ws_pong(ssl, payload)
         return nil
       end
+      # Handle pong (opcode 10) — response to our keepalive ping
+      if opcode == 0xA
+        read_ws_payload(ssl) # consume payload
+        return nil
+      end
       # Close frame (opcode 8)
       return nil if opcode == 8
       # Only process text frames (opcode 1)
@@ -194,6 +226,14 @@ module RailsConsoleAi
       ssl.write(frame)
     end
 
+    def send_ws_ping(ssl)
+      mask_key = 4.times.map { rand(256) }
+      frame = [0x89].pack("C") # FIN + ping opcode
+      frame << [0x80].pack("C") # masked, zero-length payload
+      frame << mask_key.pack("C*")
+      ssl.write(frame)
+    end
+
     # --- Slack Web API (minimal, uses Net::HTTP) ---
 
     def slack_api(method, token: @bot_token, **params)
@@ -309,6 +349,7 @@ module RailsConsoleAi
 
       session[:thread] = Thread.new do
         Thread.current.report_on_exception = false
+        Thread.current[:log_prefix] = "[#{channel_id}/#{thread_ts}] @#{user_name}"
         begin
           channel.display_dim("_session: #{channel_id}/#{thread_ts}_")
           if restored
@@ -351,6 +392,7 @@ module RailsConsoleAi
       # Otherwise treat as a new message in the conversation
       session[:thread] = Thread.new do
         Thread.current.report_on_exception = false
+        Thread.current[:log_prefix] = channel.instance_variable_get(:@log_prefix)
         begin
           engine.process_message(text)
         rescue => e

data/lib/rails_console_ai/version.rb

@@ -1,3 +1,3 @@
 module RailsConsoleAi
-  VERSION = '0.15.0'.freeze
+  VERSION = '0.17.0'.freeze
 end

data/lib/rails_console_ai.rb

@@ -94,7 +94,8 @@ module RailsConsoleAi
       table = 'rails_console_ai_sessions'
 
       if conn.table_exists?(table)
-        $stdout.puts "\e[32mRailsConsoleAi: #{table} already exists. Run RailsConsoleAi.teardown! first to recreate.\e[0m"
+        $stdout.puts "\e[32mRailsConsoleAi: #{table} already exists — checking for pending migrations.\e[0m"
+        migrate!
       else
         conn.create_table(table) do |t|
           t.text    :query,         null: false
@@ -151,6 +152,7 @@ module RailsConsoleAi
       if migrations.empty?
         $stdout.puts "\e[32mRailsConsoleAi: #{table} is up to date.\e[0m"
       else
+        RailsConsoleAi::Session.reset_column_information if defined?(RailsConsoleAi::Session)
         $stdout.puts "\e[32mRailsConsoleAi: added columns: #{migrations.join(', ')}.\e[0m"
       end
     rescue => e

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rails_console_ai
 version: !ruby/object:Gem::Version
-  version: 0.15.0
+  version: 0.17.0
 platform: ruby
 authors:
 - Cortfr
@@ -109,6 +109,7 @@ files:
 - lib/rails_console_ai/conversation_engine.rb
 - lib/rails_console_ai/engine.rb
 - lib/rails_console_ai/executor.rb
+- lib/rails_console_ai/prefixed_io.rb
 - lib/rails_console_ai/providers/anthropic.rb
 - lib/rails_console_ai/providers/base.rb
 - lib/rails_console_ai/providers/bedrock.rb