aia 0.9.5 → 0.9.7

data/examples/headlines

@@ -1,21 +1,30 @@
-#!/usr/bin/env aia run --no-out_file
-# File: examples/headlines
+#!/usr/bin/env aia run --no-out_file --exec
+# NOTE: the --exec option is REQUIRED to run this executable prompt file
+#       This option signals that the contents of this file are to be appended
+#       to the contents of the given prompt ID file.  In this case it is the
+#       "run" prompt ID's text file.
+#
+#       All other AIA options are, well, optional.  The --no-out_file is
+#       used here to cause the response to this executable prompt to be
+#       sent to STDOUT like a good little *nix CLI tool.  Its not necessary.
+#       if you do not use it, the output will go to the default out_file.
+#       You could also specify a specific file to write the output to built
+#       but it is more convential to use the *nix STDOUT redirect "> output.md"
+#
 # Desc: retrieves the news.google.com website
 #       extracts and formats the headlines
-#       and prints them to the console
+#       and prints them to STDOUT
+#
+# Method:
+#   There are several ways you can accomplish this task using
+#   the shell or ERB integration.  For this example lets use
+#   a built-in directive.
 
-//config shell = true
-# //config
+Extract and summarize the headlines from the following markdown:
 
-# Puts the webpage into index.html
-# //shell wget2 https://news.google.com
+//webpage https://news.google.com
 
-$(wget2 https://news.google.com)
+__END__
 
-
-# Lets hear the headlines as well as read them.
-//config speak = true
-
-Extract and summarize the headlines from the following text:
-
-$(html2text index.html)
+The //webpage directive makes use of the website https://pure.md
+to convert any given URL to a markdown file.

data/examples/tools/run_shell_command.rb

@@ -1,5 +1,6 @@
 # experiments/ai_misc/coding_agent_with_ruby_llm/tools/run_shell_command.rb
 
+require "io/console"
 require "ruby_llm/tool"
 
 module Tools
@@ -8,10 +9,21 @@ module Tools
     param :command, desc: "The command to execute"
 
     def execute(command:)
-      puts "AI wants to execute the following shell command: '#{command}'"
-      print "Do you want to execute it? (y/n) "
-      response = gets.chomp
-      return { error: "User declined to execute the command" } unless response == "y"
+      print "\n\n"
+      puts "AI wants to execute the following shell command:"
+      puts "="*command.size
+      puts command
+      puts "="*command.size
+      print "\n\n"
+
+      sleep 0.5
+      print "Execute the command? (y/N):"
+      allowed = STDIN.getch == "y"
+
+      unless allowed
+        print "Command aborted" + " "*30 if defined?(AIA)
+        return { error: "User declined to execute the command" }
+      end
 
       `#{command}`
     rescue => e

data/lib/aia/chat_processor_service.rb

@@ -62,7 +62,7 @@ module AIA
 
       # Only output to STDOUT if we're in chat mode
 
-      if AIA.chat? || 'STDOUT' == AIA.config.out_file.upcase
+      if AIA.chat? || AIA.config.out_file.nil? || 'STDOUT' == AIA.config.out_file.upcase
         print "\nAI:\n  "
         puts response
       else
@@ -94,21 +94,22 @@ module AIA
 
     def determine_operation_type
       mode = AIA.config.client.model.modalities
-      if mode.supports?(:text_to_image)
+
+      if mode.text_to_image?
         :text_to_image
-      elsif mode.supports?(:image_to_text)
+      elsif mode.image_to_text?
         :image_to_text
-      elsif mode.supports?(:audio_to_text)
+      elsif mode.audio_to_text?
         :audio_to_text
-      elsif mode.supports?(:text_to_audio)
+      elsif mode.text_to_audio?
         :text_to_audio
-      elsif mode.supports?(:audio_to_audio)
+      elsif mode.audio_to_audio?
         :audio_to_audio
-      elsif mode.supports?(:image_to_image)
+      elsif mode.image_to_image?
         :image_to_image
-      elsif mode.supports?(:audio_to_image)
+      elsif mode.audio_to_image?
         :audio_to_image
-      elsif mode.supports?(:image_to_audio)
+      elsif mode.image_to_audio?
         :image_to_audio
       else
         :text_to_text

data/lib/aia/config.rb

@@ -124,6 +124,17 @@ module AIA
       remaining_args = config.remaining_args.dup
       config.remaining_args = nil
 
+      # Check for STDIN content
+      stdin_content = nil
+      if !STDIN.tty? && !STDIN.closed?
+        begin
+          stdin_content = STDIN.read
+          STDIN.reopen('/dev/tty')  # Reopen STDIN for interactive use
+        rescue => _
+          # If we can't reopen, continue without error
+        end
+      end
+
       # Is first remaining argument a prompt ID?
       unless remaining_args.empty?
         maybe_id      = remaining_args.first
@@ -134,6 +145,11 @@ module AIA
         end
       end
 
+      # Store STDIN content for later processing in session.rb
+      if stdin_content && !stdin_content.strip.empty?
+        config.stdin_content = stdin_content
+      end
+
       unless remaining_args.empty?
         bad_files = remaining_args.reject { |filename| AIA.good_file?(filename) }
         if bad_files.any?
@@ -141,9 +157,21 @@ module AIA
           exit 1
         end
 
-        config.context_files = remaining_args
+        config.context_files ||= []
+        config.context_files += remaining_args
+      end
+
+      # Check if the last context file is an executable prompt
+      if  config.executable_prompt     &&
+          config.context_files         &&
+          !config.context_files.empty?
+        config.executable_prompt_file = config.context_files.pop
       end
 
+      # TODO: Consider that if there is no prompt ID but there is an executable prompt
+      #       then maybe that is all that is needed.
+
+
       if config.prompt_id.nil? && !config.chat && !config.fuzzy
         STDERR.puts "Error: A prompt ID is required unless using --chat, --fuzzy, or providing context files. Use -h or --help for help."
         exit 1
@@ -325,10 +353,68 @@ module AIA
             end
           end
 
+          opts.on('--available_models [QUERY]', 'List (then exit) available models that match the optional query - a comma separated list of AND components like: openai,mini') do |query|
+
+            # SMELL: mostly duplications the code in the vailable_models directive
+            #        assumes that the adapter is for the ruby_llm gem
+            #        should this be moved to the Utilities class as a common method?
+
+            if query.nil?
+              query = []
+            else
+              query = query.split(',')
+            end
+
+            header    = "\nAvailable LLMs"
+            header   += " for #{query.join(' and ')}" if query
+
+            puts header + ':'
+            puts
+
+            q1 = query.select{|q| q.include?('_to_')}.map{|q| ':'==q[0] ? q[1...] : q}
+            q2 = query.reject{|q| q.include?('_to_')}
+
+
+            # query   = nil
+            counter = 0
+
+            RubyLLM.models.all.each do |llm|
+              inputs  = llm.modalities.input.join(',')
+              outputs = llm.modalities.output.join(',')
+              entry   = "- #{llm.id} (#{llm.provider}) #{inputs} to #{outputs}"
+
+              if query.nil? || query.empty?
+                counter += 1
+                puts entry
+                next
+              end
+
+              show_it = true
+              q1.each{|q| show_it &&= llm.modalities.send("#{q}?")}
+              q2.each{|q| show_it &&= entry.include?(q)}
+
+              if show_it
+                counter += 1
+                puts entry
+              end
+            end
+
+            puts if counter > 0
+            puts "#{counter} LLMs matching your query"
+            puts
+
+            exit
+          end
+
           opts.on("-m MODEL", "--model MODEL", "Name of the LLM model to use") do |model|
             config.model = model
           end
 
+          opts.on("-x", "--[no-]exec", "Used to designate an executable prompt file") do |value|
+            config.executable_prompt = value
+          end
+
+
           opts.on("--terse", "Adds a special instruction to the prompt asking the AI to keep responses short and to the point") do
             config.terse = true
           end
@@ -383,7 +469,13 @@ module AIA
           end
 
           opts.on("-o", "--[no-]out_file [FILE]", "Output file (default: temp.md)") do |file|
-            config.out_file = file ? File.expand_path(file, Dir.pwd) : 'temp.md'
+            if file == false  # --no-out_file was used
+              config.out_file = nil
+            elsif file.nil?   # No argument provided
+              config.out_file = 'temp.md'
+            else              # File name provided
+              config.out_file = File.expand_path(file, Dir.pwd)
+            end
           end
 
           opts.on("-a", "--[no-]append", "Append to output file instead of overwriting") do |append|
@@ -422,8 +514,8 @@ module AIA
             config.debug = $DEBUG_ME = false
           end
 
-          opts.on("-v", "--verbose", "Be verbose") do
-            config.verbose = true
+          opts.on("-v", "--[no-]verbose", "Be verbose") do |value|
+            config.verbose = value
           end
 
           opts.on("--speak", "Simple implementation. Uses the speech model to convert text to audio, then plays the audio. Fun with --chat. Supports configuration of speech model and voice.") do

data/lib/aia/directive_processor.rb

@@ -1,6 +1,8 @@
 # lib/aia/directive_processor.rb
 
+require 'active_support/all'
 require 'faraday'
+require 'word_wrapper'      # Pure ruby word wrapping
 
 module AIA
   class DirectiveProcessor
@@ -145,7 +147,7 @@ module AIA
     end
 
     desc "Specify the next prompt ID to process after this one"
-    def next(args = [])
+    def next(args = [], context_manager=nil)
       if args.empty?
         ap AIA.config.next
       else
@@ -154,8 +156,33 @@ module AIA
       ''
     end
 
+    desc "Show a list of tools and their description"
+    def tools(args = [], context_manager=nil)
+      indent  = 4
+      spaces  = " "*indent
+      width   = TTY::Screen.width - indent - 2
+
+      if !AIA.config.tools.empty?
+        puts
+        puts "Available Tools"
+        puts "==============="
+
+        AIA.config.tools.split(',').map(&:strip).each do |tool|
+          klass = tool.constantize
+          puts "\n#{klass.name}"
+          puts "-"*klass.name.size
+          puts WordWrapper::MinimumRaggedness.new(width, klass.description).wrap.split("\n").map{|s| spaces+s+"\n"}.join
+        end
+      else
+        puts "No tools configured"
+      end
+      puts
+
+      ''
+    end
+
     desc "Specify a sequence pf prompt IDs to process after this one"
-    def pipeline(args = [])
+    def pipeline(args = [], context_manager=nil)
       if args.empty?
         ap AIA.config.pipeline
       else
@@ -299,16 +326,21 @@ module AIA
     desc "All Available models or query on [partial LLM or provider name] Examples: //llms ; //llms openai ; //llms claude"
     def available_models( args=nil, context_manager=nil)
       query     = args
-      header    = "Available LLMs"
 
-      if query
-        header += " for #{query.join(' and ')}"
+      if 1 == query.size
+        query = query.first.split(',')
       end
 
+      header    = "\nAvailable LLMs"
+      header   += " for #{query.join(' and ')}" if query
+
       puts header + ':'
+      puts
 
-      q1 = query.select{|q| !q.start_with?(':')}
-      q2 = query.select{|q|  q.start_with?(':')}
+      q1 = query.select{|q| q.include?('_to_')}.map{|q| ':'==q[0] ? q[1...] : q}
+      q2 = query.reject{|q| q.include?('_to_')}
+
+      counter = 0
 
       RubyLLM.models.all.each do |llm|
         inputs  = llm.modalities.input.join(',')
@@ -316,17 +348,25 @@ module AIA
         entry   = "- #{llm.id} (#{llm.provider}) #{inputs} to #{outputs}"
 
         if query.nil? || query.empty?
+          counter += 1
           puts entry
           next
         end
 
         show_it = true
-        q1.each{|q| show_it &&= entry.include?(q)}
-        q2.each{|q| show_it &&= llm.modalities.supports?(q)}
+        q1.each{|q| show_it &&= llm.modalities.send("#{q}?")}
+        q2.each{|q| show_it &&= entry.include?(q)}
 
-        puts entry if show_it
+        if show_it
+          counter += 1
+          puts entry
+        end
       end
 
+      puts if counter > 0
+      puts "#{counter} LLMs matching your query"
+      puts
+
       ""
     end
     alias_method :am,         :available_models

data/lib/aia/ruby_llm_adapter.rb

@@ -74,6 +74,8 @@ module AIA
         exit 1
       end
 
+      return unless @chat.model.supports_functions?
+
       if !AIA.config.tool_paths.empty? && !@chat.model.supports?(:function_calling)
         STDERR.puts "ERROR: The model #{@model} does not support tools"
         exit 1
@@ -96,17 +98,17 @@ module AIA
       modes = @chat.model.modalities
 
       # TODO: Need to consider how to handle multi-mode models
-      if modes.supports? :text_to_text
+      if modes.text_to_text?
         text_to_text(prompt)
 
-      elsif modes.supports? :image_to_text
+      elsif modes.image_to_text?
         image_to_text(prompt)
-      elsif modes.supports? :text_to_image
+      elsif modes.text_to_image?
         text_to_image(prompt)
 
-      elsif modes.supports? :text_to_audio
+      elsif modes.text_to_audio?
         text_to_audio(prompt)
-      elsif modes.supports? :audio_to_text
+      elsif modes.audio_to_text?
         audio_to_text(prompt)
 
       else

data/lib/aia/session.rb

@@ -41,7 +41,7 @@ module AIA
       @directive_processor = DirectiveProcessor.new
       @chat_processor      = ChatProcessorService.new(@ui_presenter, @directive_processor)
 
-      if AIA.config.out_file && !AIA.append? && File.exist?(AIA.config.out_file)
+      if AIA.config.out_file && !AIA.config.out_file.nil? && !AIA.append? && File.exist?(AIA.config.out_file)
         File.open(AIA.config.out_file, 'w') {} # Truncate the file
       end
     end
@@ -111,8 +111,17 @@ module AIA
         prompt.text << TERSE_PROMPT
       end
 
-      prompt.save
-      # Substitute variables and get final prompt text
+      if AIA.config.stdin_content && !AIA.config.stdin_content.strip.empty?
+        prompt.text << "\n\n" << AIA.config.stdin_content
+      end
+
+      if AIA.config.executable_prompt_file
+        prompt.text << "\n\n" << File.read(AIA.config.executable_prompt_file)
+                                  .lines[1..]
+                                  .join
+      end
+
+      # Substitute variables, execute dynamic content and get final prompt text
       prompt_text = prompt.to_s
 
       # Add context files if any

data/lib/aia/ui_presenter.rb

@@ -26,7 +26,7 @@ module AIA
       puts "\nAI: "
       format_chat_response(response)
 
-      if AIA.config.out_file
+      if AIA.config.out_file && !AIA.config.out_file.nil?
         File.open(AIA.config.out_file, 'a') do |file|
           file.puts "\nAI: "
           format_chat_response(response, file)

data/lib/aia/utility.rb

@@ -21,7 +21,7 @@ module AIA
          (O)       using #{AIA.config.adapter} (v#{RubyLLM::VERSION})
        __||__    \\) model db was last refreshed on
      [/______\\]  /    #{AIA.config.last_refresh}
-    / \\__AI__/ \\/      #{AIA.config.tool_paths.empty? ? 'I forgot my toolbox' : 'I brought some tools'}
+    / \\__AI__/ \\/      #{AIA.config.tool_paths.empty? ? 'You can share my tools' : 'I will also use your tools'}
    /    /__\\
   (\\   /____\\   #{AIA.config.tool_paths.empty? ? '' : 'My Toolbox contains:'}
         ROBOT

data/lib/aia.rb

@@ -5,15 +5,17 @@
 # provides an interface for interacting with AI models and managing prompts.
 
 require 'ruby_llm'
+require 'ruby_llm/mcp'
 require 'prompt_manager'
 
+
 require 'debug_me'
 include DebugMe
 $DEBUG_ME = false
 DebugMeDefaultOptions[:skip1] = true
 
 require_relative 'extensions/openstruct_merge'    # adds self.merge self.get_value
-require_relative 'extensions/ruby_llm/modalities' # adds model.modalities.supports? :text-to-text etc.
+require_relative 'extensions/ruby_llm/modalities' # adds model.modalities.text_to_text? etc.
 
 require_relative 'refinements/string.rb'        # adds #include_any? #include_all?
 
@@ -37,6 +39,10 @@ require_relative 'aia/session'
 module AIA
   at_exit do
     STDERR.puts "Exiting AIA application..."
+    # Clean up temporary STDIN file if it exists
+    if @config&.stdin_temp_file && File.exist?(@config.stdin_temp_file)
+      File.unlink(@config.stdin_temp_file)
+    end
   end
 
   @config = nil

data/lib/extensions/ruby_llm/modalities.rb

@@ -1,26 +1,34 @@
 # lib/extensions/ruby_llm/modalities.rb
-# A models "modes" are often expressed in terms like:
-#   text-to-text
-#   text_to_audio
-#   audio to image
-#   image2image
-# This new supports? method tests the models modalities against
-# these common expressions
 
 class RubyLLM::Model::Modalities
-  def supports?(query_mode)
-    parts = query_mode
-              .to_s
-              .downcase
-              .split(/2|-to-| to |_to_/)
-              .map(&:strip)
-
-    if 2 == parts.size
-      input.include?(parts[0]) && output.include?(parts[1])
-    elsif 1 == parts.size
-      input.include?(parts[0]) || output.include?(parts[0])
-    else
-      false
-    end
-  end
+  #
+  def text_to_text? = input.include?('text') && output.include?('text')
+  def text_to_embeddings? = input.include?('text') && output.include?('embeddings')
+  def text_to_audio? = input.include?('text') && output.include?('audio')
+  def text_to_image? = input.include?('text') && output.include?('image')
+  def text_to_moderation? = input.include?('text') && output.include?('moderation')
+  #
+  def image_to_text? = input.include?('image') && output.include?('text')
+  def image_to_embeddings? = input.include?('image') && output.include?('embeddings')
+  def image_to_audio? = input.include?('image') && output.include?('audio')
+  def image_to_image? = input.include?('image') && output.include?('image')
+  def image_to_moderation? = input.include?('image') && output.include?('moderation')
+  #
+  def pdf_to_text? = input.include?('pdf') && output.include?('text')
+  def pdf_to_embeddings? = input.include?('pdf') && output.include?('embeddings')
+  def pdf_to_audio? = input.include?('pdf') && output.include?('audio')
+  def pdf_to_image? = input.include?('pdf') && output.include?('image')
+  def pdf_to_moderation? = input.include?('pdf') && output.include?('moderation')
+  #
+  def audio_to_text? = input.include?('audio') && output.include?('text')
+  def audio_to_embeddings? = input.include?('audio') && output.include?('embeddings')
+  def audio_to_audio? = input.include?('audio') && output.include?('audio')
+  def audio_to_image? = input.include?('audio') && output.include?('image')
+  def audio_to_moderation? = input.include?('audio') && output.include?('moderation')
+  #
+  def file_to_text? = input.include?('file') && output.include?('text')
+  def file_to_embeddings? = input.include?('file') && output.include?('embeddings')
+  def file_to_audio? = input.include?('file') && output.include?('audio')
+  def file_to_image? = input.include?('file') && output.include?('image')
+  def file_to_moderation? = input.include?('file') && output.include?('moderation')
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: aia
 version: !ruby/object:Gem::Version
-  version: 0.9.5
+  version: 0.9.7
 platform: ruby
 authors:
 - Dewayne VanHoozer
@@ -9,6 +9,20 @@ bindir: bin
 cert_chain: []
 date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
+- !ruby/object:Gem::Dependency
+  name: activesupport
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: amazing_print
   requirement: !ruby/object:Gem::Requirement
@@ -57,14 +71,28 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 1.3.0
+        version: 1.3.1
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 1.3.0
+        version: 1.3.1
+- !ruby/object:Gem::Dependency
+  name: ruby_llm-mcp
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 - !ruby/object:Gem::Dependency
   name: reline
   requirement: !ruby/object:Gem::Requirement
@@ -292,6 +320,7 @@ files:
 - examples/tools/list_files.rb
 - examples/tools/read_file.rb
 - examples/tools/run_shell_command.rb
+- images/aia.png
 - justfile
 - lib/aia.rb
 - lib/aia/aia_completion.bash