aia 0.9.6 → 0.9.8

data/lib/aia/session.rb

@@ -1,149 +1,208 @@
 # lib/aia/session.rb
 
-require 'tty-spinner'
-require 'tty-screen'
-require 'reline'
-require 'prompt_manager'
-require 'json'
-require 'fileutils'
-require 'amazing_print'
-require_relative 'directive_processor'
-require_relative 'history_manager'
-require_relative 'context_manager'
-require_relative 'ui_presenter'
-require_relative 'chat_processor_service'
-require_relative 'prompt_handler'
-require_relative 'utility'
+require "tty-spinner"
+require "tty-screen"
+require "reline"
+require "prompt_manager"
+require "json"
+require "fileutils"
+require "amazing_print"
+require_relative "directive_processor"
+require_relative "history_manager"
+require_relative "context_manager"
+require_relative "ui_presenter"
+require_relative "chat_processor_service"
+require_relative "prompt_handler"
+require_relative "utility"
 
 module AIA
   class Session
     KW_HISTORY_MAX = 5 # Maximum number of history entries per keyword
-    TERSE_PROMPT   = "\nKeep your response short and to the point.\n"
+    TERSE_PROMPT = "\nKeep your response short and to the point.\n"
 
     def initialize(prompt_handler)
-      @prompt_handler  = prompt_handler
+      @prompt_handler = prompt_handler
       @chat_prompt_id = nil  # Initialize to nil
+      @include_context_flag = true  # Flag to include context files, stdin, etc. only once
 
       # 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
+        prompt_instance = nil
         @history_manager = nil
       elsif AIA.chat? && AIA.config.prompt_id.empty?
-        prompt_instance  = nil
+        prompt_instance = nil
         @history_manager = nil
       else
-        prompt_instance  = @prompt_handler.get_prompt(AIA.config.prompt_id)
+        prompt_instance = @prompt_handler.get_prompt(AIA.config.prompt_id)
         @history_manager = HistoryManager.new(prompt: prompt_instance)
       end
 
-      @context_manager     = ContextManager.new(system_prompt: AIA.config.system_prompt)
-      @ui_presenter        = UIPresenter.new
+      @context_manager = ContextManager.new(system_prompt: AIA.config.system_prompt)
+      @ui_presenter = UIPresenter.new
       @directive_processor = DirectiveProcessor.new
-      @chat_processor      = ChatProcessorService.new(@ui_presenter, @directive_processor)
+      @chat_processor = ChatProcessorService.new(@ui_presenter, @directive_processor)
 
-      if AIA.config.out_file && !AIA.append? && File.exist?(AIA.config.out_file)
-        File.open(AIA.config.out_file, 'w') {} # Truncate the file
+      if AIA.config.out_file && !AIA.config.out_file.nil? && !AIA.append? && File.exist?(AIA.config.out_file)
+        File.open(AIA.config.out_file, "w") { } # Truncate the file
       end
     end
 
-    # Starts the session, processing the initial prompt and handling user
-    # interactions. It manages the flow of prompts, context, and responses.
+    # Starts the session, processing all prompts in the pipeline and then
+    # optionally starting an interactive chat session.
     def start
-      prompt_id = AIA.config.prompt_id
-      role_id   = AIA.config.role
+      # Handle special chat-only cases first
+      if should_start_chat_immediately?
+        AIA::Utility.robot
+        start_chat
+        return
+      end
 
-      # Handle chat mode
+      # Process all prompts in the pipeline
+      process_all_prompts
+
+      # Start chat mode after all prompts are processed
       if AIA.chat?
         AIA::Utility.robot
-        # 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
+        @ui_presenter.display_separator
+        start_chat(skip_context_files: true)
+      end
+    end
+
+    private
+
+    # Check if we should start chat immediately without processing any prompts
+    def should_start_chat_immediately?
+      return false unless AIA.chat?
+
+      # If pipeline is empty or only contains empty prompt_ids, go straight to chat
+      AIA.config.pipeline.empty? || AIA.config.pipeline.all? { |id| id.nil? || id.empty? }
+    end
+
+    # Process all prompts in the pipeline sequentially
+    def process_all_prompts
+      prompt_count = 0
+      total_prompts = AIA.config.pipeline.size
+
+      until AIA.config.pipeline.empty?
+        prompt_count += 1
+        prompt_id = AIA.config.pipeline.shift
+
+        puts "\n--- Processing prompt #{prompt_count}/#{total_prompts}: #{prompt_id} ---" if AIA.verbose? && total_prompts > 1
+
+        process_single_prompt(prompt_id)
       end
+    end
+
+    # Process a single prompt with all its requirements
+    def process_single_prompt(prompt_id)
+      role_id = AIA.config.role
 
+      # Skip empty prompt IDs
+      return if prompt_id.nil? || prompt_id.empty?
 
-      # --- Get and process the initial prompt ---
+      # Get and process the prompt
       begin
         prompt = @prompt_handler.get_prompt(prompt_id, role_id)
       rescue StandardError => e
-        puts "Error: #{e.message}"
+        puts "Error processing prompt '#{prompt_id}': #{e.message}"
         return
       end
 
-      # Collect variable values if needed
+      # Collect variable values (only when flag is true)
+      if @include_context_flag
+        collect_variable_values(prompt)
+        enhance_prompt_with_extras(prompt)
+      end
+
+      # Get final prompt text
+      prompt_text = prompt.to_s
+
+      # Add context files (only when flag is true)
+      if @include_context_flag
+        prompt_text = add_context_files(prompt_text)
+        # SMELL: TODO? empty the AIA.config.context_files array
+        @include_context_flag = false
+      end
+
+      # Send prompt to AI and get response
+      send_prompt_and_get_response(prompt_text)
+    end
+
+    # Collect variable values from user input
+    def collect_variable_values(prompt)
       variables = prompt.parameters.keys
+      return if variables.nil? || variables.empty?
 
-      if variables && !variables.empty?
-        variable_values = {}
-        history_manager = AIA::HistoryManager.new prompt: prompt
-
-        variables.each do |var_name|
-          # History is based on the prompt ID and the variable name (without brackets)
-          history = prompt.parameters[var_name]
-
-          # Ask the user for the variable
-          value = history_manager.request_variable_value(
-            variable_name:  var_name,
-            history_values: history
-          )
-          # Store the value using the original BRACKETED key from prompt.parameters
-          if history.include? value
-            history.delete(value)
-          end
-          history << value
-          if history.size > HistoryManager::MAX_VARIABLE_HISTORY
-            history.shift
-          end
-          variable_values[var_name] = history
-        end
+      variable_values = {}
+      history_manager = AIA::HistoryManager.new prompt: prompt
+
+      variables.each do |var_name|
+        history = prompt.parameters[var_name]
 
-        # Assign collected values back for prompt_manager substitution
-        prompt.parameters = variable_values
+        value = history_manager.request_variable_value(
+          variable_name: var_name,
+          history_values: history,
+        )
+
+        # Update history
+        history.delete(value) if history.include?(value)
+        history << value
+        history.shift if history.size > HistoryManager::MAX_VARIABLE_HISTORY
+
+        variable_values[var_name] = history
       end
 
+      prompt.parameters = variable_values
+    end
+
+    # Add terse instructions, stdin content, and executable prompt file content
+    def enhance_prompt_with_extras(prompt)
       # Add terse instruction if needed
-      if AIA.terse?
-        prompt.text << TERSE_PROMPT
-      end
+      prompt.text << TERSE_PROMPT if AIA.terse?
 
-      prompt.save
-      # Substitute variables and get final prompt text
-      prompt_text = prompt.to_s
+      # Add STDIN content
+      if AIA.config.stdin_content && !AIA.config.stdin_content.strip.empty?
+        prompt.text << "\n\n" << AIA.config.stdin_content
+      end
 
-      # Add context files if any
-      if AIA.config.context_files && !AIA.config.context_files.empty?
-        context = AIA.config.context_files.map do |file|
-          File.read(file) rescue "Error reading file: #{file}"
-        end.join("\n\n")
-        prompt_text = "#{prompt_text}\n\nContext:\n#{context}"
+      # Add executable prompt file content
+      if AIA.config.executable_prompt_file
+        prompt.text << "\n\n" << File.read(AIA.config.executable_prompt_file)
+          .lines[1..]
+          .join
       end
+    end
 
-      # Add initial user prompt to context *before* sending to AI
-      @context_manager.add_to_context(role: 'user', content: prompt_text)
+    # Add context files to prompt text
+    def add_context_files(prompt_text)
+      return prompt_text unless AIA.config.context_files && !AIA.config.context_files.empty?
+
+      context = AIA.config.context_files.map do |file|
+        File.read(file) rescue "Error reading file: #{file}"
+      end.join("\n\n")
+
+      "#{prompt_text}\n\nContext:\n#{context}"
+    end
 
-      # Process the initial prompt
+    # Send prompt to AI and handle the response
+    def send_prompt_and_get_response(prompt_text)
+      # Add prompt to conversation context
+      @context_manager.add_to_context(role: "user", content: prompt_text)
+
+      # Process the prompt
       @ui_presenter.display_thinking_animation
-      # Send the current context (which includes the user prompt)
       response = @chat_processor.process_prompt(@context_manager.get_context)
 
       # Add AI response to context
-      @context_manager.add_to_context(role: 'assistant', content: response)
+      @context_manager.add_to_context(role: "assistant", content: response)
 
       # Output the response
-      @chat_processor.output_response(response) # Handles display
-
-      # Process next prompts/pipeline (if any)
-      @chat_processor.process_next_prompts(response, @prompt_handler)
+      @chat_processor.output_response(response)
 
-      # --- Enter chat mode AFTER processing initial prompt ---
-      if AIA.chat?
-        @ui_presenter.display_separator # Add separator
-        start_chat(skip_context_files: true) # start_chat will use the now populated context
+      # Process any directives in the response
+      if @directive_processor.directive?(response)
+        directive_result = @directive_processor.process(response, @context_manager)
+        puts "\nDirective output: #{directive_result}" if directive_result && !directive_result.strip.empty?
       end
     end
 
@@ -156,14 +215,14 @@ module AIA
 
       # Generate chat prompt ID
       now = Time.now
-      @chat_prompt_id = "chat_#{now.strftime('%Y%m%d_%H%M%S')}"
+      @chat_prompt_id = "chat_#{now.strftime("%Y%m%d_%H%M%S")}"
 
       # Create the temporary prompt
       begin
         # Create the unique? prompt ID in the file storage system with its initial text
         PromptManager::Prompt.create(
           id: @chat_prompt_id,
-          text: "Today's date is #{now.strftime('%Y-%m-%d')} and the current time is #{now.strftime('%H:%M:%S')}"
+          text: "Today's date is #{now.strftime("%Y-%m-%d")} and the current time is #{now.strftime("%H:%M:%S")}",
         )
 
         # Capture self for the handlers
@@ -171,7 +230,7 @@ module AIA
 
         # Set up cleanup handlers only after prompt is created
         at_exit { session_instance.send(:cleanup_chat_prompt) }
-        Signal.trap('INT') {
+        Signal.trap("INT") {
           session_instance.send(:cleanup_chat_prompt)
           exit
         }
@@ -181,9 +240,9 @@ module AIA
         @chat_prompt = PromptManager::Prompt.new(
           id: @chat_prompt_id,
           directives_processor: @directive_processor,
-          erb_flag:             true,
-          envar_flag:           true,
-          external_binding:     binding,
+          erb_flag: true,
+          envar_flag: true,
+          external_binding: binding,
         )
 
         Reline::HISTORY.clear
@@ -196,14 +255,14 @@ module AIA
 
           if !context.empty?
             # Add context files content to context
-            @context_manager.add_to_context(role: 'user', content: context)
+            @context_manager.add_to_context(role: "user", content: context)
 
             # Process the context
             @ui_presenter.display_thinking_animation
             response = @chat_processor.process_prompt(@context_manager.get_context)
 
             # Add AI response to context
-            @context_manager.add_to_context(role: 'assistant', content: response)
+            @context_manager.add_to_context(role: "assistant", content: response)
 
             # Output the response
             @chat_processor.output_response(response)
@@ -216,18 +275,18 @@ module AIA
         if !STDIN.tty?
           original_stdin = STDIN.dup
           piped_input = STDIN.read.strip
-          STDIN.reopen('/dev/tty')
+          STDIN.reopen("/dev/tty")
 
           if !piped_input.empty?
             @chat_prompt.text = piped_input
             processed_input = @chat_prompt.to_s
 
-            @context_manager.add_to_context(role: 'user', content: processed_input)
+            @context_manager.add_to_context(role: "user", content: processed_input)
 
             @ui_presenter.display_thinking_animation
             response = @chat_processor.process_prompt(@context_manager.get_context)
 
-            @context_manager.add_to_context(role: 'assistant', content: response)
+            @context_manager.add_to_context(role: "assistant", content: response)
             @chat_processor.output_response(response)
             @chat_processor.speak(response) if AIA.speak?
             @ui_presenter.display_separator
@@ -240,10 +299,10 @@ module AIA
         loop do
           follow_up_prompt = @ui_presenter.ask_question
 
-          break if follow_up_prompt.nil? || follow_up_prompt.strip.downcase == 'exit' || follow_up_prompt.strip.empty?
+          break if follow_up_prompt.nil? || follow_up_prompt.strip.downcase == "exit" || follow_up_prompt.strip.empty?
 
           if AIA.config.out_file
-            File.open(AIA.config.out_file, 'a') do |file|
+            File.open(AIA.config.out_file, "a") do |file|
               file.puts "\nYou: #{follow_up_prompt}"
             end
           end
@@ -251,28 +310,28 @@ module AIA
           if @directive_processor.directive?(follow_up_prompt)
             directive_output = @directive_processor.process(follow_up_prompt, @context_manager)
 
-            if follow_up_prompt.strip.start_with?('//clear')
-               # The directive processor has called context_manager.clear_context
-               # but we need a more aggressive approach to fully clear all context
+            if follow_up_prompt.strip.start_with?("//clear")
+              # The directive processor has called context_manager.clear_context
+              # but we need a more aggressive approach to fully clear all context
 
-               # First, clear the context manager's context
-               @context_manager.clear_context(keep_system_prompt: true)
+              # First, clear the context manager's context
+              @context_manager.clear_context(keep_system_prompt: true)
 
-               # Second, try clearing the client's context
-               if AIA.config.client && AIA.config.client.respond_to?(:clear_context)
-                 AIA.config.client.clear_context
-               end
+              # Second, try clearing the client's context
+              if AIA.config.client && AIA.config.client.respond_to?(:clear_context)
+                AIA.config.client.clear_context
+              end
 
-               # Third, completely reinitialize the client to ensure fresh state
-               # This is the most aggressive approach to ensure no context remains
-               begin
-                 AIA.config.client = AIA::RubyLLMAdapter.new
-               rescue => e
-                 STDERR.puts "Error reinitializing client: #{e.message}"
-               end
+              # Third, completely reinitialize the client to ensure fresh state
+              # This is the most aggressive approach to ensure no context remains
+              begin
+                AIA.config.client = AIA::RubyLLMAdapter.new
+              rescue => e
+                STDERR.puts "Error reinitializing client: #{e.message}"
+              end
 
-               @ui_presenter.display_info("Chat context cleared.")
-               next
+              @ui_presenter.display_info("Chat context cleared.")
+              next
             elsif directive_output.nil? || directive_output.strip.empty?
               next
             else
@@ -284,19 +343,18 @@ module AIA
           @chat_prompt.text = follow_up_prompt
           processed_prompt = @chat_prompt.to_s
 
-          @context_manager.add_to_context(role: 'user', content: processed_prompt)
+          @context_manager.add_to_context(role: "user", content: processed_prompt)
           conversation = @context_manager.get_context
 
           @ui_presenter.display_thinking_animation
           response = @chat_processor.process_prompt(conversation)
 
           @ui_presenter.display_ai_response(response)
-          @context_manager.add_to_context(role: 'assistant', content: response)
+          @context_manager.add_to_context(role: "assistant", content: response)
           @chat_processor.speak(response)
 
           @ui_presenter.display_separator
         end
-
       ensure
         @ui_presenter.display_chat_end
       end

data/lib/aia/ui_presenter.rb

@@ -26,7 +26,7 @@ module AIA
       puts "\nAI: "
       format_chat_response(response)
 
-      if AIA.config.out_file
+      if AIA.config.out_file && !AIA.config.out_file.nil?
         File.open(AIA.config.out_file, 'a') do |file|
           file.puts "\nAI: "
           format_chat_response(response, file)

data/lib/aia.rb

@@ -6,8 +6,6 @@
 
 require 'ruby_llm'
 require 'ruby_llm/mcp'
-require 'shared_tools'
-require 'shared_tools/ruby_llm'
 require 'prompt_manager'
 
 
@@ -41,6 +39,10 @@ require_relative 'aia/session'
 module AIA
   at_exit do
     STDERR.puts "Exiting AIA application..."
+    # Clean up temporary STDIN file if it exists
+    if @config&.stdin_temp_file && File.exist?(@config.stdin_temp_file)
+      File.unlink(@config.stdin_temp_file)
+    end
   end
 
   @config = nil

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: aia
 version: !ruby/object:Gem::Version
-  version: 0.9.6
+  version: 0.9.8
 platform: ruby
 authors:
 - Dewayne VanHoozer
@@ -10,7 +10,7 @@ cert_chain: []
 date: 1980-01-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
-  name: amazing_print
+  name: activesupport
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -24,7 +24,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: faraday
+  name: amazing_print
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -38,49 +38,49 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: prompt_manager
+  name: faraday
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.5.6
+        version: '0'
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.5.6
+        version: '0'
 - !ruby/object:Gem::Dependency
-  name: ruby_llm
+  name: prompt_manager
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 1.3.1
+        version: 0.5.7
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 1.3.1
+        version: 0.5.7
 - !ruby/object:Gem::Dependency
-  name: ruby_llm-mcp
+  name: ruby_llm
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 1.3.1
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0'
+        version: 1.3.1
 - !ruby/object:Gem::Dependency
-  name: reline
+  name: ruby_llm-mcp
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -94,7 +94,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0'
 - !ruby/object:Gem::Dependency
-  name: shared_tools
+  name: reline
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -318,6 +318,7 @@ files:
 - examples/headlines
 - examples/tools/edit_file.rb
 - examples/tools/list_files.rb
+- examples/tools/pdf_page_reader.rb
 - examples/tools/read_file.rb
 - examples/tools/run_shell_command.rb
 - images/aia.png