aia 0.8.1 → 0.8.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 54144943e09896335df01ec5d720b7f753aa183e5a01eebc866f6272b45f937b
-  data.tar.gz: c345ed440cf3ebb81f6c63d984731fb169e3893b6094297c9ee635c3cff00c2b
+  metadata.gz: 15a24480e7b5958570ae21ef3e3f958cdb991a1f03cb88ebe3e8ca52a6029f9a
+  data.tar.gz: dcef2697a2d2fdec85e371421a5c16a48c70d3688ea7bb6646f001df5e795d68
 SHA512:
-  metadata.gz: 4f293b0ec3c8e45c9e07783cd5f453cc05d64ca5b804c1ef662fd4b9f23f0cbdc3f603801b667adf406bc549a856c4ae973739115d5f35884a6df9f3c9fab1b2
-  data.tar.gz: 399e4581d7459a1df068a7cc739a05782bc159dabc7f5ac7f6bedf5e4a485ea4ebae709347dc0a791d1290e2bbffacf32615d7364f700eb8fae57840ef73eaf6
+  metadata.gz: 97f8c423247051e31eef8b6a66fdde22fe72438097da41bb30ec53ef56508266e07f5502fa85cf8cfb63ebad22e995d7c7b4cbe55745f20e87b2b535cbd0b55c
+  data.tar.gz: c48ebed9374edc9a8dcf86b2dd592031c1bef02b60f5d61e113a9c90bf31865030231ebc267a6308ca46d71f3e864d02e44bfc93f221112102309ddf2186d563

data/.version

@@ -1 +1 @@
-0.8.1
+0.8.3

data/CHANGELOG.md

@@ -3,6 +3,18 @@
 
 ## Released
 
+### [0.8.2] 2025-04-18
+- fixed problems with pre-loaded context and chat repl
+- piped content into `aia --chat` is now a part of the context/instructions
+- content via "aia --chat < some_file" is added to the context/instructions
+- `aia --chat context_file.txt context_file2.txt` now works
+- `aia --chat prompt_id context)file.txt` also works
+
+
+
+### [0.8.1] 2025-04-17
+- bumped version to 0.8.1 after correcting merge conflicts
+
 ### [0.8.0] WIP - 2025-04-15
 - Updated PromptManager to v0.5.1 which has some of the functionality that was originally developed in the AIA.
 - Enhanced README.md to include a comprehensive table of configuration options with defaults and associated environment variables.

data/README.md

@@ -7,7 +7,7 @@
      (\____/) AI Assistant    interaction with AI models. It automates the
       (_oo_)   Fancy LLM      management of pre-compositional prompts and
         (O)     is Online     executes generative AI (Gen-AI) commands on those
-      __||__    \)            prompts. AIA includes enhanced feathres such as
+      __||__    \)            prompts. AIA includes enhanced features such as
     [/______\]  /               * embedded directives * shell integration
    / \__AI__/ \/                * embedded Ruby       * history management
   /    /__\                     * interactive chat    * prompt workflows

data/lib/aia/config.rb

@@ -146,11 +146,16 @@ module AIA
 
       exit if and_exit
 
-      # Only require a prompt_id if we're not in chat mode and not using fuzzy search
-      if !config.chat && !config.fuzzy && config.prompt_id.empty?
-        STDERR.puts "Error: A prompt ID is required unless using --chat or --fuzzy. Use -h or --help for help."
+      # Only require a prompt_id if we're not in chat mode, not using fuzzy search, and no context files
+      if !config.chat && !config.fuzzy && config.prompt_id.empty? && (!config.context_files || config.context_files.empty?)
+        STDERR.puts "Error: A prompt ID is required unless using --chat, --fuzzy, or providing context files. Use -h or --help for help."
         exit 1
       end
+      
+      # If we're in chat mode with context files but no prompt_id, that's valid
+      if config.chat && config.prompt_id.empty? && config.context_files && !config.context_files.empty?
+        # This is a valid use case - no action needed
+      end
 
       # Tailor the PromptManager::Prompt
       if config.parameter_regex
@@ -192,7 +197,9 @@ module AIA
       config = OpenStruct.new
 
       opt_parser = OptionParser.new do |opts|
-        opts.banner = "Usage: aia [options] PROMPT_ID [CONTEXT_FILE]*"
+        opts.banner = "Usage: aia [options] [PROMPT_ID] [CONTEXT_FILE]*\n" +
+                     "       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
           config.chat = true
@@ -391,14 +398,24 @@ module AIA
         exit 1
       end
 
-      # First remaining arg is the prompt ID
+      # Handle remaining args
       unless remaining_args.empty?
-        config.prompt_id = remaining_args.shift
+        # If in chat mode and all args are existing files, treat them all as context files
+        if config.chat && remaining_args.all? { |arg| File.exist?(arg) }
+          config.context_files = remaining_args
+        # If first arg is empty string and we're in chat mode, treat all args as context files
+        elsif config.chat && remaining_args.first == ""
+          remaining_args.shift # Remove the empty string
+          config.context_files = remaining_args unless remaining_args.empty?
+        else
+          # First remaining arg is the prompt ID
+          config.prompt_id = remaining_args.shift
+          
+          # Remaining args are context files
+          config.context_files = remaining_args unless remaining_args.empty?
+        end
       end
 
-      # Remaining args are context files
-      config.context_files = remaining_args unless remaining_args.empty?
-
 
       config
     end

data/lib/aia/directive_processor.rb

@@ -63,7 +63,7 @@ module AIA
       if EXCLUDED_METHODS.include?(method_name)
         return "Error: #{method_name} is not a valid directive: #{key}"
       elsif respond_to?(method_name, true)
-        return send(method_name, args)
+        return send(method_name, args, context_manager)
       else
         return "Error: Unknown directive '#{key}'"
       end
@@ -125,8 +125,9 @@ module AIA
     end
 
     desc "Inserts the contents of a file  Example: //include path/to/file"
-    def include(args)
-      file_path = args.shift
+    def include(args, context_manager=nil)
+      # echo takes care of envars and tilde expansion
+      file_path = `echo #{args.shift}`.strip
 
       if @included_files.include?(file_path)
         ""
@@ -143,7 +144,7 @@ module AIA
     alias_method :import,       :include
 
     desc "Without arguments it will print a list of all config items and their values _or_ //config item (for one item's value) _or_ //config item = value (to set a value of an item)"
-    def config(args = [])
+    def config(args = [], context_manager=nil)
       args = Array(args)
 
       if args.empty?
@@ -171,33 +172,42 @@ module AIA
     alias_method :cfg, :config
 
     desc "Shortcut for //config top_p _and_ //config top_p = value"
-    def top_p(*args)
-      send(:config, args.prepend('top_p'))
+    def top_p(args, context_manager=nil)
+      send(:config, args.prepend('top_p'), context_manager)
     end
     alias_method :topp, :top_p
 
+    desc "Review the current context"
+    def review(args, context_manager=nil)
+      ap context_manager.get_context
+      ''
+    end
+    alias_method :context, :review
+
     desc "Shortcut for //config model _and_ //config model = value"
-    def model(*args)
-      send(:config, args.prepend('model'))
+    def model(args, context_manager=nil)
+      send(:config, args.prepend('model'), context_manager)
     end
 
     desc "Shortcut for //config temperature _and_ //config temperature = value"
-    def temperature(*args)
-      send(:config, args.prepend('temperature'))
+    def temperature(args, context_manager=nil)
+      send(:config, args.prepend('temperature'), context_manager)
     end
     alias_method :temp, :temperature
 
     desc "Clears the conversation history (aka context) same as //config clear = true"
-    def clear(args, context_manager)
+    def clear(args, context_manager=nil)
       if context_manager.nil?
         return "Error: Context manager not available for //clear directive."
       end
+
       context_manager.clear_context
-      nil
+
+      ''
     end
 
     desc "Shortcut for a one line of ruby code; result is added to the context"
-    def ruby(*args)
+    def ruby(args, context_manager=nil)
       ruby_code = args.join(' ')
 
       begin
@@ -211,25 +221,34 @@ module AIA
     end
     alias_method :rb, :ruby
 
+
+    desc "Executes one line of shell code; result is added to the context"
+    def shell(args, context_manager=nil)
+      shell_code = args.join(' ')
+
+      `#{shell_code}`
+    end
+    alias_method :sh, :shell
+
     desc "Use the system's say command to speak text //say some text"
-    def say(*args)
+    def say(args, context_manager=nil)
       `say #{args.join(' ')}`
       ""
     end
 
     desc "Inserts an instruction to keep responses short and to the point."
-    def terse(*args)
+    def terse(args, context_manager=nil)
       AIA::Session::TERSE_PROMPT
     end
 
     desc "Display the ASCII art AIA robot."
-    def robot(*args)
+    def robot(args, context_manager=nil)
       AIA::Utility.robot
       ""
     end
 
     desc "Generates this help content"
-    def help(*args)
+    def help(args=nil, context_manager=nil)
       puts
       puts "Available Directives"
       puts "===================="

data/lib/aia/session.rb

@@ -23,7 +23,11 @@ module AIA
     def initialize(prompt_handler)
       @prompt_handler  = prompt_handler
 
-      if AIA.chat? && AIA.config.prompt_id.empty?
+      # Special handling for chat mode with context files but no prompt ID
+      if AIA.chat? && AIA.config.prompt_id.empty? && AIA.config.context_files && !AIA.config.context_files.empty?
+        prompt_instance  = nil
+        @history_manager = nil
+      elsif AIA.chat? && AIA.config.prompt_id.empty?
         prompt_instance  = nil
         @history_manager = nil
       else
@@ -47,10 +51,15 @@ module AIA
       prompt_id = AIA.config.prompt_id
       role_id   = AIA.config.role
 
-      # Handle chat mode *only* if NO initial prompt is given
+      # Handle chat mode
       if AIA.chat?
         AIA::Utility.robot
-        if prompt_id.empty? && role_id.empty?
+        # If we're in chat mode with only context files, go straight to chat
+        if prompt_id.empty? && role_id.empty? && AIA.config.context_files && !AIA.config.context_files.empty?
+          start_chat
+          return
+        elsif prompt_id.empty? && role_id.empty?
+          # Even with an empty prompt_id, we might have context files
           start_chat
           return
         end
@@ -137,12 +146,12 @@ module AIA
       # --- Enter chat mode AFTER processing initial prompt ---
       if AIA.chat?
         @ui_presenter.display_separator # Add separator
-        start_chat # start_chat will use the now populated context
+        start_chat(skip_context_files: true) # start_chat will use the now populated context
       end
     end
 
     # Starts the interactive chat session.
-    def start_chat
+    def start_chat(skip_context_files: false)
       # Consider if display_chat_header is needed if robot+separator already shown
       # For now, let's keep it, maybe add an indicator message
       puts "\nEntering interactive chat mode..."
@@ -150,11 +159,67 @@ module AIA
 
       Reline::HISTORY.clear # Keep Reline history for user input editing, separate from chat context
 
+      # Load context files if any and not skipping
+      if !skip_context_files && AIA.config.context_files && !AIA.config.context_files.empty?
+        context_content = AIA.config.context_files.map do |file|
+          File.read(file) rescue "Error reading file: #{file}"
+        end.join("\n\n")
+
+        if !context_content.empty?
+          # Add context files content to context
+          @context_manager.add_to_context(role: 'user', content: context_content)
+
+          # Process the context
+          operation_type = @chat_processor.determine_operation_type(AIA.config.model)
+          @ui_presenter.display_thinking_animation
+          response = @chat_processor.process_prompt(@context_manager.get_context, operation_type)
+
+          # Add AI response to context
+          @context_manager.add_to_context(role: 'assistant', content: response)
+
+          # Output the response
+          @chat_processor.output_response(response)
+          @chat_processor.speak(response)
+          @ui_presenter.display_separator
+        end
+      end
+
+      # Check for piped input (STDIN not a TTY and has data)
+      if !STDIN.tty?
+        # Save the original STDIN
+        orig_stdin = STDIN.dup
+
+        # Read the piped input
+        piped_input = STDIN.read.strip
+
+        # Reopen STDIN to the terminal
+        STDIN.reopen('/dev/tty')
+
+        if !piped_input.empty?
+          # Add piped input to context
+          @context_manager.add_to_context(role: 'user', content: piped_input)
+
+          # Process the piped input
+          operation_type = @chat_processor.determine_operation_type(AIA.config.model)
+          @ui_presenter.display_thinking_animation
+          response = @chat_processor.process_prompt(@context_manager.get_context, operation_type)
+
+          # Add AI response to context
+          @context_manager.add_to_context(role: 'assistant', content: response)
+
+          # Output the response
+          @chat_processor.output_response(response)
+          @chat_processor.speak(response)
+          @ui_presenter.display_separator
+        end
+      end
+
       loop do
         # Get user input
         prompt = @ui_presenter.ask_question
 
 
+
         break if prompt.nil? || prompt.strip.downcase == 'exit' || prompt.strip.empty?
 
         if AIA.config.out_file
@@ -167,7 +232,7 @@ module AIA
           directive_output = @directive_processor.process(prompt, @context_manager) # Pass context_manager
 
           # Add check for specific directives like //clear that might modify context
-          if prompt.strip.start_with?('//clear', '#!clear:')
+          if prompt.strip.start_with?('//clear')
              # Context is likely cleared within directive_processor.process now
              # or add @context_manager.clear_context here if not handled internally
              @ui_presenter.display_info("Chat context cleared.")
@@ -191,10 +256,6 @@ module AIA
         # Use ContextManager to get the conversation
         conversation = @context_manager.get_context # System prompt handled internally
 
-        # FIXME: remove this comment once verified
-        # is conversation the same thing as the context for a chat session? YES
-        # if so need to somehow delete it when the //clear directive is entered. - Addressed above/in DirectiveProcessor
-
         operation_type = @chat_processor.determine_operation_type(AIA.config.model)
         @ui_presenter.display_thinking_animation
         response = @chat_processor.process_prompt(conversation, operation_type)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: aia
 version: !ruby/object:Gem::Version
-  version: 0.8.1
+  version: 0.8.3
 platform: ruby
 authors:
 - Dewayne VanHoozer
@@ -292,7 +292,6 @@ files:
 - lib/aia/prompt_handler.rb
 - lib/aia/session.rb
 - lib/aia/shell_command_executor.rb
-- lib/aia/tools/ai_client_backend.rb
 - lib/aia/ui_presenter.rb
 - lib/aia/utility.rb
 - lib/aia/version.rb

data/lib/aia/tools/ai_client_backend.rb

@@ -1,92 +0,0 @@
-# lib/aia/tools/ai_client_backend.rb
-
-# This is WIP in the `develop` branch
-# do not use.
-
-require 'ai_client'
-require_relative 'backend_common'
-
-class AIA::AiClientBackend < AIA::Tools
-  include AIA::BackendCommon
-
-  meta(
-    name:     'ai_client',
-    role:     :backend,
-    desc:     'AI Client integration for unified model access',
-    url:      'https://github.com/path/to/ai_client', # TODO: Update URL
-    install:  'gem install ai_client',
-  )
-
-  attr_reader :client, :raw_response
-  
-  DEFAULT_PARAMETERS = ''
-  DIRECTIVES = %w[
-    model
-    temperature
-    max_tokens
-    top_p
-    frequency_penalty
-    presence_penalty
-  ]
-
-  def initialize(text: "", files: [])
-    super
-    @client = AiClient.new
-  end
-
-  def build_command
-    # No-op - ai_client doesn't use command line
-    @parameters = ""
-  end
-
-  def run
-    handle_model(AIA.config.model)
-  rescue => e
-    puts "Error handling model #{AIA.config.model}: #{e.message}"
-  end
-
-  private
-
-  def handle_model(model_name)
-    case model_name
-    when /vision/
-      image2text
-    when /^gpt/
-      text2text
-    when /^dall-e/
-      text2image
-    when /^tts/
-      text2audio
-    when /^whisper/
-      audio2text
-    else
-      raise "Unsupported model: #{model_name}"
-    end
-  end
-
-  def text2text
-    response = client.complete(
-      prompt: text,
-      model: AIA.config.model,
-      temperature: AIA.config.temp
-    )
-    response.completion
-  end
-
-  # Placeholder methods to maintain API compatibility
-  def image2text
-    raise "Not yet implemented for ai_client"
-  end
-
-  def text2image
-    raise "Not yet implemented for ai_client"
-  end
-
-  def text2audio
-    raise "Not yet implemented for ai_client"
-  end
-
-  def audio2text
-    raise "Not yet implemented for ai_client"
-  end
-end