rails-profiler 0.29.0 → 0.30.0

data/lib/profiler/mcp/server.rb

@@ -59,6 +59,7 @@ module Profiler
       private
 
       def build_tools
+        require_relative "slave_support"
         require_relative "file_cache"
         require_relative "path_extractor"
         require_relative "body_formatter"
@@ -82,6 +83,9 @@ module Profiler
         require_relative "tools/delete_env_var"
         require_relative "tools/reset_env_var"
         require_relative "tools/reset_all_env_vars"
+        require_relative "tools/list_slaves"
+
+        slave_param = { type: "string", description: "Name of a connected slave profiler to target. Omit to use this profiler's own data." }
 
         [
           define_tool(
@@ -95,7 +99,8 @@ module Profiler
                 profile_type: { type: "string", description: "Filter by type: 'http' or 'job'" },
                 limit: { type: "number", description: "Maximum number of results (default 20)" },
                 fields: { type: "array", items: { type: "string" }, description: "Columns to include. Valid values: time, type, method, path, duration, queries, status, token. Omit for all." },
-                cursor: { type: "string", description: "Pagination cursor: ISO8601 timestamp of the last item seen. Returns profiles older than this." }
+                cursor: { type: "string", description: "Pagination cursor: ISO8601 timestamp of the last item seen. Returns profiles older than this." },
+                slave: slave_param
               }
             },
             handler: Tools::QueryProfiles
@@ -112,7 +117,8 @@ module Profiler
                 json_path: { type: "string", description: "JSONPath expression to extract from response body (e.g. '$.data.items[0]'). Only applied when save_bodies is true." },
                 xml_path: { type: "string", description: "XPath expression to extract from response body (e.g. '//items/item[1]/name'). Only applied when save_bodies is true." },
                 log_min_level: { type: "string", description: "Minimum log level to include in the logs section: DEBUG, INFO, WARN, ERROR, FATAL. Only applied when 'logs' section is requested." },
-                env_filter: { type: "string", description: "Required when requesting the env section. Case-insensitive substring filter on ENV key name (e.g. 'RAILS', 'DATABASE')." }
+                env_filter: { type: "string", description: "Required when requesting the env section. Case-insensitive substring filter on ENV key name (e.g. 'RAILS', 'DATABASE')." },
+                slave: slave_param
               },
               required: ["token"]
             },
@@ -124,7 +130,8 @@ module Profiler
             input_schema: {
               properties: {
                 token: { type: "string", description: "Profile token, or 'latest' for the most recent profile (required)" },
-                summary_only: { type: "boolean", description: "Return only the summary statistics section, skipping slow query and N+1 details." }
+                summary_only: { type: "boolean", description: "Return only the summary statistics section, skipping slow query and N+1 details." },
+                slave: slave_param
               },
               required: ["token"]
             },
@@ -136,7 +143,8 @@ module Profiler
             input_schema: {
               properties: {
                 token: { type: "string", description: "Profile token, or 'latest' for the most recent profile (required)" },
-                query_index: { type: "integer", description: "Zero-based index of the query within the profile's database queries list (required)" }
+                query_index: { type: "integer", description: "Zero-based index of the query within the profile's database queries list (required)" },
+                slave: slave_param
               },
               required: ["token", "query_index"]
             },
@@ -147,7 +155,8 @@ module Profiler
             description: "Get detailed AJAX sub-request breakdown for a profile. Use 'latest' as token to get the most recent profile.",
             input_schema: {
               properties: {
-                token: { type: "string", description: "Profile token, or 'latest' for the most recent profile (required)" }
+                token: { type: "string", description: "Profile token, or 'latest' for the most recent profile (required)" },
+                slave: slave_param
               },
               required: ["token"]
             },
@@ -158,7 +167,8 @@ module Profiler
             description: "Get variable dumps captured during a profile. Use 'latest' as token to get the most recent profile.",
             input_schema: {
               properties: {
-                token: { type: "string", description: "Profile token, or 'latest' for the most recent profile (required)" }
+                token: { type: "string", description: "Profile token, or 'latest' for the most recent profile (required)" },
+                slave: slave_param
               },
               required: ["token"]
             },
@@ -174,7 +184,8 @@ module Profiler
                 save_bodies: { type: "boolean", description: "Save request/response bodies to temp files and return paths instead of inlining content." },
                 max_body_size: { type: "number", description: "Truncate inlined body content at N characters. Ignored when save_bodies is true." },
                 json_path: { type: "string", description: "JSONPath expression to extract from response bodies (e.g. '$.data.items[0]'). Only applied when save_bodies is true." },
-                xml_path: { type: "string", description: "XPath expression to extract from response bodies (e.g. '//items/item[1]/name'). Only applied when save_bodies is true." }
+                xml_path: { type: "string", description: "XPath expression to extract from response bodies (e.g. '//items/item[1]/name'). Only applied when save_bodies is true." },
+                slave: slave_param
               },
               required: ["token"]
             },
@@ -190,7 +201,8 @@ module Profiler
                 action: { type: "string", description: "Filter by mailer action (partial match, e.g. 'welcome_email')" },
                 delivery_mode: { type: "string", description: "Filter by delivery mode: 'deliver_now', 'deliver_later', or 'queued'" },
                 save_bodies: { type: "boolean", description: "Save email bodies to temp files and return paths instead of inlining content." },
-                max_body_size: { type: "number", description: "Truncate inlined body content at N characters." }
+                max_body_size: { type: "number", description: "Truncate inlined body content at N characters." },
+                slave: slave_param
               },
               required: ["token"]
             },
@@ -205,7 +217,8 @@ module Profiler
                 status: { type: "string", description: "Filter by status (completed, failed)" },
                 limit: { type: "number", description: "Maximum number of results (default 20)" },
                 fields: { type: "array", items: { type: "string" }, description: "Columns to include. Valid values: time, job_class, queue, status, duration, token. Omit for all." },
-                cursor: { type: "string", description: "Pagination cursor: ISO8601 timestamp of the last item seen. Returns jobs older than this." }
+                cursor: { type: "string", description: "Pagination cursor: ISO8601 timestamp of the last item seen. Returns jobs older than this." },
+                slave: slave_param
               }
             },
             handler: Tools::QueryJobs
@@ -221,7 +234,8 @@ module Profiler
                 has_error: { type: "boolean", description: "Filter to only emails with delivery errors" },
                 limit: { type: "number", description: "Maximum number of results (default 20)" },
                 fields: { type: "array", items: { type: "string" }, description: "Columns to include. Valid values: time, profile, mailer, action, subject, to, mode, duration, status, token. Omit for all." },
-                cursor: { type: "string", description: "Pagination cursor: ISO8601 timestamp of the last item seen." }
+                cursor: { type: "string", description: "Pagination cursor: ISO8601 timestamp of the last item seen." },
+                slave: slave_param
               }
             },
             handler: Tools::QueryMailers
@@ -236,7 +250,8 @@ module Profiler
                 min_duration: { type: "number", description: "Minimum duration in milliseconds" },
                 limit: { type: "number", description: "Maximum number of results (default 20)" },
                 fields: { type: "array", items: { type: "string" }, description: "Columns to include. Valid values: time, test_name, status, duration, queries, n1, token. Omit for all." },
-                cursor: { type: "string", description: "Pagination cursor: ISO8601 timestamp of the last item seen." }
+                cursor: { type: "string", description: "Pagination cursor: ISO8601 timestamp of the last item seen." },
+                slave: slave_param
               }
             },
             handler: Tools::QueryTestProfiles
@@ -246,7 +261,8 @@ module Profiler
             description: "Get detailed data for a test profile: metadata, SQL queries, N+1 patterns, cache, exception. Use 'latest' as token for the most recent test.",
             input_schema: {
               properties: {
-                token: { type: "string", description: "Test profile token, or 'latest' for the most recent test profile (required)" }
+                token: { type: "string", description: "Test profile token, or 'latest' for the most recent test profile (required)" },
+                slave: slave_param
               },
               required: ["token"]
             },
@@ -273,7 +289,8 @@ module Profiler
                 max_output: {
                   type: "number",
                   description: "Maximum characters of output to return (tail). Default: 4000."
-                }
+                },
+                slave: slave_param
               }
             },
             handler: Tools::RunTests
@@ -288,7 +305,8 @@ module Profiler
                 min_duration: { type: "number", description: "Minimum duration in milliseconds" },
                 limit: { type: "number", description: "Maximum number of results (default 20)" },
                 fields: { type: "array", items: { type: "string" }, description: "Columns to include. Valid values: time, expression, return_value, status, duration, queries, token. Omit for all." },
-                cursor: { type: "string", description: "Pagination cursor: ISO8601 timestamp of the last item seen. Returns profiles older than this." }
+                cursor: { type: "string", description: "Pagination cursor: ISO8601 timestamp of the last item seen. Returns profiles older than this." },
+                slave: slave_param
               }
             },
             handler: Tools::QueryConsoleProfiles
@@ -298,7 +316,8 @@ module Profiler
             description: "Clear profiler history. Omit type to clear everything, or pass 'http', 'job', 'test', or 'console' to clear only that type.",
             input_schema: {
               properties: {
-                type: { type: "string", description: "Optional: 'http' to clear only requests, 'job' to clear only jobs, 'test' to clear only test profiles, 'console' to clear only console sessions" }
+                type: { type: "string", description: "Optional: 'http' to clear only requests, 'job' to clear only jobs, 'test' to clear only test profiles, 'console' to clear only console sessions" },
+                slave: slave_param
               }
             },
             handler: Tools::ClearProfiles
@@ -309,7 +328,8 @@ module Profiler
             input_schema: {
               properties: {
                 include_all: { type: "boolean", description: "If true, return all ENV variables (not just overrides). Default: false." },
-                filter: { type: "string", description: "Case-insensitive substring filter on key name." }
+                filter: { type: "string", description: "Case-insensitive substring filter on key name." },
+                slave: slave_param
               }
             },
             handler: Tools::ListEnvVars
@@ -320,7 +340,8 @@ module Profiler
             input_schema: {
               properties: {
                 key:   { type: "string", description: "Environment variable name (required)" },
-                value: { type: "string", description: "New value (required)" }
+                value: { type: "string", description: "New value (required)" },
+                slave: slave_param
               },
               required: ["key", "value"]
             },
@@ -331,7 +352,8 @@ module Profiler
             description: "Delete an environment variable for this session (persisted across restarts until reset).",
             input_schema: {
               properties: {
-                key: { type: "string", description: "Environment variable name (required)" }
+                key: { type: "string", description: "Environment variable name (required)" },
+                slave: slave_param
               },
               required: ["key"]
             },
@@ -342,7 +364,8 @@ module Profiler
             description: "Restore an overridden environment variable to its original value.",
             input_schema: {
               properties: {
-                key: { type: "string", description: "Environment variable name to restore (required)" }
+                key: { type: "string", description: "Environment variable name to restore (required)" },
+                slave: slave_param
               },
               required: ["key"]
             },
@@ -351,8 +374,14 @@ module Profiler
           define_tool(
             name: "reset_all_env_vars",
             description: "Restore all overridden environment variables to their original values.",
-            input_schema: { properties: {} },
+            input_schema: { properties: { slave: slave_param } },
             handler: Tools::ResetAllEnvVars
+          ),
+          define_tool(
+            name: "list_slaves",
+            description: "List all slave profilers connected to this master profiler, with their connection status.",
+            input_schema: { properties: {} },
+            handler: Tools::ListSlaves
           )
         ]
       end

data/lib/profiler/mcp/slave_support.rb

@@ -0,0 +1,23 @@
+# frozen_string_literal: true
+
+module Profiler
+  module MCP
+    module SlaveSupport
+      def self.resolve_storage(params)
+        if (slave_name = params["slave"])
+          require_relative "../cluster/slave_proxy"
+          Cluster::SlaveProxy.new(slave_name)
+        else
+          Profiler.storage
+        end
+      end
+
+      def self.with_slave_proxy(params)
+        return nil unless params["slave"]
+
+        require_relative "../cluster/slave_proxy"
+        Cluster::SlaveProxy.new(params["slave"])
+      end
+    end
+  end
+end

data/lib/profiler/mcp/tools/analyze_queries.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require_relative "../slave_support"
+
 module Profiler
   module MCP
     module Tools
@@ -15,10 +17,11 @@ module Profiler
             ]
           end
 
+          storage = MCP::SlaveSupport.resolve_storage(params)
           profile = if token == "latest"
-            Profiler.storage.list(limit: 1).first
+            storage.list(limit: 1).first
           else
-            Profiler.storage.load(token)
+            storage.load(token)
           end
           unless profile
             return [

data/lib/profiler/mcp/tools/clear_profiles.rb

@@ -1,5 +1,9 @@
 # frozen_string_literal: true
 
+require_relative "../slave_support"
+
+require "uri"
+
 module Profiler
   module MCP
     module Tools
@@ -11,7 +15,13 @@ module Profiler
             return [{ type: "text", text: "Error: type must be 'http', 'job', 'test', or 'console'" }]
           end
 
-          Profiler.storage.clear(type: type)
+          if (proxy = MCP::SlaveSupport.with_slave_proxy(params))
+            path = "/_profiler/api/profiles/clear"
+            path += "?type=#{URI.encode_www_form_component(type)}" if type
+            proxy.delete_json(path)
+          else
+            Profiler.storage.clear(type: type)
+          end
 
           label = type ? "#{type} profiles" : "all profiles"
           [{ type: "text", text: "Cleared #{label}." }]

data/lib/profiler/mcp/tools/delete_env_var.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require_relative "../slave_support"
+
 module Profiler
   module MCP
     module Tools
@@ -9,6 +11,11 @@ module Profiler
 
           return [{ type: "text", text: "Error: key cannot be blank." }] if key.empty?
 
+          if (proxy = MCP::SlaveSupport.with_slave_proxy(params))
+            proxy.patch_json("/_profiler/api/env_vars", { key: key, value: "" })
+            return [{ type: "text", text: "Deleted #{key} on slave '#{params["slave"]}'. Override persisted across restarts until reset." }]
+          end
+
           Profiler.env_override_store.delete(key)
           ENV.delete(key)
 

data/lib/profiler/mcp/tools/explain_query.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require_relative "../slave_support"
+
 require "profiler/explain_runner"
 
 module Profiler
@@ -18,6 +20,12 @@ module Profiler
             return [{ type: "text", text: "Error: query_index parameter is required" }]
           end
 
+          if (proxy = MCP::SlaveSupport.with_slave_proxy(params))
+            result = proxy.post_json("/_profiler/api/explain", { token: token, query_index: query_index.to_i })
+            text = result["error"] ? "Error: #{result["error"]}" : result.inspect
+            return [{ type: "text", text: text }]
+          end
+
           unless Profiler.configuration.enabled
             return [{ type: "text", text: "Error: EXPLAIN is only available when the profiler is enabled" }]
           end

data/lib/profiler/mcp/tools/get_profile_ajax.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require_relative "../slave_support"
+
 module Profiler
   module MCP
     module Tools
@@ -10,10 +12,11 @@ module Profiler
             return [{ type: "text", text: "Error: token parameter is required" }]
           end
 
+          storage = MCP::SlaveSupport.resolve_storage(params)
           profile = if token == "latest"
-            Profiler.storage.list(limit: 1).first
+            storage.list(limit: 1).first
           else
-            Profiler.storage.load(token)
+            storage.load(token)
           end
           unless profile
             return [{ type: "text", text: "Profile not found: #{token}" }]

data/lib/profiler/mcp/tools/get_profile_detail.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require_relative "../slave_support"
+
 require "shellwords"
 require "cgi"
 
@@ -18,10 +20,11 @@ module Profiler
             ]
           end
 
+          storage = MCP::SlaveSupport.resolve_storage(params)
           profile = if token == "latest"
-            Profiler.storage.list(limit: 1).first
+            storage.list(limit: 1).first
           else
-            Profiler.storage.load(token)
+            storage.load(token)
           end
           unless profile
             return [

data/lib/profiler/mcp/tools/get_profile_dumps.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require_relative "../slave_support"
+
 module Profiler
   module MCP
     module Tools
@@ -10,10 +12,11 @@ module Profiler
             return [{ type: "text", text: "Error: token parameter is required" }]
           end
 
+          storage = MCP::SlaveSupport.resolve_storage(params)
           profile = if token == "latest"
-            Profiler.storage.list(limit: 1).first
+            storage.list(limit: 1).first
           else
-            Profiler.storage.load(token)
+            storage.load(token)
           end
           unless profile
             return [{ type: "text", text: "Profile not found: #{token}" }]

data/lib/profiler/mcp/tools/get_profile_http.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require_relative "../slave_support"
+
 require "shellwords"
 require "uri"
 
@@ -13,10 +15,11 @@ module Profiler
             return [{ type: "text", text: "Error: token parameter is required" }]
           end
 
+          storage = MCP::SlaveSupport.resolve_storage(params)
           profile = if token == "latest"
-            Profiler.storage.list(limit: 1).first
+            storage.list(limit: 1).first
           else
-            Profiler.storage.load(token)
+            storage.load(token)
           end
           unless profile
             return [{ type: "text", text: "Profile not found: #{token}" }]

data/lib/profiler/mcp/tools/get_profile_mailers.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require_relative "../slave_support"
+
 module Profiler
   module MCP
     module Tools
@@ -12,10 +14,11 @@ module Profiler
             return [{ type: "text", text: "Error: token parameter is required" }]
           end
 
+          storage = MCP::SlaveSupport.resolve_storage(params)
           profile = if token == "latest"
-            Profiler.storage.list(limit: 1).first
+            storage.list(limit: 1).first
           else
-            Profiler.storage.load(token)
+            storage.load(token)
           end
           unless profile
             return [{ type: "text", text: "Profile not found: #{token}" }]

data/lib/profiler/mcp/tools/get_test_profile_detail.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require_relative "../slave_support"
+
 module Profiler
   module MCP
     module Tools
@@ -10,11 +12,12 @@ module Profiler
             return [{ type: "text", text: "Error: token parameter is required" }]
           end
 
+          storage = MCP::SlaveSupport.resolve_storage(params)
           profile = if token == "latest"
-            profiles = Profiler.storage.list(limit: 200)
+            profiles = storage.list(limit: 200)
             profiles.find { |p| p.profile_type == "test" }
           else
-            Profiler.storage.load(token)
+            storage.load(token)
           end
 
           unless profile && profile.profile_type == "test"

data/lib/profiler/mcp/tools/list_env_vars.rb

@@ -1,10 +1,19 @@
 # frozen_string_literal: true
 
+require_relative "../slave_support"
+
 module Profiler
   module MCP
     module Tools
       class ListEnvVars
         def self.call(params)
+          if (proxy = MCP::SlaveSupport.with_slave_proxy(params))
+            data = proxy.get_json("/_profiler/api/env_vars",
+              include_all: params["include_all"], filter: params["filter"])
+            text = format_slave_env_vars(data, params)
+            return [{ type: "text", text: text }]
+          end
+
           include_all = params["include_all"]
           filter = params["filter"]&.downcase
 
@@ -51,6 +60,35 @@ module Profiler
 
           [{ type: "text", text: text }]
         end
+
+        private
+
+        def self.format_slave_env_vars(data, params)
+          overrides = data["overrides"] || {}
+          filter = params["filter"]&.downcase
+
+          if params["include_all"]
+            vars = (data["variables"] || {}).sort.to_h
+            vars = vars.select { |k, _| k.downcase.include?(filter) } if filter
+            return "No environment variables match filter '#{params["filter"]}'." if vars.empty?
+
+            rows = vars.map do |key, value|
+              override = overrides[key]
+              overridden = override ? "✓ (was: #{override["original"] || "(unset)"})" : ""
+              "| #{key} | #{value} | #{overridden} |"
+            end
+            "**ENV variables (#{vars.size})**\n\n| Key | Current Value | Overridden |\n|-----|--------------|------------|\n" + rows.join("\n")
+          else
+            overrides = overrides.select { |k, _| k.downcase.include?(filter) } if filter
+            return "No overrides active." if overrides.empty?
+
+            rows = overrides.map do |key, entry|
+              current = entry["value"] == "__PROFILER_DELETED__" ? "(deleted)" : entry["value"]
+              "| #{key} | #{current} | #{entry["original"] || "(unset)"} |"
+            end
+            "**Active ENV overrides (#{overrides.size})**\n\n| Key | Current Value | Original Value |\n|-----|--------------|----------------|\n" + rows.join("\n")
+          end
+        end
       end
     end
   end

data/lib/profiler/mcp/tools/list_slaves.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+module Profiler
+  module MCP
+    module Tools
+      class ListSlaves
+        def self.call(_params)
+          slaves = Profiler.slave_registry.all
+
+          if slaves.empty?
+            return [{ type: "text", text: "No slave profilers connected." }]
+          end
+
+          lines = ["# Connected Slave Profilers\n"]
+          lines << "| Name | URL | Status | Registered At | Last Heartbeat |"
+          lines << "|------|-----|--------|--------------|----------------|"
+
+          slaves.each do |s|
+            lines << "| #{s[:name]} | #{s[:url]} | #{s[:status]} | #{s[:registered_at]} | #{s[:last_heartbeat_at]} |"
+          end
+
+          [{ type: "text", text: lines.join("\n") }]
+        end
+      end
+    end
+  end
+end

data/lib/profiler/mcp/tools/query_console_profiles.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require_relative "../slave_support"
+
 module Profiler
   module MCP
     module Tools
@@ -9,7 +11,8 @@ module Profiler
         def self.call(params)
           limit = params["limit"]&.to_i || 20
           fetch_size = [limit * 5, 500].min
-          profiles = Profiler.storage.list(limit: fetch_size)
+          storage = MCP::SlaveSupport.resolve_storage(params)
+          profiles = storage.list(limit: fetch_size)
 
           consoles = profiles.select { |p| p.profile_type == "console" }
 

data/lib/profiler/mcp/tools/query_jobs.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require_relative "../slave_support"
+
 module Profiler
   module MCP
     module Tools
@@ -9,7 +11,8 @@ module Profiler
         def self.call(params)
           limit = params["limit"]&.to_i || 20
           fetch_size = [limit * 5, 500].min
-          profiles = Profiler.storage.list(limit: fetch_size)
+          storage = MCP::SlaveSupport.resolve_storage(params)
+          profiles = storage.list(limit: fetch_size)
 
           jobs = profiles.select { |p| p.profile_type == "job" }
 

data/lib/profiler/mcp/tools/query_mailers.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require_relative "../slave_support"
+
 module Profiler
   module MCP
     module Tools
@@ -10,7 +12,8 @@ module Profiler
           def call(params)
             limit = params["limit"]&.to_i || 20
             fetch_size = [limit * 10, 1000].min
-            profiles = Profiler.storage.list(limit: fetch_size)
+            storage = MCP::SlaveSupport.resolve_storage(params)
+            profiles = storage.list(limit: fetch_size)
 
             emails = []
 

data/lib/profiler/mcp/tools/query_profiles.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require_relative "../slave_support"
+
 module Profiler
   module MCP
     module Tools
@@ -7,9 +9,10 @@ module Profiler
         ALL_FIELDS = %w[time type method path duration queries status token].freeze
 
         def self.call(params)
+          storage = MCP::SlaveSupport.resolve_storage(params)
           limit = params["limit"]&.to_i || 20
           fetch_size = [limit * 5, 500].min
-          profiles = Profiler.storage.list(limit: fetch_size)
+          profiles = storage.list(limit: fetch_size)
 
           profiles = profiles.select { |p| p.path&.include?(params["path"]) } if params["path"]
           profiles = profiles.select { |p| p.method == params["method"]&.upcase } if params["method"]

data/lib/profiler/mcp/tools/query_test_profiles.rb

@@ -1,5 +1,7 @@
 # frozen_string_literal: true
 
+require_relative "../slave_support"
+
 module Profiler
   module MCP
     module Tools
@@ -9,7 +11,8 @@ module Profiler
         def self.call(params)
           limit = params["limit"]&.to_i || 20
           fetch_size = [limit * 5, 500].min
-          profiles = Profiler.storage.list(limit: fetch_size)
+          storage = MCP::SlaveSupport.resolve_storage(params)
+          profiles = storage.list(limit: fetch_size)
 
           tests = profiles.select { |p| p.profile_type == "test" }