aia 0.9.6 → 0.9.8

data/lib/aia/config.rb

@@ -54,7 +54,6 @@ module AIA
       append:   false, # Default to not append to existing out_file
 
       # workflow
-      next:     nil,
       pipeline: [],
 
       # PromptManager::Prompt Tailoring
@@ -124,6 +123,18 @@ module AIA
       remaining_args = config.remaining_args.dup
       config.remaining_args = nil
 
+      # Check for STDIN content
+      stdin_content = ''
+
+      if !STDIN.tty? && !STDIN.closed?
+        begin
+          stdin_content << "\n" + 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
@@ -162,6 +190,7 @@ module AIA
       if config.prompt_id.nil? || config.prompt_id.empty?
         if !config.role.nil? || !config.role.empty?
           config.prompt_id = config.role
+          config.pipeline.prepend config.prompt_id
           config.role      = ''
         end
       end
@@ -214,6 +243,25 @@ module AIA
         PromptManager::Prompt.parameter_regex = Regexp.new(config.parameter_regex)
       end
 
+      if !config.prompt_id.empty? && config.prompt_id != config.pipeline.first
+        config.pipeline.prepend config.prompt_id
+      end
+
+      unless config.pipeline.empty?
+        config.pipeline.each do |prompt_id|
+          # Skip empty prompt IDs (can happen in chat-only mode)
+          next if prompt_id.nil? || prompt_id.empty?
+
+          prompt_file_path = File.join(config.prompts_dir, "#{prompt_id}.txt")
+          unless File.exist?(prompt_file_path)
+            STDERR.puts "Error: Prompt ID '#{prompt_id}' does not exist at #{prompt_file_path}"
+            and_exit = true
+          end
+        end
+      end
+
+      exit(1) if and_exit
+
       config
     end
 
@@ -309,7 +357,7 @@ module AIA
                       "       aia --chat [PROMPT_ID] [CONTEXT_FILE]*\n" +
                       "       aia --chat [CONTEXT_FILE]*"
 
-          opts.on("--chat", "Begin a chat session with the LLM after the initial prompt response; will set --no-out_file so that the LLM response comes to STDOUT.") do
+          opts.on("--chat", "Begin a chat session with the LLM after processing all prompts in the pipeline.") do
             config.chat = true
             puts "Debug: Setting chat mode to true" if config.debug
           end
@@ -325,10 +373,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
@@ -362,7 +468,7 @@ module AIA
             end
           end
 
-          opts.on("-p", "--prompts_dir DIR", "Directory containing prompt files") do |dir|
+          opts.on("--prompts_dir DIR", "Directory containing prompt files") do |dir|
             config.prompts_dir = dir
           end
 
@@ -383,7 +489,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|
@@ -399,11 +511,13 @@ module AIA
           end
 
           opts.on("-n", "--next PROMPT_ID", "Next prompt to process") do |next_prompt|
-            config.next = next_prompt
+            config.pipeline ||= []
+            config.pipeline << next_prompt
           end
 
-          opts.on("--pipeline PROMPTS", "Pipeline of prompts to process") do |pipeline|
-            config.pipeline = pipeline.split(',')
+          opts.on("-p PROMPTS", "--pipeline PROMPTS", "Pipeline of comma-seperated prompt IDs to process") do |pipeline|
+            config.pipeline ||= []
+            config.pipeline += pipeline.split(',').map(&:strip)
           end
 
           opts.on("-f", "--fuzzy", "Use fuzzy matching for prompt search") do
@@ -422,8 +536,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,12 +156,39 @@ 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
+      elsif 1 == args.size
+        AIA.config.pipeline += args.first.split(',').map(&:strip).reject{|id| id.empty?}
       else
-        AIA.config.pipeline += args.map {|id| id.gsub(',', '').strip}
+        AIA.config.pipeline += args.map{|id| id.gsub(',', '').strip}.reject{|id| id.empty?}
       end
       ''
     end
@@ -299,6 +328,11 @@ 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
+
+      if 1 == query.size
+        query = query.first.split(',')
+      end
+
       header    = "\nAvailable LLMs"
       header   += " for #{query.join(' and ')}" if query
 

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