actionmcp 0.7.2 → 0.8.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bea2e23118051ba1a05eb00dfb44395841068fe6358b341e01a1951f7064a259
-  data.tar.gz: e1ff02d641a4684c0369fa252139edeb8bf1dae41193deeca122832bcb0ac318
+  metadata.gz: 7a031f1969028d88d34f7f19bc18777fee99067dee692ce73754bbb001477f14
+  data.tar.gz: 35c84b344020e97ce87d9c7e033220138732c29bd7d7cee0634e4f9c7f38cc13
 SHA512:
-  metadata.gz: '097b59b77ef2028887ef159f211a314a9a9fa83edb7ff26f50d62fc56a834bf5df96e7781c5cd715d511ee957736f242971f76baac3d8b1fffd213f6373afe5c'
-  data.tar.gz: 13582b839164819da648df7e319df9e8aac565f2792bee75c44d23d1fd7af531e01d6c7a8e82f63d46fc9cd68f7d42fe10622748202a6ca10dc99ff332d8d7a5
+  metadata.gz: 953dd645cb4006c027ca2643aa8274a5011db73910159f4a0a3d5c60e80df82deae14f37ec41de869e863e7b4a8e1440579863bc33daa21185dc169abc4ddcbe
+  data.tar.gz: 22f1e9622b4330afb13bc1fd943142904f8fcbd07df734e571707c428b2d11c343775900221a33a74df354cae4a47fa36bd13e97d3809aa930b4ad2af40b66b5

data/app/controllers/action_mcp/messages_controller.rb

@@ -21,6 +21,9 @@ module ActionMCP
     end
 
     def handle_post_message(params, response)
+      if params[:method] == "initialize"
+        mcp_session.initialize!
+      end
       json_rpc_handler.call(params)
 
       response.status = :accepted
@@ -31,7 +34,7 @@ module ActionMCP
     end
 
     def mcp_session
-      Session.find(params[:session_id])
+      @mcp_session ||= Session.find_or_create_by(id: params[:session_id])
     end
 
     def clean_params

data/app/controllers/action_mcp/sse_controller.rb

@@ -20,12 +20,21 @@ module ActionMCP
 
         # Start listener and process messages via the transport
         listener = SSEListener.new(mcp_session)
+        message_received = false
         if listener.start do |message|
           # Send with proper SSE formatting
           sse.write(message)
+          message_received = true
+        end
+        sleep 1
+        # Heartbeat loop
+        unless message_received
+          Rails.logger.warn "No message received within 1 second, closing connection for session: #{session_id}"
+          error = JsonRpc::Response.new(id: SecureRandom.uuid_v7, error: JsonRpc::JsonRpcError.new(:server_error, message: "No message received within 1 second").to_h).to_h
+          sse.write(error)
+          return
         end
 
-          # Heartbeat loop
           until response.stream.closed?
             sleep HEARTBEAT_INTERVAL
             # mcp_session.send_ping!
@@ -39,8 +48,8 @@ module ActionMCP
       rescue => e
         Rails.logger.error "SSE: Unexpected error: #{e.class} - #{e.message}\n#{e.backtrace.join("\n")}"
       ensure
-        listener.stop
         response.stream.close
+        listener.stop
         Rails.logger.debug "SSE: Connection closed for session: #{session_id}"
       end
     end
@@ -59,12 +68,16 @@ module ActionMCP
     end
 
     def mcp_session
-      @mcp_session ||= Session.create
+      @mcp_session ||= Session.new
     end
 
     def session_id
       @session_id ||= mcp_session.id
     end
+
+    def cache_key
+      "action_mcp:session:#{session_id}"
+    end
   end
 
   class SSEListener
@@ -100,7 +113,7 @@ module ActionMCP
       }
 
       # Subscribe using the ActionCable adapter
-      success = adapter.subscribe(session_key, message_callback, success_callback)
+      adapter.subscribe(session_key, message_callback, success_callback)
 
       # Give some time for the subscription to be established
       sleep 0.5
@@ -110,8 +123,9 @@ module ActionMCP
 
     def stop
       @stopped = true
-
-      @session.close!
+      if (mcp_session = Session.find_by(id: session_key))
+        mcp_session.close
+      end
       Rails.logger.debug "Unsubscribed from: #{session_key}"
     end
   end

data/lib/action_mcp/version.rb

@@ -2,7 +2,7 @@
 
 require_relative "gem_version"
 module ActionMCP
-  VERSION = "0.7.2"
+  VERSION = "0.8.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.7.2
+  version: 0.8.0
 platform: ruby
 authors:
 - Abdelkader Boudih