swarm_cli 3.0.0.alpha4 → 3.0.0.alpha5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: a4a46dc28301d9944d4780989720a455736cdd45f5f770c7ac1b3d0d44e4740e
-  data.tar.gz: daf3e6f7887eef22b502e74da0c527f19c10c99e4a99da7441367cc131ad62bc
+  metadata.gz: 6e35de7c5ee992cc18c100e24ea7448499422ea05b483fc5a422c5a8072af429
+  data.tar.gz: 920f46cc3531c44cc5aaa866d9c9daabe42abb77deec8eec2b34bca83a1f5816
 SHA512:
-  metadata.gz: 2a9f88d5825f7c5d2ac65d5c891d5f44c35902fb54ba1caf691c8a79a22ecccce5957a6c8e03144ab51077f34b02c440afd6f79ff596f27cdddef5a1c30aad10
-  data.tar.gz: c77b804edb2b3c2f348630424ad803d4d1da6d047f0717ba74d593f3989f0a0fe854407163feab417bf7816a96b37fafc852e23133bfc3789becd5e0653e9d17
+  metadata.gz: 9ac7c114df87419e179bfeb7681ff213b512b10f0de3c331768ea3237f4dffe2cee7a60c10a7556c4df3e660fabf20432b228b441ef3773a1b2306df311f1818
+  data.tar.gz: d3681393fd04b53ec9ab1f1858b2a36d6ee3e01af54facefe84c8c589213eaa68337afe8415684ad3c8517684bf554d61fbc13fddfe9ea06e4a967f0f0cd14ea

data/lib/swarm_cli/v3/cli.rb

@@ -16,6 +16,9 @@ module SwarmCLI
       HISTORY_SIZE = 1000
       CONFIG_FILE = File.expand_path("~/.config/swarm/cli.json")
 
+      DEFAULT_REFINE_PROMPT = "Review what you just did. Identify weaknesses, errors, " \
+        "or areas for improvement, then make the improvements."
+
       class << self
         # Parse arguments and run the appropriate mode.
         #
@@ -124,6 +127,10 @@ module SwarmCLI
             @options[:defrag] = true
           end
 
+          opts.on("--refine N", Integer, "Refine the response N additional times") do |n|
+            @options[:refine] = n
+          end
+
           opts.on("--reboot-from PID", Integer, "Internal: consume reboot signal") do |pid|
             @options[:reboot_from] = pid
           end
@@ -190,7 +197,13 @@ module SwarmCLI
 
       def run_prompt(agent)
         expanded_prompt = expand_file_mentions(@options[:prompt])
-        ask_with_display(agent, expanded_prompt)
+
+        if @options[:refine]
+          refine_with_display(agent, expanded_prompt, @options[:refine])
+        else
+          ask_with_display(agent, expanded_prompt)
+        end
+
         check_and_reboot!(agent)
       end
 
@@ -324,7 +337,7 @@ module SwarmCLI
         puts ANSIColors.dim("\u{1F9E0} Memory: #{mem_status}")
         puts ANSIColors.dim("Option+Enter for newline. Ctrl-C interrupts agent, double Ctrl-C quits.")
         puts ANSIColors.dim("Type while agent is working to steer. /queue <prompt> for follow-up tasks.")
-        puts ANSIColors.dim("Commands: /clear /memory /defrag /queue /reboot /exit")
+        puts ANSIColors.dim("Commands: /help /clear /memory /defrag /queue /refine /reboot /exit")
         puts
       end
 
@@ -370,6 +383,19 @@ module SwarmCLI
                 @defrag_active = false
               end
               next
+            when ->(v) { v.is_a?(Array) && v[0] == :refine }
+              # [:refine, count, custom_prompt] - run interactive refinement
+              refine_count = cmd_result[1]
+              refine_prompt = cmd_result[2]
+
+              if agent.running? || agent_active
+                display.agent_print(ANSIColors.yellow("Agent is busy. Wait for it to finish before refining."))
+              else
+                agent_active = true
+                ctx = { agent: agent, display: display, reader: reader, count: refine_count, custom_prompt: refine_prompt }
+                dispatch_refine(reactor, ctx) { agent_active = false }
+              end
+              next
             when Array
               # [:process, prompt] - command wants us to process this prompt
               line = cmd_result[1]
@@ -583,12 +609,133 @@ module SwarmCLI
         ask_with_display_sync(agent, merged_prompt, display)
       end
 
+      # ----------------------------------------------------------------
+      # Refinement
+      # ----------------------------------------------------------------
+
+      # Dispatch an interactive refine task asynchronously.
+      #
+      # @param reactor [Async::Reactor] the async reactor
+      # @param context [Hash] contains :agent, :display, :reader, :count, :custom_prompt
+      # @yield Called in ensure block to reset agent_active flag
+      # @return [void]
+      def dispatch_refine(reactor, context)
+        reactor.async do
+          refine_with_display_sync(context[:agent], context[:count], context[:custom_prompt], context[:display])
+          check_and_reboot_sync!(context[:agent], context[:display], context[:reader])
+        rescue StandardError => e
+          context[:display].agent_print(ANSIColors.red("Error: #{e.class}: #{e.message}"))
+        ensure
+          yield
+        end
+      end
+
+      # Run iterative refinement in non-interactive mode.
+      #
+      # Uses agent.loop() with max_iterations = refine_count + 1 (the first
+      # iteration IS the user's prompt, subsequent iterations use the default
+      # refinement prompt). Only the final response is printed.
+      #
+      # @param agent [SwarmSDK::V3::Agent] the agent
+      # @param prompt [String] the initial prompt
+      # @param refine_count [Integer] number of additional refinement iterations
+      # @return [void]
+      def refine_with_display(agent, prompt, refine_count)
+        result = agent.loop(
+          kickoff: prompt,
+          iterate: DEFAULT_REFINE_PROMPT,
+          max_iterations: refine_count + 1,
+          converge: false,
+        ) do |event|
+          case event[:type]
+          when "content_chunk"
+            # Suppress intermediate content — we'll print the final response
+            nil
+          else
+            line = EventRenderer.format(event)
+            puts line if line
+          end
+        end
+
+        return unless result&.final_response
+
+        puts
+        puts ANSIColors.cyan("#{agent.name}> ")
+        puts TTY::Markdown.parse(result.final_response.content.to_s)
+      end
+
+      # Run iterative refinement in interactive mode.
+      #
+      # Uses agent.loop() with the existing conversation context (STM).
+      # Each iteration streams content and renders markdown. The iterate
+      # prompt is either the custom text or the default refinement prompt.
+      #
+      # @param agent [SwarmSDK::V3::Agent] the agent
+      # @param count [Integer] number of refinement iterations
+      # @param custom_prompt [String, nil] custom refinement prompt (nil = default)
+      # @param display [Display] the display coordinator
+      # @return [void]
+      def refine_with_display_sync(agent, count, custom_prompt, display)
+        iterate_prompt = custom_prompt || DEFAULT_REFINE_PROMPT
+
+        display.start_working
+        full_content = +""
+        output_chars = 0
+
+        begin
+          agent.loop(
+            kickoff: iterate_prompt,
+            iterate: iterate_prompt,
+            max_iterations: count,
+            converge: false,
+          ) do |event|
+            case event[:type]
+            when "content_chunk"
+              full_content << event[:content] if event[:content]
+              output_chars += event[:content].length if event[:content]
+              display.update_activity("\u{2193} ~#{output_chars / 4} tokens")
+            when "loop_iteration_completed"
+              # Render accumulated content after each iteration
+              unless full_content.empty?
+                rendered = TTY::Markdown.parse(full_content)
+                display.agent_print("\n#{ANSIColors.cyan("#{agent.name}> ")}\n#{rendered}")
+                full_content.clear
+                output_chars = 0
+              end
+            end
+          end
+        ensure
+          display.stop_working
+        end
+
+        # Render any remaining content from the last iteration
+        return if full_content.empty?
+
+        rendered = TTY::Markdown.parse(full_content)
+        display.agent_print("\n#{ANSIColors.cyan("#{agent.name}> ")}\n#{rendered}")
+      end
+
       # ----------------------------------------------------------------
       # Slash commands (interactive mode)
       # ----------------------------------------------------------------
 
+      # Help text for all available slash commands.
+      COMMAND_HELP = [
+        ["/help", "Show this help message"],
+        ["/clear", "Clear conversation and queue (memory preserved)"],
+        ["/memory", "Show memory and token stats"],
+        ["/defrag", "Run memory defragmentation"],
+        ["/queue <prompt>", "Queue a follow-up prompt for after the current task"],
+        ["/refine [N] [prompt]", "Refine the last response N times (default: 3)"],
+        ["/reboot", "Restart the process to reload code changes"],
+        ["/exit, /quit", "Exit the chat"],
+      ].freeze
+
       def handle_slash_command(input, agent, display, reader, agent_active = false)
         case input
+        when "/help"
+          show_help(display)
+          true
         when "/clear"
           agent.clear
           @follow_up_queue&.clear
@@ -626,6 +773,10 @@ module SwarmCLI
         when "/queue"
           display.agent_print(ANSIColors.red("Usage: /queue <prompt>"))
           true
+        when %r{^/refine(?:\s+(\d+))?(?:\s+(.+))?$}
+          count = (Regexp.last_match(1) || 3).to_i
+          custom_prompt = Regexp.last_match(2)
+          [:refine, count, custom_prompt]
         when %r{^/([a-zA-Z0-9_-]+)(?:\s+(.+))?}
           # Check if this is a skill invocation (supports hyphens in names)
           skill_name = Regexp.last_match(1)
@@ -642,7 +793,7 @@ module SwarmCLI
           else
             # Unknown command
             display.agent_print(ANSIColors.red("Unknown command: #{input}"))
-            display.agent_print(ANSIColors.dim("Commands: /clear /memory /defrag /queue /reboot /exit"))
+            display.agent_print(ANSIColors.dim("Type /help for available commands."))
             true
           end
         else
@@ -650,6 +801,22 @@ module SwarmCLI
         end
       end
 
+      # Display available commands with descriptions.
+      #
+      # @param display [Display] the display coordinator
+      # @return [void]
+      def show_help(display)
+        max_cmd = COMMAND_HELP.map { |cmd, _| cmd.length }.max
+        lines = [ANSIColors.cyan("--- Available Commands ---")]
+        COMMAND_HELP.each do |cmd, desc|
+          lines << "  #{ANSIColors.bold(cmd.ljust(max_cmd))}  #{desc}"
+        end
+        lines << ""
+        lines << ANSIColors.dim("Option+Enter for newline. Ctrl-C interrupts agent, double Ctrl-C quits.")
+        lines << ANSIColors.dim("Type while agent is working to steer. Use @file to include file contents.")
+        display.agent_print(lines.join("\n"))
+      end
+
       def show_memory_stats_sync(agent, display)
         unless agent.definition.memory_enabled?
           display.agent_print(ANSIColors.yellow("Memory not enabled for this agent."))

data/lib/swarm_cli/v3/command_completer.rb

@@ -19,10 +19,12 @@ module SwarmCLI
     class CommandCompleter
       # Available slash commands
       COMMANDS = [
+        "/help",
         "/clear",
         "/memory",
         "/defrag",
         "/queue",
+        "/refine",
         "/reboot",
         "/exit",
         "/quit",

data/lib/swarm_cli/v3/event_renderer.rb

@@ -64,6 +64,14 @@ module SwarmCLI
             ANSIColors.cyan("  \u{1F9F9} Starting defrag (#{event[:total_cards]} cards, #{event[:total_clusters]} clusters)")
           when "memory_defrag_complete"
             ANSIColors.green("  \u{2705} Defrag complete in #{event[:elapsed_ms]}ms")
+          when "loop_started"
+            ANSIColors.cyan("  \u{1F504} Starting refinement (max #{event[:max_iterations]} iterations)")
+          when "loop_iteration_completed"
+            delta = event[:delta_score] ? " (similarity: #{(event[:delta_score] * 100).round(1)}%)" : ""
+            ANSIColors.dim("  \u{1F504} Iteration #{event[:iteration]} complete#{delta}")
+          when "loop_completed"
+            label = event[:converged] ? "converged" : "#{event[:iterations]} iterations"
+            ANSIColors.green("  \u{2705} Refinement complete (#{label})")
           end
         end
 

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: swarm_cli
 version: !ruby/object:Gem::Version
-  version: 3.0.0.alpha4
+  version: 3.0.0.alpha5
 platform: ruby
 authors:
 - Paulo Arruda