smart_message 0.0.10 → 0.0.12

data/examples/city_scenario/smart_message_ai_agent.rb

@@ -0,0 +1,364 @@
+#!/usr/bin/env ruby
+# smart_message_ai_agent.rb
+#
+# Abstract base class for building AI agents that can dynamically discover,
+# select, and use SmartMessage classes for communication based on context.
+#
+# This pattern enables AI agents to:
+# 1. Discover available message types through introspection
+# 2. Use LLM to select appropriate messages for scenarios
+# 3. Generate valid message property values using AI
+# 4. Handle validation errors with AI-assisted correction
+# 5. Publish messages through SmartMessage transport
+
+require 'ruby_llm'
+require 'json'
+require 'securerandom'
+
+class SmartMessageAIAgent
+  attr_reader :service_name, :llm, :ai_available, :logger
+
+  def initialize(service_name, logger = nil)
+    @service_name = service_name
+    @logger = logger || default_logger
+    @message_cache = {}
+    @retry_attempts = 3
+    
+    setup_ai
+    logger.info("#{self.class} initialized as '#{@service_name}' - AI-powered messaging agent ready")
+  end
+
+  # Override this method to configure RubyLLM with your API keys
+  def configure_rubyllm
+    RubyLLM.configure do |config|
+      config.anthropic_api_key = ENV.fetch('ANTHROPIC_API_KEY', nil)
+      config.openai_api_key = ENV.fetch('OPENAI_API_KEY', nil)
+      # Add other provider configurations as needed
+      
+      config.log_file = "log/#{@service_name}_llm.log"
+      config.log_level = :info
+    end
+  end
+
+  # Override this to specify which module contains your message classes
+  # Default: Messages
+  def message_module
+    Messages
+  end
+
+  # Override this to filter which message classes are available to the agent
+  # Return true to include the message class, false to exclude
+  def include_message_class?(message_class)
+    true
+  end
+
+  # Override this to provide scenario-specific context for message selection
+  def build_selection_context(scenario)
+    "An AI agent needs to send a message for this scenario: #{scenario[:description]}"
+  end
+
+  # Override this to provide scenario-specific context for property generation
+  def build_property_context(scenario, message_class)
+    "Generate realistic values for a #{message_class} message based on: #{scenario[:description]}"
+  end
+
+  # Override this to provide fallback values for specific message types
+  def generate_fallback_values(message_class, scenario = nil)
+    {
+      'timestamp' => Time.now.iso8601,
+      'from' => @service_name,
+      'details' => "Generated by #{@service_name} agent"
+    }
+  end
+
+  # Core method to send a message based on a scenario
+  def send_message_for_scenario(scenario)
+    logger.info("Processing scenario: #{scenario[:type]}")
+    
+    # Step 1: Discover available messages
+    available_messages = discover_message_types
+    
+    # Step 2: Select appropriate message type
+    selected_class = select_message_type(available_messages, scenario)
+    return nil unless selected_class
+    
+    # Step 3: Generate and publish message with retry logic
+    publish_with_retry(selected_class, scenario)
+  end
+
+  # Discover all available SmartMessage classes
+  def discover_message_types
+    return @message_cache if @message_cache.any?
+    
+    logger.info("Discovering message types from #{message_module}")
+    
+    message_module.constants.each do |const_name|
+      const = message_module.const_get(const_name)
+      
+      if const.is_a?(Class) && const < SmartMessage::Base && include_message_class?(const)
+        description = const.respond_to?(:description) ? const.description : "No description available"
+        @message_cache[const_name.to_s] = {
+          class: const,
+          description: description
+        }
+        logger.debug("Discovered: #{const_name} - #{description[0..100]}")
+      end
+    end
+    
+    logger.info("Discovered #{@message_cache.size} message types")
+    @message_cache
+  end
+
+  # Use AI to select the most appropriate message type
+  def select_message_type(available_messages, scenario)
+    logger.info("Selecting message type for scenario: #{scenario[:type]}")
+    
+    if @ai_available
+      selected = ai_select_message(available_messages, scenario)
+      return selected if selected
+    end
+    
+    # Fallback selection logic
+    fallback_select_message(available_messages, scenario)
+  end
+
+  # Generate message instance with AI-provided or fallback values
+  def generate_message_instance(message_class, scenario, validation_errors = [])
+    properties = collect_property_metadata(message_class)
+    
+    values = if @ai_available
+      ai_generate_properties(message_class, properties, scenario, validation_errors)
+    else
+      generate_fallback_values(message_class, scenario)
+    end
+    
+    # Ensure 'from' is set to our service name
+    values['from'] = @service_name
+    
+    # Create instance
+    kwargs = values.transform_keys(&:to_sym)
+    message_class.new(**kwargs)
+  rescue => e
+    logger.error("Failed to create message instance: #{e.message}")
+    raise
+  end
+
+  # Publish message with retry on validation errors
+  def publish_with_retry(message_class, scenario)
+    validation_errors = []
+    
+    @retry_attempts.times do |attempt|
+      begin
+        message = generate_message_instance(message_class, scenario, validation_errors)
+        message.publish
+        logger.info("Successfully published #{message_class} (attempt #{attempt + 1})")
+        return message
+      rescue => e
+        if validation_error = parse_validation_error(e.message)
+          validation_errors = build_validation_context(validation_error)
+          logger.warn("Validation error on attempt #{attempt + 1}: #{validation_error[:message]}")
+          next if attempt < @retry_attempts - 1
+        end
+        raise
+      end
+    end
+  end
+
+  private
+
+  def setup_ai
+    begin
+      configure_rubyllm
+      @llm = RubyLLM.chat
+      @ai_available = true
+      logger.info("AI model initialized successfully")
+    rescue => e
+      @ai_available = false
+      logger.warn("AI not available: #{e.message}. Using fallback logic.")
+    end
+  end
+
+  def default_logger
+    require 'logger'
+    Logger.new("log/#{@service_name}.log")
+  end
+
+  def ai_select_message(available_messages, scenario)
+    descriptions = available_messages.map { |name, info| "#{name}: #{info[:description]}" }.join("\n\n")
+    
+    prompt = <<~PROMPT
+      #{build_selection_context(scenario)}
+      
+      Available message types:
+      #{descriptions}
+      
+      Which message type is most appropriate? Respond with ONLY the class name.
+    PROMPT
+    
+    begin
+      response = @llm.ask(prompt)
+      class_name = response.content.strip
+      
+      if available_messages[class_name]
+        logger.info("AI selected: #{class_name}")
+        return available_messages[class_name][:class]
+      end
+    rescue => e
+      logger.error("AI selection failed: #{e.message}")
+    end
+    
+    nil
+  end
+
+  def fallback_select_message(available_messages, scenario)
+    # Simple keyword matching fallback
+    type_keyword = scenario[:type].to_s.downcase
+    
+    # Try exact match
+    match = available_messages.find { |name, _| name.downcase.include?(type_keyword) }
+    return match[1][:class] if match
+    
+    # Try partial match on description
+    match = available_messages.find { |_, info| info[:description].downcase.include?(type_keyword) }
+    return match[1][:class] if match
+    
+    # Return first available
+    first = available_messages.first
+    first[1][:class] if first
+  end
+
+  def collect_property_metadata(message_class)
+    return {} unless message_class.respond_to?(:property_descriptions)
+    
+    properties = {}
+    message_class.property_descriptions.each do |prop, desc|
+      enhanced_desc = desc.to_s
+      
+      # Add validation constraints if available
+      if message_class.respond_to?(:property_validations)
+        validation = message_class.property_validations[prop]
+        if validation && validation[:validation_message]
+          enhanced_desc += " (#{validation[:validation_message]})"
+        end
+      end
+      
+      # Check for valid value constants
+      const_name = "VALID_#{prop.to_s.upcase}S"
+      if message_class.const_defined?(const_name)
+        valid_values = message_class.const_get(const_name)
+        enhanced_desc += " Valid values: #{valid_values.join(', ')}"
+      end
+      
+      properties[prop] = enhanced_desc
+    end
+    
+    properties
+  end
+
+  def ai_generate_properties(message_class, properties, scenario, validation_errors)
+    properties_text = properties.map { |prop, desc| "#{prop}: #{desc}" }.join("\n")
+    
+    error_context = if validation_errors.any?
+      "\n\nPREVIOUS VALIDATION ERRORS:\n" + validation_errors.join("\n") +
+      "\n\nFix these specific properties with valid values."
+    else
+      ""
+    end
+    
+    prompt = <<~PROMPT
+      #{build_property_context(scenario, message_class)}
+      
+      Properties to fill:
+      #{properties_text}
+      
+      The 'from' field should be '#{@service_name}'.
+      #{error_context}
+      
+      Respond with a JSON object containing the property values.
+    PROMPT
+    
+    begin
+      response = @llm.ask(prompt)
+      JSON.parse(response.content)
+    rescue => e
+      logger.error("AI property generation failed: #{e.message}")
+      generate_fallback_values(message_class, scenario)
+    end
+  end
+
+  def parse_validation_error(error_message)
+    # Parse validation errors in various formats
+    if error_message =~ /Messages::(\w+)#(\w+):\s*(.+)/
+      return {
+        class_name: $1,
+        property: $2,
+        message: $3,
+        valid_values: extract_valid_values($3)
+      }
+    elsif error_message =~ /property\s+'(\w+)'\s+is\s+required/i
+      return {
+        property: $1,
+        message: error_message
+      }
+    elsif error_message.include?("ValidationError")
+      return {
+        property: "unknown",
+        message: error_message
+      }
+    end
+    
+    nil
+  end
+
+  def extract_valid_values(message)
+    if message =~ /must be(?:\s+one\s+of)?:\s*(.+)/i
+      $1.strip.split(/,\s*/).map(&:strip)
+    end
+  end
+
+  def build_validation_context(validation_error)
+    context = ["Property '#{validation_error[:property]}' has invalid value."]
+    context << "Error: #{validation_error[:message]}"
+    context << "Valid values: #{validation_error[:valid_values].join(', ')}" if validation_error[:valid_values]
+    context
+  end
+end
+
+# Example concrete implementation
+class ExampleAIAgent < SmartMessageAIAgent
+  def initialize
+    super('example_agent')
+  end
+
+  # Customize which messages this agent can use
+  def include_message_class?(message_class)
+    # Only include emergency-related messages
+    message_class.to_s.include?('Emergency') || message_class.to_s.include?('Alert')
+  end
+
+  # Provide domain-specific context
+  def build_selection_context(scenario)
+    "An emergency response agent observed: #{scenario[:description]}. Select the appropriate emergency message type."
+  end
+
+  def build_property_context(scenario, message_class)
+    "Generate emergency report values for #{message_class}. Incident: #{scenario[:description]}. Use high severity for serious incidents."
+  end
+
+  # Domain-specific fallback values
+  def generate_fallback_values(message_class, scenario = nil)
+    base_values = super
+    
+    # Add emergency-specific defaults
+    if message_class.to_s.include?('Emergency')
+      base_values.merge!({
+        'severity' => 'high',
+        'response_required' => true,
+        'location' => 'Unknown Location',
+        'reported_by' => @service_name
+      })
+    end
+    
+    base_values
+  end
+end

data/examples/city_scenario/start_demo.sh

@@ -0,0 +1,236 @@
+#!/bin/bash
+
+# SmartMessage City Demo Launcher for iTerm2
+# Creates a new iTerm2 window with separate tabs for each city service
+
+DEMO_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
+
+echo "Removing old log files..."
+rm -f $DEMO_DIR/log/*.log
+
+# Check if iTerm2 is available
+if ! ls /Applications/iTerm.app &>/dev/null; then
+    echo "Error: iTerm2 is not installed."
+    echo "Please install iTerm2 from https://iterm2.com/"
+    exit 1
+fi
+
+echo "Starting SmartMessage City Demo in iTerm2..."
+
+# Create the iTerm2 window and tabs using AppleScript
+osascript <<EOF
+tell application "iTerm2"
+    activate
+
+    -- Create new window
+    set newWindow to (create window with default profile)
+
+    -- Tab 1: Health Department (already created)
+    tell current session of current tab of newWindow
+        set name to "Health Department"
+        write text "cd '$DEMO_DIR'"
+        write text "clear"
+        write text "echo 'Starting Health Department...'"
+        write text "ruby health_department.rb; exit"
+    end tell
+
+    -- Tab 2: Police Department
+    tell newWindow
+        set newTab to (create tab with default profile)
+        tell current session of newTab
+            set name to "Police Department"
+            write text "cd '$DEMO_DIR'"
+            write text "clear"
+            write text "echo 'Starting Police Department...'"
+            write text "ruby police_department.rb; exit"
+        end tell
+    end tell
+
+    -- Tab 3: Fire Department
+    tell newWindow
+        set newTab to (create tab with default profile)
+        tell current session of newTab
+            set name to "Fire Department"
+            write text "cd '$DEMO_DIR'"
+            write text "clear"
+            write text "echo 'Starting Fire Department...'"
+            write text "ruby fire_department.rb; exit"
+        end tell
+    end tell
+
+    -- Tab 4: Local Bank
+    tell newWindow
+        set newTab to (create tab with default profile)
+        tell current session of newTab
+            set name to "Local Bank"
+            write text "cd '$DEMO_DIR'"
+            write text "clear"
+            write text "echo 'Starting Local Bank...'"
+            write text "ruby local_bank.rb; exit"
+        end tell
+    end tell
+
+    -- Tab 5: House #1
+    tell newWindow
+        set newTab to (create tab with default profile)
+        tell current session of newTab
+            set name to "House #1"
+            write text "cd '$DEMO_DIR'"
+            write text "clear"
+            write text "echo 'Starting House #1...'"
+            write text "ruby house.rb '456 Oak Street'; exit"
+        end tell
+    end tell
+
+    -- Tab 6: House #2
+    tell newWindow
+        set newTab to (create tab with default profile)
+        tell current session of newTab
+            set name to "House #2"
+            write text "cd '$DEMO_DIR'"
+            write text "clear"
+            write text "echo 'Starting House #2...'"
+            write text "ruby house.rb '789 Pine Lane'; exit"
+        end tell
+    end tell
+
+    -- Tab 7: Emergency Dispatch (911)
+    tell newWindow
+        set newTab to (create tab with default profile)
+        tell current session of newTab
+            set name to "911 Dispatch"
+            write text "cd '$DEMO_DIR'"
+            write text "clear"
+            write text "echo 'Starting Emergency Dispatch Center (911)...'"
+            write text "ruby emergency_dispatch_center.rb; exit"
+        end tell
+    end tell
+
+    -- Tab 8: Citizen Caller
+    tell newWindow
+        set newTab to (create tab with default profile)
+        tell current session of newTab
+            set name to "Citizen"
+            write text "cd '$DEMO_DIR'"
+            write text "clear"
+            write text "echo 'Starting Citizen 911 Caller...'"
+            write text "sleep 3; ruby citizen.rb auto; exit"
+        end tell
+    end tell
+
+    -- Tab 9: Redis Monitor
+    tell newWindow
+        set newTab to (create tab with default profile)
+        tell current session of newTab
+            set name to "Redis Monitor"
+            write text "cd '$DEMO_DIR'"
+            write text "clear"
+            write text "echo 'Starting Redis Message Monitor...'"
+            write text "ruby redis_monitor.rb; exit"
+        end tell
+    end tell
+
+    -- Tab 10: Redis Statistics
+    tell newWindow
+        set newTab to (create tab with default profile)
+        tell current session of newTab
+            set name to "Redis Statistics"
+            write text "cd '$DEMO_DIR'"
+            write text "clear"
+            write text "echo 'Starting Redis Statistics Dashboard...'"
+            write text "ruby redis_stats.rb; exit"
+        end tell
+    end tell
+
+    -- Tab 11: Control Panel
+    tell newWindow
+        set newTab to (create tab with default profile)
+        tell current session of newTab
+            set name to "Control Panel"
+            write text "cd '$DEMO_DIR'"
+            write text "clear"
+            write text "cat << 'CONTROL_PANEL_EOF'
+===== SmartMessage City Demo Control Panel =====
+
+CITY SERVICES:
+  Tab 1: Health Department (broadcasts health checks every 5s)
+  Tab 2: Police Department (responds to alarms & 911 calls)
+  Tab 3: Fire Department (responds to fires & 911 emergencies)
+  Tab 4: Local Bank (occasional silent alarms)
+  Tab 5: House #1 (456 Oak Street)
+  Tab 6: House #2 (789 Pine Lane)
+  Tab 7: 911 Dispatch (emergency call routing center)
+  Tab 8: Citizen (makes 911 calls automatically)
+  Tab 9: Redis Monitor (real-time message traffic)
+  Tab 10: Redis Statistics (performance dashboard)
+  Tab 11: Control Panel (this tab)
+
+CONTROLS:
+  ./stop_demo.sh     - Stop all city services
+  Cmd+W             - Close current tab
+  Cmd+1,2,3,4,5,6,7,8,9,0 - Switch to tab
+  Ctrl+C            - Stop service in current tab
+
+MESSAGE FLOWS:
+  Health Checks: Health Dept -> All Services
+  Health Status: All Services -> Health Dept (colored output)
+  Bank Alarms:   Bank -> Police (silent alarm system)
+  House Fires:   Houses -> Fire Dept (emergency response)
+  911 Calls:     Citizen -> 911 Dispatch -> Police/Fire
+  Emergencies:   All -> Health Dept (incident resolution)
+
+WHAT TO WATCH:
+  Tab 1: Health status with GREEN/YELLOW/ORANGE/RED colors
+  Tab 2: Police dispatching units to crimes & accidents
+  Tab 3: Fire trucks responding to fires, medical & rescue calls
+  Tab 4: Bank triggering occasional silent alarms
+  Tab 5/6: Houses occasionally catching fire
+  Tab 7: 911 dispatch routing emergency calls
+  Tab 8: Citizens calling 911 with various emergencies
+  Tab 9: Real-time Redis message traffic (color-coded)
+  Tab 10: Redis performance metrics & pub/sub statistics
+
+STATUS COLORS:
+  🟢 Green: healthy    🟡 Yellow: warning
+  🟠 Orange: critical  🔴 Red: failed
+
+Ready! Watch the city come alive with emergency services!
+========================================================
+CONTROL_PANEL_EOF"
+            write text "# Type 'exit' or use ./stop_demo.sh to close all tabs"
+            write text "bash"
+        end tell
+    end tell
+
+    -- Switch back to first tab
+    tell newWindow
+        select (first tab)
+    end tell
+
+end tell
+EOF
+
+if [ $? -eq 0 ]; then
+    echo ""
+    echo "✅ City Demo started successfully in iTerm2!"
+    echo ""
+    echo "🏥 Tab 1: Health Department - monitors all city services"
+    echo "🚔 Tab 2: Police Department - responds to alarms & 911 calls"
+    echo "🚒 Tab 3: Fire Department - responds to fires & 911 emergencies"
+    echo "🏦 Tab 4: Local Bank - triggers occasional alarms"
+    echo "🏠 Tab 5/6: Houses - occasionally catch fire"
+    echo "📞 Tab 7: 911 Dispatch - emergency call routing center"
+    echo "👤 Tab 8: Citizen - makes 911 calls automatically"
+    echo "🔍 Tab 9: Redis Monitor - real-time message traffic"
+    echo "📊 Tab 10: Redis Statistics - performance dashboard"
+    echo ""
+    echo "📱 Use Cmd+1,2,3,4,5,6,7,8,9,0 to switch between tabs"
+    echo "🛑 Run ./stop_demo.sh to stop all services"
+    echo ""
+    echo "🌟 Watch Tab 1 for colored health status updates!"
+    echo "📞 Watch Tab 7 for 911 dispatch routing & Tab 8 for citizen emergencies!"
+    echo "🔍 Check Tab 9 for real-time message traffic & Tab 10 for Redis stats!"
+else
+    echo "��� Failed to start demo. Please check that iTerm2 is installed and running."
+    exit 1
+fi

data/examples/city_scenario/stop_demo.sh

@@ -0,0 +1,106 @@
+#!/bin/bash
+
+# SmartMessage City Demo Shutdown Script for iTerm2
+# Stops all city services and closes the iTerm2 demo window
+
+echo "Stopping SmartMessage City Demo..."
+
+# Check if iTerm2 is running (process name is "iTerm" not "iTerm2")
+if ! pgrep -x "iTerm" > /dev/null; then
+    echo "iTerm2 is not running."
+    # Still continue to check for orphaned processes
+fi
+
+# Function to find and close demo window
+echo "Looking for SmartMessage city demo window in iTerm2..."
+
+WINDOW_CLOSED=$(osascript <<'EOF'
+tell application "iTerm2"
+    set windowFound to false
+    
+    repeat with theWindow in windows
+        tell theWindow
+            set tabNames to {}
+            repeat with theTab in tabs
+                set end of tabNames to (name of current session of theTab)
+            end repeat
+            
+            -- Check if this window has our city demo tabs
+            if "Health Department" is in tabNames or "Police Department" is in tabNames or "Fire Department" is in tabNames or "Local Bank" is in tabNames or "911 Dispatch" is in tabNames then
+                set windowFound to true
+                
+                -- Send Ctrl+C to stop programs and exit shells
+                repeat with theTab in tabs
+                    tell current session of theTab
+                        write text (character id 3) -- Ctrl+C
+                        delay 0.5
+                        write text "exit" -- Exit the shell to close cleanly
+                    end tell
+                end repeat
+                
+                delay 2
+                -- Force close the window
+                close theWindow
+                exit repeat
+            end if
+        end tell
+    end repeat
+    
+    return windowFound
+end tell
+EOF
+)
+
+if [ "$WINDOW_CLOSED" = "true" ]; then
+    echo "✅ City demo window found and closed successfully."
+else
+    echo "⚠️  No city demo window found. Checking for orphaned processes..."
+fi
+
+# Clean up any remaining city service processes
+echo "Checking for remaining city service processes..."
+
+# Include all demo programs in the cleanup
+ORPHANS=$(pgrep -f "(health_department|police_department|fire_department|local_bank|house|emergency_dispatch_center|citizen|redis_monitor|redis_stats)\.rb")
+
+if [ -n "$ORPHANS" ]; then
+    echo "Found orphaned city service processes. Cleaning up..."
+    echo "$ORPHANS" | while read pid; do
+        echo "  Stopping process $pid..."
+        kill -TERM "$pid" 2>/dev/null || true
+    done
+    
+    # Wait a moment for graceful termination
+    sleep 1
+    
+    # Force kill any remaining processes
+    REMAINING=$(pgrep -f "(health_department|police_department|fire_department|local_bank|house|emergency_dispatch_center|citizen|redis_monitor|redis_stats)\.rb")
+    if [ -n "$REMAINING" ]; then
+        echo "Force killing remaining processes..."
+        echo "$REMAINING" | while read pid; do
+            echo "  Force killing process $pid..."
+            kill -KILL "$pid" 2>/dev/null || true
+        done
+    fi
+    
+    echo "✅ Orphaned processes cleaned up."
+else
+    echo "✅ No orphaned processes found."
+fi
+
+# Clean up Redis channels and reset message counts
+echo "Cleaning up Redis channels..."
+redis-cli <<EOF > /dev/null 2>&1
+# Unsubscribe all clients from SmartMessage channels
+CLIENT LIST | grep -o 'id=[0-9]*' | cut -d= -f2 | xargs -I {} CLIENT KILL ID {}
+# Note: We can't actually "clear" channels in Redis, but killing clients removes subscriptions
+# The publish counts remain in Redis stats but that's historical data
+EOF
+
+echo "✅ Redis channels cleaned up."
+
+echo ""
+echo "🛑 SmartMessage city demo has been stopped."
+echo "   All emergency services are offline."
+echo ""
+echo "💡 To start the city demo again, run: ./start_demo.sh"