ruby_llm-mcp 0.6.2 → 0.6.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f6b3a7fc52258ae0e1a8e5635b7c7347165b2c78da97ba1cfa1bc3633390eb39
-  data.tar.gz: d2220a770be8038c8fa794196742f8c35d390e4020b838a9d7279c449b5894b0
+  metadata.gz: fa780ea2ca1b1ba71c5efa1784b14e9473ad351af94f0dc35d94aab4d6057325
+  data.tar.gz: 1fd76e02e0f6e22087943eff186dbb05cd059e931927442982752cbb9250aa98
 SHA512:
-  metadata.gz: 4d6d145ab20653923ce4a557f6cc6d44b10f395a7c54dde228d69fe9f7e175e6d56ae6b6382ceae6dd32878305757271843e09a81ea07aaca2a1741a0f8f29a3
-  data.tar.gz: 50872333e5206d74d30dd612d8e09db65665957cca705e6f20cf3f72460896332e0c5ea271ab2fa88659421dc4d1a07b0fe217ed9b925d237993b59baf4f94a2
+  metadata.gz: 5c37a73381c487a02d61bb8365e39b064b37a0d0668bd46eae1bd25ad225982c8f15229867cd4110e89689ec5757e395ce3825795298aabc3f4ed3b8cd49e0e5
+  data.tar.gz: ecb7f5524d9af14c3386abb3ddae91bee454046449ae2e0bba758c95060157c057222a57ce1b9eae5af197e5ec8187e34474fb3801b7cdabb92fdb3cbe14ddd2

data/lib/ruby_llm/mcp/transports/sse.rb

@@ -30,7 +30,6 @@ module RubyLLM
                                      "Accept" => "text/event-stream",
                                      "Content-Type" => "application/json",
                                      "Cache-Control" => "no-cache",
-                                     "Connection" => "keep-alive",
                                      "X-CLIENT-ID" => @client_id
                                    })
 
@@ -167,41 +166,82 @@ module RubyLLM
         end
 
         def stream_events_from_server
-          sse_client = HTTPX.plugin(:stream)
-          sse_client = sse_client.with(
-            headers: @headers
-          )
+          sse_client = create_sse_client
+          response = sse_client.get(@event_url, stream: true)
+          validate_sse_response!(response)
+          process_event_stream(response)
+        end
 
-          if @version == :http1
-            sse_client = sse_client.with(
-              ssl: { alpn_protocols: ["http/1.1"] }
-            )
-          end
+        def create_sse_client
+          sse_client = HTTPX.plugin(:stream).with(headers: @headers)
+          return sse_client unless @version == :http1
 
-          response = sse_client.get(@event_url, stream: true)
+          sse_client.with(ssl: { alpn_protocols: ["http/1.1"] })
+        end
+
+        def validate_sse_response!(response)
+          return unless response.status >= 400
+
+          error_body = read_error_body(response)
+          error_message = "HTTP #{response.status} error from SSE endpoint: #{error_body}"
+          RubyLLM::MCP.logger.error error_message
+
+          handle_client_error!(error_message, response.status) if response.status < 500
+
+          raise StandardError, error_message
+        end
+
+        def handle_client_error!(error_message, status_code)
+          @running = false
+          raise Errors::TransportError.new(
+            message: error_message,
+            code: status_code
+          )
+        end
 
+        def process_event_stream(response)
           event_buffer = []
           response.each_line do |event_line|
-            unless @running
-              response.body.close
-              next
-            end
+            break unless handle_event_line?(event_line, event_buffer, response)
+          end
+        end
 
-            # Strip the line and check if it's empty (indicates end of event)
-            line = event_line.strip
+        def handle_event_line?(event_line, event_buffer, response)
+          unless @running
+            response.body.close
+            return false
+          end
 
-            if line.empty?
-              # End of event - process the accumulated buffer
-              if event_buffer.any?
-                events = parse_event(event_buffer.join("\n"))
-                events.each { |event| process_event(event) }
-                event_buffer.clear
-              end
-            else
-              # Accumulate the line for the current event
-              event_buffer << line
+          line = event_line.strip
+
+          if line.empty?
+            process_buffered_event(event_buffer)
+          else
+            event_buffer << line
+          end
+
+          true
+        end
+
+        def process_buffered_event(event_buffer)
+          return unless event_buffer.any?
+
+          events = parse_event(event_buffer.join("\n"))
+          events.each { |event| process_event(event) }
+          event_buffer.clear
+        end
+
+        def read_error_body(response)
+          # Try to read the error body from the response
+          body = ""
+          begin
+            response.each do |chunk|
+              body << chunk
             end
+          rescue StandardError
+            # If we can't read the body, just use what we have
           end
+          body.strip.empty? ? "No error details provided" : body.strip
         end
 
         def handle_connection_error(message, error)

data/lib/ruby_llm/mcp/version.rb

@@ -2,6 +2,6 @@
 
 module RubyLLM
   module MCP
-    VERSION = "0.6.2"
+    VERSION = "0.6.3"
   end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm-mcp
 version: !ruby/object:Gem::Version
-  version: 0.6.2
+  version: 0.6.3
 platform: ruby
 authors:
 - Patrick Vice