Merge pull request #32 from patvice/changing-zeitwerk-setup

Changing Zeitwerk setup

Author: patvice

lib/ruby_llm/chat.rb

@@ -1,6 +1,7 @@
 # frozen_string_literal: true
 
-# This is an override of the RubyLLM::Chat class to convient methods for easy MCP support
+# This is an override of the RubyLLM::Chat class to add convenient methods to more
+# easily work with the MCP clients.
 module RubyLLM
   class Chat
     def with_resources(*resources, **args)

lib/ruby_llm/mcp.rb

@@ -4,13 +4,6 @@
 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
@@ -41,3 +34,11 @@ def logger
     end
   end
 end
+
+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

ruby_llm-mcp.gemspec

@@ -33,7 +33,7 @@ Gem::Specification.new do |spec|
 
   # Specify which files should be added to the gem when it is released.
   # The `git ls-files -z` loads the files in the RubyGem that have been added into git.
-  spec.files = Dir.glob("lib/**/*") + ["README.md", "LICENSE"]
+  spec.files = Dir.glob("lib/ruby_llm/**/*") + ["README.md", "LICENSE"]
   spec.require_paths = ["lib"]
 
   spec.add_dependency "httpx", "~> 1.4"