support complations for resources

Author: patvice

Gemfile

@@ -13,7 +13,7 @@ group :development do
   gem "rdoc"
   gem "reline"
   gem "rspec", "~> 3.12"
-  gem "rubocop", ">= 1.0"
-  gem "rubocop-rake", ">= 0.6"
-  gem "rubocop-rspec"
+  gem "rubocop", ">= 1.76"
+  gem "rubocop-rake", ">= 0.7"
+  gem "rubocop-rspec", ">= 3.6"
 end

Gemfile.lock

@@ -36,7 +36,7 @@ GEM
       pp (>= 0.6.0)
       rdoc (>= 4.0.0)
       reline (>= 0.4.2)
-    json (2.12.0)
+    json (2.12.2)
     language_server-protocol (3.17.0.5)
     lint_roller (1.1.0)
     logger (1.7.0)
@@ -76,18 +76,18 @@ GEM
       diff-lcs (>= 1.2.0, < 2.0)
       rspec-support (~> 3.13.0)
     rspec-support (3.13.3)
-    rubocop (1.75.6)
+    rubocop (1.76.0)
       json (~> 2.3)
       language_server-protocol (~> 3.17.0.2)
       lint_roller (~> 1.1.0)
       parallel (~> 1.10)
       parser (>= 3.3.0.2)
       rainbow (>= 2.2.2, < 4.0)
       regexp_parser (>= 2.9.3, < 3.0)
-      rubocop-ast (>= 1.44.0, < 2.0)
+      rubocop-ast (>= 1.45.0, < 2.0)
       ruby-progressbar (~> 1.7)
       unicode-display_width (>= 2.4.0, < 4.0)
-    rubocop-ast (1.44.1)
+    rubocop-ast (1.45.1)
       parser (>= 3.3.7.2)
       prism (~> 1.4)
     rubocop-rake (0.7.1)
@@ -125,9 +125,9 @@ DEPENDENCIES
   rdoc
   reline
   rspec (~> 3.12)
-  rubocop (>= 1.0)
-  rubocop-rake (>= 0.6)
-  rubocop-rspec
+  rubocop (>= 1.76)
+  rubocop-rake (>= 0.7)
+  rubocop-rspec (>= 3.6)
   ruby_llm-mcp!
 
 BUNDLED WITH

examples/resources/list_resource_template.rb

@@ -0,0 +1,30 @@
+# frozen_string_literal: true
+
+require "bundler/setup"
+require "ruby_llm/mcp"
+require "debug"
+require "dotenv"
+
+Dotenv.load
+
+RubyLLM.configure do |config|
+  config.openai_api_key = ENV.fetch("OPENAI_API_KEY", nil)
+end
+
+client = RubyLLM::MCP.client(
+  name: "streamable_mcp",
+  transport_type: :streamable,
+  config: {
+    url: "http://localhost:3005/mcp"
+  }
+)
+resources = client.resources
+
+resources.each do |resource|
+  puts "Resource: #{resource.name}"
+  puts "Description: #{resource.description}"
+  puts "MIME Type: #{resource.mime_type}"
+  puts "Template: #{resource.template?}"
+  puts "Content: #{resource.content}"
+  puts "--------------------------------"
+end

lib/ruby_llm/mcp/capabilities.rb

@@ -20,6 +20,10 @@ def resource_subscribe?
       def tools_list_changed?
         @capabilities.dig("tools", "listChanged") || false
       end
+
+      def completion?
+        @capabilities["completion"].present?
+      end
     end
   end
 end

lib/ruby_llm/mcp/client.rb

@@ -36,8 +36,8 @@ def initialize(name:, transport_type:, request_timeout: 8000, reverse_proxy_url:
         notification_request
       end
 
-      def request(body, add_id: true, wait_for_response: true, **options)
-        @transport.request(body, add_id: add_id, wait_for_response: wait_for_response, **options)
+      def request(body, **options)
+        @transport.request(body, **options)
       end
 
       def tools(refresh: false)
@@ -50,6 +50,11 @@ def resources(refresh: false)
         @resources ||= fetch_and_create_resources
       end
 
+      def resource_templates(refresh: false)
+        @resource_templates = nil if refresh
+        @resource_templates ||= fetch_and_create_resources(set_as_template: true)
+      end
+
       def execute_tool(name:, parameters:)
         response = execute_tool_request(name: name, parameters: parameters)
         result = response["result"]
@@ -59,8 +64,12 @@ def execute_tool(name:, parameters:)
         result["content"].map { |content| content["text"] }.join("\n")
       end
 
-      def resource_read_request(uri:)
-        @resource_read_response = RubyLLM::MCP::Requests::ResourceRead.new(self, uri: uri).call
+      def resource_read_request(**args)
+        RubyLLM::MCP::Requests::ResourceRead.new(self, **args).call
+      end
+
+      def completion(**args)
+        RubyLLM::MCP::Requests::Completion.new(self, **args).call
       end
 
       private
@@ -71,23 +80,23 @@ def initialize_request
       end
 
       def notification_request
-        @notification_response = RubyLLM::MCP::Requests::Notification.new(self).call
+        RubyLLM::MCP::Requests::Notification.new(self).call
       end
 
       def tool_list_request
-        @tool_request = RubyLLM::MCP::Requests::ToolList.new(self).call
+        RubyLLM::MCP::Requests::ToolList.new(self).call
       end
 
-      def execute_tool_request(name:, parameters:)
-        @execute_tool_response = RubyLLM::MCP::Requests::ToolCall.new(self, name: name, parameters: parameters).call
+      def execute_tool_request(**args)
+        RubyLLM::MCP::Requests::ToolCall.new(self, **args).call
       end
 
       def resources_list_request
-        @resources_request = RubyLLM::MCP::Requests::ResourceList.new(self).call
+        RubyLLM::MCP::Requests::ResourceList.new(self).call
       end
 
       def resource_template_list_request
-        @resource_template_list_response = RubyLLM::MCP::Requests::ResourceTemplateList.new(self).call
+        RubyLLM::MCP::Requests::ResourceTemplateList.new(self).call
       end
 
       def fetch_and_create_tools
@@ -99,12 +108,12 @@ def fetch_and_create_tools
         end
       end
 
-      def fetch_and_create_resources
+      def fetch_and_create_resources(set_as_template: false)
         resources_response = resources_list_request
         resources_response = resources_response["result"]["resources"]
 
         @resources = resources_response.map do |resource|
-          RubyLLM::MCP::Resource.new(self, resource)
+          RubyLLM::MCP::Resource.new(self, resource, template: set_as_template)
         end
       end
     end

lib/ruby_llm/mcp/requests/completion.rb

@@ -0,0 +1,50 @@
+# frozen_string_literal: true
+
+module RubyLLM
+  module MCP
+    module Requests
+      class Completion
+        def initialize(client, type:, name:, argument:, value:)
+          @client = client
+          @type = type
+          @name = name
+          @argument = argument
+          @value = value
+        end
+
+        def call
+          @client.request(request_body)
+        end
+
+        private
+
+        def request_body
+          {
+            jsonrpc: "2.0",
+            id: 1,
+            method: "completion/complete",
+            params: {
+              ref: {
+                type: ref_type,
+                name: @name
+              },
+              argument: {
+                name: @argument,
+                value: @value
+              }
+            }
+          }
+        end
+
+        def ref_type
+          case @type
+          when :prompt
+            "ref/prompt"
+          when :resource
+            "ref/resource"
+          end
+        end
+      end
+    end
+  end
+end

lib/ruby_llm/mcp/resource.rb

@@ -27,6 +27,18 @@ def content(parsable_information: {})
           response.dig("result", "contents", 0, "blob")
       end
 
+      def arguments_search(argument, value)
+        if template? && @mcp_client.capabilities.completion?
+          response = @mcp_client.completion(type: :resource, name: @name, argument: argument, value: value)
+          response = response.dig("result", "completion")
+
+          Struct.new(:arg_values, :total, :has_more)
+                .new(response["values"], response["total"], response["hasMore"])
+        else
+          []
+        end
+      end
+
       def template?
         @template
       end