rails_console_ai 0.21.0 → 0.22.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8faec693c3d3c83dafcaa65437081d5ba8e235f04527a28182104c4ecba5b80d
-  data.tar.gz: 171445a3ddc0c50093b1688260545df1558548a29cb81a6d3262d15d8ec4cf3c
+  metadata.gz: dba28b6d7543df66792877bddc51336f105c441a06b089de34ab0c7f89ca8f26
+  data.tar.gz: 4a54272fafd704abd003f06f2ab065d11f47b7c9a42b2f7c99c55b07b8712f8a
 SHA512:
-  metadata.gz: 1d0689ebfc82073deb58f8bd03fce39503541d63f042f531322bfc11ef0d6cf65f69fa9ae59c8b89a8a549ea9b4b7893e8505da83887ec4cbc5c1a064ffa5d99
-  data.tar.gz: e8f635c26035fcf73f2ccda5bd3d403a2ec33ef806c701c28f67ee8888a8189625a2569f45c369f7b29189fda3b162f03067be26590afbe5428be9da15487552
+  metadata.gz: 27c8413cc63c84c94d0fad464c3f3d6cd0b376a744122a2358b5899e0382f82e696a3e188647db952235986ee2a973c85b73065939a898cd7229955a61c377cf
+  data.tar.gz: d7e3041e7e37dd2ed304537bdfb02693647d3d32a9f1b6888718d9e38fcafe0c1b92000606c970716cd5074e7424b81e37d894d099b93ac6155268703ad54738

data/CHANGELOG.md

@@ -2,6 +2,13 @@
 
 All notable changes to this project will be documented in this file.
 
+## [0.22.0]
+
+- Fix blank content block handling in Bedrock provider
+- Fix `recall_output` to expand in place and restore preview after LLM responds, preventing context bloat
+- Remove safety guard bypass from prompt
+- Fix issue where LLM couldn't recall multiple outputs at once
+
 ## [0.21.0]
 
 - Add Slack @mention support and channel name tracking

data/lib/rails_console_ai/context_builder.rb

@@ -82,9 +82,6 @@ module RailsConsoleAi
           (e.g. `api = SalesforceApi.new(step1)`).
         - If the user asks you to provide code for them to run later (not execute now), put it
           in a ```ruby code block in your text response.
-        - Use `RailsConsoleAi.configuration.safety_guards.without_guards { }` to wrap any
-          operation that should bypass safety guards (e.g. calling a known-safe admin method).
-
         You have skills — predefined procedures for specific operations. When a user's request
         matches a skill, call activate_skill first to load the recipe and enable its guard
         bypasses, then follow the recipe.

data/lib/rails_console_ai/conversation_engine.rb

@@ -242,11 +242,16 @@ module RailsConsoleAi
       output_parts << "Return value: #{exec_result.inspect}" if exec_result
 
       result_str = output_parts.join("\n\n")
-      result_str = result_str[0..1000] + '...' if result_str.length > 1000
 
       context_msg = "User directly executed code: `#{raw_code}`"
-      context_msg += "\n#{result_str}" unless output_parts.empty?
-      output_id = output_parts.empty? ? nil : @executor.store_output(result_str)
+      if result_str.length > LARGE_OUTPUT_THRESHOLD
+        output_id = @executor.store_output(result_str)
+        preview = result_str[0, LARGE_OUTPUT_PREVIEW_CHARS]
+        context_msg += "\n#{preview}\n\n[Output truncated at #{LARGE_OUTPUT_PREVIEW_CHARS} of #{result_str.length} chars — use recall_output tool with id #{output_id} to retrieve the full output]"
+      elsif !output_parts.empty?
+        output_id = @executor.store_output(result_str)
+        context_msg += "\n#{result_str}"
+      end
       @history << { role: :user, content: context_msg, output_id: output_id }
 
       @interactive_query ||= "> #{raw_code}"
@@ -314,9 +319,15 @@ module RailsConsoleAi
           output_parts << "Return value: #{exec_result.inspect}" if exec_result
           unless output_parts.empty?
             result_str = output_parts.join("\n\n")
-            result_str = result_str[0..1000] + '...' if result_str.length > 1000
             output_id = @executor.store_output(result_str)
-            @history << { role: :user, content: "Code was executed (safety override). #{result_str}", output_id: output_id }
+            context_msg = "Code was executed (safety override). "
+            if result_str.length > LARGE_OUTPUT_THRESHOLD
+              context_msg += result_str[0, LARGE_OUTPUT_PREVIEW_CHARS]
+              context_msg += "\n\n[Output truncated at #{LARGE_OUTPUT_PREVIEW_CHARS} of #{result_str.length} chars — use recall_output tool with id #{output_id} to retrieve the full output]"
+            else
+              context_msg += result_str
+            end
+            @history << { role: :user, content: context_msg, output_id: output_id }
           end
           :success
         else
@@ -336,9 +347,15 @@ module RailsConsoleAi
 
         unless output_parts.empty?
           result_str = output_parts.join("\n\n")
-          result_str = result_str[0..1000] + '...' if result_str.length > 1000
           output_id = @executor.store_output(result_str)
-          @history << { role: :user, content: "Code was executed. #{result_str}", output_id: output_id }
+          context_msg = "Code was executed. "
+          if result_str.length > LARGE_OUTPUT_THRESHOLD
+            context_msg += result_str[0, LARGE_OUTPUT_PREVIEW_CHARS]
+            context_msg += "\n\n[Output truncated at #{LARGE_OUTPUT_PREVIEW_CHARS} of #{result_str.length} chars — use recall_output tool with id #{output_id} to retrieve the full output]"
+          else
+            context_msg += result_str
+          end
+          @history << { role: :user, content: context_msg, output_id: output_id }
         end
 
         :success
@@ -615,6 +632,8 @@ module RailsConsoleAi
 
           This session has safety guards that block side effects (database writes, HTTP mutations, etc.).
           If an operation is blocked, the user will be prompted to allow it or disable guards.
+          Do NOT attempt to bypass or work around safety guards in your code — just write the
+          operation normally and let the safety system handle it.
         PROMPT
       end
     end
@@ -803,6 +822,28 @@ module RailsConsoleAi
         last_tool_names = result.tool_calls.map { |tc| tc[:name] }
         result.tool_calls.each do |tc|
           break if @channel.cancelled?
+
+          # Intercept recall_output/recall_outputs: expand in place instead of adding large messages
+          if tc[:name] == 'recall_output' || tc[:name] == 'recall_outputs'
+            ids = if tc[:name] == 'recall_outputs'
+                    Array(tc[:arguments]['ids']).map(&:to_i)
+                  else
+                    [tc[:arguments]['id'].to_i]
+                  end
+            @channel.display_tool_call("#{tc[:name]}(#{ids.join(', ')})")
+            expanded = expand_outputs_in_place(messages, ids)
+            tool_result = if expanded.any?
+                            "Expanded #{expanded.length} output(s) in conversation. The full content is now visible in the original message(s) above."
+                          else
+                            "No matching outputs found with id(s) #{ids.join(', ')}."
+                          end
+            @channel.display_dim("     #{tool_result}")
+            tool_msg = provider.format_tool_result(tc[:id], tool_result)
+            messages << tool_msg
+            new_messages << tool_msg
+            next
+          end
+
           if tc[:name] == 'ask_user' || tc[:name] == 'execute_plan'
             # Display any pending LLM text before prompting the user
             if last_thinking
@@ -848,6 +889,10 @@ module RailsConsoleAi
         exhausted = true if round == max_rounds - 1
       end
 
+      # Re-truncate any outputs that were expanded for the LLM — the LLM has
+      # seen them and responded, so collapse back to save context on future calls.
+      re_truncate_expanded(messages)
+
       if exhausted
         $stdout.puts "\e[33m  Hit tool round limit (#{max_rounds}). Forcing final answer. Increase with: RailsConsoleAi.configure { |c| c.max_tool_rounds = 200 }\e[0m"
         messages << { role: :user, content: "You've used all available tool rounds. Please provide your best answer now based on what you've learned so far." }
@@ -1110,16 +1155,14 @@ module RailsConsoleAi
                                .select { |m, _| m[:output_id] }
                                .map { |_, i| i }
 
-      if output_indices.length <= RECENT_OUTPUTS_TO_KEEP
-        return messages.map { |m| m.except(:output_id) }
-      end
+      return messages if output_indices.length <= RECENT_OUTPUTS_TO_KEEP
 
       trim_indices = output_indices[0..-(RECENT_OUTPUTS_TO_KEEP + 1)]
       messages.each_with_index.map do |msg, i|
         if trim_indices.include?(i)
           trim_message(msg)
         else
-          msg.except(:output_id)
+          msg
         end
       end
     end
@@ -1135,12 +1178,52 @@ module RailsConsoleAi
             block
           end
         end
-        { role: msg[:role], content: trimmed_content }
+        msg.merge(content: trimmed_content)
       elsif msg[:role].to_s == 'tool'
-        msg.except(:output_id).merge(content: ref)
+        msg.merge(content: ref)
       else
         first_line = msg[:content].to_s.lines.first&.strip || msg[:content]
-        { role: msg[:role], content: "#{first_line}\n#{ref}" }
+        msg.merge(content: "#{first_line}\n#{ref}")
+      end
+    end
+
+    def expand_outputs_in_place(messages, ids)
+      expanded = []
+      messages.each do |msg|
+        next unless msg[:output_id] && ids.include?(msg[:output_id])
+        full_output = @executor.recall_output(msg[:output_id])
+        next unless full_output
+        # Save original content so re_truncate_expanded can restore it
+        msg[:pre_expand_content] = msg[:content]
+        # Replace content with full output (handle Anthropic, OpenAI, and user message formats)
+        if msg[:content].is_a?(Array)
+          msg[:content] = msg[:content].map do |block|
+            if block.is_a?(Hash) && block['type'] == 'tool_result'
+              block.merge('content' => full_output)
+            else
+              block
+            end
+          end
+        elsif msg[:role].to_s == 'tool'
+          msg[:content] = full_output
+        else
+          # User messages (e.g., direct execution) — preserve first line, replace rest
+          first_line = msg[:content].to_s.lines.first&.chomp || ''
+          msg[:content] = "#{first_line}\n#{full_output}"
+        end
+        msg[:expanded] = true
+        expanded << msg[:output_id]
+      end
+      expanded
+    end
+
+    # Restore messages that were temporarily expanded back to their original
+    # (preview/truncated) content. Called after the LLM has seen the expanded
+    # content and responded.
+    def re_truncate_expanded(messages)
+      messages.each do |msg|
+        next unless msg.delete(:expanded)
+        msg[:content] = msg.delete(:pre_expand_content)
       end
     end
 

data/lib/rails_console_ai/providers/bedrock.rb

@@ -121,6 +121,8 @@ module RailsConsoleAi
                     else
                       [{ text: msg[:content].to_s }]
                     end
+          # Bedrock rejects empty text blocks in content arrays
+          content.reject! { |block| block.is_a?(Hash) && block.key?(:text) && !block.key?(:tool_use) && !block.key?(:tool_result) && block[:text].to_s.empty? }
           { role: msg[:role].to_s, content: content }
         end
 

data/lib/rails_console_ai/tools/registry.rb

@@ -186,7 +186,7 @@ module RailsConsoleAi
         if @executor
           register(
             name: 'recall_output',
-            description: 'Retrieve a previous code execution output that was omitted from the conversation to save context. Use the output id shown in the "[Output omitted]" placeholder.',
+            description: 'Retrieve a previous code execution output that was omitted or truncated. The output will be expanded in place in the conversation. Use the output id shown in the "[Output omitted]" or "[Output truncated]" placeholder.',
             parameters: {
               'type' => 'object',
               'properties' => {
@@ -199,6 +199,19 @@ module RailsConsoleAi
               result || "No output found with id #{args['id']}"
             }
           )
+
+          register(
+            name: 'recall_outputs',
+            description: 'Retrieve multiple previous code execution outputs that were omitted from the conversation. Use the output ids shown in "[Output omitted]" or "[Output truncated]" placeholders.',
+            parameters: {
+              'type' => 'object',
+              'properties' => {
+                'ids' => { 'type' => 'array', 'items' => { 'type' => 'integer' }, 'description' => 'The output ids to retrieve' }
+              },
+              'required' => ['ids']
+            },
+            handler: ->(args) { "recall_outputs handled by conversation engine" }
+          )
         end
 
         unless @mode == :init

data/lib/rails_console_ai/version.rb

@@ -1,3 +1,3 @@
 module RailsConsoleAi
-  VERSION = '0.21.0'.freeze
+  VERSION = '0.22.0'.freeze
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: rails_console_ai
 version: !ruby/object:Gem::Version
-  version: 0.21.0
+  version: 0.22.0
 platform: ruby
 authors:
 - Cortfr