aia 0.9.23 → 0.10.2

data/lib/aia/config/base.rb

@@ -1,288 +0,0 @@
-# lib/aia/config/base.rb
-
-require 'ostruct'
-require 'date'
-require_relative 'defaults'
-require_relative 'cli_parser'
-require_relative 'file_loader'
-require_relative 'validator'
-
-module AIA
-  module ConfigModules
-    module Base
-      class << self
-        # Delegate to other config modules
-        def cli_options
-          CLIParser.cli_options
-        end
-
-        def cf_options(file)
-          FileLoader.cf_options(file)
-        end
-
-        def dump_config(config, file)
-          FileLoader.dump_config(config, file)
-        end
-
-        def generate_completion_script(shell)
-          FileLoader.generate_completion_script(shell)
-        end
-
-        def tailor_the_config(config)
-          Validator.tailor_the_config(config)
-        end
-
-        def validate_pipeline_prompts(config)
-          Validator.validate_pipeline_prompts(config)
-        end
-
-        def normalize_boolean_flag(config, flag)
-          Validator.normalize_boolean_flag(config, flag)
-        end
-
-        def process_tools_option(path_list, config)
-          CLIParser.process_tools_option(path_list, config)
-        end
-
-        def validate_and_set_context_files(config, remaining_args)
-          Validator.validate_and_set_context_files(config, remaining_args)
-        end
-
-        def setup_mode_options(opts, config)
-          CLIParser.setup_mode_options(opts, config)
-        end
-
-        def parse_config_content(content, ext)
-          FileLoader.parse_config_content(content, ext)
-        end
-
-        def normalize_last_refresh_date(config)
-          FileLoader.normalize_last_refresh_date(config)
-        end
-
-        def process_prompt_id_from_args(config, remaining_args)
-          Validator.process_prompt_id_from_args(config, remaining_args)
-        end
-
-        def process_role_configuration(config)
-          Validator.process_role_configuration(config)
-        end
-
-        def prepare_pipeline(config)
-          Validator.prepare_pipeline(config)
-        end
-
-        def setup_model_options(opts, config)
-          CLIParser.setup_model_options(opts, config)
-        end
-
-        def setup_ai_parameters(opts, config)
-          CLIParser.setup_ai_parameters(opts, config)
-        end
-
-        def read_and_process_config_file(file)
-          FileLoader.read_and_process_config_file(file)
-        end
-
-        def process_stdin_content
-          Validator.process_stdin_content
-        end
-
-        def process_allowed_tools_option(tools_list, config)
-          CLIParser.process_allowed_tools_option(tools_list, config)
-        end
-
-        def process_rejected_tools_option(tools_list, config)
-          CLIParser.process_rejected_tools_option(tools_list, config)
-        end
-
-        def normalize_boolean_flags(config)
-          Validator.normalize_boolean_flags(config)
-        end
-
-        def handle_executable_prompt(config)
-          Validator.handle_executable_prompt(config)
-        end
-
-        def handle_fuzzy_search_prompt_id(config)
-          Validator.handle_fuzzy_search_prompt_id(config)
-        end
-
-        def create_option_parser(config)
-          CLIParser.create_option_parser(config)
-        end
-
-        def apply_file_config_to_struct(config, file_config)
-          FileLoader.apply_file_config_to_struct(config, file_config)
-        end
-
-        def configure_prompt_manager(config)
-          Validator.configure_prompt_manager(config)
-        end
-
-        def setup
-          default_config  = Defaults::DEFAULT_CONFIG.dup
-          cli_config      = cli_options
-          envar_config    = envar_options(default_config, cli_config)
-
-          file = envar_config.config_file   unless envar_config.config_file.nil?
-          file = cli_config.config_file     unless cli_config.config_file.nil?
-
-          cf_config     = cf_options(file)
-
-          config        = OpenStruct.merge(
-                            default_config,
-                            cf_config    || {},
-                            envar_config || {},
-                            cli_config   || {}
-                          )
-
-          config = tailor_the_config(config)
-          load_libraries(config)
-          load_tools(config)
-          load_mcp_servers(config)
-
-          if config.dump_file
-            dump_config(config, config.dump_file)
-          end
-
-          config
-        end
-
-        def load_libraries(config)
-          return if config.require_libs.empty?
-
-          exit_on_error = false
-
-          config.require_libs.each do |library|
-            begin
-              require(library)
-            rescue => e
-              STDERR.puts "Error loading library '#{library}' #{e.message}"
-            exit_on_error = true
-            end
-          end
-
-          exit(1) if exit_on_error
-
-          config
-        end
-
-        def load_tools(config)
-          return if config.tool_paths.empty?
-
-          require_all_tools(config)
-
-          config
-        end
-
-        def require_all_tools(config)
-          exit_on_error = false
-
-          config.tool_paths.each do |tool_path|
-            begin
-              # expands path based on PWD
-              absolute_tool_path = File.expand_path(tool_path)
-              require(absolute_tool_path)
-            rescue => e
-              STDERR.puts "Error loading tool '#{tool_path}' #{e.message}"
-            exit_on_error = true
-            end
-          end
-
-          exit(1) if exit_on_error
-        end
-
-        def load_mcp_servers(config)
-          servers = config.mcp_servers
-
-          return config if servers.nil? || (servers.respond_to?(:empty?) && servers.empty?)
-
-          servers.each do |server|
-            name    = server[:name]    || server["name"]
-            command = server[:command] || server["command"]
-            args    = server[:args]    || server["args"] || []
-            env     = server[:env]     || server["env"]  || {}
-            timeout = server[:timeout] || server["timeout"] || 8000  # default 8 seconds in ms
-
-            unless name && command
-              STDERR.puts "WARNING: MCP server entry missing name or command: #{server.inspect}"
-              next
-            end
-
-            # Resolve command path if not absolute
-            resolved_command = resolve_command_path(command)
-            unless resolved_command
-              STDERR.puts "WARNING: MCP server '#{name}' command not found: #{command}"
-              next
-            end
-
-            begin
-              RubyLLM::MCP.add_client(
-                name:            name,
-                transport_type:  :stdio,
-                request_timeout: timeout,
-                config: {
-                  command:  resolved_command,
-                  args:     args,
-                  env:      env
-                }
-              )
-            rescue => e
-              STDERR.puts "ERROR: Failed to load MCP server '#{name}': #{e.message}"
-            end
-          end
-
-          config
-        end
-
-        def resolve_command_path(command)
-          # If already absolute path, verify it exists
-          if command.start_with?('/')
-            return File.executable?(command) ? command : nil
-          end
-
-          # Search in PATH
-          ENV['PATH'].split(File::PATH_SEPARATOR).each do |dir|
-            full_path = File.join(dir, command)
-            return full_path if File.executable?(full_path)
-          end
-
-          nil
-        end
-
-        # envar values are always String object so need other config
-        # layers to know the prompter type for each key's value
-        def envar_options(default, cli_config)
-          config = OpenStruct.merge(default, cli_config)
-          envars = ENV.keys.select { |key, _| key.start_with?('AIA_') }
-          envars.each do |envar|
-            key   = envar.sub(/^AIA_/, '').downcase.to_sym
-            value = ENV[envar]
-
-            value = case config[key]
-                    when TrueClass, FalseClass
-                      value.downcase == 'true'
-                    when Integer
-                      value.to_i
-                    when Float
-                      value.to_f
-                    when Array
-                      # Special handling for :model to support inline role syntax (ADR-005 v2)
-                      if key == :model && value.include?('=')
-                        CLIParser.parse_models_with_roles(value)
-                      else
-                        value.split(',').map(&:strip)
-                      end
-                    else
-                      value # defaults to String
-                    end
-            config[key] = value
-          end
-
-          config
-        end
-      end
-    end
-  end
-end

data/lib/aia/config/defaults.rb

@@ -1,91 +0,0 @@
-# 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: [],
-
-        # MCP Servers (nil means not configured, set in config file)
-        mcp_servers: nil,
-      }).freeze
-    end
-  end
-end

data/lib/aia/config/file_loader.rb

@@ -1,163 +0,0 @@
-# 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|
-            # Special handling for model array with roles (ADR-005 v2)
-            if (key == :model || key == 'model') && value.is_a?(Array) && value.first.is_a?(Hash)
-              config[:model] = process_model_array_with_roles(value)
-            else
-              config[key] = value
-            end
-          end
-        end
-
-        # Process model array with roles from config file (ADR-005 v2)
-        # Format: [{model: "gpt-4o", role: "architect"}, ...]
-        # Also supports models without roles: [{model: "gpt-4o"}, ...]
-        def process_model_array_with_roles(models_array)
-          return [] if models_array.nil? || models_array.empty?
-
-          model_specs = []
-          model_counts = Hash.new(0)
-
-          models_array.each do |spec|
-            model_name = spec[:model] || spec['model']
-            role_name = spec[:role] || spec['role']
-
-            model_counts[model_name] += 1
-            instance = model_counts[model_name]
-
-            model_specs << {
-              model: model_name,
-              role: role_name,
-              instance: instance,
-              internal_id: instance > 1 ? "#{model_name}##{instance}" : model_name
-            }
-          end
-
-          model_specs
-        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/context_manager.rb

@@ -1,134 +0,0 @@
-# lib/aia/context_manager.rb
-
-module AIA
-  # Manages the conversation context for chat sessions.
-  class ContextManager
-    attr_reader :context, :checkpoints
-
-    # Initializes the ContextManager with an optional system prompt.
-    def initialize(system_prompt: nil)
-      @context = []
-      @checkpoints = {}
-      @checkpoint_counter = 0
-      add_system_prompt(system_prompt) if system_prompt && !system_prompt.strip.empty?
-    end
-
-    # Adds a message to the conversation context.
-    #
-    # @param role [String] The role of the message sender ('user' or 'assistant').
-    # @param content [String] The content of the message.
-    def add_to_context(role:, content:)
-      @context << { role: role, content: content }
-    end
-
-    # Returns the current conversation context.
-    # Optionally adds the system prompt if it wasn't added during initialization
-    # or needs to be re-added after clearing.
-    #
-    # @param system_prompt [String, nil] The system prompt to potentially prepend.
-    # @return [Array<Hash>] The conversation context array.
-    def get_context(system_prompt: nil)
-      # Add or replace system prompt if provided and not empty
-      if system_prompt && !system_prompt.strip.empty?
-        add_system_prompt(system_prompt)
-      end
-      @context
-    end
-
-    # Clears the conversation context, optionally keeping the system prompt.
-    #
-    # @param keep_system_prompt [Boolean] Whether to retain the initial system prompt.
-    def clear_context(keep_system_prompt: true)
-      if keep_system_prompt && !@context.empty? && @context.first[:role] == 'system'
-        @context = [@context.first]
-      else
-        @context = []
-      end
-
-      # Clear all checkpoints when clearing context
-      @checkpoints.clear
-      @checkpoint_counter = 0
-
-      # Attempt to clear the LLM client's context as well
-      begin
-        if AIA.config.client && AIA.config.client.respond_to?(:clear_context)
-          AIA.config.client.clear_context
-        end
-
-        if AIA.config.respond_to?(:llm) && AIA.config.llm && AIA.config.llm.respond_to?(:clear_context)
-          AIA.config.llm.clear_context
-        end
-
-        if defined?(RubyLLM) && RubyLLM.respond_to?(:chat) && RubyLLM.chat.respond_to?(:clear_history)
-          RubyLLM.chat.clear_history
-        end
-      rescue => e
-        STDERR.puts "ERROR: context_manager clear_context error #{e.message}"
-      end
-    end
-
-    # Creates a checkpoint of the current context with an optional name.
-    #
-    # @param name [String, nil] The name of the checkpoint. If nil, uses an incrementing integer.
-    # @return [String] The name of the created checkpoint.
-    def create_checkpoint(name: nil)
-      if name.nil?
-        @checkpoint_counter += 1
-        name = @checkpoint_counter.to_s
-      end
-
-      # Store a deep copy of the current context and its position
-      @checkpoints[name] = {
-        context: @context.map(&:dup),
-        position: @context.size
-      }
-      @last_checkpoint_name = name
-      name
-    end
-
-    # Restores the context to a previously saved checkpoint.
-    #
-    # @param name [String, nil] The name of the checkpoint to restore. If nil, uses the last checkpoint.
-    # @return [Boolean] True if restore was successful, false otherwise.
-    def restore_checkpoint(name: nil)
-      name = @last_checkpoint_name if name.nil?
-
-      return false if name.nil? || !@checkpoints.key?(name)
-
-      # Restore the context from the checkpoint
-      checkpoint_data = @checkpoints[name]
-      @context = checkpoint_data[:context].map(&:dup)
-      true
-    end
-
-    # Returns the list of available checkpoint names.
-    #
-    # @return [Array<String>] The names of all checkpoints.
-    def checkpoint_names
-      @checkpoints.keys
-    end
-
-    # Returns checkpoint information mapped to context positions.
-    #
-    # @return [Hash<Integer, Array<String>>] Position to checkpoint names mapping.
-    def checkpoint_positions
-      positions = {}
-      @checkpoints.each do |name, data|
-        position = data[:position]
-        positions[position] ||= []
-        positions[position] << name
-      end
-      positions
-    end
-
-    private
-
-    # Adds or replaces the system prompt at the beginning of the context.
-    def add_system_prompt(system_prompt)
-       # Remove existing system prompt if present
-       @context.shift if !@context.empty? && @context.first[:role] == 'system'
-       # Add the new system prompt at the beginning
-       @context.unshift({ role: 'system', content: system_prompt })
-    end
-  end
-end

data/mcp_servers/imcp.json

@@ -1,7 +0,0 @@
-{
-  "mcpServers" : {
-    "iMCP" : {
-      "command" : "/Applications/iMCP.app/Contents/MacOS/imcp-server"
-    }
-  }
-}

data/mcp_servers/launcher.json

@@ -1,11 +0,0 @@
-{
-  "mcpServers": {
-    "launcher": {
-      "command": "npx",
-      "args": [
-        "y",
-        "@joshuarileydev/mac-apps-launcher-mcp-server"
-      ]
-    }
-  }
-}

data/mcp_servers/timeserver.json

@@ -1,8 +0,0 @@
-{
-  "mcpServers": {
-    "timeserver": {
-      "command": "python",
-      "args": ["-m", "mcp_simple_timeserver"]
-    }
-  }
-}