agent99 0.0.4 → 0.0.5

data/examples/README.md

@@ -1,5 +1,7 @@
 # Agent99 Framework Examples - 86 and the Chief
 
+FYI ... I am in the process of extracting this examples directory out into its own repository:  [MadBomber/agent99_examples](https://github.com/MadBomber/agent99_examples)
+
 This folder contains example implementations using the Agent99 framework. The framework provides a foundation for building AI agents that can communicate with each other through a message-based system.
 
 ## Files
@@ -90,6 +92,40 @@ Note: To use the example_agent.rb, first run the AgentWatcher, then copy example
 
 ## Usage
 
+There are two ways to run the Agent99 examples:
+
+### 🚀 Automated Demo Runner (Recommended)
+
+The easiest way to run examples is with the comprehensive demo runner:
+
+```bash
+./run_demo.rb --list                    # List all available scenarios
+./run_demo.rb                          # Run default 'basic' scenario  
+./run_demo.rb -s basic                  # Basic Maxwell/Chief interaction
+./run_demo.rb -s control                # Control agent demonstration  
+./run_demo.rb -s watcher                # Dynamic agent loading demo
+./run_demo.rb -s security               # Security demonstration (KAOS spy)
+./run_demo.rb -s all                    # Run multiple scenarios in sequence
+./run_demo.rb -v -s basic               # Verbose output
+./run_demo.rb --help                    # Show all options
+```
+
+The demo runner automatically:
+- ✅ Checks dependencies (RabbitMQ, boxes command, etc.)
+- 🏗️ Starts infrastructure (registry service, RabbitMQ if available)
+- 🎬 Orchestrates multiple agents with proper timing
+- 🧹 Handles cleanup on exit or interrupt
+- 📊 Provides progress feedback and duration estimates
+
+**Available Scenarios:**
+- **basic** (~10s): Maxwell Agent86 and Chief interaction
+- **control** (~15s): Control agent managing other agents  
+- **watcher** (~20s): Agent watcher dynamically loading new agents
+- **security** (~10s): KAOS spy demonstration (educational security example)
+- **all** (~60s): Run multiple scenarios in sequence
+
+### 📋 Manual Setup (Original Method)
+
 From the examples directory you will need to start three different processes.  You will want to keep them all in the forgound so it would be best to start them in different terminal windows.
 
 Start the sample registry first: `./registry.rb`
@@ -104,6 +140,20 @@ But first the Chief asks the registry for the UUIDs of all agents who can handle
 
 Run the chief a few times in a roll.  Some times the agent to whom the Chief issues his requests does not always respond the you would expect.
 
+### 🔧 Optional Dependencies
+
+For the best experience, install these optional dependencies:
+
+```bash
+# For message broker (recommended)
+brew install rabbitmq-server
+
+# For enhanced chief agent output
+brew install boxes
+```
+
+**Note:** The framework works without these dependencies using fallback implementations.
+
 ![Agent99 Framework Diagram](diagram.png)
 
 

data/examples/bad_agent.rb

@@ -0,0 +1,32 @@
+#!/usr/bin/env ruby
+# examples/bad_agent.rb
+#
+# This agent is meant to cause errors.
+
+require_relative '../lib/agent99'
+
+class BadAgent < Agent99::Base
+  # this information is made available when the agent
+  # registers with the central registry service.  It is
+  # made available during the discovery process.
+  #
+  def info
+    {
+      name:             self.class.to_s,
+      type:             :server,
+      kapabilities:     %w[ rubber_stamp yes_man example ],
+      # request_schema:   {}, # ExampleRequest.schema,
+      # response_schema:  {}, # Agent99::RESPONSE.schema
+      # control_schema:   {}, # Agent99::CONTROL.schema
+      # error_schema:     {}, # Agent99::ERROR.schema
+    }
+  end  
+
+  
+  def receive_request
+    logger.info "Example agent received request: #{payload}"
+    send_response(status: 'success')
+  end
+end
+
+BadAgent.new.run

data/examples/registry.rb

@@ -28,17 +28,9 @@ end
 post '/register' do
   request.body.rewind
   request_data  = JSON.parse(request.body.read, symbolize_names: true)
-  debug_me{[
-    :request_data
-  ]}
   agent_name    = request_data[:name]
   agent_uuid    = SecureRandom.uuid
   
-  debug_me{[
-    :agent_name,
-    :agent_uuid
-  ]}
-
   # Ensure capabilities are lowercase
   request_data[:capabilities].map!{|c| c.downcase}
   

data/examples/run_demo.rb

@@ -0,0 +1,433 @@
+#!/usr/bin/env ruby
+# examples/run_demo.rb
+#
+# Comprehensive demo script for Agent99 framework examples
+# This script orchestrates running multiple examples automatically
+#
+
+require 'optparse'
+require 'fileutils'
+require 'timeout'
+
+class Agent99Demo
+  EXAMPLES_DIR = File.dirname(__FILE__)
+  
+  # Available demo scenarios
+  SCENARIOS = {
+    'basic' => {
+      description: 'Basic Maxwell Agent86 and Chief interaction',
+      agents: ['maxwell_agent86.rb', 'chief_agent.rb'],
+      duration: 10
+    },
+    'control' => {
+      description: 'Control agent managing other agents',
+      agents: ['maxwell_agent86.rb', 'control.rb'],
+      duration: 15
+    },
+    'watcher' => {
+      description: 'Agent watcher dynamically loading new agents',
+      agents: ['agent_watcher.rb'],
+      duration: 20,
+      special: :watcher_demo
+    },
+    'security' => {
+      description: 'KAOS spy demonstration (security example)',
+      agents: ['maxwell_agent86.rb', 'kaos_spy.rb'],
+      duration: 10,
+      warning: 'This demonstrates a malicious agent for educational purposes'
+    },
+    'all' => {
+      description: 'Run multiple scenarios in sequence',
+      agents: [],
+      duration: 60,
+      special: :run_all_scenarios
+    }
+  }
+  
+  def initialize
+    @pids = []
+    @registry_pid = nil
+    @rabbitmq_pid = nil
+    @options = {
+      scenario: 'basic',
+      verbose: false,
+      no_cleanup: false,
+      list_only: false
+    }
+    setup_signal_handlers
+  end
+  
+  def run(args = ARGV)
+    parse_options(args)
+    
+    if @options[:list_only]
+      list_scenarios
+      return
+    end
+    
+    puts "🚀 Starting Agent99 Demo: #{@options[:scenario]}"
+    puts "📁 Working directory: #{EXAMPLES_DIR}"
+    puts
+    
+    scenario = SCENARIOS[@options[:scenario]]
+    unless scenario
+      puts "❌ Unknown scenario: #{@options[:scenario]}"
+      list_scenarios
+      exit 1
+    end
+    
+    if scenario[:warning]
+      puts "⚠️  WARNING: #{scenario[:warning]}"
+      puts "Continue? (y/N): "
+      response = gets.chomp.downcase
+      unless response == 'y' || response == 'yes'
+        puts "Demo cancelled."
+        exit 0
+      end
+      puts
+    end
+    
+    begin
+      check_dependencies
+      start_infrastructure
+      
+      case scenario[:special]
+      when :run_all_scenarios
+        run_all_scenarios
+      when :watcher_demo
+        run_watcher_demo
+      else
+        run_scenario(scenario)
+      end
+      
+    rescue Interrupt
+      puts "\n🛑 Demo interrupted by user"
+    rescue => e
+      puts "❌ Error running demo: #{e.message}"
+      puts e.backtrace.first(5) if @options[:verbose]
+    ensure
+      cleanup unless @options[:no_cleanup]
+    end
+  end
+  
+  private
+  
+  def parse_options(args)
+    OptionParser.new do |opts|
+      opts.banner = "Usage: #{$0} [options]"
+      opts.separator ""
+      opts.separator "Agent99 Framework Demo Runner"
+      opts.separator ""
+      opts.separator "Options:"
+      
+      opts.on("-s", "--scenario SCENARIO", "Demo scenario to run (#{SCENARIOS.keys.join(', ')})") do |s|
+        @options[:scenario] = s
+      end
+      
+      opts.on("-l", "--list", "List available scenarios") do
+        @options[:list_only] = true
+      end
+      
+      opts.on("-v", "--verbose", "Verbose output") do
+        @options[:verbose] = true
+      end
+      
+      opts.on("--no-cleanup", "Don't cleanup processes on exit (for debugging)") do
+        @options[:no_cleanup] = true
+      end
+      
+      opts.on("-h", "--help", "Show this help") do
+        puts opts
+        exit
+      end
+      
+      opts.separator ""
+      opts.separator "Examples:"
+      opts.separator "  #{$0} -s basic     # Run basic Maxwell/Chief demo"
+      opts.separator "  #{$0} -s security  # Run security demonstration"
+      opts.separator "  #{$0} -l           # List all available scenarios"
+    end.parse!(args)
+  end
+  
+  def list_scenarios
+    puts "Available demo scenarios:"
+    puts
+    SCENARIOS.each do |name, config|
+      puts "  #{name.ljust(12)} - #{config[:description]}"
+      puts "#{' ' * 17}Duration: ~#{config[:duration]}s"
+      puts "#{' ' * 17}Warning: #{config[:warning]}" if config[:warning]
+      puts
+    end
+  end
+  
+  def check_dependencies
+    puts "🔍 Checking dependencies..."
+    
+    # Check if Ruby scripts exist
+    required_files = %w[registry.rb]
+    scenario = SCENARIOS[@options[:scenario]]
+    required_files += scenario[:agents] if scenario[:agents]
+    
+    required_files.each do |file|
+      path = File.join(EXAMPLES_DIR, file)
+      unless File.exist?(path)
+        puts "❌ Missing required file: #{file}"
+        exit 1
+      end
+    end
+    
+    # Check for rabbitmq (optional warning)
+    begin
+      system("which rabbitmq-server > /dev/null 2>&1")
+      unless $?.success?
+        puts "⚠️  RabbitMQ not found. Install with: brew install rabbitmq-server"
+        puts "   Continuing anyway - agents will use fallback message client"
+      end
+    rescue
+      puts "⚠️  Could not check for RabbitMQ"
+    end
+    
+    # Check for boxes command (used by chief_agent)
+    begin
+      system("boxes --help > /dev/null 2>&1")
+      unless $?.success?
+        puts "ℹ️  'boxes' command not found. Install with: brew install boxes"
+        puts "   Chief agent output will be plain text"
+      end
+    rescue
+    end
+    
+    puts "✅ Dependencies check complete"
+    puts
+  end
+  
+  def start_infrastructure
+    puts "🏗️  Starting infrastructure..."
+    
+    Dir.chdir(EXAMPLES_DIR) do
+      # Start RabbitMQ in background (if available)
+      if system("which rabbitmq-server > /dev/null 2>&1")
+        puts "   Starting RabbitMQ server..."
+        @rabbitmq_pid = spawn("rabbitmq-server", out: "/dev/null", err: "/dev/null")
+        sleep 3  # Give RabbitMQ time to start
+      end
+      
+      # Start registry
+      puts "   Starting registry service on http://localhost:4567..."
+      @registry_pid = spawn("ruby registry.rb", out: @options[:verbose] ? $stdout : "/dev/null")
+      sleep 2  # Give registry time to start
+      
+      # Test registry connection
+      begin
+        require 'net/http'
+        response = Net::HTTP.get_response(URI('http://localhost:4567/healthcheck'))
+        if response.code == '200'
+          puts "✅ Registry service started successfully"
+        else
+          raise "Registry returned status #{response.code}"
+        end
+      rescue => e
+        puts "❌ Failed to connect to registry: #{e.message}"
+        exit 1
+      end
+    end
+    
+    puts
+  end
+  
+  def run_scenario(scenario)
+    puts "🎬 Running scenario: #{scenario[:description]}"
+    puts "   Duration: ~#{scenario[:duration]} seconds"
+    puts
+    
+    Dir.chdir(EXAMPLES_DIR) do
+      agent_pids = []
+      
+      # Start each agent
+      scenario[:agents].each_with_index do |agent, index|
+        puts "   Starting agent: #{agent}"
+        
+        if agent == 'chief_agent.rb'
+          # Chief agent runs once and exits, so we handle it specially
+          sleep 1  # Give other agents time to register
+          puts "   🎯 Running Chief Agent (one-shot)..."
+          system("ruby #{agent}")
+        else
+          # Regular agents run continuously
+          pid = spawn("ruby #{agent}", 
+                     out: @options[:verbose] ? $stdout : "/dev/null",
+                     err: @options[:verbose] ? $stderr : "/dev/null")
+          agent_pids << pid
+          @pids << pid
+          sleep 1  # Stagger startup
+        end
+      end
+      
+      unless scenario[:agents].include?('chief_agent.rb')
+        # For continuous scenarios, run for specified duration
+        puts "   ⏱️  Running for #{scenario[:duration]} seconds..."
+        puts "   Press Ctrl+C to stop early"
+        sleep scenario[:duration]
+      end
+    end
+    
+    puts "✅ Scenario completed"
+  end
+  
+  def run_watcher_demo
+    puts "🎬 Running Agent Watcher Demo"
+    puts "   This demo shows dynamic agent loading"
+    puts
+    
+    Dir.chdir(EXAMPLES_DIR) do
+      # Ensure agents directory exists
+      FileUtils.mkdir_p('agents')
+      
+      # Start the agent watcher
+      puts "   Starting Agent Watcher..."
+      watcher_pid = spawn("ruby agent_watcher.rb", 
+                         out: @options[:verbose] ? $stdout : "/dev/null")
+      @pids << watcher_pid
+      
+      sleep 3
+      
+      puts "   📂 Copying example_agent.rb to agents/ directory..."
+      FileUtils.cp('example_agent.rb', 'agents/example_agent_demo.rb')
+      
+      sleep 5
+      
+      puts "   📂 Adding second agent..."
+      # Create a simple second agent
+      File.write('agents/demo_agent2.rb', generate_demo_agent_code('DemoAgent2', ['demo', 'test2']))
+      
+      sleep 5
+      
+      puts "   📂 Adding third agent..."
+      File.write('agents/demo_agent3.rb', generate_demo_agent_code('DemoAgent3', ['demo', 'test3']))
+      
+      puts "   ⏱️  Letting agents run for 15 seconds..."
+      sleep 15
+      
+      # Cleanup created files
+      FileUtils.rm_rf('agents/example_agent_demo.rb') rescue nil
+      FileUtils.rm_rf('agents/demo_agent2.rb') rescue nil
+      FileUtils.rm_rf('agents/demo_agent3.rb') rescue nil
+    end
+    
+    puts "✅ Agent Watcher demo completed"
+  end
+  
+  def run_all_scenarios
+    puts "🎬 Running All Scenarios"
+    puts
+    
+    %w[basic control security].each do |scenario_name|
+      puts "=" * 60
+      scenario = SCENARIOS[scenario_name]
+      
+      if scenario[:warning]
+        puts "⚠️  Skipping #{scenario_name}: #{scenario[:warning]}"
+        puts "   Run with -s #{scenario_name} to run individually"
+        next
+      end
+      
+      run_scenario(scenario)
+      puts
+      
+      # Brief pause between scenarios
+      sleep 3
+      cleanup_agents
+    end
+    
+    puts "=" * 60
+    puts "✅ All scenarios completed"
+  end
+  
+  def generate_demo_agent_code(class_name, capabilities)
+    <<~RUBY
+      require_relative '../lib/agent99'
+      
+      class #{class_name} < Agent99::Base
+        def info
+          {
+            name: self.class.to_s,
+            type: :server,
+            capabilities: #{capabilities.inspect}
+          }
+        end
+        
+        private
+        
+        def receive_request
+          send_response({ result: "Hello from \#{self.class.name}!" })
+        end
+      end
+      
+      # Auto-start when loaded
+      agent = #{class_name}.new
+      agent.run
+    RUBY
+  end
+  
+  def setup_signal_handlers
+    %w[INT TERM QUIT].each do |signal|
+      Signal.trap(signal) do
+        puts "\n🛑 Received #{signal} signal, cleaning up..."
+        cleanup
+        exit 0
+      end
+    end
+  end
+  
+  def cleanup
+    puts "🧹 Cleaning up processes..."
+    cleanup_agents
+    cleanup_infrastructure
+    puts "✅ Cleanup complete"
+  end
+  
+  def cleanup_agents
+    @pids.each do |pid|
+      begin
+        Process.kill('TERM', pid)
+        Process.wait(pid, Process::WNOHANG)
+      rescue Errno::ESRCH, Errno::ECHILD
+        # Process already terminated
+      rescue => e
+        puts "⚠️  Error terminating process #{pid}: #{e.message}" if @options[:verbose]
+      end
+    end
+    @pids.clear
+  end
+  
+  def cleanup_infrastructure
+    if @registry_pid
+      begin
+        Process.kill('TERM', @registry_pid)
+        Process.wait(@registry_pid, Process::WNOHANG)
+      rescue Errno::ESRCH, Errno::ECHILD
+      rescue => e
+        puts "⚠️  Error terminating registry: #{e.message}" if @options[:verbose]
+      end
+    end
+    
+    if @rabbitmq_pid
+      begin
+        Process.kill('TERM', @rabbitmq_pid)
+        Process.wait(@rabbitmq_pid, Process::WNOHANG)
+      rescue Errno::ESRCH, Errno::ECHILD
+      rescue => e
+        puts "⚠️  Error terminating RabbitMQ: #{e.message}" if @options[:verbose]
+      end
+      
+      # Also try rabbitmqctl stop as backup
+      system("rabbitmqctl stop > /dev/null 2>&1") rescue nil
+    end
+  end
+end
+
+# Main execution
+if __FILE__ == $PROGRAM_NAME
+  demo = Agent99Demo.new
+  demo.run
+end

data/lib/agent99/amqp_message_client.rb

@@ -2,7 +2,7 @@
 
 require 'bunny'
 require 'json'
-require 'json_schema'
+require 'simple_json_schema_builder'
 require 'logger'
 
 class Agent99::AmqpMessageClient
@@ -34,10 +34,10 @@ class Agent99::AmqpMessageClient
       config: CONFIG,
       logger: Logger.new($stdout))
     @config     = config
+    @logger     = logger
     @connection = create_amqp_connection
     @channel    = @connection.create_channel
     @exchange   = @channel.default_exchange
-    @logger     = logger
   end
 
   def setup(agent_id:, logger:)

data/lib/agent99/base.rb

@@ -2,7 +2,7 @@
 
 require 'logger'
 require 'json'
-require 'json_schema'
+require 'simple_json_schema_builder'
 
 require_relative 'timestamp'
 require_relative 'registry_client'

data/lib/agent99/message_processing.rb

@@ -128,19 +128,13 @@ module Agent99::MessageProcessing
   # @return [Array] An array of validation errors, empty if validation succeeds
   #
   def validate_schema
-    return unless info[:request_schema]
-    
-    schema = JsonSchema.parse!(info[:request_schema])
-    schema.expand_references!
-    validator = JsonSchema::Validator.new(schema)
-    
-    validator.validate(@payload)
-    []
-  
-  rescue JsonSchema::ValidationError => e
+    # TODO: Implement proper JSON schema validation
+    # For now, skip validation to avoid JsonSchema dependency issues
+    return []
+  rescue => e
     handle_error("Validation error", e)
-    send_response(type: 'error', errors: e.messages)
-    e.messages
+    send_response(type: 'error', errors: [e.message])
+    [e.message]
   end
 
 

data/lib/agent99/registry_client.rb

@@ -13,7 +13,8 @@ class Agent99::RegistryClient
     )
     @base_url     = base_url
     @logger       = logger
-    @http_client  = Net::HTTP.new(URI.parse(base_url).host, URI.parse(base_url).port)
+    uri = URI.parse(base_url)
+    @http_client  = Net::HTTP.new(uri.host, uri.port)
   end
 
   def register(info:)
@@ -58,10 +59,12 @@ class Agent99::RegistryClient
   
   rescue JSON::ParserError => e
     logger.error "JSON parsing error: #{e.message}"
+    logger.debug "Response body that failed parsing: #{response&.body}"
     nil
   
   rescue StandardError => e
     logger.error "Request error: #{e.message}"
+    logger.debug "Error details: #{e.class} - #{e.backtrace&.first(5)}"
     nil
   end
 

data/lib/agent99/version.rb

@@ -1,7 +1,7 @@
 # frozen_string_literal: true
 
 module Agent99
-  VERSION = "0.0.4"
+  VERSION = "0.0.5"
 
   def self.version  = VERSION
 end

data/lib/agent99.rb

@@ -4,7 +4,7 @@ require 'debug_me'
 include DebugMe
 
 require 'json'
-require 'json_schema'
+require 'simple_json_schema_builder'
 require 'securerandom'
 
 module Agent99; end  # Establish a namespace