feat: rake task to test chat locally

lib/tasks/captain_chat.rake

@@ -0,0 +1,198 @@
+require 'io/console'
+require 'readline'
+
+namespace :captain do
+  desc 'Start interactive chat with Captain assistant - Usage: rake captain:chat[assistant_id] or rake captain:chat -- assistant_id'
+  task :chat, [:assistant_id] => :environment do |_, args|
+    assistant_id = args[:assistant_id] || ARGV[1]
+
+    unless assistant_id
+      puts '❌ Please provide an assistant ID'
+      puts 'Usage: rake captain:chat[assistant_id]'
+      puts "\nAvailable assistants:"
+      Captain::Assistant.includes(:account).each do |assistant|
+        puts "  ID: #{assistant.id} - #{assistant.name} (Account: #{assistant.account.name})"
+      end
+      exit 1
+    end
+
+    assistant = Captain::Assistant.find_by(id: assistant_id)
+    unless assistant
+      puts "❌ Assistant with ID #{assistant_id} not found"
+      exit 1
+    end
+
+    # Clear ARGV to prevent gets from reading files
+    ARGV.clear
+
+    chat_session = CaptainChatSession.new(assistant)
+    chat_session.start
+  end
+end
+
+class CaptainChatSession
+  def initialize(assistant)
+    @assistant = assistant
+    @message_history = []
+  end
+
+  def start
+    show_assistant_info
+    puts "💡 Type 'exit', 'quit', or 'bye' to end the session"
+    puts "💡 Type 'clear' to clear message history"
+    puts "#{'-' * 50}"
+
+    loop do
+      puts '' # Add spacing before prompt
+      user_input = Readline.readline('👤 You: ', true)
+      next unless user_input # Handle Ctrl+D
+
+      user_input = user_input.strip
+      case user_input.downcase
+      when 'exit', 'quit', 'bye'
+        break
+      when 'clear'
+        clear_history
+        next
+      when ''
+        next
+      end
+
+      process_user_message(user_input)
+    end
+
+    puts "\nChat session ended"
+    puts "Final conversation log has #{@message_history.length} messages"
+  end
+
+  private
+
+  def show_assistant_info
+    puts "🤖 Starting chat with #{@assistant.name}"
+    puts "🏢 Account: #{@assistant.account.name}"
+    puts "🆔 Assistant ID: #{@assistant.id}"
+
+    # Show scenarios
+    scenarios = @assistant.scenarios.enabled
+    if scenarios.any?
+      puts "⚡ Enabled Scenarios (#{scenarios.count}):"
+      scenarios.each do |scenario|
+        tools_count = scenario.tools&.length || 0
+        puts "   • #{scenario.title} (#{tools_count} tools)"
+        if scenario.description.present?
+          description = scenario.description.length > 60 ? "#{scenario.description[0..60]}..." : scenario.description
+          puts "     #{description}"
+        end
+      end
+    else
+      puts '⚡ No scenarios enabled'
+    end
+
+    # Show available tools
+    available_tools = Captain::Assistant.available_tool_ids
+    if available_tools.any?
+      puts "🔧 Available Tools (#{available_tools.count}): #{available_tools.join(', ')}"
+    else
+      puts '🔧 No tools available'
+    end
+
+    puts ''
+  end
+
+  def process_user_message(user_input)
+    # Add user message to history
+    add_to_history('user', user_input)
+
+    begin
+      print "🤖 #{@assistant.name}: "
+
+      # Track system messages to group them
+      @current_system_messages = []
+
+      # Define callbacks for detailed visibility
+      callbacks = {
+        on_agent_thinking: proc { |agent, input|
+          agent_name = extract_name(agent)
+          @current_system_messages << "#{agent_name} is thinking..."
+          add_to_history('system', "#{agent_name} is thinking...")
+        },
+        on_tool_start: proc { |tool, args|
+          tool_name = extract_tool_name(tool)
+          @current_system_messages << "Using tool: #{tool_name}"
+          add_to_history('system', "Using tool: #{tool_name}")
+        },
+        on_tool_complete: proc { |tool, result|
+          tool_name = extract_tool_name(tool)
+          @current_system_messages << "Tool #{tool_name} completed"
+          add_to_history('system', "Tool #{tool_name} completed")
+        },
+        on_agent_handoff: proc { |from, to, reason|
+          @current_system_messages << "Handoff: #{extract_name(from)} → #{extract_name(to)} (#{reason})"
+          add_to_history('system', "Agent handoff: #{extract_name(from)} → #{extract_name(to)} (#{reason})")
+        }
+      }
+
+      # Generate response using AgentRunnerService with callbacks
+      runner = Captain::Assistant::AgentRunnerService.new(assistant: @assistant, callbacks: callbacks)
+      result = runner.generate_response(message_history: @message_history)
+
+      response_text = result['response'] || 'No response generated'
+      reasoning = result['reasoning']
+
+      # Print grouped system messages with dimmer styling
+      puts dim_text("\n" + @current_system_messages.join("\n")) if @current_system_messages.any?
+
+      # Print the response
+      puts response_text
+
+      puts dim_italic_text("(Reasoning: #{reasoning})") if reasoning && reasoning != 'Processed by agent'
+
+      # Add assistant response to history
+      add_to_history('assistant', response_text, reasoning: reasoning)
+
+    rescue StandardError => e
+      error_msg = "Error: #{e.message}"
+      puts "❌ #{error_msg}"
+      add_to_history('system', error_msg)
+    end
+  end
+
+  def add_to_history(role, content, agent_name: nil, reasoning: nil)
+    message = {
+      role: role,
+      content: content,
+      timestamp: Time.current,
+      agent_name: agent_name || (role == 'assistant' ? @assistant.name : nil)
+    }
+    message[:reasoning] = reasoning if reasoning
+
+    @message_history << message
+  end
+
+  def clear_history
+    @message_history.clear
+    puts 'Message history cleared'
+  end
+
+  def dim_text(text)
+    # ANSI escape code for very dim gray text (bright black/dark gray)
+    "\e[90m#{text}\e[0m"
+  end
+
+  def dim_italic_text(text)
+    # ANSI escape codes for dim gray + italic text
+    "\e[90m\e[3m#{text}\e[0m"
+  end
+
+  def extract_tool_name(tool)
+    return tool if tool.is_a?(String)
+
+    tool.class.name.split('::').last.gsub('Tool', '')
+  rescue StandardError
+    tool.to_s
+  end
+
+  def extract_name(obj)
+    obj.respond_to?(:name) ? obj.name : obj.to_s
+  end
+end