aia 0.9.11 → 0.9.12

data/lib/aia/config/defaults.rb

@@ -0,0 +1,88 @@
+# lib/aia/config/defaults.rb
+
+require 'yaml'
+require 'toml-rb'
+require 'date'
+require 'prompt_manager'
+
+module AIA
+  module ConfigModules
+    module Defaults
+      DEFAULT_CONFIG = OpenStruct.new({
+        adapter:      'ruby_llm', # 'ruby_llm' or ???
+        #
+        aia_dir:      File.join(ENV['HOME'], '.aia'),
+        config_file:  File.join(ENV['HOME'], '.aia', 'config.yml'),
+        out_file:     'temp.md',
+        log_file:     File.join(ENV['HOME'], '.prompts', '_prompts.log'),
+        context_files: [],
+        #
+        prompts_dir:  File.join(ENV['HOME'], '.prompts'),
+        prompt_extname: PromptManager::Storage::FileSystemAdapter::PROMPT_EXTENSION,
+        #
+        roles_prefix: 'roles',
+        roles_dir:    File.join(ENV['HOME'], '.prompts', 'roles'),
+        role:         '',
+
+        #
+        system_prompt: '',
+
+        # Tools
+        tools:          '',  # Comma-separated string of loaded tool names (set by adapter)
+        allowed_tools:  nil, # nil means all tools are allowed; otherwise an Array of Strings which are the tool names
+        rejected_tools: nil, # nil means no tools are rejected
+        tool_paths:     [],  # Strings - absolute and relative to tools
+
+        # Flags
+        markdown: true,
+        shell:    true,
+        erb:      true,
+        chat:     false,
+        clear:    false,
+        terse:    false,
+        verbose:  false,
+        debug:    $DEBUG_ME,
+        fuzzy:    false,
+        speak:    false,
+        append:   false, # Default to not append to existing out_file
+
+        # workflow
+        pipeline: [],
+
+        # PromptManager::Prompt Tailoring
+        parameter_regex: PromptManager::Prompt.parameter_regex.to_s,
+
+        # LLM tuning parameters
+        temperature:          0.7,
+        max_tokens:           2048,
+        top_p:                1.0,
+        frequency_penalty:    0.0,
+        presence_penalty:     0.0,
+
+        # Audio Parameters
+        voice:                'alloy',
+        speak_command:        'afplay', # 'afplay' for audio files on MacOS
+
+        # Image Parameters
+        image_size:           '1024x1024',
+        image_quality:        'standard',
+        image_style:          'vivid',
+
+        # Models
+        model:                ['gpt-4o-mini'],
+        consensus:            nil, # nil/false = individual responses; true = consensus response
+        speech_model:         'tts-1',
+        transcription_model:  'whisper-1',
+        embedding_model:      'text-embedding-ada-002',
+        image_model:          'dall-e-3',
+
+        # Model Regristery
+        refresh:              7, # days between refreshes of model info; 0 means every startup
+        last_refresh:         Date.today - 1,
+
+        # Ruby libraries to require for Ruby binding
+        require_libs: [],
+      }).freeze
+    end
+  end
+end

data/lib/aia/config/file_loader.rb

@@ -0,0 +1,131 @@
+# lib/aia/config/file_loader.rb
+
+require 'yaml'
+require 'toml-rb'
+require 'erb'
+require 'date'
+
+module AIA
+  module ConfigModules
+    module FileLoader
+      class << self
+        def load_config_file(file, config)
+          if File.exist?(file)
+            ext = File.extname(file).downcase
+            content = File.read(file)
+
+            # Process ERB if filename ends with .erb
+            if file.end_with?('.erb')
+              content = ERB.new(content).result
+              file = file.chomp('.erb')
+              File.write(file, content)
+            end
+
+            file_config = case ext
+                          when '.yml', '.yaml'
+                            YAML.safe_load(content, permitted_classes: [Symbol], symbolize_names: true)
+                          when '.toml'
+                            TomlRB.parse(content)
+                          else
+                            raise "Unsupported config file format: #{ext}"
+                          end
+
+            file_config.each do |key, value|
+              config[key.to_sym] = value
+            end
+          else
+            raise "Config file not found: #{file}"
+          end
+        end
+
+        def cf_options(file)
+          config = OpenStruct.new
+
+          if File.exist?(file)
+            content = read_and_process_config_file(file)
+            file_config = parse_config_content(content, File.extname(file).downcase)
+            apply_file_config_to_struct(config, file_config)
+          else
+            STDERR.puts "WARNING:Config file not found: #{file}"
+          end
+
+          normalize_last_refresh_date(config)
+          config
+        end
+
+        def read_and_process_config_file(file)
+          content = File.read(file)
+
+          # Process ERB if filename ends with .erb
+          if file.end_with?('.erb')
+            content = ERB.new(content).result
+            processed_file = file.chomp('.erb')
+            File.write(processed_file, content)
+          end
+
+          content
+        end
+
+        def parse_config_content(content, ext)
+          case ext
+          when '.yml', '.yaml'
+            YAML.safe_load(content, permitted_classes: [Symbol], symbolize_names: true)
+          when '.toml'
+            TomlRB.parse(content)
+          else
+            raise "Unsupported config file format: #{ext}"
+          end
+        end
+
+        def apply_file_config_to_struct(config, file_config)
+          file_config.each do |key, value|
+            config[key] = value
+          end
+        end
+
+        def normalize_last_refresh_date(config)
+          return unless config.last_refresh&.is_a?(String)
+
+          config.last_refresh = Date.strptime(config.last_refresh, '%Y-%m-%d')
+        end
+
+        def dump_config(config, file)
+          # Implementation for config dump
+          ext = File.extname(file).downcase
+
+          config.last_refresh = config.last_refresh.to_s if config.last_refresh.is_a? Date
+
+          config_hash = config.to_h
+
+          # Remove prompt_id to prevent automatic initial pompting in --chat mode
+          config_hash.delete(:prompt_id)
+
+          # Remove dump_file key to prevent automatic exit on next load
+          config_hash.delete(:dump_file)
+
+          content = case ext
+                    when '.yml', '.yaml'
+                      YAML.dump(config_hash)
+                    when '.toml'
+                      TomlRB.dump(config_hash)
+                    else
+                      raise "Unsupported config file format: #{ext}"
+                    end
+
+          File.write(file, content)
+          puts "Config successfully dumped to #{file}"
+        end
+
+        def generate_completion_script(shell)
+          script_path = File.join(File.dirname(__FILE__), "../../aia_completion.#{shell}")
+
+          if File.exist?(script_path)
+            puts File.read(script_path)
+          else
+            STDERR.puts "ERROR: The shell '#{shell}' is not supported or the completion script is missing."
+          end
+        end
+      end
+    end
+  end
+end

data/lib/aia/config/validator.rb

@@ -0,0 +1,184 @@
+# lib/aia/config/validator.rb
+
+require 'ostruct'
+
+module AIA
+  module ConfigModules
+    module Validator
+      class << self
+        def tailor_the_config(config)
+          remaining_args = config.remaining_args.dup
+          config.remaining_args = nil
+
+          stdin_content = process_stdin_content
+          config.stdin_content = stdin_content if stdin_content && !stdin_content.strip.empty?
+
+          process_prompt_id_from_args(config, remaining_args)
+          validate_and_set_context_files(config, remaining_args)
+          handle_executable_prompt(config)
+          validate_required_prompt_id(config)
+          process_role_configuration(config)
+          handle_fuzzy_search_prompt_id(config)
+          normalize_boolean_flags(config)
+          handle_completion_script(config)
+          validate_final_prompt_requirements(config)
+          configure_prompt_manager(config)
+          prepare_pipeline(config)
+          validate_pipeline_prompts(config)
+
+          config
+        end
+
+        def process_stdin_content
+          stdin_content = ''
+
+          if !STDIN.tty? && !STDIN.closed?
+            begin
+              stdin_content << "\n" + STDIN.read
+              STDIN.reopen('/dev/tty')  # Reopen STDIN for interactive use
+            rescue => _
+              # If we can't reopen, continue without error
+            end
+          end
+
+          stdin_content
+        end
+
+        def process_prompt_id_from_args(config, remaining_args)
+          return if remaining_args.empty?
+
+          maybe_id = remaining_args.first
+          maybe_id_plus = File.join(config.prompts_dir, maybe_id + config.prompt_extname)
+
+          if AIA.bad_file?(maybe_id) && AIA.good_file?(maybe_id_plus)
+            config.prompt_id = remaining_args.shift
+          end
+        end
+
+        def validate_and_set_context_files(config, remaining_args)
+          return if remaining_args.empty?
+
+          bad_files = remaining_args.reject { |filename| AIA.good_file?(filename) }
+          if bad_files.any?
+            STDERR.puts "Error: The following files do not exist: #{bad_files.join(', ')}"
+            exit 1
+          end
+
+          config.context_files ||= []
+          config.context_files += remaining_args
+        end
+
+        def handle_executable_prompt(config)
+          return unless config.executable_prompt && config.context_files && !config.context_files.empty?
+
+          config.executable_prompt_file = config.context_files.pop
+        end
+
+        def validate_required_prompt_id(config)
+          return unless config.prompt_id.nil? && !config.chat && !config.fuzzy
+
+          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
+
+        def process_role_configuration(config)
+          return if config.role.empty?
+
+          unless config.roles_prefix.empty?
+            unless config.role.start_with?(config.roles_prefix)
+              config.role.prepend "#{config.roles_prefix}/"
+            end
+          end
+
+          config.roles_dir ||= File.join(config.prompts_dir, config.roles_prefix)
+
+          if config.prompt_id.nil? || config.prompt_id.empty?
+            if !config.role.nil? && !config.role.empty?
+              config.prompt_id = config.role
+              config.pipeline.prepend config.prompt_id
+              config.role = ''
+            end
+          end
+        end
+
+        def handle_fuzzy_search_prompt_id(config)
+          return unless config.fuzzy && config.prompt_id.empty?
+
+          # When fuzzy search is enabled but no prompt ID is provided,
+          # set a special value to trigger fuzzy search without an initial query
+          # SMELL: This feels like a cludge
+          config.prompt_id = '__FUZZY_SEARCH__'
+        end
+
+        def normalize_boolean_flags(config)
+          normalize_boolean_flag(config, :chat)
+          normalize_boolean_flag(config, :fuzzy)
+          normalize_boolean_flag(config, :consensus)
+        end
+
+        def normalize_boolean_flag(config, flag)
+          return if [TrueClass, FalseClass].include?(config[flag].class)
+
+          config[flag] = case config[flag]
+                         when nil, '', 'false', false
+                           false
+                         when 'true', true
+                           true
+                         else
+                           # For any other non-empty string value, treat as true
+                           true
+                         end
+        end
+
+        def handle_completion_script(config)
+          return unless config.completion
+
+          FileLoader.generate_completion_script(config.completion)
+          exit
+        end
+
+        def validate_final_prompt_requirements(config)
+          # 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.nil? || 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
+          # This is handled implicitly - no action needed
+        end
+
+        def configure_prompt_manager(config)
+          return unless config.parameter_regex
+
+          PromptManager::Prompt.parameter_regex = Regexp.new(config.parameter_regex)
+        end
+
+        def prepare_pipeline(config)
+          return if config.prompt_id.nil? || config.prompt_id.empty? || config.prompt_id == config.pipeline.first
+
+          config.pipeline.prepend config.prompt_id
+        end
+
+        def validate_pipeline_prompts(config)
+          return if config.pipeline.empty?
+
+          and_exit = false
+
+          config.pipeline.each do |prompt_id|
+            # Skip empty prompt IDs (can happen in chat-only mode)
+            next if prompt_id.nil? || prompt_id.empty?
+
+            prompt_file_path = File.join(config.prompts_dir, "#{prompt_id}.txt")
+            unless File.exist?(prompt_file_path)
+              STDERR.puts "Error: Prompt ID '#{prompt_id}' does not exist at #{prompt_file_path}"
+              and_exit = true
+            end
+          end
+
+          exit(1) if and_exit
+        end
+      end
+    end
+  end
+end