ruby_llm-mcp 0.4.1 → 0.5.1

data/lib/ruby_llm/mcp.rb

@@ -4,19 +4,59 @@ require "ruby_llm"
 require "zeitwerk"
 require_relative "chat"
 
-loader = Zeitwerk::Loader.for_gem_extension(RubyLLM)
-loader.inflector.inflect("mcp" => "MCP")
-loader.inflector.inflect("sse" => "SSE")
-loader.inflector.inflect("openai" => "OpenAI")
-loader.inflector.inflect("streamable_http" => "StreamableHTTP")
-loader.setup
-
 module RubyLLM
   module MCP
     module_function
 
-    def client(*args, **kwargs)
-      @client ||= Client.new(*args, **kwargs)
+    def clients(config = RubyLLM::MCP.config.mcp_configuration)
+      @clients ||= {}
+      config.map do |options|
+        @clients[options[:name]] ||= Client.new(**options)
+      end
+    end
+
+    def add_client(options)
+      @clients ||= {}
+      @clients[options[:name]] ||= Client.new(**options)
+    end
+
+    def remove_client(name)
+      @clients ||= {}
+      client = @clients.delete(name)
+      client&.stop
+      client
+    end
+
+    def client(...)
+      Client.new(...)
+    end
+
+    def establish_connection(&)
+      clients.each(&:start)
+      if block_given?
+        begin
+          yield clients
+        ensure
+          close_connection
+        end
+      else
+        clients
+      end
+    end
+
+    def close_connection
+      clients.each do |client|
+        client.stop if client.alive?
+      end
+    end
+
+    def tools(blacklist: [], whitelist: [])
+      tools = @clients.values.map(&:tools)
+                      .flatten
+                      .reject { |tool| blacklist.include?(tool.name) }
+
+      tools = tools.select { |tool| whitelist.include?(tool.name) } if whitelist.any?
+      tools.uniq(&:name)
     end
 
     def support_complex_parameters!
@@ -41,3 +81,14 @@ module RubyLLM
     end
   end
 end
+
+require_relative "mcp/railtie" if defined?(Rails::Railtie)
+
+loader = Zeitwerk::Loader.for_gem_extension(RubyLLM)
+loader.inflector.inflect("mcp" => "MCP")
+loader.inflector.inflect("sse" => "SSE")
+loader.inflector.inflect("openai" => "OpenAI")
+loader.inflector.inflect("streamable_http" => "StreamableHTTP")
+loader.inflector.inflect("http_client" => "HTTPClient")
+
+loader.setup

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm-mcp
 version: !ruby/object:Gem::Version
-  version: 0.4.1
+  version: 0.5.1
 platform: ruby
 authors:
 - Patrick Vice
 autorequire:
 bindir: bin
 cert_chain: []
-date: 2025-07-02 00:00:00.000000000 Z
+date: 2025-07-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: httpx
@@ -65,10 +65,13 @@ extra_rdoc_files: []
 files:
 - LICENSE
 - README.md
+- lib/generators/ruby_llm/mcp/install_generator.rb
+- lib/generators/ruby_llm/mcp/templates/README.txt
+- lib/generators/ruby_llm/mcp/templates/initializer.rb
+- lib/generators/ruby_llm/mcp/templates/mcps.yml
 - lib/ruby_llm/chat.rb
 - lib/ruby_llm/mcp.rb
 - lib/ruby_llm/mcp/attachment.rb
-- lib/ruby_llm/mcp/capabilities.rb
 - lib/ruby_llm/mcp/client.rb
 - lib/ruby_llm/mcp/completion.rb
 - lib/ruby_llm/mcp/configuration.rb
@@ -77,37 +80,50 @@ files:
 - lib/ruby_llm/mcp/error.rb
 - lib/ruby_llm/mcp/errors.rb
 - lib/ruby_llm/mcp/logging.rb
+- lib/ruby_llm/mcp/notification_handler.rb
+- lib/ruby_llm/mcp/notifications/cancelled.rb
+- lib/ruby_llm/mcp/notifications/initialize.rb
+- lib/ruby_llm/mcp/notifications/roots_list_change.rb
 - lib/ruby_llm/mcp/parameter.rb
 - lib/ruby_llm/mcp/progress.rb
 - lib/ruby_llm/mcp/prompt.rb
 - lib/ruby_llm/mcp/providers/anthropic/complex_parameter_support.rb
 - lib/ruby_llm/mcp/providers/gemini/complex_parameter_support.rb
 - lib/ruby_llm/mcp/providers/openai/complex_parameter_support.rb
-- lib/ruby_llm/mcp/requests/base.rb
-- lib/ruby_llm/mcp/requests/cancelled_notification.rb
+- lib/ruby_llm/mcp/railtie.rb
 - lib/ruby_llm/mcp/requests/completion_prompt.rb
 - lib/ruby_llm/mcp/requests/completion_resource.rb
 - lib/ruby_llm/mcp/requests/initialization.rb
-- lib/ruby_llm/mcp/requests/initialize_notification.rb
 - lib/ruby_llm/mcp/requests/logging_set_level.rb
-- lib/ruby_llm/mcp/requests/meta.rb
 - lib/ruby_llm/mcp/requests/ping.rb
-- lib/ruby_llm/mcp/requests/ping_response.rb
 - lib/ruby_llm/mcp/requests/prompt_call.rb
 - lib/ruby_llm/mcp/requests/prompt_list.rb
 - lib/ruby_llm/mcp/requests/resource_list.rb
 - lib/ruby_llm/mcp/requests/resource_read.rb
 - lib/ruby_llm/mcp/requests/resource_template_list.rb
 - lib/ruby_llm/mcp/requests/resources_subscribe.rb
+- lib/ruby_llm/mcp/requests/shared/meta.rb
+- lib/ruby_llm/mcp/requests/shared/pagination.rb
 - lib/ruby_llm/mcp/requests/tool_call.rb
 - lib/ruby_llm/mcp/requests/tool_list.rb
 - lib/ruby_llm/mcp/resource.rb
 - lib/ruby_llm/mcp/resource_template.rb
+- lib/ruby_llm/mcp/response_handler.rb
+- lib/ruby_llm/mcp/responses/error.rb
+- lib/ruby_llm/mcp/responses/ping.rb
+- lib/ruby_llm/mcp/responses/roots_list.rb
+- lib/ruby_llm/mcp/responses/sampling_create_message.rb
 - lib/ruby_llm/mcp/result.rb
+- lib/ruby_llm/mcp/roots.rb
+- lib/ruby_llm/mcp/sample.rb
+- lib/ruby_llm/mcp/server_capabilities.rb
 - lib/ruby_llm/mcp/tool.rb
-- lib/ruby_llm/mcp/transport/sse.rb
-- lib/ruby_llm/mcp/transport/stdio.rb
-- lib/ruby_llm/mcp/transport/streamable_http.rb
+- lib/ruby_llm/mcp/transport.rb
+- lib/ruby_llm/mcp/transports/http_client.rb
+- lib/ruby_llm/mcp/transports/sse.rb
+- lib/ruby_llm/mcp/transports/stdio.rb
+- lib/ruby_llm/mcp/transports/streamable_http.rb
+- lib/ruby_llm/mcp/transports/timeout.rb
 - lib/ruby_llm/mcp/version.rb
 - lib/tasks/release.rake
 homepage: https://github.com/patvice/ruby_llm-mcp

data/lib/ruby_llm/mcp/requests/base.rb

@@ -1,31 +0,0 @@
-# frozen_string_literal: true
-
-require "json"
-
-module RubyLLM
-  module MCP
-    module Requests
-      class Base
-        attr_reader :coordinator
-
-        def initialize(coordinator)
-          @coordinator = coordinator
-        end
-
-        def call
-          raise "Not implemented"
-        end
-
-        private
-
-        def validate_response!(response, body)
-          # TODO: Implement response validation
-        end
-
-        def raise_error(error)
-          raise "MCP Error: code: #{error['code']} message: #{error['message']} data: #{error['data']}"
-        end
-      end
-    end
-  end
-end

data/lib/ruby_llm/mcp/requests/meta.rb

@@ -1,30 +0,0 @@
-# frozen_string_literal: true
-
-require "securerandom"
-
-module RubyLLM
-  module MCP
-    module Requests
-      module Meta
-        def merge_meta(body)
-          meta = {}
-          meta.merge!(progress_token) if @coordinator.client.tracking_progress?
-
-          body[:params] ||= {}
-          body[:params].merge!({ _meta: meta }) unless meta.empty?
-          body
-        end
-
-        private
-
-        def progress_token
-          { progressToken: generate_progress_token }
-        end
-
-        def generate_progress_token
-          SecureRandom.uuid
-        end
-      end
-    end
-  end
-end