aia 0.9.21 → 0.9.23

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2d974eb8f23ef8b3af71447951101047c7bdf04a203d4690655b8bfbe940fb2f
-  data.tar.gz: d63ff74f8837d642a454e116e2efb29f5a57d2d3e37a0e1181378d93d15f4660
+  metadata.gz: 38ae30b2341fb1e96738601b0c7cac5459772217d005aa5e22f18ecde270e2e0
+  data.tar.gz: 34555fad44d99436b4171887c0804c88f88e6fdc572e8a2d05476f485b0b80c6
 SHA512:
-  metadata.gz: 1c20c13f8d2c80abcc1dafc0401d6bfb3c1c1045ff4dc0821b3de256ba59241938a076f0cb74583758e1bbadd2e31cd933a2137440eee1a78e4adea794e66ea6
-  data.tar.gz: f7b6d1301362e22ccb3ef07956f5a9fe3766504ec23a2df82b420ab330b2e628406a1ce98b43dced48ab4142d49ccc5f7737f6fef0e9c9bded5d4ecda8626c8a
+  metadata.gz: 67182004edf9443ad3fc9190c1d57daa18c8f997d0425fa642e6f4c0c94d74490a1943ae1f79e7b18f55f65c0237a8fba31e765b750c374c8885926cbc08ca62
+  data.tar.gz: 3a86c4b7734a955180e7282a5462f7cda7ff83d1eac49c7106e63a69eaf9554a318ef2195ef61c4a61aec2340be6fdac8d19fd0d6a681f450e6c6c55b7c02940

data/.version

@@ -1 +1 @@
-0.9.21
+0.9.23

data/CHANGELOG.md

@@ -1,6 +1,55 @@
 # Changelog
 ## [Unreleased]
 
+## [0.9.23] 2025-12-06
+
+### New Features
+- **MCP Server Configuration**: Added native support for defining MCP (Model Context Protocol) servers in the config file
+  - Configure MCP servers in `~/.aia/config.yml` under the `mcp_servers` key
+  - Supports `name`, `command`, `args`, `env`, and `timeout` options per server
+  - Automatic PATH resolution for commands (no absolute paths required)
+  - Configurable timeouts for slow-starting servers (default: 8000ms)
+  - Environment variable support for MCP server processes
+
+### Improvements
+- **Robot Display**: Added MCP server names to the robot ASCII art display
+  - Shows "MCP: server1, server2, ..." when MCP servers are configured
+
+### Technical Changes
+- Added `load_mcp_servers` method to `lib/aia/config/base.rb` for automatic MCP client registration
+- Added `resolve_command_path` method for PATH-based command resolution
+- Added `mcp_servers?` and `mcp_server_names` helper methods to `lib/aia/utility.rb`
+- Fixed `OpenStruct.merge` to skip nil values, preventing config file values from being overwritten
+- Added `mcp_servers: nil` default to prevent merge issues with empty arrays
+
+### Configuration Example
+```yaml
+# ~/.aia/config.yml
+:mcp_servers:
+  - name: "my-server"
+    command: "my_mcp_server.rb"
+    args: ["stdio"]
+    timeout: 30000
+    env:
+      MY_VAR: "value"
+```
+
+## [0.9.22] 2025-11-12
+
+### Bug Fixes
+- **TEST SUITE**: Fixed all Mocha test isolation issues causing stub contamination between tests
+  - Added proper teardown methods with `super` calls to 13 test files to ensure Mocha cleanup
+  - Fixed PromptHandlerTest missing teardown and config fields (erb, shell, roles_dir)
+  - Fixed ModelsDirectiveTest to use consistent stubbing approach instead of mixing real and stubbed config
+  - Fixed MultiModelIsolationTest to use stubs instead of direct instance variable manipulation
+  - Fixed AIAIntegrationTest, ChatProcessorServiceTest, ContextManagerTest, DirectiveProcessorTest, RubyLLMAdapterTest, SessionTest, LocalProvidersTest, UtilityTest, AIAMockingTest, and AIAPropertyBasedTest to include proper Mocha cleanup
+  - Test results improved from 2 failures, 2 errors to 0 failures, 0 errors (325 runs, 1018 assertions)
+
+### Technical Changes
+- Enhanced test isolation by ensuring all tests using Mocha stubs properly clean up via `super` in teardown
+- Standardized stub usage pattern across test suite for consistency
+- Eliminated stub leakage that caused "unexpected invocation" and "AIA was instantiated in one test but receiving invocations in another" errors
+
 ## [0.9.21] 2025-10-08
 ### Bug Fixes
 - **Checkpoint Directive Output**: Fixed `//checkpoint` directive to return empty string instead of status message (lib/aia/directives/configuration.rb:155)

data/README.md

@@ -100,6 +100,7 @@ For more information on AIA visit these locations:
       - [Example Workflow](#example-workflow)
     - [Roles and System Prompts](#roles-and-system-prompts)
     - [RubyLLM::Tool Support](#rubyllmtool-support)
+    - [MCP Server Configuration](#mcp-server-configuration)
   - [Examples & Tips](#examples--tips)
     - [Practical Examples](#practical-examples)
       - [Code Review Prompt](#code-review-prompt)
@@ -922,6 +923,170 @@ aia --tools examples/tools/mcp/imcp.rb --chat
 
 These MCP clients require the `ruby_llm-mcp` gem and provide access to external services and data sources through the Model Context Protocol.
 
+### MCP Server Configuration
+
+AIA supports defining MCP (Model Context Protocol) servers directly in your configuration file. This allows MCP tools to be automatically loaded at startup without needing to specify them on the command line each time.
+
+#### Configuration Format
+
+Add MCP servers to your `~/.aia/config.yml` file:
+
+```yaml
+:mcp_servers:
+  - name: "server-name"
+    command: "server_command"
+    args: ["arg1", "arg2"]
+    timeout: 30000  # milliseconds (default: 8000)
+    env:
+      ENV_VAR: "value"
+```
+
+#### Configuration Options
+
+| Option | Required | Default | Description |
+|--------|----------|---------|-------------|
+| `name` | Yes | - | Unique identifier for the MCP server |
+| `command` | Yes | - | Executable command (absolute path or found in PATH) |
+| `args` | No | `[]` | Array of command-line arguments |
+| `timeout` | No | `8000` | Connection timeout in milliseconds |
+| `env` | No | `{}` | Environment variables for the server process |
+
+#### Example: GitHub MCP Server
+
+The GitHub MCP server provides access to GitHub repositories, issues, pull requests, and more:
+
+```yaml
+# ~/.aia/config.yml
+:mcp_servers:
+  - name: "github"
+    command: "github-mcp-server"
+    args: ["stdio"]
+    timeout: 15000
+    env:
+      GITHUB_PERSONAL_ACCESS_TOKEN: "ghp_your_token_here"
+```
+
+**Setup:**
+```bash
+# Install GitHub MCP server (macOS)
+brew install github-mcp-server
+
+# Or via npm
+npm install -g @anthropic/github-mcp-server
+
+# Set your GitHub token (recommended: use environment variable instead of config)
+export GITHUB_PERSONAL_ACCESS_TOKEN="ghp_your_token_here"
+```
+
+#### Example: Hierarchical Temporal Memory (HTM)
+
+```shell
+gem install htm
+```
+
+See the [full HTM documentation](https://madbomber.github.io/htm) for database configuration and system environment variable usage.
+
+A custom Ruby-based MCP server for accessing database-backed long term memory:
+
+```yaml
+# ~/.aia/config.yml
+:mcp_servers:
+  - name: "htm"
+    command: "htm_mcp.rb"
+    args: ["stdio"]
+    timeout: 30000
+    env:
+      HTM_DBURL: "postgres://localhost:5432/htm_development"
+      ...
+```
+
+**Notes:**
+- The `command` can be just the executable name if it's in your PATH
+- AIA automatically resolves command paths, so you don't need absolute paths
+- Environment variables in the `env` section are passed only to that MCP server process
+
+#### Example: Multiple MCP Servers
+
+You can configure multiple MCP servers to provide different capabilities:
+
+```yaml
+# ~/.aia/config.yml
+:mcp_servers:
+  - name: "github"
+    command: "github-mcp-server"
+    args: ["stdio"]
+    env:
+      GITHUB_PERSONAL_ACCESS_TOKEN: "ghp_your_token_here"
+
+  - name: "htm"
+    command: "htm_mcp.rb"
+    args: ["stdio"]
+    timeout: 30000
+    env:
+      HTM_DBURL: "postgres://localhost:5432/htm_development"
+
+  - name: "filesystem"
+    command: "filesystem-mcp-server"
+    args: ["stdio", "--root", "/Users/me/projects"]
+```
+
+#### Verifying MCP Server Configuration
+
+When MCP servers are configured, AIA displays them in the startup robot:
+
+```
+       ,      ,
+       (\____/) AI Assistant (v0.9.23) is Online
+        (_oo_)   gpt-4o-mini (supports tools)
+         (O)       using ruby_llm (v1.9.0 MCP v0.6.1)
+       __||__    \) model db was last refreshed on
+     [/______\]  /    2025-06-03
+    / \__AI__/ \/      You can share my tools
+   /    /__\              MCP: github, htm
+  (\   /____\
+```
+
+Use the `//tools` directive in chat mode to see all available tools including those from MCP servers:
+
+```bash
+aia --chat
+> //tools
+
+Available Tools:
+- github_create_issue: Create a new GitHub issue
+- github_list_repos: List repositories for the authenticated user
+- htm_query: Execute a query against the HTM database
+- htm_insert: Insert a record into HTM
+...
+```
+
+#### Troubleshooting MCP Servers
+
+If an MCP server fails to load, AIA will display a warning:
+
+```
+WARNING: MCP server 'github' command not found: github-mcp-server
+WARNING: MCP server entry missing name or command: {...}
+ERROR: Failed to load MCP server 'htm': Connection timeout
+```
+
+**Common Issues:**
+
+| Problem | Solution |
+|---------|----------|
+| Command not found | Ensure the command is in your PATH or use absolute path |
+| Connection timeout | Increase the `timeout` value |
+| Missing environment variables | Add required env vars to the `env` section |
+| Server hangs on startup | Check that all required environment variables are set |
+
+**Debug Mode:**
+
+Enable debug mode to see detailed MCP server loading information:
+
+```bash
+aia --debug --chat
+```
+
 **Shared Tools Collection:**
 AIA can use the [shared_tools gem](https://github.com/madbomber/shared_tools) which provides a curated collection of commonly-used  tools (aka functions) via the --require option.
 

data/lib/aia/config/base.rb

@@ -140,6 +140,7 @@ module AIA
           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)
@@ -192,6 +193,64 @@ module AIA
           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)

data/lib/aia/config/defaults.rb

@@ -82,6 +82,9 @@ module AIA
 
         # 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

data/lib/aia/directives/models.rb

@@ -34,10 +34,15 @@ module AIA
           current_models = AIA.config.model
           current_models = [current_models] if current_models.is_a?(String)
 
-          using_local_provider = current_models.any? { |m| m.start_with?('ollama/', 'lms/') }
+          # Extract model names (handles both String and Hash formats)
+          model_names = current_models.map do |m|
+            m.is_a?(Hash) ? m[:model] : m
+          end
+
+          using_local_provider = model_names.any? { |m| m.start_with?('ollama/', 'lms/') }
 
           if using_local_provider
-            show_local_models(current_models, args)
+            show_local_models(model_names, args)
           else
             show_rubyllm_models(args)
           end

data/lib/aia/utility.rb

@@ -13,6 +13,15 @@ module AIA
         AIA.config&.tool_paths && !AIA.config.tool_paths.empty?
       end
 
+      def mcp_servers?
+        AIA.config&.mcp_servers && !AIA.config.mcp_servers.empty?
+      end
+
+      def mcp_server_names
+        return [] unless mcp_servers?
+        AIA.config.mcp_servers.map { |s| s[:name] || s["name"] }.compact
+      end
+
       def supports_tools?
         AIA.config&.client&.model&.supports_functions? || false
       end
@@ -46,6 +55,8 @@ module AIA
           'unknown-model'
         end
 
+        mcp_line = mcp_servers? ? "MCP: #{mcp_server_names.join(', ')}" : ''
+
         puts <<-ROBOT
 
        ,      ,
@@ -55,7 +66,7 @@ module AIA
        __||__    \\) model db was last refreshed on
      [/______\\]  /    #{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')}
-   /    /__\\
+   /    /__\\              #{mcp_line}
   (\\   /____\\   #{user_tools? && tools? ? 'My Toolbox contains:' : ''}
         ROBOT
         if user_tools? && tools?

data/lib/aia.rb

@@ -7,9 +7,8 @@
 require 'ruby_llm'
 require 'ruby_llm/mcp'
 
-# FIXME: Actually fix ruby_llm; this is supposed to a temporary fix for the issues
-#        discovered with shared_tools/ruby_llm/mcp/github_mcp_server
-RubyLLM::MCP.support_complex_parameters!
+# NOTE: Complex parameters are now supported natively in ruby_llm-mcp >= 0.8.0
+#       The support_complex_parameters! method has been removed.
 
 require 'prompt_manager'
 

data/lib/extensions/openstruct_merge.rb

@@ -21,7 +21,11 @@ class OpenStruct
   end
 
   # Sets value in result OpenStruct, handling nested OpenStruct and Hash objects
+  # Skip nil values to avoid overwriting existing values with nil
   def self.set_value(result, key, value)
+    # Skip nil values - don't overwrite existing values with nil
+    return if value.nil?
+
     if value.is_a?(OpenStruct) || value.is_a?(Hash)
       current_value = result[key]
       current_value = {} if current_value.nil?

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: aia
 version: !ruby/object:Gem::Version
-  version: 0.9.21
+  version: 0.9.23
 platform: ruby
 authors:
 - Dewayne VanHoozer
@@ -99,28 +99,28 @@ dependencies:
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 1.5.1
+        version: 1.9.1
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 1.5.1
+        version: 1.9.1
 - !ruby/object:Gem::Dependency
   name: ruby_llm-mcp
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.6.1
+        version: 0.8.0
   type: :runtime
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: 0.6.1
+        version: 0.8.0
 - !ruby/object:Gem::Dependency
   name: reline
   requirement: !ruby/object:Gem::Requirement
@@ -477,7 +477,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.7.2
+rubygems_version: 4.0.0
 specification_version: 4
 summary: Multi-model AI CLI with dynamic prompts, consensus responses, shell & Ruby
   integration, and seamless chat workflows.