aia 0.9.11 → 0.9.13

data/lib/aia/session.rb

@@ -313,26 +313,39 @@ module AIA
     def handle_piped_input
       return if STDIN.tty?
 
-      original_stdin = STDIN.dup
-      piped_input = STDIN.read.strip
-      STDIN.reopen("/dev/tty")
+      # Additional check: see if /dev/tty is available before attempting to use it
+      return unless File.exist?("/dev/tty") && File.readable?("/dev/tty") && File.writable?("/dev/tty")
 
-      return if piped_input.empty?
+      begin
+        original_stdin = STDIN.dup
+        piped_input = STDIN.read.strip
+        STDIN.reopen("/dev/tty")
 
-      @chat_prompt.text = piped_input
-      processed_input = @chat_prompt.to_s
+        return if piped_input.empty?
 
-      @context_manager.add_to_context(role: "user", content: processed_input)
+        @chat_prompt.text = piped_input
+        processed_input = @chat_prompt.to_s
 
-      @ui_presenter.display_thinking_animation
-      response = @chat_processor.process_prompt(@context_manager.get_context)
+        @context_manager.add_to_context(role: "user", content: processed_input)
 
-      @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
+        @ui_presenter.display_thinking_animation
+        response = @chat_processor.process_prompt(@context_manager.get_context)
+
+        @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
 
-      STDIN.reopen(original_stdin)
+        STDIN.reopen(original_stdin)
+      rescue Errno::ENXIO => e
+        # Handle case where /dev/tty is not available (e.g., in some containerized environments)
+        warn "Warning: Unable to handle piped input due to TTY unavailability: #{e.message}"
+        return
+      rescue StandardError => e
+        # Handle any other errors gracefully
+        warn "Warning: Error handling piped input: #{e.message}"
+        return
+      end
     end
 
     def run_chat_loop
@@ -359,11 +372,34 @@ module AIA
         conversation = @context_manager.get_context
 
         @ui_presenter.display_thinking_animation
-        response = @chat_processor.process_prompt(conversation)
+        response_data = @chat_processor.process_prompt(conversation)
+
+        # Handle new response format with metrics
+        if response_data.is_a?(Hash)
+          content = response_data[:content]
+          metrics = response_data[:metrics]
+          multi_metrics = response_data[:multi_metrics]
+        else
+          content = response_data
+          metrics = nil
+          multi_metrics = nil
+        end
 
-        @ui_presenter.display_ai_response(response)
-        @context_manager.add_to_context(role: "assistant", content: response)
-        @chat_processor.speak(response)
+        @ui_presenter.display_ai_response(content)
+        
+        # Display metrics if enabled and available (chat mode only)
+        if AIA.config.show_metrics
+          if multi_metrics
+            # Display metrics for each model in multi-model mode
+            @ui_presenter.display_multi_model_metrics(multi_metrics)
+          elsif metrics
+            # Display metrics for single model
+            @ui_presenter.display_token_metrics(metrics)
+          end
+        end
+        
+        @context_manager.add_to_context(role: "assistant", content: content)
+        @chat_processor.speak(content)
 
         @ui_presenter.display_separator
       end

data/lib/aia/ui_presenter.rb

@@ -117,5 +117,211 @@ module AIA
         yield
       end
     end
+
+    def display_token_metrics(metrics)
+      return unless metrics
+      
+      output_lines = []
+      output_lines << "═" * 55
+      output_lines << "Model: #{metrics[:model_id]}"
+      
+      if AIA.config.show_cost
+        output_lines.concat(format_metrics_with_cost(metrics))
+      else
+        output_lines.concat(format_metrics_basic(metrics))
+      end
+      
+      output_lines << "═" * 55
+      
+      # Output to STDOUT
+      output_lines.each { |line| puts line }
+      
+      # Also write to file if configured
+      if AIA.config.out_file && !AIA.config.out_file.nil?
+        File.open(AIA.config.out_file, 'a') do |file|
+          output_lines.each { |line| file.puts line }
+        end
+      end
+    end
+    
+    def display_multi_model_metrics(metrics_list)
+      return unless metrics_list && !metrics_list.empty?
+      
+      output_lines = []
+      
+      # Determine table width based on whether costs are shown
+      if AIA.config.show_cost
+        table_width = 80
+      else
+        table_width = 60
+      end
+      
+      output_lines << "═" * table_width
+      output_lines << "Multi-Model Token Usage"
+      output_lines << "─" * table_width
+      
+      # Build header row
+      if AIA.config.show_cost
+        output_lines << sprintf("%-20s %10s %10s %10s %12s %10s", 
+                                "Model", "Input", "Output", "Total", "Cost", "x1000")
+        output_lines << "─" * table_width
+      else
+        output_lines << sprintf("%-20s %10s %10s %10s", 
+                                "Model", "Input", "Output", "Total")
+        output_lines << "─" * table_width
+      end
+      
+      # Process each model
+      total_input = 0
+      total_output = 0
+      total_cost = 0.0
+      
+      metrics_list.each do |metrics|
+        model_name = metrics[:model_id]
+        # Truncate model name if too long
+        model_name = model_name[0..17] + ".." if model_name.length > 19
+        
+        input_tokens = metrics[:input_tokens] || 0
+        output_tokens = metrics[:output_tokens] || 0
+        total_tokens = input_tokens + output_tokens
+        
+        if AIA.config.show_cost
+          cost_data = calculate_cost(metrics)
+          if cost_data[:available]
+            cost_str = "$#{'%.5f' % cost_data[:total_cost]}"
+            x1000_str = "$#{'%.2f' % (cost_data[:total_cost] * 1000)}"
+            total_cost += cost_data[:total_cost]
+          else
+            cost_str = "N/A"
+            x1000_str = "N/A"
+          end
+          
+          output_lines << sprintf("%-20s %10d %10d %10d %12s %10s",
+                                  model_name, input_tokens, output_tokens, total_tokens, cost_str, x1000_str)
+        else
+          output_lines << sprintf("%-20s %10d %10d %10d",
+                                  model_name, input_tokens, output_tokens, total_tokens)
+        end
+        
+        total_input += input_tokens
+        total_output += output_tokens
+      end
+      
+      # Display totals row
+      output_lines << "─" * table_width
+      total_tokens = total_input + total_output
+      
+      if AIA.config.show_cost && total_cost > 0
+        cost_str = "$#{'%.5f' % total_cost}"
+        x1000_str = "$#{'%.2f' % (total_cost * 1000)}"
+        output_lines << sprintf("%-20s %10d %10d %10d %12s %10s",
+                               "TOTAL", total_input, total_output, total_tokens, cost_str, x1000_str)
+      else
+        output_lines << sprintf("%-20s %10d %10d %10d",
+                               "TOTAL", total_input, total_output, total_tokens)
+      end
+      
+      output_lines << "═" * table_width
+      
+      # Output to STDOUT
+      output_lines.each { |line| puts line }
+      
+      # Also write to file if configured
+      if AIA.config.out_file && !AIA.config.out_file.nil?
+        File.open(AIA.config.out_file, 'a') do |file|
+          output_lines.each { |line| file.puts line }
+        end
+      end
+    end
+    
+    private
+    
+    def display_metrics_basic(metrics)
+      puts "Input tokens:  #{metrics[:input_tokens] || 'N/A'}"
+      puts "Output tokens: #{metrics[:output_tokens] || 'N/A'}"
+      
+      if metrics[:input_tokens] && metrics[:output_tokens]
+        total = metrics[:input_tokens] + metrics[:output_tokens]
+        puts "Total tokens:  #{total}"
+      end
+    end
+    
+    def format_metrics_basic(metrics)
+      lines = []
+      lines << "Input tokens:  #{metrics[:input_tokens] || 'N/A'}"
+      lines << "Output tokens: #{metrics[:output_tokens] || 'N/A'}"
+      
+      if metrics[:input_tokens] && metrics[:output_tokens]
+        total = metrics[:input_tokens] + metrics[:output_tokens]
+        lines << "Total tokens:  #{total}"
+      end
+      
+      lines
+    end
+    
+    def display_metrics_with_cost(metrics)
+      cost_data = calculate_cost(metrics)
+      
+      if cost_data[:available]
+        puts "Input tokens:  #{metrics[:input_tokens]} ($#{'%.5f' % cost_data[:input_cost]})"
+        puts "Output tokens: #{metrics[:output_tokens]} ($#{'%.5f' % cost_data[:output_cost]})"
+        puts "Total cost:    $#{'%.5f' % cost_data[:total_cost]}"
+        puts "Cost x1000:    $#{'%.2f' % (cost_data[:total_cost] * 1000)}"
+      else
+        puts "Input tokens:  #{metrics[:input_tokens] || 'N/A'}"
+        puts "Output tokens: #{metrics[:output_tokens] || 'N/A'}"
+        total = (metrics[:input_tokens] || 0) + (metrics[:output_tokens] || 0)
+        puts "Total tokens:  #{total}"
+        puts "Cost:          N/A (pricing not available)"
+      end
+    end
+    
+    def format_metrics_with_cost(metrics)
+      lines = []
+      cost_data = calculate_cost(metrics)
+      
+      if cost_data[:available]
+        lines << "Input tokens:  #{metrics[:input_tokens]} ($#{'%.5f' % cost_data[:input_cost]})"
+        lines << "Output tokens: #{metrics[:output_tokens]} ($#{'%.5f' % cost_data[:output_cost]})"
+        lines << "Total cost:    $#{'%.5f' % cost_data[:total_cost]}"
+        lines << "Cost x1000:    $#{'%.2f' % (cost_data[:total_cost] * 1000)}"
+      else
+        lines << "Input tokens:  #{metrics[:input_tokens] || 'N/A'}"
+        lines << "Output tokens: #{metrics[:output_tokens] || 'N/A'}"
+        total = (metrics[:input_tokens] || 0) + (metrics[:output_tokens] || 0)
+        lines << "Total tokens:  #{total}"
+        lines << "Cost:          N/A (pricing not available)"
+      end
+      
+      lines
+    end
+    
+    def calculate_cost(metrics)
+      return { available: false } unless metrics[:model_id] && metrics[:input_tokens] && metrics[:output_tokens]
+      
+      # Look up model info from RubyLLM
+      begin
+        model_info = RubyLLM::Models.find(metrics[:model_id])
+        return { available: false } unless model_info
+        
+        input_price = model_info.input_price_per_million
+        output_price = model_info.output_price_per_million
+        
+        return { available: false } unless input_price && output_price
+        
+        input_cost = metrics[:input_tokens] * input_price / 1_000_000.0
+        output_cost = metrics[:output_tokens] * output_price / 1_000_000.0
+        total_cost = input_cost + output_cost
+        
+        {
+          available: true,
+          input_cost: input_cost,
+          output_cost: output_cost,
+          total_cost: total_cost
+        }
+      rescue StandardError => e
+        { available: false, error: e.message }
+      end
+    end
   end
 end

data/lib/aia/utility.rb

@@ -6,11 +6,15 @@ module AIA
   class Utility
     class << self
       def tools?
-        !AIA.config.tool_names.empty?
+        AIA.config&.tool_names && !AIA.config.tool_names.empty?
+      end
+
+      def user_tools?
+        AIA.config&.tool_paths && !AIA.config.tool_paths.empty?
       end
 
       def supports_tools?
-        AIA.config.client.model.supports_functions?
+        AIA.config&.client&.model&.supports_functions? || false
       end
 
 
@@ -28,15 +32,15 @@ module AIA
 
        ,      ,
        (\\____/) AI Assistant (v#{AIA::VERSION}) is Online
-        (_oo_)   #{AIA.config.model}#{supports_tools? ? ' (supports tools)' : ''}
-         (O)       using #{AIA.config.adapter} (v#{RubyLLM::VERSION}#{mcp_version})
+        (_oo_)   #{AIA.config&.model || 'unknown-model'}#{supports_tools? ? ' (supports tools)' : ''}
+         (O)       using #{AIA.config&.adapter || 'unknown-adapter'} (v#{RubyLLM::VERSION}#{mcp_version})
        __||__    \\) model db was last refreshed on
-     [/______\\]  /    #{AIA.config.last_refresh}
-    / \\__AI__/ \\/      #{tools? ? 'You can share my tools' : 'I did not bring any tools'}
+     [/______\\]  /    #{AIA.config&.last_refresh || 'unknown'}
+    / \\__AI__/ \\/      #{user_tools? ? 'I will also use your tools' : (tools? ? 'You can share my tools' : 'I did not bring any tools')}
    /    /__\\
-  (\\   /____\\   #{tools? ? 'My Toolbox contains:' : ''}
+  (\\   /____\\   #{user_tools? && tools? ? 'My Toolbox contains:' : ''}
         ROBOT
-        if tools?
+        if user_tools? && tools?
           tool_names = AIA.config.respond_to?(:tool_names) ? AIA.config.tool_names : AIA.config.tools
           if tool_names && !tool_names.to_s.empty?
             puts WordWrapper::MinimumRaggedness.new(

data/lib/aia.rb

@@ -89,8 +89,17 @@ module AIA
     build_flags
 
     # Load Fzf if fuzzy search is enabled and fzf is installed
-    if @config.fuzzy && system('which fzf >/dev/null 2>&1')
-      require_relative 'aia/fzf'
+    if @config.fuzzy
+      begin
+        # Cache fzf availability check for better performance
+        if system('which fzf >/dev/null 2>&1')
+          require_relative 'aia/fzf'
+        else
+          warn "Warning: Fuzzy search enabled but fzf not found. Install fzf for enhanced search capabilities."
+        end
+      rescue StandardError => e
+        warn "Warning: Failed to load fzf: #{e.message}"
+      end
     end
 
     prompt_handler = PromptHandler.new

data/lib/extensions/ruby_llm/.irbrc

@@ -0,0 +1,56 @@
+require 'ruby_llm'
+require_relative 'modalities'
+
+RubyLLM.configure do |config|
+  config.openai_api_key = ENV['OPENAI_API_KEY']
+end
+
+C = RubyLLM::Chat.new
+M = C.model.modalities
+
+__END__
+
+# I = ["text", "image", "pdf", "audio", "file"]
+# O = ["text", "embeddings", "audio", "image", "moderation"]
+
+# I.each do |i|
+#   puts '#'
+#   O.each do |o|
+#     puts <<~HEREDOC
+#       def #{i}_to_#{o}? = input.include?('#{i}') && output.include?('#{o}')
+#     HEREDOC
+#   end
+# end
+
+
+
+#
+def text_to_text? = input.include?('text') && output.include?('text')
+def text_to_embeddings? = input.include?('text') && output.include?('embeddings')
+def text_to_audio? = input.include?('text') && output.include?('audio')
+def text_to_image? = input.include?('text') && output.include?('image')
+def text_to_moderation? = input.include?('text') && output.include?('moderation')
+#
+def image_to_text? = input.include?('image') && output.include?('text')
+def image_to_embeddings? = input.include?('image') && output.include?('embeddings')
+def image_to_audio? = input.include?('image') && output.include?('audio')
+def image_to_image? = input.include?('image') && output.include?('image')
+def image_to_moderation? = input.include?('image') && output.include?('moderation')
+#
+def pdf_to_text? = input.include?('pdf') && output.include?('text')
+def pdf_to_embeddings? = input.include?('pdf') && output.include?('embeddings')
+def pdf_to_audio? = input.include?('pdf') && output.include?('audio')
+def pdf_to_image? = input.include?('pdf') && output.include?('image')
+def pdf_to_moderation? = input.include?('pdf') && output.include?('moderation')
+#
+def audio_to_text? = input.include?('audio') && output.include?('text')
+def audio_to_embeddings? = input.include?('audio') && output.include?('embeddings')
+def audio_to_audio? = input.include?('audio') && output.include?('audio')
+def audio_to_image? = input.include?('audio') && output.include?('image')
+def audio_to_moderation? = input.include?('audio') && output.include?('moderation')
+#
+def file_to_text? = input.include?('file') && output.include?('text')
+def file_to_embeddings? = input.include?('file') && output.include?('embeddings')
+def file_to_audio? = input.include?('file') && output.include?('audio')
+def file_to_image? = input.include?('file') && output.include?('image')
+def file_to_moderation? = input.include?('file') && output.include?('moderation')

data/mkdocs.yml

@@ -0,0 +1,165 @@
+# MkDocs Configuration for AIA Documentation
+site_name: AIA - AI Assistant
+site_description: A powerful CLI tool for dynamic prompt management and AI interaction
+site_author: Dewayne VanHoozer
+site_url: https://madbomber.github.io/aia
+copyright: Copyright © 2024 Dewayne VanHoozer
+
+# Repository information
+repo_name: madbomber/aia
+repo_url: https://github.com/madbomber/aia
+edit_uri: edit/main/docs/
+
+# Configuration
+theme:
+  name: material
+  
+  # Color scheme - Default to light mode
+  palette:
+    # Palette toggle for light mode (default)
+    - scheme: default
+      primary: blue
+      accent: blue
+      toggle:
+        icon: material/brightness-7
+        name: Switch to dark mode
+    
+    # Palette toggle for dark mode
+    - scheme: slate
+      primary: blue
+      accent: blue
+      toggle:
+        icon: material/brightness-4
+        name: Switch to light mode
+
+  # Typography
+  font:
+    text: Roboto
+    code: Roboto Mono
+
+  # Theme features
+  features:
+    # Navigation
+    - navigation.instant
+    - navigation.tracking
+    - navigation.tabs
+    - navigation.tabs.sticky
+    - navigation.sections
+    - navigation.path
+    - navigation.indexes
+    - navigation.top
+
+    # Table of contents
+    - toc.follow
+
+    # Search
+    - search.suggest
+    - search.highlight
+    - search.share
+
+    # Header
+    - header.autohide
+
+    # Content
+    - content.code.copy
+    - content.code.annotate
+    - content.tabs.link
+    - content.tooltips
+    - content.action.edit
+    - content.action.view
+
+# Plugins
+plugins:
+  - search:
+      separator: '[\s\-,:!=\[\]()"`/]+|\.(?!\d)|&[lg]t;|(?!\b)(?=[A-Z][a-z])'
+
+# Extensions
+markdown_extensions:
+  # Python Markdown
+  - abbr
+  - admonition
+  - attr_list
+  - def_list
+  - footnotes
+  - md_in_html
+  - toc:
+      permalink: true
+      title: On this page
+
+  # Python Markdown Extensions
+  - pymdownx.arithmatex:
+      generic: true
+  - pymdownx.betterem:
+      smart_enable: all
+  - pymdownx.caret
+  - pymdownx.details
+  - pymdownx.emoji:
+      emoji_generator: !!python/name:material.extensions.emoji.to_svg
+      emoji_index: !!python/name:material.extensions.emoji.twemoji
+  - pymdownx.highlight:
+      anchor_linenums: true
+      line_spans: __span
+      pygments_lang_class: true
+  - pymdownx.inlinehilite
+  - pymdownx.keys
+  - pymdownx.magiclink:
+      repo_url_shorthand: true
+      user: madbomber
+      repo: aia
+  - pymdownx.mark
+  - pymdownx.smartsymbols
+  - pymdownx.superfences:
+      custom_fences:
+        - name: mermaid
+          class: mermaid
+          format: !!python/name:pymdownx.superfences.fence_code_format
+  - pymdownx.tabbed:
+      alternate_style: true
+  - pymdownx.tasklist:
+      custom_checkbox: true
+  - pymdownx.tilde
+
+# Extra CSS and JavaScript
+extra_css:
+  - assets/css/custom.css
+
+extra_javascript:
+  - assets/js/mathjax.js
+  - https://polyfill.io/v3/polyfill.min.js?features=es6
+  - https://cdn.jsdelivr.net/npm/mathjax@3/es5/tex-mml-chtml.js
+
+# Social media and extra configuration
+extra:
+  social:
+    - icon: fontawesome/brands/github
+      link: https://github.com/madbomber/aia
+      name: AIA on GitHub
+    - icon: fontawesome/solid/gem
+      link: https://rubygems.org/gems/aia
+      name: AIA on RubyGems
+  
+  version:
+    provider: mike
+
+# Navigation
+nav:
+  - Home: index.md
+  - Installation: installation.md
+  - Configuration: configuration.md
+  - Guides:
+    - guides/index.md
+    - Getting Started: guides/getting-started.md
+    - Chat Mode: guides/chat.md
+    - Working with Models: guides/models.md
+    - Available Models: guides/available-models.md
+    - Image Generation: guides/image-generation.md
+    - Tools Integration: guides/tools.md
+  - Reference:
+    - CLI Parameters: cli-reference.md
+    - Directives: directives-reference.md
+  - Advanced:
+    - Prompt Management: prompt_management.md
+    - Advanced Prompting: advanced-prompting.md
+    - Workflows & Pipelines: workflows-and-pipelines.md
+    - MCP Integration: mcp-integration.md
+    - Tools & MCP Examples: tools-and-mcp-examples.md