aia 0.5.18 → 0.8.1

data/lib/aia/cli.rb

@@ -1,452 +0,0 @@
-# lib/aia/cli.rb
-
-HOME    = Pathname.new(ENV['HOME'])
-MY_NAME = 'aia'
-
-
-require 'hashie'
-require 'pathname'
-require 'yaml'
-require 'toml-rb'
-
-
-class AIA::Cli
-  CF_FORMATS    = %w[yml yaml toml]
-  ENV_PREFIX    = self.name.split('::').first.upcase + "_"
-  MAN_PAGE_PATH = Pathname.new(__dir__) + '../../man/aia.1'
-  
-
-  def initialize(args)
-    args = args.split(' ') if args.is_a? String
-
-    setup_options_with_defaults(args) # 1. defaults
-    load_env_options                  # 2. over-ride with envars
-    process_command_line_arguments    # 3. over-ride with command line options
-
-    # 4. over-ride everything with config file
-    load_config_file unless AIA.config.config_file.nil?
-
-    convert_to_pathname_objects
-    error_on_invalid_option_combinations
-    setup_prompt_manager
-    execute_immediate_commands
-  end
-
-
-  def convert_pathname_objects!(converting_to_pathname: true)
-    path_keys = AIA.config.keys.grep(/_(dir|file)\z/)
-    path_keys.each do |key|
-      case AIA.config[key]
-      when String
-        AIA.config[key] = string_to_pathname(AIA.config[key])
-      when Pathname
-        AIA.config[key] = pathname_to_string(AIA.config[key]) unless converting_to_pathname
-      end
-    end
-  end
-
-
-  def error_on_invalid_option_combinations
-    # --chat is intended as an interactive exchange
-    if AIA.config.chat?
-      unless AIA.config.next.empty?
-        abort "ERROR: Cannot use --next with --chat"
-      end
-      unless STDOUT == AIA.config.out_file
-        abort "ERROR: Cannot use --out_file with --chat"
-      end
-      unless AIA.config.pipeline.empty?
-        abort "ERROR: Cannot use --pipeline with --chat"
-      end
-    end
-
-    # --next says which prompt to process next
-    # but --pipeline gives an entire sequence of prompts for processing
-    unless AIA.config.next.empty?
-      unless AIA.config.pipeline.empty?
-        abort "ERROR: Cannot use --pipeline with --next"
-      end
-    end
-  end
-
-  def string_to_pathname(string)
-    ['~/', '$HOME/'].each do |prefix|
-      if string.start_with? prefix
-        string = string.gsub(prefix, HOME.to_s+'/')
-        break
-      end
-    end
-
-    pathname = Pathname.new(string)
-    pathname.relative? ? Pathname.pwd + pathname : pathname
-  end
-
-
-  def pathname_to_string(pathname)
-    pathname.to_s
-  end
-
-
-  def convert_to_pathname_objects
-    convert_pathname_objects!(converting_to_pathname: true)
-  end
-
-
-  def convert_from_pathname_objects
-    convert_pathname_objects!(converting_to_pathname: false)
-  end
-
-
-  def load_env_options
-    known_keys = @options.keys
-
-    keys  = ENV.keys
-              .select{|k| k.start_with?(ENV_PREFIX)}
-              .map{|k| k.gsub(ENV_PREFIX,'').downcase.to_sym}
-
-    keys.each do |key|
-      envar_key       = ENV_PREFIX + key.to_s.upcase
-      if known_keys.include?(key)
-        AIA.config[key] = ENV[envar_key]
-      elsif known_keys.include?("#{key}?".to_sym)
-        key = "#{key}?".to_sym
-        AIA.config[key] = %w[true t yes yea y 1].include?(ENV[envar_key].strip.downcase) ? true : false
-      else
-        # This is a new config key
-        AIA.config[key] = ENV[envar_key]
-      end
-    end
-  end
-
-
-  def replace_erb_in_config_file
-    content = Pathname.new(AIA.config.config_file).read
-    content = ERB.new(content).result(binding)
-    AIA.config.config_file  = AIA.config.config_file.to_s.gsub('.erb', '')
-    Pathname.new(AIA.config.config_file).write content
-  end
-
-
-  def load_config_file
-    if AIA.config.config_file.to_s.end_with?(".erb")
-      replace_erb_in_config_file
-    end
-
-    AIA.config.config_file = Pathname.new(AIA.config.config_file)
-    if AIA.config.config_file.exist?
-      AIA.config.merge! parse_config_file
-    else
-      abort "Config file does not exist: #{AIA.config.config_file}"
-    end
-  end
-
-
-  def setup_options_with_defaults(args)
-    # TODO: This structure if flat; consider making it
-    #       at least two levels totake advantage of
-    #       YAML and TOML capabilities to isolate
-    #       common options within a section.
-    #
-    @options    = {
-      #           Default
-      # Key       Value,      switches
-      arguments:  [args], # NOTE: after process, prompt_id and context_files will be left
-      directives: [[]],   # an empty Array as the default value
-      extra:      [''],   # 
-      #
-      model:      ["gpt-4-1106-preview",  "--llm --model"],
-      speech_model: ["tts-1", "--sm --speech_model"],
-      voice:        ["alloy", "--voice"],
-      #
-      transcription_model:  ["wisper-1", "--tm --transcription_model"],
-      #
-      dump_file:  [nil,       "--dump"],
-      completion: [nil,       "--completion"],
-      #
-      chat?:      [false,     "--chat"],
-      debug?:     [false,     "-d --debug"],
-      edit?:      [false,     "-e --edit"],
-      erb?:       [false,     "--erb"],
-      fuzzy?:     [false,     "-f --fuzzy"],
-      help?:      [false,     "-h --help"],
-      markdown?:  [true,      "-m --markdown --no-markdown --md --no-md"],
-      render?:    [false,     "--render"],
-      shell?:     [false,     "--shell"],
-      speak?:     [false,     "--speak"],
-      terse?:     [false,     "--terse"],
-      verbose?:   [false,     "-v --verbose"],
-      version?:   [false,     "--version"],
-      #
-      next:       ['',        "-n --next"],
-      pipeline:   [[],        "--pipeline"],
-      role:       ['',        "-r --role"],
-      #
-      config_file:[nil,                       "-c --config_file"],
-      prompts_dir:["~/.prompts",              "-p --prompts_dir"],
-      roles_dir:  ["~/.prompts/roles",        "--roles_dir"],
-      out_file:   [STDOUT,                    "-o --out_file --no-out_file"],
-      log_file:   ["~/.prompts/_prompts.log", "-l --log_file --no-log_file"],
-      #
-      backend:    ['mods',    "-b --be --backend --no-backend"],
-      #
-      # text2image related ...
-      #
-      image_size:     ['', '--is --image_size'],
-      image_quality:  ['', '--iq --image_quality'],
-    }
-    
-    AIA.config = AIA::Config.new(@options.transform_values { |values| values.first })
-  end
-
-
-  def arguments
-    AIA.config.arguments
-  end
-
-
-  def execute_immediate_commands
-    show_usage        if AIA.config.help?
-    show_version      if AIA.config.version?
-    dump_config_file  if AIA.config.dump_file
-    show_completion   if AIA.config.completion
-  end
-
-
-  def dump_config_file
-    a_hash = prepare_config_as_hash
-
-    dump_file = Pathname.new AIA.config.dump_file
-    extname   = dump_file.extname.to_s.downcase
-
-    case extname
-    when '.yml', '.yaml'
-      dump_file.write YAML.dump(a_hash)
-    when '.toml'
-      dump_file.write TomlRB.dump(a_hash)
-    else
-      abort "Invalid config file format (#{extname}) request.  Only #{CF_FORMATS.join(', ')} are supported."
-    end
-
-    exit
-  end
-
-
-  def prepare_config_as_hash
-    convert_from_pathname_objects
-    
-    a_hash          = AIA.config.to_h
-    a_hash['dump']  = nil
-
-    %w[ arguments config_file dump_file ].each do |unwanted_key|
-      a_hash.delete(unwanted_key)
-    end
-    
-    a_hash
-  end
-
-
-  def process_command_line_arguments
-    # get the options meant for the backend AI command
-    # doing this first in case there are any options that conflict
-    # between frontend and backend.
-    extract_extra_options
-
-    @options.keys.each do |option|
-      check_for option
-    end
-
-    bad_options = arguments.select{|a| a.start_with?('-')}
-
-    unless bad_options.empty?
-      puts <<~EOS
-
-        ERROR: Unknown options: #{bad_options.join(' ')}
-
-      EOS
-      
-      show_error_usage
-
-      exit
-    end
-  
-    # After all other arguments 
-    # are processed, check for role parameter.
-    check_for_role_parameter
-  end
-
-
-  def check_for(option_sym)
-    # sometimes @options has stuff that is not a command line option
-    return if @options[option_sym].nil? || @options[option_sym].size <= 1
-
-    boolean   = option_sym.to_s.end_with?('?')
-    switches  = @options[option_sym][1].split
-
-    switches.each do |switch|
-      if arguments.include?(switch)
-        index = arguments.index(switch)
-
-        if boolean
-          AIA.config[option_sym] = switch.include?('-no-') ? false : true
-          arguments.slice!(index,1)
-        else
-          if switch.include?('-no-')
-            AIA.config[option_sym] = switch.include?('out_file') ? STDOUT : nil
-            arguments.slice!(index,1)
-          else
-            value = arguments[index + 1]
-            if value.nil? || value.start_with?('-')
-              abort "ERROR: #{option_sym} requires a parameter value"
-            elsif "--pipeline" == switch
-              prompt_sequence = value.split(',')
-              AIA.config[option_sym] = prompt_sequence
-              arguments.slice!(index,2)
-            else
-              AIA.config[option_sym] = value
-              arguments.slice!(index,2)
-            end
-          end
-        end
-        
-        break
-      end
-    end
-  end
-
-
-  def check_for_role_parameter
-    role = AIA.config.role
-    return if role.empty?
-
-    role_path = string_to_pathname(AIA.config.roles_dir) + "#{role}.txt"
-
-    unless role_path.exist?
-      puts "Role prompt '#{role}' not found. Invoking fzf to choose a role..."
-      invoke_fzf_to_choose_role
-    end
-  end
-
-
-  def invoke_fzf_to_choose_role
-    roles_path = string_to_pathname AIA.config.roles_dir
-
-    available_roles = roles_path
-                        .children
-                        .select { |f| '.txt' == f.extname}
-                        .map{|role| role.basename.to_s.gsub('.txt','')}
-    
-    fzf = AIA::Fzf.new(
-      list:       available_roles,
-      directory:  roles_path,
-      prompt:     'Select Role:',
-      extension:  '.txt'
-    )
-
-    chosen_role = fzf.run
-
-    if chosen_role.nil?
-      abort("No role selected. Exiting...")
-    else
-      AIA.config.role = chosen_role
-      puts "Role changed to '#{chosen_role}'."
-    end
-  end
-
-
-  def show_error_usage
-    puts <<~ERROR_USAGE
-
-      Usage: aia [options] PROMPT_ID [CONTEXT_FILE(s)] [-- EXTERNAL_OPTIONS]"
-      Try 'aia --help' for more information."
-    
-    ERROR_USAGE
-  end
-
-
-  # aia usage is maintained in a man page
-  def show_usage
-    @options[:help?][0] = false 
-    puts `man #{MAN_PAGE_PATH}`
-    show_verbose_usage if AIA.config.verbose?
-    exit
-  end
-  alias_method :show_help, :show_usage
-
-
-  def show_verbose_usage
-    puts <<~EOS
-
-      ======================================
-      == Currently selected Backend: #{AIA.config.backend} ==
-      ======================================
-
-    EOS
-    puts `mods --help` if "mods" == AIA.config.backend
-    puts `sgpt --help` if "sgpt" == AIA.config.backend
-    puts
-  end
-  # alias_method :show_verbose_help, :show_verbose_usage
-
-
-  def show_completion
-    shell   = AIA.config.completion
-    script  = Pathname.new(__dir__) + "aia_completion.#{shell}"
-
-    if script.exist?
-      puts
-      puts script.read
-      puts
-    else
-      STDERR.puts <<~EOS
-
-        ERROR: The shell '#{shell}' is not supported.
-
-      EOS
-    end
-
-    exit    
-  end
-
-
-  def show_version
-    puts AIA::VERSION
-    exit
-  end
-
-
-  def setup_prompt_manager
-    @prompt     = nil
-
-    PromptManager::Prompt.storage_adapter = 
-      PromptManager::Storage::FileSystemAdapter.config do |config|
-        config.prompts_dir        = AIA.config.prompts_dir
-        config.prompt_extension   = '.txt'
-        config.params_extension   = '.json'
-        config.search_proc        = nil
-        # TODO: add the rgfzf script for search_proc
-      end.new
-  end
-
-
-  # Get the additional CLI arguments intended for the
-  # backend gen-AI processor.
-  def extract_extra_options
-    extra_index = arguments.index('--')
-
-    if extra_index
-      AIA.config.extra = arguments.slice!(extra_index..-1)[1..].join(' ')
-    end
-  end
-
-
-  def parse_config_file
-    case AIA.config.config_file.extname.downcase
-    when '.yaml', '.yml'
-      YAML.safe_load(AIA.config.config_file.read)
-    when '.toml'
-      TomlRB.parse(AIA.config.config_file.read)
-    else
-      abort "Unsupported config file type: #{AIA.config.config_file.extname}"
-    end
-  end
-end

data/lib/aia/directives.rb

@@ -1,142 +0,0 @@
-# lib/aia/directives.rb
-
-require 'hashie'
-
-=begin
-  AIA.config.directives is an Array of Arrays.  An
-  entry looks like this:
-    [directive, parameters]
-  where both are String objects
-=end
-
-
-class AIA::Directives
-  def execute_my_directives
-    return if AIA.config.directives.nil? || AIA.config.directives.empty?
-    
-    result    = ""
-    not_mine  = []
-
-    AIA.config.directives.each do |entry|
-      directive   = entry[0].to_sym
-      parameters  = entry[1]
-
-      if respond_to? directive
-        output  = send(directive, parameters)
-        result << "#{output}\n" unless output.nil?
-      else
-        not_mine << entry
-      end
-    end
-
-    AIA.config.directives = not_mine
-
-    result.empty? ? nil : result
-  end
-
-
-  def box(what)
-    f   = what[0]
-    bar = "#{f}"*what.size
-    puts "#{bar}\n#{what}\n#{bar}"
-  end
-
-
-  # Allows a prompt to change its configuration environment
-  def config(what)
-    parts = what.split(' ')
-    item  = parts.shift
-    parts.shift if %w[:= =].include? parts[0]
-
-    if '<<' == parts[0]
-      parts.shift
-      value = parts.join
-      if AIA.config(item).is_a?(Array)
-        AIA.config[item] << value
-      else
-        AIA.config[item] = [ value ]
-      end
-    else
-      value = parts.join
-      if item.end_with?('?')
-        AIA.config[item] = %w[1 y yea yes t true].include?(value.downcase)
-      elsif item.end_with?('_file')
-        if "STDOUT" == value.upcase
-          AIA.config[item] = STDOUT
-        elsif "STDERR" == value.upcase
-          AIA.config[item] = STDERR
-        else
-          AIA.config[item] = value.start_with?('/') ? 
-            Pathname.new(value) :
-            Pathname.pwd + value
-        end
-      elsif %w[next pipeline].include? item.downcase
-        pipeline(value)
-      else
-        AIA.config[item] = value
-      end
-    end
-
-    nil
-  end
-
-
-  # TODO: we need a way to submit CLI arguments into
-  #       the next prompt(s) from the main prompt.
-  #       currently the config for subsequent prompts
-  #       is expected to be set within those prompts.
-  #       Maybe something like:
-  #         //next prompt_id CLI args
-  #       This would mean that the pipeline would be:
-  #         //pipeline id1 cli args, id2 cli args, id3 cli args
-  #
-  
-  # TODO: Change AIA.config.pipline Array to be an Array of arrays
-  #       where each entry is:
-  #         [prompt_id, cli_args]
-  #       This means that:
-  #         entry = AIA.config.pipeline.shift
-  #         entry.is_A?(Sring) ? 'old format' : 'new format'
-  #
-
-  # //next id
-  # //pipeline id1,id2, id3   ,   id4
-  def pipeline(what)
-    return if what.empty?
-    AIA.config.pipeline << what.split(',').map(&:strip)
-    AIA.config.pipeline.flatten!
-  end
-  alias_method :next, :pipeline
-
-  # when path_to_file is relative it will be
-  # relative to the PWD.
-  #
-  # TODO: Consider an AIA_INCLUDE_DIR --include_dir
-  # option to be used for all relative include paths
-  #
-  def include(path_to_file)
-    path = Pathname.new path_to_file
-    if path.exist? && path.readable?
-      content = path.readlines.reject do |a_line|
-        a_line.strip.start_with?(AIA::Prompt::COMMENT_SIGNAL) ||
-        a_line.strip.start_with?(AIA::Prompt::DIRECTIVE_SIGNAL)
-      end.join("\n")
-    else
-      abort "ERROR: could not include #{path_to_file}"
-    end
-
-    content
-  end
-
-
-  def shell(command)
-    `#{command}`
-  end
-
-
-  def ruby(code)
-    output = eval(code)
-
-    output.is_a?(String) ? output : nil
-  end
-end

data/lib/aia/dynamic_content.rb

@@ -1,26 +0,0 @@
-# aia/lib/aia/dynamic_content.rb
-
-require 'erb'
-
-module AIA::DynamicContent
-
-  # inserts environment variables (envars) and dynamic content into a prompt
-  # replaces patterns like $HOME and ${HOME} with the value of ENV['HOME']
-  # replaces patterns like $(shell command) with the output of the shell command
-  #
-  def render_env(a_string)
-    a_string.gsub(/\$(\w+|\{\w+\})/) do |match|
-      ENV[match.tr('$', '').tr('{}', '')]
-    end.gsub(/\$\((.*?)\)/) do |match|
-      `#{match[2..-2]}`.chomp
-    end
-  end
-
-
-  # Need to use instance variables in assignments
-  # to maintain binding from one follow up prompt
-  # to another.
-  def render_erb(the_prompt_text)
-    ERB.new(the_prompt_text).result(binding)
-  end
-end

data/lib/aia/logging.rb

@@ -1,62 +0,0 @@
-# lib/aia/logging.rb
-
-require 'logger'
-
-class AIA::Logging
-  attr_accessor :logger
-
-  def initialize(log_file_path)
-    @logger = if log_file_path
-                Logger.new(
-                  log_file_path,  # path/to/file
-                  'weekly',       # rotation interval
-                  'a'             # append to existing file
-                )
-              else
-                # SMELL: Looks like you get logging whether you want it or not
-                # TODO: when path is nil create a fake logger
-                #       that does nothing
-                Logger.new(STDOUT) # Fall back to standard output if path is nil or invalid
-              end
-
-    configure_logger
-  end
-
-  def prompt_result(prompt, result)
-    logger.info( <<~EOS
-      PROMPT ID #{prompt.id}
-      PATH:     #{prompt.path}
-      KEYWORDS: #{prompt.keywords.join(', ')}
-        
-        #{prompt.to_s}
-
-      RESULT:
-      #{result}
-
-
-    EOS
-    )
-  rescue StandardError => e
-    logger.error("Failed to log the result. Error: #{e.message}")
-  end
-
-
-  def debug(msg)    = logger.debug(msg)
-  def info(msg)     = logger.info(msg)
-  def warn(msg)     = logger.warn(msg)
-  def error(msg)    = logger.error(msg)
-  def fatal(msg)    = logger.fatal(msg)
-
-  private
-
-  def configure_logger
-    @logger.formatter = proc do |severity, datetime, _progname, msg|
-      formatted_datetime = datetime.strftime("%Y-%m-%d %H:%M:%S")
-      "[#{formatted_datetime}] #{severity}: #{msg}\n"
-    end
-    @logger.level = Logger::DEBUG
-  end
-end
-
-
-