aia 0.9.24 → 0.10.2

data/lib/aia/config/config_section.rb

@@ -0,0 +1,87 @@
+# frozen_string_literal: true
+
+# lib/aia/config/config_section.rb
+#
+# ConfigSection provides method access to nested configuration hashes.
+# This allows dot-notation access like: config.llm.temperature
+# instead of: config[:llm][:temperature]
+
+module AIA
+  class ConfigSection
+    def initialize(hash = {})
+      @data = {}
+      (hash || {}).each do |key, value|
+        @data[key.to_sym] = value.is_a?(Hash) ? ConfigSection.new(value) : value
+      end
+    end
+
+    def method_missing(method, *args, &block)
+      key = method.to_s
+      if key.end_with?('=')
+        @data[key.chomp('=').to_sym] = args.first
+      elsif @data.key?(method)
+        @data[method]
+      else
+        nil
+      end
+    end
+
+    def respond_to_missing?(method, include_private = false)
+      key = method.to_s.chomp('=').to_sym
+      @data.key?(key) || super
+    end
+
+    def to_h
+      @data.transform_values do |v|
+        v.is_a?(ConfigSection) ? v.to_h : v
+      end
+    end
+
+    def [](key)
+      @data[key.to_sym]
+    end
+
+    def []=(key, value)
+      @data[key.to_sym] = value
+    end
+
+    def merge(other)
+      other_hash = other.is_a?(ConfigSection) ? other.to_h : other
+      ConfigSection.new(deep_merge(to_h, other_hash || {}))
+    end
+
+    def keys
+      @data.keys
+    end
+
+    def values
+      @data.values
+    end
+
+    def each(&block)
+      @data.each(&block)
+    end
+
+    def empty?
+      @data.empty?
+    end
+
+    def key?(key)
+      @data.key?(key.to_sym)
+    end
+
+    alias has_key? key?
+
+    private
+
+    def deep_merge(base, overlay)
+      base.merge(overlay) do |_key, old_val, new_val|
+        if old_val.is_a?(Hash) && new_val.is_a?(Hash)
+          deep_merge(old_val, new_val)
+        else
+          new_val
+        end
+      end
+    end
+  end
+end

data/lib/aia/config/defaults.yml

@@ -0,0 +1,219 @@
+# AIA Configuration Schema and Defaults
+#
+# This file is the SINGLE SOURCE OF TRUTH for AIA configuration.
+# All attributes must be declared here to be recognized.
+# It is bundled with the gem and loaded automatically at lowest priority.
+#
+# Loading priority (lowest to highest):
+# 1. This file (bundled defaults)
+# 2. User config (~/.aia/config.yml)
+# 3. Environment variables (AIA_*)
+# 4. CLI arguments
+# 5. Embedded directives (//config)
+#
+# Environment Variable Naming:
+# Use double underscore for nested keys:
+#   AIA_LLM__TEMPERATURE=0.5
+#   AIA_PROMPTS__DIR=~/my-prompts
+#   AIA_FLAGS__DEBUG=true
+
+service:
+  name: aia
+
+# -----------------------------------------------------------------------------
+# LLM Configuration
+# Access: AIA.config.llm.adapter, AIA.config.llm.temperature, etc.
+# Env: AIA_LLM__ADAPTER, AIA_LLM__TEMPERATURE, etc.
+# -----------------------------------------------------------------------------
+llm:
+  adapter: ruby_llm
+  temperature: 0.7
+  max_tokens: 2048
+  top_p: 1.0
+  frequency_penalty: 0.0
+  presence_penalty: 0.0
+
+# -----------------------------------------------------------------------------
+# Models Configuration
+# Access: AIA.config.models (array of ModelSpec objects)
+# Each model has: name, role, instance, internal_id
+#
+# Example YAML:
+#   models:
+#     - name: gpt-4o
+#       role: architect
+#     - name: claude-3-opus
+#       role: reviewer
+#
+# Example CLI: --model "gpt-4o=architect,claude-3-opus=reviewer"
+# -----------------------------------------------------------------------------
+models:
+  - name: gpt-4o-mini
+    role: ~
+
+# -----------------------------------------------------------------------------
+# Prompts Configuration
+# Access: AIA.config.prompts.dir, AIA.config.prompts.roles_prefix, etc.
+# Env: AIA_PROMPTS__DIR, AIA_PROMPTS__ROLES_PREFIX, etc.
+# -----------------------------------------------------------------------------
+prompts:
+  dir: ~/.prompts
+  extname: .txt
+  roles_prefix: roles
+  roles_dir: ~/.prompts/roles
+  role: ~
+  system_prompt: ~
+  parameter_regex: ~
+
+# -----------------------------------------------------------------------------
+# Output Configuration
+# Access: AIA.config.output.file, AIA.config.output.append, etc.
+# Env: AIA_OUTPUT__FILE, AIA_OUTPUT__APPEND, etc.
+# -----------------------------------------------------------------------------
+output:
+  file: temp.md
+  append: false
+  markdown: true
+  history_file: ~/.prompts/_prompts.log
+
+# -----------------------------------------------------------------------------
+# Audio Configuration
+# Access: AIA.config.audio.voice, AIA.config.audio.speak_command, etc.
+# Env: AIA_AUDIO__VOICE, AIA_AUDIO__SPEAK_COMMAND, etc.
+# -----------------------------------------------------------------------------
+audio:
+  voice: alloy
+  speak_command: afplay
+  speech_model: tts-1
+  transcription_model: whisper-1
+
+# -----------------------------------------------------------------------------
+# Image Configuration
+# Access: AIA.config.image.model, AIA.config.image.size, etc.
+# Env: AIA_IMAGE__MODEL, AIA_IMAGE__SIZE, etc.
+# -----------------------------------------------------------------------------
+image:
+  model: dall-e-3
+  size: 1024x1024
+  quality: standard
+  style: vivid
+
+# -----------------------------------------------------------------------------
+# Embedding Configuration
+# Access: AIA.config.embedding.model
+# Env: AIA_EMBEDDING__MODEL
+# -----------------------------------------------------------------------------
+embedding:
+  model: text-embedding-ada-002
+
+# -----------------------------------------------------------------------------
+# Tools Configuration
+# Access: AIA.config.tools.paths, AIA.config.tools.allowed, etc.
+# Env: AIA_TOOLS__PATHS, AIA_TOOLS__ALLOWED, etc.
+# -----------------------------------------------------------------------------
+tools:
+  paths: []
+  allowed: ~
+  rejected: ~
+
+# -----------------------------------------------------------------------------
+# Flags (Boolean Options)
+# Access: AIA.config.flags.chat, AIA.config.flags.debug, etc.
+# Env: AIA_FLAGS__CHAT=true, AIA_FLAGS__DEBUG=true, etc.
+# -----------------------------------------------------------------------------
+flags:
+  chat: false
+  cost: false
+  debug: false
+  verbose: false
+  fuzzy: false
+  tokens: false
+  no_mcp: false
+  speak: false
+  terse: false
+  shell: true
+  erb: true
+  clear: false
+  consensus: false
+
+# -----------------------------------------------------------------------------
+# Logger Configuration
+# Access: AIA.config.logger.aia.file, AIA.config.logger.llm.level, etc.
+# Env: AIA_LOGGER__AIA__FILE, AIA_LOGGER__LLM__LEVEL, etc.
+#
+# Configures logging for three systems:
+#   aia: AIA application logging
+#   llm: RubyLLM gem logging
+#   mcp: RubyLLM::MCP gem logging
+#
+# file: STDOUT, STDERR, or a file path (e.g., ~/.aia/aia.log)
+#       All three loggers can write to the same file safely.
+# level: debug, info, warn, error, fatal
+# flush: true = immediate write (no buffering), false = buffered writes
+# -----------------------------------------------------------------------------
+logger:
+  aia:
+    file: STDOUT
+    level: warn
+    flush: true
+  llm:
+    file: STDOUT
+    level: warn
+    flush: true
+  mcp:
+    file: STDOUT
+    level: warn
+    flush: true
+
+# -----------------------------------------------------------------------------
+# Pipeline/Workflow Configuration
+# Access: AIA.config.pipeline (array of prompt IDs)
+# -----------------------------------------------------------------------------
+pipeline: []
+
+# -----------------------------------------------------------------------------
+# Model Registry Configuration
+# Access: AIA.config.registry.refresh
+# Env: AIA_REGISTRY__REFRESH
+# Note: last_refresh is derived from models.json file modification time
+# -----------------------------------------------------------------------------
+registry:
+  refresh: 7 # days between refreshes (0 = disable periodic refresh)
+
+# -----------------------------------------------------------------------------
+# Required Ruby Libraries
+# Access: AIA.config.require_libs (array)
+# -----------------------------------------------------------------------------
+require_libs: []
+
+# -----------------------------------------------------------------------------
+# MCP Servers Configuration
+# Access: AIA.config.mcp_servers (array of server configs)
+#
+# Example:
+#   mcp_servers:
+#     - name: my-server
+#       command: /path/to/server
+#       args: []
+#       env: {}
+#       timeout: 8000
+# -----------------------------------------------------------------------------
+mcp_servers: []
+
+# -----------------------------------------------------------------------------
+# Paths Configuration
+# Access: AIA.config.paths.aia_dir, AIA.config.paths.config_file
+# Env: AIA_PATHS__AIA_DIR, AIA_PATHS__CONFIG_FILE
+#
+# Note: anyway_config uses XDG Base Directory Specification by default.
+# User config is loaded from ~/.config/aia/aia.yml
+# -----------------------------------------------------------------------------
+paths:
+  aia_dir: ~/.config/aia
+  config_file: ~/.config/aia/aia.yml
+
+# -----------------------------------------------------------------------------
+# Context Files (set at runtime)
+# Access: AIA.config.context_files (array of file paths)
+# -----------------------------------------------------------------------------
+context_files: []

data/lib/aia/config/defaults_loader.rb

@@ -0,0 +1,147 @@
+# frozen_string_literal: true
+
+# lib/aia/config/defaults_loader.rb
+#
+# Configuration Loaders for Anyway Config
+#
+# Provides two custom loaders:
+# 1. DefaultsLoader - Loads bundled defaults from defaults.yml
+# 2. UserConfigLoader - Loads user config from ~/.config/aia/aia.yml
+#
+# Loading priority (lowest to highest):
+# 1. Bundled defaults (DefaultsLoader)
+# 2. User config (UserConfigLoader)
+# 3. Environment variables (AIA_*)
+# 4. CLI arguments (applied separately)
+
+require 'anyway_config'
+require 'yaml'
+
+module AIA
+  module Loaders
+    # Loads bundled default configuration values from defaults.yml
+    class DefaultsLoader < Anyway::Loaders::Base
+      DEFAULTS_PATH = File.expand_path('../defaults.yml', __FILE__).freeze
+
+      class << self
+        # Returns the path to the bundled defaults file
+        #
+        # @return [String] path to defaults.yml
+        def defaults_path
+          DEFAULTS_PATH
+        end
+
+        # Check if defaults file exists
+        #
+        # @return [Boolean]
+        def defaults_exist?
+          File.exist?(DEFAULTS_PATH)
+        end
+
+        # Load and parse the raw YAML content
+        #
+        # @return [Hash] parsed YAML with symbolized keys
+        def load_raw_yaml
+          return {} unless defaults_exist?
+
+          content = File.read(defaults_path)
+          YAML.safe_load(
+            content,
+            permitted_classes: [Symbol, Date],
+            symbolize_names: true,
+            aliases: true
+          ) || {}
+        rescue Psych::SyntaxError => e
+          warn "AIA: Failed to parse bundled defaults #{defaults_path}: #{e.message}"
+          {}
+        end
+
+        # Returns the schema (all configuration keys and their defaults)
+        #
+        # @return [Hash] the complete defaults
+        def schema
+          load_raw_yaml
+        end
+
+        # Get a list of all top-level configuration sections
+        #
+        # @return [Array<Symbol>] list of section names
+        def sections
+          schema.keys
+        end
+      end
+
+      # Called by Anyway Config to load configuration
+      #
+      # @param name [Symbol] the config name (unused, always :aia)
+      # @return [Hash] configuration hash
+      def call(name:, **_options)
+        return {} unless self.class.defaults_exist?
+
+        trace!(:bundled_defaults, path: self.class.defaults_path) do
+          self.class.load_raw_yaml
+        end
+      end
+    end
+
+    # Loads user configuration from XDG config directory
+    # Follows XDG Base Directory Specification: ~/.config/aia/aia.yml
+    class UserConfigLoader < Anyway::Loaders::Base
+      class << self
+        # Returns the path to the user config file
+        # Uses XDG_CONFIG_HOME if set, otherwise defaults to ~/.config
+        #
+        # @return [String] path to user config file
+        def user_config_path
+          xdg_config_home = ENV.fetch('XDG_CONFIG_HOME', File.expand_path('~/.config'))
+          File.join(xdg_config_home, 'aia', 'aia.yml')
+        end
+
+        # Check if user config file exists
+        #
+        # @return [Boolean]
+        def user_config_exist?
+          File.exist?(user_config_path)
+        end
+
+        # Load and parse the user YAML content
+        #
+        # @return [Hash] parsed YAML with symbolized keys
+        def load_user_yaml
+          return {} unless user_config_exist?
+
+          content = File.read(user_config_path)
+          YAML.safe_load(
+            content,
+            permitted_classes: [Symbol, Date],
+            symbolize_names: true,
+            aliases: true
+          ) || {}
+        rescue Psych::SyntaxError => e
+          warn "AIA: Failed to parse user config #{user_config_path}: #{e.message}"
+          {}
+        end
+      end
+
+      # Called by Anyway Config to load configuration
+      #
+      # @param name [Symbol] the config name (unused, always :aia)
+      # @return [Hash] configuration hash
+      def call(name:, **_options)
+        return {} unless self.class.user_config_exist?
+
+        trace!(:user_config, path: self.class.user_config_path) do
+          self.class.load_user_yaml
+        end
+      end
+    end
+  end
+end
+
+# Register loaders in priority order (lowest to highest)
+# 1. bundled_defaults - gem's default values
+# 2. user_config - user's ~/.config/aia/aia.yml
+# 3. yml - standard anyway_config yml loader (for config/aia.yml in app directory)
+# 4. env - environment variables
+Anyway.loaders.insert_before :yml, :bundled_defaults, AIA::Loaders::DefaultsLoader
+Anyway.loaders.insert_after :bundled_defaults, :user_config, AIA::Loaders::UserConfigLoader

data/lib/aia/config/mcp_parser.rb

@@ -0,0 +1,151 @@
+# frozen_string_literal: true
+
+# lib/aia/config/mcp_parser.rb
+#
+# Parses MCP server JSON configuration files and converts them
+# to the format expected by AIA's config system.
+#
+# Supports two JSON formats:
+#
+# 1. Simple format (single server):
+#    {
+#      "type": "stdio",
+#      "command": ["npx", "-y", "@server/name", "/path"]
+#    }
+#
+# 2. mcpServers format (one or more servers):
+#    {
+#      "mcpServers": {
+#        "server_name": {
+#          "command": "python",
+#          "args": ["-m", "module_name"],
+#          "env": {},
+#          "timeout": 8000
+#        }
+#      }
+#    }
+
+require 'json'
+
+module AIA
+  module McpParser
+    class << self
+      # Parse MCP server configuration files and return array of server configs
+      #
+      # @param file_paths [Array<String>] paths to JSON configuration files
+      # @return [Array<Hash>] array of server configurations
+      def parse_files(file_paths)
+        return [] if file_paths.nil? || file_paths.empty?
+
+        servers = []
+
+        file_paths.each do |file_path|
+          expanded_path = File.expand_path(file_path)
+
+          unless File.exist?(expanded_path)
+            warn "Warning: MCP config file not found: #{file_path}"
+            next
+          end
+
+          begin
+            json_content = File.read(expanded_path)
+            parsed = JSON.parse(json_content)
+            servers.concat(convert_to_config_format(parsed, file_path))
+          rescue JSON::ParserError => e
+            warn "Warning: Invalid JSON in MCP config file '#{file_path}': #{e.message}"
+          rescue StandardError => e
+            warn "Warning: Error reading MCP config file '#{file_path}': #{e.message}"
+          end
+        end
+
+        servers
+      end
+
+      private
+
+      # Convert parsed JSON to the config format
+      #
+      # @param parsed [Hash] parsed JSON content
+      # @param file_path [String] original file path (for deriving server name)
+      # @return [Array<Hash>] array of server configurations
+      def convert_to_config_format(parsed, file_path)
+        if parsed.key?('mcpServers')
+          convert_mcp_servers_format(parsed['mcpServers'])
+        else
+          convert_simple_format(parsed, file_path)
+        end
+      end
+
+      # Convert mcpServers format to config format
+      #
+      # @param mcp_servers [Hash] the mcpServers hash from JSON
+      # @return [Array<Hash>] array of server configurations
+      def convert_mcp_servers_format(mcp_servers)
+        mcp_servers.map do |name, config|
+          server = { name: name }
+
+          if config['command']
+            server[:command] = config['command']
+          end
+
+          if config['args']
+            server[:args] = Array(config['args'])
+          end
+
+          if config['env']
+            server[:env] = config['env']
+          end
+
+          if config['timeout']
+            server[:timeout] = config['timeout'].to_i
+          end
+
+          if config['url']
+            server[:url] = config['url']
+          end
+
+          if config['headers']
+            server[:headers] = config['headers']
+          end
+
+          server
+        end
+      end
+
+      # Convert simple format to config format
+      #
+      # @param parsed [Hash] parsed JSON with type and command
+      # @param file_path [String] file path for deriving server name
+      # @return [Array<Hash>] array with single server configuration
+      def convert_simple_format(parsed, file_path)
+        # Derive name from filename (e.g., "filesystem.json" -> "filesystem")
+        name = File.basename(file_path, '.*')
+
+        server = { name: name }
+
+        if parsed['command'].is_a?(Array)
+          # Command is an array: first element is command, rest are args
+          server[:command] = parsed['command'].first
+          server[:args] = parsed['command'][1..] || []
+        elsif parsed['command']
+          server[:command] = parsed['command']
+          server[:args] = parsed['args'] || []
+        end
+
+        if parsed['env']
+          server[:env] = parsed['env']
+        end
+
+        if parsed['timeout']
+          server[:timeout] = parsed['timeout'].to_i
+        end
+
+        if parsed['type']
+          server[:type] = parsed['type']
+        end
+
+        [server]
+      end
+    end
+  end
+end

data/lib/aia/config/model_spec.rb

@@ -0,0 +1,67 @@
+# frozen_string_literal: true
+
+# lib/aia/config/model_spec.rb
+#
+# ModelSpec represents a single model configuration with optional role.
+# This provides typed access to model configuration instead of raw hashes.
+#
+# Example:
+#   spec = ModelSpec.new(name: 'gpt-4o', role: 'architect')
+#   spec.name        # => 'gpt-4o'
+#   spec.role        # => 'architect'
+#   spec.internal_id # => 'gpt-4o'
+
+module AIA
+  class ModelSpec
+    attr_accessor :name, :role, :instance, :internal_id
+
+    def initialize(hash = {})
+      hash = hash.transform_keys(&:to_sym) if hash.respond_to?(:transform_keys)
+
+      @name = hash[:name]
+      @role = hash[:role]
+      @instance = hash[:instance] || 1
+      @internal_id = hash[:internal_id] || @name
+    end
+
+    def to_h
+      {
+        name: @name,
+        role: @role,
+        instance: @instance,
+        internal_id: @internal_id
+      }
+    end
+
+    def to_s
+      if @role
+        "#{@name}=#{@role}"
+      else
+        @name.to_s
+      end
+    end
+
+    def ==(other)
+      return false unless other.is_a?(ModelSpec)
+      name == other.name && role == other.role && instance == other.instance
+    end
+
+    def eql?(other)
+      self == other
+    end
+
+    def hash
+      [name, role, instance].hash
+    end
+
+    # Check if this model has a role assigned
+    def role?
+      !@role.nil? && !@role.empty?
+    end
+
+    # Check if this is a duplicate instance of the same model
+    def duplicate?
+      @instance > 1
+    end
+  end
+end