aia 0.8.0 → 0.8.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 3a77cfcc71cea66fafd6eb8c17666eee983f7903a60f528e3bb396d8bdac0488
-  data.tar.gz: c9424c2a40f886a231b10fe569276ba0b8e3e9feb435b38a0257852a91103629
+  metadata.gz: fd634957d83d1ea9d6a15fd9b82e5e535ea54952a16505490e49dd89aa1cd09e
+  data.tar.gz: 3c53dddc42841b8bd8106a7b108ebc86a5b358a1784edef20a08ba58c659ab45
 SHA512:
-  metadata.gz: 6f96bdaafd932c34637cbd05415dc4a5c856ecaf386780426d10db5f1dade03a2368c7a767698c385754467b3882bf50faa972a8dedfcd0a5b040777b86d4bb7
-  data.tar.gz: 0cfd3aa1b9750a7ae83dbb21ef737647f64b9cf67cb2a428864b7f22edbccaf317c9dbee205e2995cfdb3809b883f792722aac281a82869477caba3e70d04d23
+  metadata.gz: 39e70a7f9a50fbad4885e181fbaa7497d0a0d997efcf6354212f4c433e588598dc40716acd1842fe82ee11de6f5e22d5157a9fdb7fe5e5f9bdea03d912a7179b
+  data.tar.gz: bd7e23daeca3400394c6c0bdcb3b55eecf01fe61ee7a98441bc18a92e769e8a59953dda51800167f8a5212f5f7abc8657311d5a97df20d2b5e7338ec80279c4f

data/.version

@@ -1 +1 @@
-0.8.0
+0.8.2

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/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,22 +146,78 @@ 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..."
       @ui_presenter.display_chat_header
 
       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?

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: aia
 version: !ruby/object:Gem::Version
-  version: 0.8.0
+  version: 0.8.2
 platform: ruby
 authors:
 - Dewayne VanHoozer