aia 0.9.8 → 0.9.9

data/lib/aia/utility.rb

@@ -5,6 +5,15 @@ require 'word_wrapper'      # Pure ruby word wrapping
 module AIA
   class Utility
     class << self
+      def tools?
+        !AIA.config.tool_names.empty?
+      end
+
+      def supports_tools?
+        AIA.config.client.model.supports_functions?
+      end
+
+
       # Displays the AIA robot ASCII art
       # Yes, its slightly frivolous but it does contain some
       # useful configuration information.
@@ -13,20 +22,31 @@ module AIA
         spaces  = " "*indent
         width   = TTY::Screen.width - indent - 2
 
+        mcp_version = defined?(RubyLLM::MCP::VERSION) ? " MCP v" + RubyLLM::MCP::VERSION : ''
+
         puts <<-ROBOT
 
        ,      ,
        (\\____/) AI Assistant (v#{AIA::VERSION}) is Online
-        (_oo_)   #{AIA.config.model}
-         (O)       using #{AIA.config.adapter} (v#{RubyLLM::VERSION})
+        (_oo_)   #{AIA.config.model}#{supports_tools? ? ' (supports tools)' : ''}
+         (O)       using #{AIA.config.adapter} (v#{RubyLLM::VERSION}#{mcp_version})
        __||__    \\) model db was last refreshed on
      [/______\\]  /    #{AIA.config.last_refresh}
-    / \\__AI__/ \\/      #{AIA.config.tool_paths.empty? ? 'You can share my tools' : 'I will also use your tools'}
+    / \\__AI__/ \\/      #{tools? ? 'You can share my tools' : 'I did not bring any tools'}
    /    /__\\
-  (\\   /____\\   #{AIA.config.tool_paths.empty? ? '' : 'My Toolbox contains:'}
+  (\\   /____\\   #{tools? ? 'My Toolbox contains:' : ''}
         ROBOT
-        if AIA.config.tools
-          puts WordWrapper::MinimumRaggedness.new(width, AIA.config.tools).wrap.split("\n").map{|s| spaces+s+"\n"}.join
+        if tools?
+          tool_names = AIA.config.respond_to?(:tool_names) ? AIA.config.tool_names : AIA.config.tools
+          if tool_names && !tool_names.to_s.empty?
+            puts WordWrapper::MinimumRaggedness.new(
+                width,
+                tool_names.to_s # String of tool names, comma separated
+              ).wrap
+              .split("\n")
+              .map{|s| spaces+s+"\n"}
+              .join
+          end
         end
       end
     end

data/lib/aia.rb

@@ -6,6 +6,11 @@
 
 require 'ruby_llm'
 require 'ruby_llm/mcp'
+
+# FIXME: Actually fix ruby_llm; this is supposed to a temporary fix for the issues
+#        discovered with shared_tools/ruby_llm/mcp/github_mcp_server
+RubyLLM::MCP.support_complex_parameters!
+
 require 'prompt_manager'
 
 
@@ -25,7 +30,6 @@ require_relative 'refinements/string.rb'        # adds #include_any? #include_al
 require_relative 'aia/utility'
 require_relative 'aia/version'
 require_relative 'aia/config'
-require_relative 'aia/shell_command_executor'
 require_relative 'aia/prompt_handler'
 require_relative 'aia/ruby_llm_adapter'
 require_relative 'aia/directive_processor'

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: aia
 version: !ruby/object:Gem::Version
-  version: 0.9.8
+  version: 0.9.9
 platform: ruby
 authors:
 - Dewayne VanHoozer
@@ -85,14 +85,14 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.5.1
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 0.5.1
 - !ruby/object:Gem::Dependency
   name: reline
   requirement: !ruby/object:Gem::Requirement
@@ -290,12 +290,19 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 description: |
-  Unleash the full power of AI from your terminal! AIA is a cutting-edge CLI
-  assistant for generative AI workflows, offering dynamic prompt management,
-  seamless shell and Ruby integration, interactive chat, and advanced automation.
-  Effortlessly craft, manage, and execute prompts with embedded directives,
-  history, and flexible configuration. Experience next-level productivity for
-  developers, power users, and AI enthusiasts—all from your command line.
+  Unlock the Power of AI Right from Your Terminal!  AIA is a
+  revolutionary CLI console application designed for generative AI
+  workflows. With AIA, you can effortlessly manage prompts,
+  integrate seamlessly with shell and embedded Ruby (ERB), and
+  engage in interactive chats, all while harnessing advanced
+  automation features.  Experience a new level of productivity with
+  dynamic prompt management, tailored directives, and comprehensive
+  history tracking. AIA supports callback functions (Tools) and
+  model context protocol (MCP) servers, making it the ultimate tool
+  for developers, power users, and AI enthusiasts alike.  Transform
+  your command line into a powerhouse of creativity and efficiency.
+  Elevate your workflow with AIA and unleash the full potential of
+  AI at your fingertips!
 email:
 - dvanhoozer@gmail.com
 executables:
@@ -315,10 +322,23 @@ files:
 - _notes.txt
 - bin/aia
 - examples/README.md
+- examples/directives/ask.rb
 - examples/headlines
 - examples/tools/edit_file.rb
+- examples/tools/incomplete/calculator_tool.rb
+- examples/tools/incomplete/composite_analysis_tool.rb
+- examples/tools/incomplete/data_science_kit.rb
+- examples/tools/incomplete/database_query_tool.rb
+- examples/tools/incomplete/devops_toolkit.rb
+- examples/tools/incomplete/error_handling_tool.rb
+- examples/tools/incomplete/pdf_page_reader.rb
+- examples/tools/incomplete/secure_tool_template.rb
+- examples/tools/incomplete/weather_tool.rb
+- examples/tools/incomplete/workflow_manager_tool.rb
 - examples/tools/list_files.rb
-- examples/tools/pdf_page_reader.rb
+- examples/tools/mcp/README.md
+- examples/tools/mcp/github_mcp_server.rb
+- examples/tools/mcp/imcp.rb
 - examples/tools/read_file.rb
 - examples/tools/run_shell_command.rb
 - images/aia.png
@@ -336,7 +356,6 @@ files:
 - lib/aia/prompt_handler.rb
 - lib/aia/ruby_llm_adapter.rb
 - lib/aia/session.rb
-- lib/aia/shell_command_executor.rb
 - lib/aia/ui_presenter.rb
 - lib/aia/utility.rb
 - lib/aia/version.rb

data/lib/aia/shell_command_executor.rb

@@ -1,109 +0,0 @@
-# lib/aia/shell_command_executor.rb
-
-module AIA
-  class ShellCommandExecutor
-    DANGEROUS_PATTERNS = [
-      # File system destructive commands
-      /\brm\s+(-[a-z]*)?f/i,               # rm with force flag
-      /\bmkfs/i,                           # format filesystems
-      /\bdd\b.*\bof=/i,                    # dd with output file
-      /\bshred\b/i,                        # securely delete files
-      # System modification commands
-      /\bsystemctl\s+(stop|disable|mask)/i, # stopping system services
-      /\bchmod\s+777\b/i,                  # setting dangerous permissions
-      /\b(halt|poweroff|shutdown|reboot)\b/i, # system power commands
-      # Network security related
-      /\btcpdump\b/i,                      # packet capturing
-      /\bifconfig\b.*\bdown\b/i,           # taking down network interfaces
-      # Process control
-      /\bkill\s+-9\b/i,                    # force killing processes
-      /\bpkill\b/i                         # pattern-based process killing
-    ].freeze
-
-
-    MAX_COMMAND_LENGTH = 500
-
-
-
-    def initialize
-      # Stub method for future implementation
-    end
-
-    # Class-level
-
-
-
-
-
-    def self.execute_command(command)
-      new.execute_command(command)
-    end
-
-
-
-    def execute_command(command)
-      return "No command specified" if blank?(command)
-
-      validation_result = validate_command(command)
-      return validation_result if validation_result
-
-      `#{command}`.chomp
-    rescue StandardError => error
-      "Error executing shell command: #{error.message}"
-    end
-
-
-
-    def dangerous_command?(command)
-      return false if blank?(command)
-      DANGEROUS_PATTERNS.any? { |pattern| command =~ pattern }
-    end
-
-    private
-
-
-
-    def blank?(str)
-      str.nil? || str.strip.empty?
-    end
-
-
-
-    def validate_command(command)
-      command_length = command.length
-
-      if command_length > MAX_COMMAND_LENGTH
-        return "Error: Command too long (#{command_length} chars). Maximum length is #{MAX_COMMAND_LENGTH}."
-      end
-
-
-      is_dangerous = dangerous_command?(command)
-
-
-      if AIA.strict_shell_safety? && is_dangerous
-        return "Error: Potentially dangerous command blocked for security reasons: '#{command}'"
-      end
-
-      if AIA.shell_confirm? && is_dangerous
-        return prompt_confirmation(command)
-      end
-
-      nil # Command is valid
-    end
-
-
-
-
-
-
-
-    def prompt_confirmation(command)
-      puts "\n⚠️  WARNING: Potentially dangerous shell command detected:"
-      puts "\n    #{command}\n"
-      print "\nDo you want to execute this command? [y/N]: "
-      confirm = STDIN.gets.chomp.downcase
-      return "Command execution canceled by user" unless confirm == 'y'
-      nil
-    end
-  end
-end