language-operator 0.1.61 → 0.1.62

data/examples/ux_helpers_demo.rb

@@ -0,0 +1,296 @@
+#!/usr/bin/env ruby
+# frozen_string_literal: true
+
+# Demo script showcasing all UxHelper components
+#
+# Usage:
+#   ruby examples/ux_helpers_demo.rb
+
+$LOAD_PATH.unshift File.expand_path('../lib', __dir__)
+
+require 'language_operator/cli/helpers/ux_helper'
+
+# Demo class showing all UX helper features
+class UxDemo
+  include LanguageOperator::CLI::Helpers::UxHelper
+
+  def run
+    show_header
+    demo_colors
+    demo_spinner
+    demo_table
+    demo_box
+    demo_highlighted_box
+    demo_list_box
+    demo_prompt if ARGV.include?('--interactive')
+  end
+
+  private
+
+  def show_header
+    puts "\n"
+    puts pastel.bold.cyan('=' * 60)
+    puts pastel.bold.cyan('  UxHelper Demo - All Available Components')
+    puts pastel.bold.cyan('=' * 60)
+    puts "\n"
+  end
+
+  def demo_colors
+    puts pastel.bold('1. Colors & Styles')
+    puts pastel.dim('-' * 40)
+    puts "  #{pastel.green('✓')} Success message"
+    puts "  #{pastel.yellow('⚠')} Warning message"
+    puts "  #{pastel.red('✗')} Error message"
+    puts "  #{pastel.cyan('ℹ')} Info message"
+    puts "  #{pastel.bold('Bold text')}"
+    puts "  #{pastel.dim('Dimmed text')}"
+    puts "  #{pastel.red.bold('Combined: red + bold')}"
+    puts "\n"
+  end
+
+  def demo_spinner
+    puts pastel.bold('2. Spinners')
+    puts pastel.dim('-' * 40)
+
+    # Example 1: Success
+    spin = spinner('Loading configuration...')
+    spin.auto_spin
+    sleep 1.5
+    spin.success('Config loaded!')
+
+    # Example 2: Processing
+    spin = spinner('Processing data...')
+    spin.auto_spin
+    sleep 1
+    spin.success('Data processed!')
+
+    # Example 3: Error handling
+    spin = spinner('Connecting to cluster...')
+    spin.auto_spin
+    sleep 1
+    if rand > 0.5
+      spin.success('Connected!')
+    else
+      spin.error('Connection failed!')
+    end
+
+    puts "\n"
+  end
+
+  def demo_table
+    puts pastel.bold('3. Tables')
+    puts pastel.dim('-' * 40)
+
+    # Agent status table
+    agents = [
+      ['agent-web', pastel.green('running'), '2h 15m', '42.3 MB'],
+      ['agent-data', pastel.green('running'), '5h 02m', '128.7 MB'],
+      ['agent-sync', pastel.yellow('pending'), '0m', '-'],
+      ['agent-test', pastel.red('stopped'), '12m', '8.1 MB']
+    ]
+
+    puts table(%w[Name Status Uptime Memory], agents)
+    puts "\n"
+  end
+
+  def demo_box
+    puts pastel.bold('4. Boxes')
+    puts pastel.dim('-' * 40)
+    puts "\n"
+
+    # Simple box
+    puts box('Simple framed message')
+    puts "\n"
+
+    # Box with title
+    puts box(
+      'Agent deployed successfully to cluster!',
+      title: 'Success'
+    )
+    puts "\n"
+
+    # Warning box
+    puts box(
+      'This action cannot be undone.',
+      title: 'Warning',
+      border: :thick
+    )
+    puts "\n"
+
+    # Multi-line box
+    puts box(<<~MSG, title: 'Next Steps', padding: 2)
+      1. Monitor logs:    aictl agent logs my-agent
+      2. Check status:    aictl agent inspect my-agent
+      3. View metrics:    aictl agent metrics my-agent
+      4. Scale replicas:  aictl agent scale my-agent --replicas=3
+    MSG
+    puts "\n"
+  end
+
+  def demo_highlighted_box
+    puts pastel.bold('5. Highlighted Boxes')
+    puts pastel.dim('-' * 40)
+    puts "\n"
+
+    # Model details
+    highlighted_box(
+      title: 'LanguageModel Details',
+      rows: {
+        'Name' => 'gpt-4-turbo',
+        'Provider' => 'OpenAI',
+        'Model' => 'gpt-4-turbo-preview',
+        'Cluster' => 'production'
+      }
+    )
+    puts "\n"
+
+    # Agent configuration with nil values (skipped)
+    highlighted_box(
+      title: 'Agent Configuration',
+      rows: {
+        'Name' => 'web-scraper',
+        'Mode' => 'scheduled',
+        'Schedule' => '0 */6 * * *',
+        'Endpoint' => nil, # This will be skipped
+        'Replicas' => '3',
+        'Status' => pastel.green('Running')
+      }
+    )
+    puts "\n"
+
+    # Resource summary with custom character
+    highlighted_box(
+      title: 'Resource Summary',
+      rows: {
+        'Created' => '5 agents',
+        'Updated' => '2 models',
+        'Deleted' => '0 personas'
+      },
+      title_char: '▶'
+    )
+    puts "\n"
+
+    # Deployment status
+    highlighted_box(
+      title: 'Deployment Status',
+      rows: {
+        'Cluster' => 'production-us-west',
+        'Namespace' => 'language-operator',
+        'Available' => pastel.green('12/12'),
+        'Ready' => pastel.green('12/12'),
+        'Up-to-date' => pastel.green('12/12')
+      }
+    )
+    puts "\n"
+
+    # Custom colors
+    highlighted_box(
+      title: 'Success',
+      rows: {
+        'Operation' => 'Model created',
+        'Status' => 'Completed'
+      },
+      color: :green
+    )
+    puts "\n"
+
+    highlighted_box(
+      title: 'Error',
+      rows: {
+        'Code' => '500',
+        'Message' => 'Connection failed'
+      },
+      color: :red
+    )
+    puts "\n"
+  end
+
+  def demo_list_box
+    puts pastel.bold('6. List Boxes')
+    puts pastel.dim('-' * 40)
+    puts "\n"
+
+    # Simple list
+    list_box(
+      title: 'Models',
+      items: ['gpt-4-turbo', 'claude-3-opus', 'llama-3-70b']
+    )
+    puts "\n"
+
+    # Detailed list
+    list_box(
+      title: 'Agents',
+      items: [
+        { name: 'bash-agent', status: pastel.green('Running') },
+        { name: 'web-scraper', status: pastel.yellow('Pending') },
+        { name: 'data-processor', status: pastel.red('Stopped') }
+      ],
+      style: :detailed
+    )
+    puts "\n"
+
+    # Conditions style
+    list_box(
+      title: 'Conditions',
+      items: [
+        { type: 'Ready', status: 'True', message: 'Agent is ready' },
+        { type: 'Synthesized', status: 'True', message: 'Code synthesized successfully' },
+        { type: 'Validated', status: 'False', message: 'Validation pending' }
+      ],
+      style: :conditions
+    )
+    puts "\n"
+
+    # Key-value pairs
+    list_box(
+      title: 'Labels',
+      items: {
+        'app' => 'language-operator',
+        'env' => 'production',
+        'version' => 'v1.0.0'
+      },
+      style: :key_value
+    )
+    puts "\n"
+
+    # Empty list
+    list_box(
+      title: 'Personas',
+      items: [],
+      empty_message: 'No personas configured'
+    )
+    puts "\n"
+  end
+
+  def demo_prompt
+    puts pastel.bold('7. Interactive Prompts')
+    puts pastel.dim('-' * 40)
+
+    name = prompt.ask('What is your name?')
+    puts "  Hello, #{pastel.cyan(name)}!"
+
+    if prompt.yes?('Do you like the new UX helpers?')
+      puts "  #{pastel.green('Great!')} We're glad you like them."
+    else
+      puts "  #{pastel.yellow('Thanks for the feedback!')} We'll keep improving."
+    end
+
+    choice = prompt.select(
+      'Which helper is your favorite?',
+      %w[pastel prompt spinner table box highlighted_box list_box]
+    )
+    puts "  You selected: #{pastel.bold(choice)}"
+
+    puts "\n"
+  end
+end
+
+# Run the demo
+if __FILE__ == $PROGRAM_NAME
+  demo = UxDemo.new
+  demo.run
+
+  puts demo.pastel.bold.cyan('Demo complete!')
+  puts demo.pastel.dim("Run with --interactive for prompt examples: ruby #{__FILE__} --interactive")
+  puts "\n"
+end

data/lib/language_operator/agent/base.rb

@@ -38,7 +38,7 @@ module LanguageOperator
         logger.info "OpenTelemetry #{otel_enabled ? 'enabled' : 'disabled'}"
 
         @workspace_path = ENV.fetch('WORKSPACE_PATH', '/workspace')
-        @mode = ENV.fetch('AGENT_MODE', 'autonomous')
+        @mode = agent_mode_with_default
         @executor = nil
       end
 
@@ -134,6 +134,16 @@ module LanguageOperator
       rescue StandardError => e
         logger.warn("Failed to flush telemetry: #{e.message}")
       end
+
+      # Get AGENT_MODE with fallback to default, handling empty/whitespace values
+      #
+      # @return [String] The agent mode to use
+      def agent_mode_with_default
+        mode = ENV.fetch('AGENT_MODE', nil)
+        return 'autonomous' if mode.nil? || mode.strip.empty?
+
+        mode
+      end
     end
   end
 end

data/lib/language_operator/agent/executor.rb

@@ -56,6 +56,16 @@ module LanguageOperator
         execute(enriched_instruction)
       end
 
+      # Cleanup executor resources including MCP connections
+      #
+      # This method delegates to the agent's connection cleanup to prevent
+      # resource leaks when executors are no longer needed.
+      #
+      # @return [void]
+      def cleanup_connections
+        @agent.cleanup_connections if @agent.respond_to?(:cleanup_connections)
+      end
+
       # Execute a single task
       #
       # @param task [String] The task to execute
@@ -282,23 +292,30 @@ module LanguageOperator
 
       def parse_float_env(key)
         val = ENV.fetch(key, nil)
-        return nil unless val
+        return nil unless val && !val.strip.empty?
 
-        val.to_f
+        Float(val.strip)
+      rescue ArgumentError
+        logger.warn("Invalid float value for #{key}: #{val}. Ignoring.")
+        nil
       end
 
       def parse_int_env(key)
         val = ENV.fetch(key, nil)
-        return nil unless val
+        return nil unless val && !val.strip.empty?
 
-        val.to_i
+        Integer(val.strip)
+      rescue ArgumentError
+        logger.warn("Invalid integer value for #{key}: #{val}. Ignoring.")
+        nil
       end
 
       def parse_array_env(key)
         val = ENV.fetch(key, nil)
-        return nil unless val
+        return nil unless val && !val.strip.empty?
 
-        val.split(',').map(&:strip)
+        result = val.split(',').map(&:strip).reject(&:empty?)
+        result.empty? ? nil : result
       end
 
       def estimate_tokens(text)

data/lib/language_operator/agent/safety/safe_executor.rb

@@ -34,32 +34,26 @@ module LanguageOperator
           # Step 2: Execute in sandboxed context
           sandbox = SandboxProxy.new(@context, self)
 
-          # Step 3: Prepend safe constant definitions to the code
-          # This makes Ruby type constants available in the evaluated scope
-          safe_constants_code = <<~RUBY
-            Numeric = ::Numeric
-            Integer = ::Integer
-            Float = ::Float
-            String = ::String
-            Array = ::Array
-            Hash = ::Hash
-            TrueClass = ::TrueClass
-            FalseClass = ::FalseClass
-            Time = ::Time
-            Date = ::Date
-          RUBY
-
-          # Step 4: Execute using instance_eval with safe constants prepended
-          # Note: We still use instance_eval but with validated code
-          # and wrapped context
-          #
-          # The string interpolation below evaluates to:
-          #   sandbox.instance_eval("Numeric = ::Numeric\nInteger = ::Integer\nFloat = ::Float\n
-          #     String = ::String\nArray = ::Array\nHash = ::Hash\nTrueClass = ::TrueClass\n
-          #     FalseClass = ::FalseClass\nTime = ::Time\nDate = ::Date\n<user code>", __FILE__, __LINE__)
-          # rubocop:disable Style/DocumentDynamicEvalDefinition
-          sandbox.instance_eval("#{safe_constants_code}\n#{code}", __FILE__, __LINE__)
-          # rubocop:enable Style/DocumentDynamicEvalDefinition
+          # Step 3: Execute using instance_eval with smart constant injection
+          # Only inject constants that won't conflict with user-defined ones
+          safe_constants = %w[Numeric Integer Float String Array Hash TrueClass FalseClass Time Date]
+
+          # Find which constants user code defines to avoid redefinition warnings
+          user_defined_constants = safe_constants.select { |const| code.include?("#{const} =") }
+
+          # Only inject constants that user code doesn't define
+          constants_to_inject = safe_constants - user_defined_constants
+
+          if constants_to_inject.any?
+            # Inject only safe constants that won't conflict
+            safe_setup = constants_to_inject.map { |name| "#{name} = ::#{name}" }.join("\n")
+            # rubocop:disable Style/DocumentDynamicEvalDefinition
+            sandbox.instance_eval("#{safe_setup}\n#{code}", __FILE__, __LINE__)
+            # rubocop:enable Style/DocumentDynamicEvalDefinition
+          else
+            # User defines all constants, just run their code
+            sandbox.instance_eval(code, file_path, 1)
+          end
         rescue ASTValidator::SecurityError => e
           # Re-raise validation errors as executor errors for clarity
           raise SecurityError, "Code validation failed: #{e.message}"
@@ -119,20 +113,28 @@ module LanguageOperator
 
           # Provide access to safe constants from the context
           def const_missing(name)
-            # Allow access to HTTP and Shell helper classes
-            if name == :HTTP
-              return ::LanguageOperator::Dsl::HTTP
-            elsif name == :Shell
-              return ::LanguageOperator::Dsl::Shell
+            # Define allowed constants explicitly (security-by-default)
+            # This allowlist must be kept in sync with ASTValidator safe constants
+            case name
+            when :HTTP
+              ::LanguageOperator::Dsl::HTTP
+            when :Shell
+              ::LanguageOperator::Dsl::Shell
+            when :String, :Array, :Hash, :Integer, :Float, :Numeric, :Symbol
+              # Allow safe Ruby built-in types
+              ::Object.const_get(name)
+            when :Time, :Date
+              # Allow safe time/date types
+              ::Object.const_get(name)
+            when :TrueClass, :FalseClass, :NilClass
+              # Allow boolean and nil types
+              ::Object.const_get(name)
+            else
+              # Security-by-default: explicitly deny access to any other constants
+              # This prevents sandbox bypass through const_missing fallback
+              ::Kernel.raise ::LanguageOperator::Agent::Safety::SafeExecutor::SecurityError,
+                             "Access to constant '#{name}' is not allowed in sandbox (security restriction)"
             end
-
-            # Ruby type constants are now injected at eval time (see SafeExecutor#eval)
-            # but keep this as fallback for dynamic constant access
-
-            # Otherwise delegate to the context's module
-            @__context__.class.const_get(name)
-          rescue ::NameError
-            ::Kernel.raise ::NameError, "uninitialized constant #{name}"
           end
 
           private