actionmcp 0.107.1 → 0.108.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: be05e5d56452910d6c725b15041b68b41b12830f38bd5661ba5c363403ef8df7
-  data.tar.gz: 4ead1672cb76e8c1ff4057062f6396cc68d290abf0ab7c0e0562f2d438ddd8a7
+  metadata.gz: 917d5120acc0a5cb82cb4a9516a65e59370f922dfe2828670e1117a5d1d26934
+  data.tar.gz: a77bd78234661ee4ef52cafff59f5d6c856e94728bca98832c926928635e8e94
 SHA512:
-  metadata.gz: f0d3ed2611e1347f24a1b6422c9d1628d85cbce23e9b9db13996aca9d9133d03e50bf6b45b26a0f4afa5795314b5285e70e545ecfd2e40ef71294bfc13241b03
-  data.tar.gz: fbdf20f97d3324d4adc6d9905ff45dc662ea9a039e6682e884a614fb02a14726b32c17c84785856188d9dea2babbe7aefd9d8d2d0c1c4171facaf69d1dcd02ce
+  metadata.gz: 6110ec397f08c3e8cbc2fe54a20f22bae3f600ad005e82f254bf6eeea7ba07c3bee1fc28a337a6263725fff213fdcbf092b08000285bdff0958b16fa5a958ccf
+  data.tar.gz: f57f192f165bf2a5d44d18189e6107eb0ab951eeaf4f88ddc92fa2e3208bcc4bfe7a7d6aed9cd0e01b2951d07e6e8b2826f75e6033054130b304707b0bec03b1

data/README.md

@@ -712,10 +712,12 @@ This will create:
 
 ## Authentication with Gateway
 
-ActionMCP provides a Gateway system for handling authentication. The Gateway allows you to authenticate users and make them available throughout your MCP components.
+ActionMCP provides a Gateway system for handling authentication. The Gateway allows you to authenticate users and make them available throughout your MCP components. For the full gateway reference including identifier classes, session persistence, profile switching, and production hardening tips, see **[GATEWAY.md](GATEWAY.md)**.
 
 ActionMCP uses a Gateway pattern with pluggable identifiers for authentication. You can implement custom authentication strategies using session-based auth, API keys, bearer tokens, or integrate with existing authentication systems like Warden, Devise, or external OAuth providers.
 
+> **Note:** Auth errors return HTTP 200 with a JSON-RPC error payload (not HTTP 401). This is correct per the MCP specification — all MCP communication uses JSON-RPC over HTTP, and protocol-level errors are expressed within the JSON-RPC envelope. The `initialize` request bypasses authentication per MCP spec.
+
 ### Creating an ApplicationGateway
 
 When you run the install generator, it creates an `ApplicationGateway` class:
@@ -986,26 +988,68 @@ class ToolTest < ActiveSupport::TestCase
   include ActionMCP::TestHelper
 
   test "CalculateSumTool returns the correct sum" do
-    assert_tool_findable("calculate_sum")
-    result = execute_tool("calculate_sum", a: 5, b: 10)
-    assert_tool_output(result, "15.0")
+    assert_mcp_tool_findable("calculate_sum")
+    result = execute_mcp_tool("calculate_sum", a: 5, b: 10)
+    assert_mcp_tool_output("15.0", result)
   end
 
   test "AnalyzeCodePrompt returns the correct analysis" do
-    assert_prompt_findable("analyze_code")
-    result = execute_prompt("analyze_code", language: "Ruby", code: "def hello; puts 'Hello, world!'; end")
-    assert_equal "Analyzing Ruby code: def hello; puts 'Hello, world!'; end", assert_prompt_output(result)
+    assert_mcp_prompt_findable("analyze_code")
+    result = execute_mcp_prompt("analyze_code", language: "Ruby", code: "def hello; puts 'Hello, world!'; end")
+    assert_mcp_prompt_output("Analyzing Ruby code: def hello; puts 'Hello, world!'; end", result)
   end
 end
 ```
 
-The TestHelper provides several assertion methods:
-- `assert_tool_findable(name)` - Verifies a tool exists and is registered
-- `assert_prompt_findable(name)` - Verifies a prompt exists and is registered
-- `execute_tool(name, **args)` - Executes a tool with arguments
-- `execute_prompt(name, **args)` - Executes a prompt with arguments
-- `assert_tool_output(result, expected)` - Asserts tool output matches expected text
-- `assert_prompt_output(result)` - Extracts and returns prompt output for assertions
+The TestHelper provides several assertion and execution methods:
+
+**Tools:**
+- `assert_mcp_tool_findable(name)` - Verifies a tool exists and is registered
+- `execute_mcp_tool(name, **args)` - Executes a tool with arguments and asserts success
+- `execute_mcp_tool_with_error(name, **args)` - Executes a tool without asserting success (for testing error cases)
+- `assert_mcp_tool_output(expected, response)` - Asserts tool output matches expected content
+
+**Prompts:**
+- `assert_mcp_prompt_findable(name)` - Verifies a prompt exists and is registered
+- `execute_mcp_prompt(name, **args)` - Executes a prompt with arguments
+- `assert_mcp_prompt_output(expected, response)` - Asserts prompt output matches expected content
+
+**Resource Templates:**
+- `assert_mcp_resource_template_findable(name)` - Verifies a resource template exists and is registered
+- `resolve_mcp_resource(uri)` - Resolves a resource URI and asserts success
+- `resolve_mcp_resource_with_error(uri)` - Resolves a resource URI without asserting success (for testing error cases)
+
+**General:**
+- `assert_mcp_error_code(code, response)` - Asserts a specific JSON-RPC error code
+
+### Testing Resource Templates
+
+```ruby
+require "test_helper"
+require "action_mcp/test_helper"
+
+class ProductResourceTest < ActiveSupport::TestCase
+  include ActionMCP::TestHelper
+
+  test "product template is registered" do
+    assert_mcp_resource_template_findable("products")
+  end
+
+  test "resolves a product resource by URI" do
+    resp = resolve_mcp_resource("ecommerce://products/1")
+
+    assert resp.success?
+    assert_not_empty resp.contents
+    assert_equal "application/json", resp.contents.first.mime_type
+  end
+
+  test "returns error for nonexistent product" do
+    resp = resolve_mcp_resource_with_error("ecommerce://products/0")
+
+    assert resp.is_error
+  end
+end
+```
 
 ## Inspecting Your MCP Server
 

data/app/controllers/action_mcp/application_controller.rb

@@ -70,6 +70,11 @@ module ActionMCP
         end
       end
 
+      if session.nil?
+        id = jsonrpc_params.respond_to?(:id) ? jsonrpc_params.id : nil
+        return render_not_found("Session not found.", id)
+      end
+
       if session.new_record?
         session.save!
         response.headers[MCP_SESSION_ID_HEADER] = session.id

data/app/models/action_mcp/session.rb

@@ -172,6 +172,7 @@ module ActionMCP
     def register_tool(tool_class_or_name)
       tool_name = normalize_name(tool_class_or_name, :tool)
       return false unless tool_exists?(tool_name)
+      return true if uses_all_tools?
 
       self.tool_registry ||= []
       unless self.tool_registry.include?(tool_name)
@@ -184,8 +185,13 @@ module ActionMCP
 
     def unregister_tool(tool_class_or_name)
       tool_name = normalize_name(tool_class_or_name, :tool)
-      self.tool_registry ||= []
 
+      if uses_all_tools?
+        return unless tool_exists?(tool_name)
+        expand_tool_registry!
+      end
+
+      self.tool_registry ||= []
       return unless self.tool_registry.delete(tool_name)
 
       save!
@@ -195,6 +201,7 @@ module ActionMCP
     def register_prompt(prompt_class_or_name)
       prompt_name = normalize_name(prompt_class_or_name, :prompt)
       return false unless prompt_exists?(prompt_name)
+      return true if uses_all_prompts?
 
       self.prompt_registry ||= []
       unless self.prompt_registry.include?(prompt_name)
@@ -207,8 +214,13 @@ module ActionMCP
 
     def unregister_prompt(prompt_class_or_name)
       prompt_name = normalize_name(prompt_class_or_name, :prompt)
-      self.prompt_registry ||= []
 
+      if uses_all_prompts?
+        return unless prompt_exists?(prompt_name)
+        expand_prompt_registry!
+      end
+
+      self.prompt_registry ||= []
       return unless self.prompt_registry.delete(prompt_name)
 
       save!
@@ -218,6 +230,7 @@ module ActionMCP
     def register_resource_template(template_class_or_name)
       template_name = normalize_name(template_class_or_name, :resource_template)
       return false unless resource_template_exists?(template_name)
+      return true if uses_all_resources?
 
       self.resource_registry ||= []
       unless self.resource_registry.include?(template_name)
@@ -230,8 +243,13 @@ module ActionMCP
 
     def unregister_resource_template(template_class_or_name)
       template_name = normalize_name(template_class_or_name, :resource_template)
-      self.resource_registry ||= []
 
+      if uses_all_resources?
+        return unless resource_template_exists?(template_name)
+        expand_resource_registry!
+      end
+
+      self.resource_registry ||= []
       return unless self.resource_registry.delete(template_name)
 
       save!
@@ -348,6 +366,20 @@ module ActionMCP
       self.resource_registry = [ "*" ]
     end
 
+    # Expand wildcard registries to explicit name lists so individual
+    # entries can be removed without breaking the wildcard check.
+    def expand_tool_registry!
+      self.tool_registry = ActionMCP.configuration.filtered_tools.map(&:name)
+    end
+
+    def expand_prompt_registry!
+      self.prompt_registry = ActionMCP.configuration.filtered_prompts.map(&:name)
+    end
+
+    def expand_resource_registry!
+      self.resource_registry = ActionMCP.configuration.filtered_resources.map(&:name)
+    end
+
     def normalize_name(class_or_name, type)
       case class_or_name
       when String

data/db/migrate/20250512154359_consolidated_migration.rb

@@ -1,113 +1,102 @@
 # frozen_string_literal: true
 
-class ConsolidatedMigration < ActiveRecord::Migration[8.0]
-  def change
-    # Only create tables if they don't exist to avoid deleting existing data
-
-    # Create sessions table
-    unless table_exists?(:action_mcp_sessions)
-      create_table :action_mcp_sessions, id: :string do |t|
-        t.string :role, null: false, default: 'server', comment: 'The role of the session'
-        t.string :status, null: false, default: 'pre_initialize'
-        t.datetime :ended_at, comment: 'The time the session ended'
-        t.string :protocol_version
-        t.json :server_capabilities, comment: 'The capabilities of the server'
-        t.json :client_capabilities, comment: 'The capabilities of the client'
-        t.json :server_info, comment: 'The information about the server'
-        t.json :client_info, comment: 'The information about the client'
-        t.boolean :initialized, null: false, default: false
-        t.integer :messages_count, null: false, default: 0
-        t.json :tool_registry, default: []
-        t.json :prompt_registry, default: []
-        t.json :resource_registry, default: []
-        t.timestamps
-      end
-    end
-
-    # Create session messages table
-    unless table_exists?(:action_mcp_session_messages)
-      create_table :action_mcp_session_messages do |t|
-        t.references :session, null: false,
-                               foreign_key: { to_table: :action_mcp_sessions,
-                                              on_delete: :cascade,
-                                              on_update: :cascade,
-                                              name: 'fk_action_mcp_session_messages_session_id' }, type: :string
-        t.string :direction, null: false, comment: 'The message recipient', default: 'client'
-        t.string :message_type, null: false, comment: 'The type of the message'
-        t.string :jsonrpc_id
-        t.json :message_json
-        t.boolean :is_ping, default: false, null: false, comment: 'Whether the message is a ping'
-        t.boolean :request_acknowledged, default: false, null: false
-        t.boolean :request_cancelled, null: false, default: false
-        t.timestamps
-      end
-    end
-
-    # Create session subscriptions table
-    unless table_exists?(:action_mcp_session_subscriptions)
-      create_table :action_mcp_session_subscriptions do |t|
-        t.references :session,
-                     null: false,
-                     foreign_key: { to_table: :action_mcp_sessions, on_delete: :cascade },
-                     type: :string
-        t.string :uri, null: false
-        t.datetime :last_notification_at
-        t.timestamps
-      end
-    end
-
-    # Add missing columns to existing tables if they exist
-
-    # For action_mcp_sessions
-    if table_exists?(:action_mcp_sessions)
-      unless column_exists?(:action_mcp_sessions, :tool_registry)
-        add_column :action_mcp_sessions, :tool_registry, :json, default: []
-      end
-
-      unless column_exists?(:action_mcp_sessions, :prompt_registry)
-        add_column :action_mcp_sessions, :prompt_registry, :json, default: []
-      end
-
-      unless column_exists?(:action_mcp_sessions, :resource_registry)
-        add_column :action_mcp_sessions, :resource_registry, :json, default: []
-      end
+class ConsolidatedMigration < ActiveRecord::Migration[8.1]
+  def up
+    create_table :action_mcp_sessions, id: :string, if_not_exists: true do |t|
+      t.string :role, null: false, default: "server", comment: "The role of the session"
+      t.string :status, null: false, default: "pre_initialize"
+      t.datetime :ended_at, comment: "The time the session ended"
+      t.string :protocol_version
+      t.json :server_capabilities, comment: "The capabilities of the server"
+      t.json :client_capabilities, comment: "The capabilities of the client"
+      t.json :server_info, comment: "The information about the server"
+      t.json :client_info, comment: "The information about the client"
+      t.boolean :initialized, null: false, default: false
+      t.integer :messages_count, null: false, default: 0
+      t.json :tool_registry, default: []
+      t.json :prompt_registry, default: []
+      t.json :resource_registry, default: []
+      t.json :consents, default: {}, null: false
+      t.json :session_data, default: {}, null: false
+      t.timestamps
     end
 
-    # For action_mcp_session_messages
-    return unless table_exists?(:action_mcp_session_messages)
-
-    unless column_exists?(:action_mcp_session_messages, :is_ping)
-      add_column :action_mcp_session_messages, :is_ping, :boolean, default: false, null: false,
-                                                                   comment: 'Whether the message is a ping'
-    end
-
-    unless column_exists?(:action_mcp_session_messages, :request_acknowledged)
-      add_column :action_mcp_session_messages, :request_acknowledged, :boolean, default: false, null: false
+    add_column :action_mcp_sessions, :consents, :json, default: {}, null: false unless column_exists?(:action_mcp_sessions, :consents)
+    add_column :action_mcp_sessions, :session_data, :json, default: {}, null: false unless column_exists?(:action_mcp_sessions, :session_data)
+
+    create_table :action_mcp_session_messages, if_not_exists: true do |t|
+      t.references :session, null: false,
+                             foreign_key: { to_table: :action_mcp_sessions,
+                                            on_delete: :cascade,
+                                            on_update: :cascade,
+                                            name: "fk_action_mcp_session_messages_session_id" },
+                             type: :string
+      t.string :direction, null: false, comment: "The message recipient", default: "client"
+      t.string :message_type, null: false, comment: "The type of the message"
+      t.string :jsonrpc_id
+      t.json :message_json
+      t.boolean :is_ping, default: false, null: false, comment: "Whether the message is a ping"
+      t.boolean :request_acknowledged, default: false, null: false
+      t.boolean :request_cancelled, null: false, default: false
+      t.timestamps
     end
 
-    unless column_exists?(:action_mcp_session_messages, :request_cancelled)
-      add_column :action_mcp_session_messages, :request_cancelled, :boolean, null: false, default: false
+    add_column :action_mcp_session_messages, :is_ping, :boolean, default: false, null: false unless column_exists?(:action_mcp_session_messages, :is_ping)
+    add_column :action_mcp_session_messages, :request_acknowledged, :boolean, default: false, null: false unless column_exists?(:action_mcp_session_messages, :request_acknowledged)
+    add_column :action_mcp_session_messages, :request_cancelled, :boolean, default: false, null: false unless column_exists?(:action_mcp_session_messages, :request_cancelled)
+
+    create_table :action_mcp_session_subscriptions, if_not_exists: true do |t|
+      t.references :session, null: false,
+                             foreign_key: { to_table: :action_mcp_sessions, on_delete: :cascade },
+                             type: :string
+      t.string :uri, null: false
+      t.datetime :last_notification_at
+      t.timestamps
     end
 
-    if column_exists?(:action_mcp_session_messages, :message_text)
-      remove_column :action_mcp_session_messages, :message_text
+    create_table :action_mcp_session_tasks, id: :string, if_not_exists: true do |t|
+      t.references :session, null: false,
+                             foreign_key: { to_table: :action_mcp_sessions,
+                                            on_delete: :cascade,
+                                            on_update: :cascade,
+                                            name: "fk_action_mcp_session_tasks_session_id" },
+                             type: :string
+      t.string :status, null: false, default: "working"
+      t.string :status_message
+      t.string :request_method, comment: "e.g., tools/call, prompts/get"
+      t.string :request_name, comment: "e.g., tool name, prompt name"
+      t.json :request_params, comment: "Original request params"
+      t.json :result_payload, comment: "Final result data"
+      t.integer :ttl, comment: "Time to live in milliseconds"
+      t.integer :poll_interval, comment: "Suggested polling interval in milliseconds"
+      t.datetime :last_updated_at, null: false
+      t.json :continuation_state, default: {}
+      t.integer :progress_percent, comment: "Task progress as percentage 0-100"
+      t.string :progress_message, comment: "Human-readable progress message"
+      t.datetime :last_step_at
+      t.timestamps
+
+      t.index :status
+      t.index %i[session_id status]
+      t.index :created_at
     end
 
-    return unless column_exists?(:action_mcp_session_messages, :direction)
-
-    # SQLite3 doesn't support changing column comments
-    return unless connection.adapter_name.downcase != 'sqlite'
-
-    change_column_comment :action_mcp_session_messages, :direction, 'The message recipient'
-  end
-
-  private
-
-  def table_exists?(table_name)
-    ActionMCP::ApplicationRecord.connection.table_exists?(table_name)
-  end
-
-  def column_exists?(table_name, column_name)
-    ActionMCP::ApplicationRecord.connection.column_exists?(table_name, column_name)
+    add_column :action_mcp_session_tasks, :continuation_state, :json, default: {} unless column_exists?(:action_mcp_session_tasks, :continuation_state)
+    add_column :action_mcp_session_tasks, :progress_percent, :integer unless column_exists?(:action_mcp_session_tasks, :progress_percent)
+    add_column :action_mcp_session_tasks, :progress_message, :string unless column_exists?(:action_mcp_session_tasks, :progress_message)
+    add_column :action_mcp_session_tasks, :last_step_at, :datetime unless column_exists?(:action_mcp_session_tasks, :last_step_at)
+
+    # Remove deprecated tables if they exist
+    drop_table :action_mcp_oauth_clients, if_exists: true
+    drop_table :action_mcp_oauth_tokens, if_exists: true
+    drop_table :action_mcp_sse_events, if_exists: true
+    drop_table :action_mcp_session_resources, if_exists: true
+
+    # Remove deprecated columns if they exist
+    remove_column :action_mcp_sessions, :oauth_access_token if column_exists?(:action_mcp_sessions, :oauth_access_token)
+    remove_column :action_mcp_sessions, :oauth_refresh_token if column_exists?(:action_mcp_sessions, :oauth_refresh_token)
+    remove_column :action_mcp_sessions, :oauth_token_expires_at if column_exists?(:action_mcp_sessions, :oauth_token_expires_at)
+    remove_column :action_mcp_sessions, :oauth_user_context if column_exists?(:action_mcp_sessions, :oauth_user_context)
+    remove_column :action_mcp_sessions, :sse_event_counter if column_exists?(:action_mcp_sessions, :sse_event_counter)
   end
 end

data/lib/action_mcp/test_helper.rb

@@ -14,8 +14,12 @@ module ActionMCP
   #
   # and you get   assert_mcp_tool_findable,
   #               assert_mcp_prompt_findable,
+  #               assert_mcp_resource_template_findable,
   #               execute_mcp_tool,
+  #               execute_mcp_tool_with_error,
   #               execute_mcp_prompt,
+  #               resolve_mcp_resource,
+  #               resolve_mcp_resource_with_error,
   #               assert_mcp_error_code,
   #               assert_mcp_tool_output,
   #               assert_mcp_prompt_output.
@@ -29,6 +33,12 @@ module ActionMCP
     include ProgressNotificationAssertions
 
     # ──── Registry assertions ────────────────────────────────────────────────
+    def assert_mcp_resource_template_findable(name, msg = nil)
+      assert ActionMCP::ResourceTemplatesRegistry.resource_templates.key?(name),
+             msg || "Resource template #{name.inspect} not found in ResourceTemplatesRegistry"
+    end
+    alias assert_resource_template_findable assert_mcp_resource_template_findable
+
     def assert_mcp_tool_findable(name, msg = nil)
       assert ActionMCP::ToolsRegistry.tools.key?(name),
              msg || "Tool #{name.inspect} not found in ToolsRegistry"
@@ -61,6 +71,31 @@ module ActionMCP
     end
     alias execute_prompt execute_mcp_prompt
 
+    def resolve_mcp_resource(uri)
+      template_class = ActionMCP::ResourceTemplatesRegistry.find_template_for_uri(uri)
+      assert template_class, "No resource template found matching URI #{uri.inspect}"
+      template = template_class.process(uri)
+      assert template, "Failed to process URI #{uri.inspect} with template #{template_class.name}"
+      resp = template.call
+      assert !resp.is_error, "Resource #{uri.inspect} returned error: #{resp.to_h[:message]}"
+      resp
+    end
+    alias resolve_resource resolve_mcp_resource
+
+    def resolve_mcp_resource_with_error(uri)
+      template_class = ActionMCP::ResourceTemplatesRegistry.find_template_for_uri(uri)
+      unless template_class
+        return ActionMCP::ResourceResponse.new.tap { |r| r.mark_as_not_found!(uri) }
+      end
+
+      unless template_class.respond_to?(:readable_uri?) && template_class.readable_uri?(uri)
+        return ActionMCP::ResourceResponse.new.tap { |r| r.mark_as_not_found!(uri) }
+      end
+      template = template_class.process(uri)
+      template.call
+    end
+    alias resolve_resource_with_error resolve_mcp_resource_with_error
+
     # ──── Negative‑path helper ───────────────────────────────────────────────
     def assert_mcp_error_code(code, response, msg = nil)
       assert response.error?, msg || "Expected response to be an error"

data/lib/action_mcp/version.rb

@@ -2,7 +2,7 @@
 
 require_relative "gem_version"
 module ActionMCP
-  VERSION = "0.107.1"
+  VERSION = "0.108.0"
 
   class << self
     alias version gem_version

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: actionmcp
 version: !ruby/object:Gem::Version
-  version: 0.107.1
+  version: 0.108.0
 platform: ruby
 authors:
 - Abdelkader Boudih
@@ -160,14 +160,6 @@ files:
 - app/models/concerns/action_mcp/mcp_message_inspect.rb
 - config/routes.rb
 - db/migrate/20250512154359_consolidated_migration.rb
-- db/migrate/20250715070713_add_consents_to_action_mcp_sess.rb
-- db/migrate/20250727000001_remove_oauth_support.rb
-- db/migrate/20251125000001_create_action_mcp_session_tasks.rb
-- db/migrate/20251126000001_add_continuation_state_to_action_mcp_session_tasks.rb
-- db/migrate/20251203000001_remove_sse_support.rb
-- db/migrate/20251204000001_add_progress_to_session_tasks.rb
-- db/migrate/20260303000001_add_session_data_to_action_mcp_sessions.rb
-- db/migrate/20260304000001_remove_session_resources.rb
 - db/test.sqlite3
 - exe/actionmcp_cli
 - lib/action_mcp.rb
@@ -303,7 +295,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 4.0.3
+rubygems_version: 4.0.6
 specification_version: 4
 summary: Lightweight Model Context Protocol (MCP) server toolkit for Ruby/Rails
 test_files: []

data/db/migrate/20250715070713_add_consents_to_action_mcp_sess.rb

@@ -1,9 +0,0 @@
-# frozen_string_literal: true
-
-class AddConsentsToActionMCPSess < ActiveRecord::Migration[8.0]
-  def change
-    return if column_exists?(:action_mcp_sessions, :consents)
-
-    add_column :action_mcp_sessions, :consents, :json, default: {}, null: false
-  end
-end

data/db/migrate/20250727000001_remove_oauth_support.rb

@@ -1,59 +0,0 @@
-# frozen_string_literal: true
-
-class RemoveOauthSupport < ActiveRecord::Migration[8.0]
-  def change
-    # Remove OAuth tables
-    drop_table :action_mcp_oauth_clients, if_exists: true do |t|
-      t.string :client_id, null: false
-      t.string :client_secret
-      t.string :client_name
-      t.json :redirect_uris, default: []
-      t.json :grant_types, default: [ "authorization_code" ]
-      t.json :response_types, default: [ "code" ]
-      t.string :token_endpoint_auth_method, default: "client_secret_basic"
-      t.text :scope
-      t.boolean :active, default: true
-      t.integer :client_id_issued_at
-      t.integer :client_secret_expires_at
-      t.string :registration_access_token
-      t.json :metadata, default: {}
-      t.datetime :created_at, null: false
-      t.datetime :updated_at, null: false
-    end
-
-    drop_table :action_mcp_oauth_tokens, if_exists: true do |t|
-      t.string :token, null: false
-      t.string :token_type, null: false
-      t.string :client_id, null: false
-      t.string :user_id
-      t.text :scope
-      t.datetime :expires_at
-      t.boolean :revoked, default: false
-      t.string :redirect_uri
-      t.string :code_challenge
-      t.string :code_challenge_method
-      t.string :access_token
-      t.json :metadata, default: {}
-      t.datetime :created_at, null: false
-      t.datetime :updated_at, null: false
-    end
-
-    # Remove OAuth columns from sessions table
-    if table_exists?(:action_mcp_sessions)
-      remove_column :action_mcp_sessions, :oauth_access_token, :string if column_exists?(:action_mcp_sessions, :oauth_access_token)
-      remove_column :action_mcp_sessions, :oauth_refresh_token, :string if column_exists?(:action_mcp_sessions, :oauth_refresh_token)
-      remove_column :action_mcp_sessions, :oauth_token_expires_at, :datetime if column_exists?(:action_mcp_sessions, :oauth_token_expires_at)
-      remove_column :action_mcp_sessions, :oauth_user_context, :json if column_exists?(:action_mcp_sessions, :oauth_user_context)
-    end
-  end
-
-  private
-
-  def table_exists?(table_name)
-    ActiveRecord::Base.connection.table_exists?(table_name)
-  end
-
-  def column_exists?(table_name, column_name)
-    ActiveRecord::Base.connection.column_exists?(table_name, column_name)
-  end
-end

data/db/migrate/20251125000001_create_action_mcp_session_tasks.rb

@@ -1,29 +0,0 @@
-# frozen_string_literal: true
-
-class CreateActionMCPSessionTasks < ActiveRecord::Migration[8.0]
-  def change
-    create_table :action_mcp_session_tasks, id: :string do |t|
-      t.references :session, null: false,
-                             foreign_key: { to_table: :action_mcp_sessions,
-                                            on_delete: :cascade,
-                                            on_update: :cascade,
-                                            name: 'fk_action_mcp_session_tasks_session_id' },
-                             type: :string
-      t.string :status, null: false, default: 'working'
-      t.string :status_message
-      t.string :request_method, comment: 'e.g., tools/call, prompts/get'
-      t.string :request_name, comment: 'e.g., tool name, prompt name'
-      t.json :request_params, comment: 'Original request params'
-      t.json :result_payload, comment: 'Final result data'
-      t.integer :ttl, comment: 'Time to live in milliseconds'
-      t.integer :poll_interval, comment: 'Suggested polling interval in milliseconds'
-      t.datetime :last_updated_at, null: false
-
-      t.timestamps
-
-      t.index :status
-      t.index %i[session_id status]
-      t.index :created_at
-    end
-  end
-end

data/db/migrate/20251126000001_add_continuation_state_to_action_mcp_session_tasks.rb

@@ -1,10 +0,0 @@
-# frozen_string_literal: true
-
-class AddContinuationStateToActionMCPSessionTasks < ActiveRecord::Migration[8.1]
-  def change
-    add_column :action_mcp_session_tasks, :continuation_state, :json, default: {}
-    add_column :action_mcp_session_tasks, :progress_percent, :integer
-    add_column :action_mcp_session_tasks, :progress_message, :string
-    add_column :action_mcp_session_tasks, :last_step_at, :datetime
-  end
-end

data/db/migrate/20251203000001_remove_sse_support.rb

@@ -1,31 +0,0 @@
-# frozen_string_literal: true
-
-class RemoveSseSupport < ActiveRecord::Migration[8.1]
-  def up
-    # Drop SSE events table
-    drop_table :action_mcp_sse_events, if_exists: true
-
-    # Remove SSE counter from sessions
-    remove_column :action_mcp_sessions, :sse_event_counter, if_exists: true
-  end
-
-  def down
-    # Re-add sse_event_counter to sessions
-    unless column_exists?(:action_mcp_sessions, :sse_event_counter)
-      add_column :action_mcp_sessions, :sse_event_counter, :integer, default: 0, null: false
-    end
-
-    # Re-create SSE events table
-    unless table_exists?(:action_mcp_sse_events)
-      create_table :action_mcp_sse_events do |t|
-        t.references :session, null: false, type: :string, foreign_key: { to_table: :action_mcp_sessions }
-        t.integer :event_id, null: false
-        t.text :data, null: false
-        t.timestamps
-      end
-
-      add_index :action_mcp_sse_events, :created_at
-      add_index :action_mcp_sse_events, [ :session_id, :event_id ], unique: true
-    end
-  end
-end

data/db/migrate/20251204000001_add_progress_to_session_tasks.rb

@@ -1,12 +0,0 @@
-# frozen_string_literal: true
-
-class AddProgressToSessionTasks < ActiveRecord::Migration[8.1]
-  def change
-    unless column_exists?(:action_mcp_session_tasks, :progress_percent)
-      add_column :action_mcp_session_tasks, :progress_percent, :integer, comment: "Task progress as percentage 0-100"
-    end
-    unless column_exists?(:action_mcp_session_tasks, :progress_message)
-      add_column :action_mcp_session_tasks, :progress_message, :string, comment: "Human-readable progress message"
-    end
-  end
-end

data/db/migrate/20260303000001_add_session_data_to_action_mcp_sessions.rb

@@ -1,9 +0,0 @@
-# frozen_string_literal: true
-
-class AddSessionDataToActionMCPSessions < ActiveRecord::Migration[8.1]
-  def change
-    return if column_exists?(:action_mcp_sessions, :session_data)
-
-    add_column :action_mcp_sessions, :session_data, :json, default: {}, null: false
-  end
-end

data/db/migrate/20260304000001_remove_session_resources.rb

@@ -1,26 +0,0 @@
-# frozen_string_literal: true
-
-class RemoveSessionResources < ActiveRecord::Migration[8.1]
-  def up
-    drop_table :action_mcp_session_resources, if_exists: true
-  end
-
-  def down
-    return if table_exists?(:action_mcp_session_resources)
-
-    create_table :action_mcp_session_resources do |t|
-      t.references :session,
-                   null: false,
-                   foreign_key: { to_table: :action_mcp_sessions, on_delete: :cascade },
-                   type: :string
-      t.string :uri, null: false
-      t.string :name
-      t.text :description
-      t.string :mime_type, null: false
-      t.boolean :created_by_tool, default: false
-      t.datetime :last_accessed_at
-      t.json :metadata
-      t.timestamps
-    end
-  end
-end