RubyGems - actionmcp - Versions diffs - 0.111.0 → 0.200.0 - Mend

actionmcp 0.111.0 → 0.200.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (83) hide show

checksums.yaml +4 -4
data/README.md +26 -18
data/Rakefile +3 -0
data/app/controllers/action_mcp/application_controller.rb +166 -109
data/app/controllers/action_mcp/mcp_app_renderer.rb +1 -0
data/app/helpers/action_mcp/apps_helper.rb +48 -0
data/app/jobs/action_mcp/tool_execution_job.rb +67 -35
data/app/models/action_mcp/session/message.rb +8 -3
data/app/models/action_mcp/session/task.rb +15 -5
data/app/models/action_mcp/session.rb +93 -19
data/config/routes.rb +1 -1
data/lib/action_mcp/apps/javascript/ext_apps.LICENSE +216 -0
data/lib/action_mcp/apps/javascript/ext_apps.js +78 -0
data/lib/action_mcp/apps/view_manifest.rb +170 -0
data/lib/action_mcp/apps.rb +17 -2
data/lib/action_mcp/configuration.rb +14 -30
data/lib/action_mcp/content/audio.rb +4 -1
data/lib/action_mcp/content/base.rb +3 -1
data/lib/action_mcp/content/image.rb +4 -1
data/lib/action_mcp/content/resource.rb +15 -14
data/lib/action_mcp/content/resource_link.rb +15 -5
data/lib/action_mcp/content/text.rb +2 -1
data/lib/action_mcp/content/validation.rb +253 -0
data/lib/action_mcp/content.rb +3 -1
data/lib/action_mcp/dev/runner.rb +203 -0
data/lib/action_mcp/engine.rb +24 -2
data/lib/action_mcp/json_rpc_handler_base.rb +7 -12
data/lib/action_mcp/logging/logger.rb +16 -11
data/lib/action_mcp/logging.rb +34 -1
data/lib/action_mcp/middleware/origin_validation.rb +9 -13
data/lib/action_mcp/output_schema_builder.rb +2 -0
data/lib/action_mcp/prompt_response.rb +12 -2
data/lib/action_mcp/protocol_validator.rb +513 -0
data/lib/action_mcp/renderable.rb +4 -4
data/lib/action_mcp/resource.rb +15 -11
data/lib/action_mcp/resource_template.rb +105 -90
data/lib/action_mcp/resource_templates_registry.rb +31 -97
data/lib/action_mcp/schema_validator.rb +43 -0
data/lib/action_mcp/server/base_session.rb +134 -28
data/lib/action_mcp/server/base_session_store.rb +1 -1
data/lib/action_mcp/server/capabilities.rb +10 -30
data/lib/action_mcp/server/client_notifications.rb +63 -0
data/lib/action_mcp/server/completions.rb +68 -0
data/lib/action_mcp/server/elicitation.rb +47 -25
data/lib/action_mcp/server/elicitation_request.rb +136 -51
data/lib/action_mcp/server/handlers/logging_handler.rb +7 -5
data/lib/action_mcp/server/handlers/prompt_handler.rb +12 -1
data/lib/action_mcp/server/handlers/resource_handler.rb +41 -7
data/lib/action_mcp/server/handlers/task_handler.rb +26 -0
data/lib/action_mcp/server/handlers/tool_handler.rb +8 -0
data/lib/action_mcp/server/json_rpc_handler.rb +5 -21
data/lib/action_mcp/server/messaging_service.rb +3 -1
data/lib/action_mcp/server/prompts.rb +5 -3
data/lib/action_mcp/server/resources.rb +35 -2
data/lib/action_mcp/server/response_collector.rb +5 -1
data/lib/action_mcp/server/roots.rb +15 -2
data/lib/action_mcp/server/sampling.rb +36 -1
data/lib/action_mcp/server/sampling_request.rb +317 -55
data/lib/action_mcp/server/tasks.rb +27 -45
data/lib/action_mcp/server/tool_result.rb +27 -0
data/lib/action_mcp/server/tools.rb +13 -8
data/lib/action_mcp/server/transport_handler.rb +3 -0
data/lib/action_mcp/server/url_elicitation_request.rb +13 -0
data/lib/action_mcp/tool.rb +133 -199
data/lib/action_mcp/tool_response.rb +35 -4
data/lib/action_mcp/tools_registry.rb +1 -1
data/lib/action_mcp/version.rb +1 -1
data/lib/action_mcp.rb +2 -5
data/lib/generators/action_mcp/identifier/identifier_generator.rb +1 -1
data/lib/generators/action_mcp/identifier/templates/identifier.rb.erb +1 -1
data/lib/generators/action_mcp/install/templates/bin/mcp +16 -4
data/lib/generators/action_mcp/install/templates/mcp/config.ru.tt +2 -2
data/lib/generators/action_mcp/install/templates/mcp.yml +4 -0
data/lib/generators/action_mcp/resource_template/resource_template_generator.rb +18 -1
data/lib/generators/action_mcp/resource_template/templates/resource_template_ui.rb.erb +27 -0
data/lib/generators/action_mcp/tool/templates/tool.rb.erb +6 -0
data/lib/generators/action_mcp/tool/tool_generator.rb +2 -0
data/lib/generators/action_mcp/view/templates/resource_template.rb.erb +16 -0
data/lib/generators/action_mcp/view/templates/view.html.erb +27 -0
data/lib/generators/action_mcp/view/view_generator.rb +57 -0
data/lib/tasks/action_mcp_apps.rake +17 -0
data/lib/tasks/action_mcp_tasks.rake +135 -74
metadata +47 -17

data/lib/generators/action_mcp/install/templates/mcp/config.ru.tt CHANGED Viewed

@@ -9,8 +9,8 @@
 #
 # Start this server with:
 #   bin/mcp                                              # Uses Falcon (recommended)
-#   bundle exec falcon serve --bind http://0.0.0.0:62770 mcp/config.ru
-#   bundle exec rails s -c mcp/config.ru -p 62770       # Uses Puma (fallback)
+#   bundle exec falcon serve --bind http://127.0.0.1:62770 mcp/config.ru
+#   bundle exec rails s -c mcp/config.ru -b 127.0.0.1 -p 62770 # Uses Puma (fallback)
 #
 # Port 62770 = MCPS0 on a phone keypad (MCP Server, instance 0)
 #

data/lib/generators/action_mcp/install/templates/mcp.yml CHANGED Viewed

@@ -24,6 +24,10 @@ shared:
   # io.modelcontextprotocol/ui with text/html;profile=mcp-app support.
   # mcp_apps_enabled: true
+  # Directory holding JavaScript view bundles compiled from JavaScript or
+  # TypeScript by @action-mcp/vite-plugin. Relative to Rails.root.
+  # mcp_apps_views_path: .action_mcp/views
   # MCP capability profiles
   profiles:
     primary:

data/lib/generators/action_mcp/resource_template/resource_template_generator.rb CHANGED Viewed

@@ -11,8 +11,12 @@ module ActionMCP
       argument :name, type: :string, required: true, banner: "ResourceTemplateName"
+      class_option :ui, type: :boolean, default: false,
+                        desc: "Generate an MCP Apps UI template (ui:// URI, :mcp_app mime type, render_ui)"
       def create_resource_template_file
-        template "resource_template.rb.erb", "app/mcp/resource_templates/#{file_name}.rb"
+        source = options[:ui] ? "resource_template_ui.rb.erb" : "resource_template.rb.erb"
+        template source, "app/mcp/resource_templates/#{file_name}.rb"
       end
       private
@@ -25,6 +29,19 @@ module ActionMCP
         base = name.underscore
         base.end_with?("_template") ? base : "#{base}_template"
       end
+      # Name without the Template suffix, for view paths and ui:// URIs.
+      def base_name
+        name.camelize.delete_suffix("Template")
+      end
+      def view_name
+        base_name.underscore
+      end
+      def uri_name
+        view_name.dasherize
+      end
     end
   end
 end

data/lib/generators/action_mcp/resource_template/templates/resource_template_ui.rb.erb ADDED Viewed

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+class <%= class_name %> < ApplicationMCPResTemplate
+  description "Interactive UI view for <%= base_name.titleize.downcase %>"
+  uri_template "ui://views/<%= uri_name %>"
+  mime_type :mcp_app
+  # Declare CSP origins and host preferences for the embedded view (MCP Apps / SEP-1865).
+  # Only origins listed here can be reached from inside the view iframe.
+  # ui csp: { connectDomains: %w[https://api.example.com] },
+  #    prefersBorder: true
+  def resolve
+    # Inline HTML keeps this template self-contained. To render an ERB view
+    # instead, run `bin/rails generate action_mcp:view <%= base_name %>` or
+    # create app/views/mcp/ui/<%= view_name %>.html.erb and use:
+    #   render_ui(template: "mcp/ui/<%= view_name %>")
+    render_ui(text: <<~HTML)
+      <!doctype html>
+      <html>
+        <body>
+          <h1><%= base_name.titleize %></h1>
+        </body>
+      </html>
+    HTML
+  end
+end

data/lib/generators/action_mcp/tool/templates/tool.rb.erb CHANGED Viewed

@@ -23,6 +23,12 @@ class <%= class_name %> < ApplicationMCPTool
   annotate(:<%= k %>, <%= v.inspect %>)
   <% end %>
 <% end %>
+<% if options[:ui] %>
+  renders_ui "<%= options[:ui] %>"
+<% else %>
+  # Link this tool to an MCP Apps UI view (generate one with `bin/rails g action_mcp:view MyView`):
+  # renders_ui "ui://views/my-view", visibility: %i[model app]
+<% end %>
     <% if properties.empty? %>
   property :input, type: "string", description: "Input", required: true

data/lib/generators/action_mcp/tool/tool_generator.rb CHANGED Viewed

@@ -14,6 +14,8 @@ module ActionMCP
       class_option :destructive, type: :boolean, default: false
       class_option :category, type: :string, default: nil
       class_option :properties, type: :array, default: [], banner: "name:type:description:required"
+      class_option :ui, type: :string, default: nil, banner: "ui://views/my-view",
+                        desc: "Link this tool to an MCP Apps UI view via renders_ui"
       def create_tool_file
         template "tool.rb.erb", "app/mcp/tools/#{file_name}.rb"

data/lib/generators/action_mcp/view/templates/resource_template.rb.erb ADDED Viewed

@@ -0,0 +1,16 @@
+# frozen_string_literal: true
+class <%= class_name %> < ApplicationMCPResTemplate
+  description "Interactive UI view for <%= base_name.titleize.downcase %>"
+  uri_template "ui://views/<%= uri_name %>"
+  mime_type :mcp_app
+  # Declare CSP origins and host preferences for the embedded view (MCP Apps / SEP-1865).
+  # Only origins listed here can be reached from inside the view iframe.
+  # ui csp: { connectDomains: %w[https://api.example.com] },
+  #    prefersBorder: true
+  def resolve
+    render_ui(template: "mcp/ui/<%= view_name %>")
+  end
+end

data/lib/generators/action_mcp/view/templates/view.html.erb ADDED Viewed

@@ -0,0 +1,27 @@
+<!doctype html>
+<html>
+  <head>
+    <meta charset="utf-8">
+    <style>
+      /* Styles must be inline or served from origins declared in the ui csp metadata. */
+      body { font-family: system-ui, sans-serif; margin: 1rem; }
+    </style>
+  </head>
+  <body>
+    <h1><%= base_name.titleize %></h1>
+    <pre id="output">Waiting for tool result…</pre>
+    <%%= mcp_app_bridge_tag %>
+    <script type="module">
+      // Register handlers before connecting; handlers set after connect() can miss events.
+      const app = await ActionMCP.connect({
+        handlers: {
+          ontoolresult: (result) => {
+            document.getElementById("output").textContent =
+              JSON.stringify(result.structuredContent ?? result, null, 2);
+          }
+        }
+      });
+      // app.callServerTool({ name: "my-tool", arguments: { ... } }) calls back into the server.
+    </script>
+  </body>
+</html>

data/lib/generators/action_mcp/view/view_generator.rb ADDED Viewed

@@ -0,0 +1,57 @@
+# frozen_string_literal: true
+require "rails/generators"
+module ActionMCP
+  module Generators
+    class ViewGenerator < Rails::Generators::NamedBase
+      namespace "action_mcp:view"
+      source_root File.expand_path("templates", __dir__)
+      desc "Creates an MCP Apps UI view: a ui:// ResourceTemplate (in app/mcp/resource_templates) " \
+           "paired with an ERB view (in app/views/mcp/ui)"
+      argument :name, type: :string, required: true, banner: "ViewName"
+      def create_resource_template_file
+        template "resource_template.rb.erb", "app/mcp/resource_templates/#{file_name}.rb"
+      end
+      def create_view_file
+        template "view.html.erb", "app/views/mcp/ui/#{view_name}.html.erb"
+      end
+      def show_instructions
+        say <<~INSTRUCTIONS
+          MCP Apps view created. Next steps:
+            1. Enable MCP Apps in config/mcp.yml: mcp_apps_enabled: true
+            2. Link a tool to this view: renders_ui "ui://views/#{uri_name}"
+            3. Edit app/views/mcp/ui/#{view_name}.html.erb
+        INSTRUCTIONS
+      end
+      private
+      def class_name
+        "#{base_name.camelize}Template"
+      end
+      def file_name
+        "#{view_name}_template"
+      end
+      # Name without a trailing Template suffix, used for the view path and URI.
+      def base_name
+        name.camelize.delete_suffix("Template")
+      end
+      def view_name
+        base_name.underscore
+      end
+      def uri_name
+        view_name.dasherize
+      end
+    end
+  end
+end

data/lib/tasks/action_mcp_apps.rake ADDED Viewed

@@ -0,0 +1,17 @@
+# frozen_string_literal: true
+namespace :action_mcp do
+  namespace :apps do
+    desc "Print registered tool schemas as JSON (consumed by @action-mcp/vite-plugin for types.d.ts)"
+    task schema: :environment do
+      Rails.application.eager_load!
+      tools = ActionMCP::ToolsRegistry.non_abstract.sort_by(&:name).map do |tool|
+        tool.klass.to_h(protocol_version: ActionMCP.configuration.protocol_version)
+             .slice(:name, :description, :inputSchema, :outputSchema)
+      end
+      puts JSON.generate({ tools: tools })
+    end
+  end
+end

data/lib/tasks/action_mcp_tasks.rake CHANGED Viewed

@@ -9,10 +9,13 @@ namespace :action_mcp do
     puts "\e[34mACTION MCP TOOLS\e[0m"  # Blue
     puts "\e[34m---------------\e[0m"   # Blue
-    ActionMCP::Tool.descendants.each do |tool|
-      next if tool.abstract?
-      puts "\e[34m#{tool.capability_name}:\e[0m #{tool.description}" # Blue name
+    tools = ActionMCP::ToolsRegistry.non_abstract.sort_by(&:name)
+    if tools.any?
+      tools.each do |tool|
+        puts "\e[34m#{tool.name}:\e[0m #{tool.description}" # Blue name
+      end
+    else
+      puts "  No tools registered"
     end
     puts "\n"
   end
@@ -25,10 +28,13 @@ namespace :action_mcp do
     puts "\e[32mACTION MCP PROMPTS\e[0m"  # Green
     puts "\e[32m-----------------\e[0m"   # Green
-    ActionMCP::Prompt.descendants.each do |prompt|
-      next if prompt.abstract?
-      puts "\e[32m#{prompt.capability_name}:\e[0m #{prompt.description}" # Green name
+    prompts = ActionMCP::PromptsRegistry.non_abstract.sort_by(&:name)
+    if prompts.any?
+      prompts.each do |prompt|
+        puts "\e[32m#{prompt.name}:\e[0m #{prompt.description}" # Green name
+      end
+    else
+      puts "  No prompts registered"
     end
     puts "\n"
   end
@@ -41,11 +47,70 @@ namespace :action_mcp do
     puts "\e[33mACTION MCP RESOURCES\e[0m" # Yellow
     puts "\e[33m--------------------\e[0m" # Yellow
-    ActionMCP::ResourceTemplate.descendants.each do |resource|
-      next if resource.abstract?
+    resources = ActionMCP::ResourceTemplatesRegistry.non_abstract.sort_by(&:name)
+    if resources.any?
+      resources.each do |resource|
+        puts "\e[33m#{resource.name}:\e[0m #{resource.description} : #{resource.klass.uri_template}" # Yellow name
+      end
+    else
+      puts "  No resources registered"
+    end
+    puts "\n"
+  end
+  # bin/rails action_mcp:list_widgets
+  desc "List MCP Apps UI widgets and their linked tools"
+  task list_widgets: :environment do
+    Rails.application.eager_load!
+    widgets = ActionMCP::ResourceTemplatesRegistry.non_abstract
+      .select { |resource| resource.klass.mime_type == ActionMCP::Apps::MIME_TYPE }
+      .sort_by(&:name)
+    widget_classes = widgets.map(&:klass)
+    tools_by_widget = Hash.new { |linked, widget| linked[widget] = [] }
+    missing_resources = Hash.new { |uris, uri| uris[uri] = [] }
+    ActionMCP::ToolsRegistry.non_abstract.sort_by(&:name).each do |tool|
+      descriptor = tool.klass.to_h(protocol_version: ActionMCP.configuration.protocol_version)
+      meta = descriptor[:_meta] || descriptor["_meta"]
+      ui = meta && (meta[:ui] || meta["ui"])
+      next unless ui.is_a?(Hash)
+      resource_uri = ui[:resourceUri] || ui["resourceUri"]
+      next unless resource_uri
+      visibility = ui[:visibility] || ui["visibility"]
+      widget = ActionMCP::ResourceTemplatesRegistry.find_template_for_uri(resource_uri, templates: widget_classes)
+      destination = widget ? tools_by_widget[widget] : missing_resources[resource_uri]
+      destination << [ tool.name, Array(visibility) ]
+    end
+    puts "\e[36mACTION MCP UI WIDGETS\e[0m" # Cyan
+    puts "\e[36m---------------------\e[0m" # Cyan
+    if widgets.any?
+      widgets.each do |widget|
+        uri = widget.klass.uri_template
+        linked_tools = tools_by_widget.fetch(widget.klass, []).map do |tool_name, visibility|
+          visibility.any? ? "#{tool_name} [visibility: #{visibility.join(', ')}]" : tool_name
+        end
+        puts "\e[36m#{widget.name}:\e[0m #{uri}"
+        puts "  #{widget.description}" if widget.description.present?
+        puts "  Linked tools: #{linked_tools.any? ? linked_tools.join(', ') : 'None'}"
+      end
+    else
+      puts "  No MCP Apps UI widgets registered"
+    end
-      puts "\e[33m#{resource.capability_name}:\e[0m #{resource.description} : #{resource.uri_template}" # Yellow name
+    if missing_resources.any?
+      puts "\n\e[31mMISSING UI WIDGET RESOURCES\e[0m" # Red
+      puts "\e[31m---------------------------\e[0m" # Red
+      missing_resources.sort.each do |uri, linked_tools|
+        puts "  #{uri}: #{linked_tools.map(&:first).join(', ')}"
+      end
     end
     puts "\n"
   end
@@ -81,9 +146,8 @@ namespace :action_mcp do
     profiles = ActionMCP.configuration.profiles
     unless profiles.key?(profile_name)
-      puts "\e[31mProfile '#{profile_name}' not found!\e[0m"
-      puts "Available profiles: #{profiles.keys.join(', ')}"
-      next
+      abort "\e[31mProfile '#{profile_name}' not found!\e[0m\n" \
+            "Available profiles: #{profiles.keys.join(', ')}"
     end
     # Temporarily activate this profile to show what would be included
@@ -142,73 +206,70 @@ namespace :action_mcp do
     puts "\n"
   end
-  desc "List all tools, prompts, resources and available profiles"
-  task list: %i[list_tools list_prompts list_resources list_profiles] do
-    # This task lists all tools, prompts, resources and profiles
+  desc "List all tools, prompts, resources, UI widgets and available profiles"
+  task list: %i[list_tools list_prompts list_resources list_widgets list_profiles] do
+    # This task lists all registered MCP components and profiles.
   end
   # bin/rails action_mcp:info
-  # bin/rails action_mcp:info[test]
-  desc "Display ActionMCP configuration for current or specified environment"
+  desc "Display the live ActionMCP configuration for the current Rails environment"
   task :info, [ :env ] => :environment do |_t, args|
-    env = args[:env] || Rails.env
-    # Load configuration for the specified environment
-    original_env = Rails.env
-    Rails.env = env.to_s
+    requested_env = args[:env]&.to_s
+    if requested_env && requested_env != Rails.env
+      abort "ActionMCP configuration is initialized when Rails boots. " \
+            "Run `RAILS_ENV=#{requested_env} bin/rails action_mcp:info` instead."
+    end
-    # Reload configuration to get the environment-specific settings
-    config = ActionMCP::Configuration.new
-    config.load_profiles
+    env = Rails.env.to_s
+    config = ActionMCP.configuration
     puts "\e[35mActionMCP Configuration (#{env})\e[0m"
     puts "\e[35m#{'=' * (25 + env.length)}\e[0m"
-    # Basic Information
     puts "\n\e[36mBasic Information:\e[0m"
     puts "  Name: #{config.name}"
     puts "  Version: #{config.version}"
     puts "  Protocol Version: #{config.protocol_version}"
     puts "  Active Profile: #{config.active_profile}"
-    # Session Storage
+    puts "\n\e[36mTransport:\e[0m"
+    puts "  Base Path: #{config.base_path}"
+    puts "  Allowed Origins: #{config.allowed_origins.join(', ')}"
+    puts "  Pagination Page Size: #{config.pagination_page_size || 'disabled'}"
     puts "\n\e[36mSession Storage:\e[0m"
     puts "  Session Store Type: #{config.session_store_type}"
     puts "  Client Session Store: #{config.client_session_store_type || 'default'}"
     puts "  Server Session Store: #{config.server_session_store_type || 'default'}"
-    # Transport Configuration
-    puts "\n\e[36mTransport Configuration:\e[0m"
-    puts "  Protocol Version: #{config.protocol_version}"
-    # Pub/Sub Adapter
-    puts "\n\e[36mPub/Sub Adapter:\e[0m"
-    puts "  Adapter: #{config.adapter || 'not configured'}"
-    if config.adapter
-      puts "  Polling Interval: #{config.polling_interval}" if config.polling_interval
-      puts "  Min Threads: #{config.min_threads}" if config.min_threads
-      puts "  Max Threads: #{config.max_threads}" if config.max_threads
-      puts "  Max Queue: #{config.max_queue}" if config.max_queue
-    end
+    puts "\n\e[36mThread Pool:\e[0m"
+    puts "  Min Threads: #{config.min_threads || 'default'}"
+    puts "  Max Threads: #{config.max_threads || 'default'}"
+    puts "  Max Queue: #{config.max_queue || 'default'}"
-    # Authentication
     puts "\n\e[36mAuthentication:\e[0m"
-    puts "  Methods: #{config.authentication_methods.join(', ')}"
-    if config.oauth_config&.any?
-      puts "  OAuth Provider: #{config.oauth_config['provider']}"
-      puts "  OAuth Scopes: #{config.oauth_config['scopes_supported']&.join(', ')}"
-    end
+    methods = config.authentication_methods
+    puts "  Methods: #{methods.any? ? methods.join(', ') : 'none'}"
+    puts "  Allowed Identity Keys: #{config.allowed_identity_keys.join(', ')}"
+    puts "\n\e[36mMCP Apps:\e[0m"
+    puts "  Enabled: #{config.mcp_apps_enabled}"
+    puts "  Views Path: #{config.mcp_apps_views_path}"
+    puts "  Widget MIME Type: #{ActionMCP::Apps::MIME_TYPE}"
+    puts "\n\e[36mTasks:\e[0m"
+    puts "  Enabled: #{config.tasks_enabled}"
+    puts "  List Enabled: #{config.tasks_list_enabled}"
+    puts "  Cancel Enabled: #{config.tasks_cancel_enabled}"
+    puts "  Result Poll Interval: #{config.tasks_result_poll_interval}"
-    # Logging
     puts "\n\e[36mLogging:\e[0m"
     puts "  Logging Enabled: #{config.logging_enabled}"
     puts "  Logging Level: #{config.logging_level}"
-    # Gateway
     puts "\n\e[36mGateway:\e[0m"
-    puts "  Gateway Class: #{config.gateway_class}"
+    puts "  Gateway Class: #{config.gateway_class || 'not configured'}"
-    # Capabilities
     puts "\n\e[36mEnabled Capabilities:\e[0m"
     capabilities = config.capabilities
     if capabilities.any?
@@ -219,15 +280,11 @@ namespace :action_mcp do
       puts "  None"
     end
-    # Available Profiles
     puts "\n\e[36mAvailable Profiles:\e[0m"
     config.profiles.each_key do |profile_name|
       puts "  - #{profile_name}"
     end
-    # Restore original environment
-    Rails.env = original_env
     puts "\n"
   end
@@ -237,15 +294,25 @@ namespace :action_mcp do
     puts "\e[35mActionMCP Statistics\e[0m"
     puts "\e[35m===================\e[0m"
-    # Debug database connection
-    puts "\n\e[36mDatabase Debug:\e[0m"
+    database_path = nil
+    puts "\n\e[36mDatabase:\e[0m"
     puts "  Rails Environment: #{Rails.env}"
     puts "  Rails Root: #{Rails.root}"
-    puts "  Database Config: #{ActionMCP::ApplicationRecord.connection_db_config.configuration_hash.inspect}"
-    if ActionMCP::ApplicationRecord.connection.adapter_name.downcase.include?("sqlite")
-      db_path = ActionMCP::ApplicationRecord.connection_db_config.database
-      puts "  Database Path: #{db_path}"
-      puts "  Database Exists?: #{File.exist?(db_path)}" if db_path
+    begin
+      connection = ActionMCP::ApplicationRecord.connection
+      db_config = ActionMCP::ApplicationRecord.connection_db_config
+      puts "  Adapter: #{connection.adapter_name}"
+      puts "  Database: #{db_config.database}"
+      if connection.adapter_name.downcase.include?("sqlite") && db_config.database != ":memory:"
+        configured_path = Pathname.new(db_config.database.to_s)
+        database_path = configured_path.absolute? ? configured_path : Rails.root.join(configured_path)
+        puts "  Database Path: #{database_path}"
+        puts "  Database Exists?: #{database_path.file?}"
+      end
+    rescue StandardError => e
+      puts "  Could not inspect database connection: #{e.message}"
     end
     # Session Statistics
@@ -332,19 +399,13 @@ namespace :action_mcp do
     # Storage Information
     puts "\n\e[36mStorage Information:\e[0m"
     puts "  Session Store Type: #{ActionMCP.configuration.session_store_type}"
-    puts "  Database Adapter: #{ActionMCP::ApplicationRecord.connection.adapter_name}"
-    # Database size (if SQLite)
     begin
-      if ActionMCP::ApplicationRecord.connection.adapter_name.downcase.include?("sqlite")
-        db_config = Rails.application.config.database_configuration[Rails.env]
-        if db_config && db_config["database"]
-          db_file = db_config["database"]
-          if File.exist?(db_file)
-            size_mb = (File.size(db_file) / 1024.0 / 1024.0).round(2)
-            puts "  Database Size: #{size_mb} MB"
-          end
-        end
+      connection = ActionMCP::ApplicationRecord.connection
+      puts "  Database Adapter: #{connection.adapter_name}"
+      if database_path&.file?
+        size_mb = (database_path.size / 1024.0 / 1024.0).round(2)
+        puts "  Database Size: #{size_mb} MB"
       end
     rescue StandardError => e
       puts "  Could not determine database size: #{e.message}"