aia 0.5.17 → 0.8.0

data/lib/aia/main.rb

@@ -1,265 +0,0 @@
-# lib/aia/main.rb
-
-module AIA ; end
-
-require_relative 'config'
-require_relative 'cli'
-require_relative 'directives'
-require_relative 'dynamic_content'
-require_relative 'prompt'
-require_relative 'logging'
-require_relative 'tools'
-require_relative 'user_query'
-
-# Everything is being handled within the context
-# of a single class.
-
-class AIA::Main
-  SPINNER_FORMAT = :bouncing_ball
-
-  include AIA::DynamicContent
-  include AIA::UserQuery
-  
-  attr_accessor :logger, :tools, :backend, :directive_output, :piped_content
-
-  attr_reader :spinner
-
-  def initialize(args= ARGV)
-    unless $stdin.tty?
-      @piped_content = $stdin.readlines.join.chomp
-      $stdin.reopen("/dev/tty")
-    end
-
-    @directive_output = ""
-    AIA::Tools.load_tools
-
-    AIA.client = AIA::Client.new
-
-    AIA::Cli.new(args)
-
-    if AIA.config.debug?
-      debug_me('== CONFIG AFTER CLI =='){[
-        "AIA.config"
-      ]}
-    end
-
-    @spinner  = TTY::Spinner.new(":spinner :title", format: SPINNER_FORMAT)
-    spinner.update(title: "composing response ... ")
-
-    @logger = AIA::Logging.new(AIA.config.log_file)
-
-    @logger.info(AIA.config) if AIA.config.debug? || AIA.config.verbose?
-
-
-    @directives_processor = AIA::Directives.new 
-
-    @prompt = AIA::Prompt.new.prompt
-
-    @prompt.text += piped_content unless piped_content.nil?
-
-    # TODO: still should verify that the tools are ion the $PATH
-    # tools.class.verify_tools
-  end
-
-
-  # Function to setup the Reline history with a maximum depth
-  def setup_reline_history(max_history_size=5)
-    Reline::HISTORY.clear
-    # Reline::HISTORY.max_size = max_history_size
-  end
-
-
-  # This will be recursive with the new options
-  # --next and --pipeline
-  def call
-    @directive_output = @directives_processor.execute_my_directives
-
-    if AIA.config.chat?
-      AIA.config.out_file = STDOUT 
-      AIA.config.extra = "--quiet" if 'mods' == AIA.config.backend
-    end
-
-    # TODO: the context_files left in the @arguments array
-    #       should be verified BEFORE asking the user for a
-    #       prompt keyword or process the prompt.  Do not
-    #       want invalid files to make it this far.
-
-    found = AIA::Tools
-              .search_for(
-                name: AIA.config.backend, 
-                role: :backend
-              )
-
-    if found.empty?
-      abort "There are no :backend tools named #{AIA.config.backend}"
-    end
-
-    if found.size > 1
-      abort "There are #{found.size} :backend tools with the name #{AIAA.config.backend}"
-    end
-
-    backend_klass = found.first.klass
-
-    abort "backend not found: #{AIA.config.backend}" if backend_klass.nil?
-
-    the_prompt = @prompt.to_s
-
-    the_prompt.prepend(@directive_output + "\n") unless @directive_output.nil? || @directive_output.empty?
-
-    if AIA.config.terse?
-      the_prompt.prepend "Be terse in your response. "
-    end
-
-    @backend  = backend_klass.new(
-                  text:           the_prompt,
-                  files:          AIA.config.arguments    # FIXME: want validated context files
-                )
-
-    result = get_and_display_result(the_prompt)
-
-    AIA.speak(result) if AIA.config.speak?
-
-    logger.prompt_result(@prompt, result)
-
-    if AIA.config.chat?
-      setup_reline_history
-      AIA.speak result
-      lets_chat 
-    end
-
-    return if AIA.config.next.empty? && AIA.config.pipeline.empty?
-
-    keep_going(result) unless AIA.config.pipeline.empty?
-  end
-
-
-  # The AIA.config.pipeline is NOT empty, so feed this result
-  # into the next prompt within the pipeline.
-  #
-  def keep_going(result)
-    temp_file = Tempfile.new('aia_pipeline')
-    temp_file.write(result)
-    temp_file.close
-
-    AIA.config.directives = []
-    AIA.config.model      = ""
-    AIA.config.arguments  = [
-                              AIA.config.pipeline.shift, 
-                              temp_file.path,
-                              # TODO: additional arguments from the pipeline
-                            ] 
-    AIA.config.next       = ""
-
-    AIA.config.files = [temp_file.path]
-
-    @prompt = AIA::Prompt.new.prompt
-    call # Recurse! until the AIA.config.pipeline is emplty
-    puts
-  ensure
-    temp_file.unlink
-  end
-
-
-  def get_and_display_result(the_prompt_text)
-    spinner.auto_spin if AIA.config.verbose?
-
-    backend.text  = the_prompt_text
-    result        = backend.run
-
-    if AIA.config.verbose?
-      spinner.success "Done." 
-    end
-
-    AIA.config.out_file.write "\nResponse:\n"
-
-    if STDOUT == AIA.config.out_file
-      if AIA.config.render?
-        AIA::Glow.new(content: result).run
-      else
-        result  = result.wrap(indent: 2)
-        AIA.config.out_file.write result
-      end
-    else
-      AIA.config.out_file.write result
-      if AIA.config.render?
-        AIA::Glow.new(file_path: AIA.config.out_file).run
-      end
-    end
-
-    result
-  end
-
-
-  def log_the_follow_up(the_prompt_text, result)
-    logger.info "Follow Up:\n#{the_prompt_text}"
-    logger.info "Response:\n#{result}"
-  end
-
-
-  def add_continue_option
-    if 'mods' == AIA.config.backend
-      continue_option   = " -C"
-      AIA.config.extra += continue_option unless AIA.config.extra.include?(continue_option)
-    end
-  end
-
-
-  def insert_terse_phrase(a_string)
-    if AIA.config.terse?
-      a_string.prepend "Be terse in your response. "
-    end
-
-    a_string
-  end
-
-  
-  def handle_directives(the_prompt_text)
-    signal = PromptManager::Prompt::DIRECTIVE_SIGNAL
-    result = the_prompt_text.start_with?(signal)
-
-    if result
-      parts       = the_prompt_text[signal.size..].split(' ')
-      directive   = parts.shift
-      parameters  = parts.join(' ')
-      AIA.config.directives << [directive, parameters]
-
-      @directive_output = @directives_processor.execute_my_directives
-    else
-      @directive_output = ""
-    end
-
-    result
-  end
-
-
-  def lets_chat
-    add_continue_option    
-
-    the_prompt_text = ask_question_with_reline("\nFollow Up: ")
-
-    until the_prompt_text.empty?
-      the_prompt_text   = render_erb(the_prompt_text) if AIA.config.erb?
-      the_prompt_text   = render_env(the_prompt_text) if AIA.config.shell?
-
-      if handle_directives(the_prompt_text)
-        if @directive_output.nil? || @directive_output.empty?
-          # Do nothing
-        else
-          the_prompt_text = @directive_output 
-          the_prompt_text = render_erb(the_prompt_text) if AIA.config.erb?
-          the_prompt_text = render_env(the_prompt_text) if AIA.config.shell?
-          result          = get_and_display_result(the_prompt_text)
-          log_the_follow_up(the_prompt_text, result)
-          AIA.speak result
-        end
-      else
-        the_prompt_text = insert_terse_phrase(the_prompt_text)
-        result          = get_and_display_result(the_prompt_text)
-        log_the_follow_up(the_prompt_text, result)
-        AIA.speak result
-      end
-
-      the_prompt_text = ask_question_with_reline("\nFollow Up: ")
-    end
-  end
-end

data/lib/aia/prompt.rb

@@ -1,275 +0,0 @@
-# lib/aia/prompt.rb
-
-require 'reline'
-
-require_relative 'dynamic_content'
-require_relative 'user_query'
-
-class AIA::Prompt
-  include AIA::DynamicContent
-  include AIA::UserQuery
-  
-  #
-  # used when no prompt_id is provided but there
-  # are extra parameters that need to be passed
-  # to the backend.  For example "aia -- --settings"
-  #
-  class Fake
-    def id          = '_fake_'
-    def path        = '_fake_'
-    def text        = ''
-    def keywords    = []
-    def directives  = []
-    def to_s        = ''
-  end
-
-  KW_HISTORY_MAX    = 5
-  COMMENT_SIGNAL    = '#'
-  DIRECTIVE_SIGNAL  = "//"
-
-  attr_reader :prompt
-
-  # setting build: false supports unit testing.
-  def initialize(build: true)
-    if AIA.config.role.empty?
-      @role = nil
-    else
-      @role = (AIA.config.roles_dir + "#{AIA.config.role}.txt").read
-    end
-
-    get_prompt
-    
-    @prompt_text_before_role = @prompt.text.dup
-
-    unless @role.nil?
-      @prompt.text.prepend @role
-    end
-
-    if build
-      @prompt.text = render_erb(@prompt.text)   if AIA.config.erb?
-      @prompt.text = render_env(@prompt.text)   if AIA.config.shell?
-      process_prompt 
-    end
-
-    AIA.config.directives = @prompt.directives
-  end
-
-
-  # Fetch the first argument which should be the prompt id
-  def get_prompt
-    prompt_id = AIA.config.arguments.shift
-
-    unless prompt_id
-      if AIA.config.extra.empty?
-        abort("Please provide a prompt id") 
-      else
-        @prompt = Fake.new
-        return
-      end
-    end
-
-    search_for_a_matching_prompt(prompt_id) unless existing_prompt?(prompt_id)
-    edit_prompt if AIA.config.edit?
-  end
-
-
-  # Check if a prompt with the given id already exists.  If so, use it.
-  def existing_prompt?(prompt_id)
-    @prompt = PromptManager::Prompt.get(id: prompt_id)
-    true
-  rescue ArgumentError
-    false
-  end
-
-
-  # Process the prompt's associated keywords and parameters
-  def process_prompt
-    unless @prompt.keywords.empty?
-      replace_keywords
-      @prompt.build
-      save(@prompt_text_before_role)
-    end
-  end
-
-
-  def save(original_text)
-    temp_text     = @prompt.text
-    @prompt.text  = original_text
-    @prompt.save
-    @prompt.text  = temp_text
-  end
-
-
-  def replace_keywords
-    puts
-    puts "ID: #{@prompt.id}"
-    
-    show_prompt_without_comments
-
-    puts "\nPress up/down arrow to scroll through history."
-    puts "Type new input or edit the current input."
-    puts  "Quit #{MY_NAME} with a CNTL-D or a CNTL-C"
-    puts
-    @prompt.keywords.each do |kw|
-      value = keyword_value(kw, @prompt.parameters[kw])
-      
-      unless value.nil? || value.strip.empty?
-        value_inx = @prompt.parameters[kw].index(value)
-        
-        if value_inx
-          @prompt.parameters[kw].delete_at(value_inx)
-        end
-
-        # The most recent value for this kw will always be
-        # in the last position
-        @prompt.parameters[kw] << value
-        @prompt.parameters[kw].shift if @prompt.parameters[kw].size > KW_HISTORY_MAX
-      end
-    end
-  end
-
-
-  # Function to setup the Reline history with a maximum depth
-  def setup_reline_history(max_history_size=5)
-    Reline::HISTORY.clear
-    # Reline::HISTORY.max_size = max_history_size
-  end
-
-
-  # query the user for a value to the keyword allow the
-  # reuse of the previous value shown as the default
-  #
-  # FIXME:  Ruby v3.3.0 drops readline in favor or reline
-  #         internally it redirects "require 'readline'" to Reline
-  #         puts lipstick on the pig so that you can continue to
-  #         use the Readline namespace
-  #
-  def keyword_value(kw, history_array)
-    setup_reline_history
-    
-    default = history_array.last
-
-    Array(history_array).each { |entry| Reline::HISTORY.push(entry) unless entry.nil? || entry.empty? }
-
-    puts "Parameter #{kw} ..."
-
-    if default&.empty?
-      user_prompt = "\n-=> "
-    else
-      user_prompt = "\n(#{default}) -=> "
-    end
-
-    a_string = ask_question_with_reline(user_prompt)
-
-    if a_string.nil?
-      puts "okay. Come back soon."
-      exit
-    end
-
-    puts
-    a_string.empty? ? default : a_string
-  end
-
-
-  # Search for a prompt with a matching id or keyword
-  def search_for_a_matching_prompt(prompt_id)
-    # TODO: using the rgfzf version of the search_proc should only
-    #       return a single prompt_id
-    found_prompts = PromptManager::Prompt.search(prompt_id)
-
-    if found_prompts.empty?
-      if AIA.config.edit?
-        create_prompt(prompt_id)
-        edit_prompt
-      else
-        abort <<~EOS
-          
-          No prompts where found for: #{prompt_id}
-          To create a prompt with this ID use the --edit option
-          like this:
-            #{MY_NAME} #{prompt_id} --edit
-
-        EOS
-      end
-    else    
-      prompt_id     = 1 == found_prompts.size ? found_prompts.first : handle_multiple_prompts(found_prompts, prompt_id)
-      @prompt       = PromptManager::Prompt.get(id: prompt_id)
-    end
-  end
-
-
-  def handle_multiple_prompts(found_these, while_looking_for_this)
-    raise ArgumentError, "Argument is not an Array" unless found_these.is_a?(Array)
-
-    # Create an instance of AIA::Fzf with appropriate parameters
-    fzf_instance = AIA::Fzf.new(
-      list:       found_these,
-      directory:  AIA.config.prompts_dir, # Assuming this is the correct directory
-      query:      while_looking_for_this,
-      subject:    'Prompt IDs',
-      prompt:     'Select one: '
-    )
-
-    # Run the fzf instance and get the selected result
-    result = fzf_instance.run
-
-    exit unless result
-
-    result
-  end
-
-
-
-
-  def create_prompt(prompt_id)
-    @prompt = PromptManager::Prompt.create(id: prompt_id)
-    # TODO: consider a configurable prompt template
-    #       ERB ???
-  end
-
-
-  def edit_prompt
-    # FIXME: replace with the editor from the configuration
-    
-    @editor   = AIA::Subl.new(
-                  file: @prompt.path
-                )
-    
-    @editor.run # blocks until file is closed
-
-    AIA.config[:edit?] = false # turn off the --edit switch
-
-    # reload the edited prompt
-    @prompt = PromptManager::Prompt.get(id: @prompt.id)
-  end
-
-
-  def show_prompt_without_comments
-    puts remove_comments.wrap(indent: 4)
-  end
-
-
-  # removes comments and directives
-  def remove_comments
-    lines           = @prompt.text
-                        .split("\n")
-                        .reject{|a_line| 
-                          a_line.strip.start_with?('#') ||
-                          a_line.strip.start_with?('//')
-                        }
-
-    # Remove empty lines at the start of the prompt
-    #
-    lines = lines.drop_while(&:empty?)
-
-    # Drop all the lines at __END__ and after
-    #
-    logical_end_inx = lines.index("__END__")
-
-    if logical_end_inx
-      lines[0...logical_end_inx] # NOTE: ... means to not include last index
-    else
-      lines
-    end.join("\n") 
-  end
-end

data/lib/aia/tools/backend_common.rb

@@ -1,58 +0,0 @@
-# aia/lib/aia/tools/backend_common.rb
-
-# Used by both the AIA::Mods and AIA::Sgpt classes
-
-module AIA::BackendCommon
-  attr_accessor :command, :text, :files, :parameters
-
-  def initialize(text: "", files: [])
-    @text       = text
-    @files      = files
-    @parameters = self.class::DEFAULT_PARAMETERS.dup
-    build_command
-  end
-
-
-  def sanitize(input)
-    Shellwords.escape(input)
-  end
-
-
-  def build_command
-    @parameters += " --model #{AIA.config.model} " if AIA.config.model
-    @parameters += AIA.config.extra
-
-    set_parameter_from_directives
-
-    @command = "#{meta.name} #{@parameters} "
-    @command += sanitize(text)
-
-    puts @command if AIA.config.debug?
-
-    @command
-  end
-
-
-  def set_parameter_from_directives
-    AIA.config.directives.each do |entry|
-      directive, value = entry
-      if self.class::DIRECTIVES.include?(directive)
-        @parameters += " --#{directive} #{sanitize(value)}" unless @parameters.include?(directive)
-      end
-    end
-  end
-
-
-  def run
-    case @files.size
-    when 0
-      @result = `#{build_command}`
-    when 1
-      @result = `#{build_command} < #{@files.first}`
-    else
-      @result = %x[cat #{@files.join(' ')} | #{build_command}]
-    end
-
-    @result
-  end
-end