ruby_llm 0.1.0.pre → 0.1.0.pre3

data/lib/ruby_llm/tool.rb

@@ -0,0 +1,75 @@
+# frozen_string_literal: true
+
+module RubyLLM
+  # Represents a tool/function that can be called by an LLM
+  class Tool
+    attr_reader :name, :description, :parameters, :handler
+
+    def self.from_method(method_object, description: nil, parameter_descriptions: {})
+      method_params = {}
+      method_object.parameters.each do |param_type, param_name|
+        next unless %i[req opt key keyreq].include?(param_type)
+
+        method_params[param_name] = {
+          type: 'string',
+          description: parameter_descriptions[param_name] || param_name.to_s.tr('_', ' '),
+          required: %i[req keyreq].include?(param_type)
+        }
+      end
+
+      new(
+        name: method_object.name.to_s,
+        description: description || "Executes the #{method_object.name} operation",
+        parameters: method_params
+      ) do |args|
+        # Create an instance if it's an instance method
+        instance = if method_object.owner.instance_methods.include?(method_object.name)
+                     method_object.owner.new
+                   else
+                     method_object.owner
+                   end
+
+        # Call the method with the arguments
+        if args.is_a?(Hash)
+          instance.method(method_object.name).call(**args)
+        else
+          instance.method(method_object.name).call(args)
+        end
+      end
+    end
+
+    def initialize(name:, description:, parameters: {}, &block)
+      @name = name
+      @description = description
+      @parameters = parameters
+      @handler = block
+
+      validate!
+    end
+
+    def call(args)
+      validated_args = validate_args!(args)
+      handler.call(validated_args)
+    end
+
+    private
+
+    def validate!
+      raise ArgumentError, 'Name must be a string' unless name.is_a?(String)
+      raise ArgumentError, 'Description must be a string' unless description.is_a?(String)
+      raise ArgumentError, 'Parameters must be a hash' unless parameters.is_a?(Hash)
+      raise ArgumentError, 'Block must be provided' unless handler.respond_to?(:call)
+    end
+
+    def validate_args!(args)
+      args = args.transform_keys(&:to_sym)
+      required_params = parameters.select { |_, v| v[:required] }.keys
+
+      required_params.each do |param|
+        raise ArgumentError, "Missing required parameter: #{param}" unless args.key?(param.to_sym)
+      end
+
+      args
+    end
+  end
+end

data/lib/ruby_llm/version.rb

@@ -1,3 +1,5 @@
+# frozen_string_literal: true
+
 module RubyLLM
-  VERSION = "0.1.0.pre"
+  VERSION = '0.1.0.pre3'
 end

data/lib/ruby_llm.rb

@@ -1,15 +1,23 @@
+# frozen_string_literal: true
+
 require 'zeitwerk'
 require 'faraday'
 require 'json'
+require 'securerandom'
+require 'logger'
 
+# Main module for RubyLLM functionality
 module RubyLLM
   class Error < StandardError; end
 
   class << self
-    attr_accessor :configuration
+    attr_writer :configuration
+
+    def configuration
+      @configuration ||= Configuration.new
+    end
 
     def configure
-      self.configuration ||= Configuration.new
       yield(configuration) if block_given?
     end
 
@@ -20,17 +28,41 @@ module RubyLLM
     def loader
       @loader ||= begin
         loader = Zeitwerk::Loader.for_gem
+
+        # Add lib directory to the load path
+        loader.push_dir(File.expand_path('..', __dir__))
+
+        # Configure custom inflections
         loader.inflector.inflect(
+          'ruby_llm' => 'RubyLLM',
           'llm' => 'LLM',
           'openai' => 'OpenAI',
           'api' => 'API'
         )
+
+        # Ignore Rails-specific files and specs
+        loader.ignore("#{__dir__}/ruby_llm/railtie.rb")
+        loader.ignore("#{__dir__}/ruby_llm/active_record")
+        loader.ignore(File.expand_path('../spec', __dir__).to_s)
+
+        # Log autoloading in debug mode
+        loader.logger = Logger.new($stdout) if ENV['RUBY_LLM_DEBUG']
+        loader.enable_reloading if ENV['RUBY_LLM_DEBUG']
+
         loader.setup
+        loader.eager_load
         loader
       end
     end
   end
 end
 
-# Register inflections
+# Initialize loader
 RubyLLM.loader
+
+# Load Rails integration if Rails is defined
+if defined?(Rails)
+  require 'active_support'
+  require 'ruby_llm/railtie'
+  require 'ruby_llm/active_record/acts_as'
+end

data/ruby_llm.gemspec

@@ -1,48 +1,59 @@
+# frozen_string_literal: true
+
 require_relative 'lib/ruby_llm/version'
 
 Gem::Specification.new do |spec|
-  spec.name          = "ruby_llm"
+  spec.name          = 'ruby_llm'
   spec.version       = RubyLLM::VERSION
-  spec.authors       = ["Carmine Paolino"]
-  spec.email         = ["carmine@paolino.me"]
-
-  spec.summary       = %q{The Ruby LLM client library - unified interface for OpenAI, Anthropic, and other LLM providers}
-  spec.description   = %q{Complete Ruby library for working with Large Language Models (LLMs). Supports OpenAI, Anthropic,
-    and others with a consistent interface. Features include tool usage, token tracking, and seamless Rails integration.}
-  spec.homepage      = "https://github.com/crmne/ruby_llm"
-  spec.license       = "MIT"
-  spec.required_ruby_version = Gem::Requirement.new(">= 2.7.0")
-
-  spec.metadata["homepage_uri"] = spec.homepage
-  spec.metadata["source_code_uri"] = spec.homepage
-  spec.metadata["changelog_uri"] = "#{spec.homepage}/blob/main/CHANGELOG.md"
-  spec.metadata["documentation_uri"] = "https://rubydoc.info/gems/ruby_llm"
-  spec.metadata["bug_tracker_uri"] = "#{spec.homepage}/issues"
+  spec.authors       = ['Carmine Paolino']
+  spec.email         = ['carmine@paolino.me']
+
+  spec.summary       = 'The Ruby LLM client library - unified interface for OpenAI, Anthropic, and other LLM providers'
+  spec.description   = 'Complete Ruby library for working with Large Language Models (LLMs). ' \
+                       'Supports OpenAI, Anthropic, and others with a consistent interface. ' \
+                       'Features include tool usage, token tracking, and seamless Rails integration.'
+  spec.homepage      = 'https://github.com/crmne/ruby_llm'
+  spec.license       = 'MIT'
+  spec.required_ruby_version = Gem::Requirement.new('>= 2.7.0')
+
+  spec.metadata['homepage_uri'] = spec.homepage
+  spec.metadata['source_code_uri'] = spec.homepage
+  spec.metadata['changelog_uri'] = "#{spec.homepage}/blob/main/CHANGELOG.md"
+  spec.metadata['documentation_uri'] = 'https://rubydoc.info/gems/ruby_llm'
+  spec.metadata['bug_tracker_uri'] = "#{spec.homepage}/issues"
 
   # 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.chdir(File.expand_path('..', __FILE__)) do
+  spec.files = Dir.chdir(File.expand_path(__dir__)) do
     `git ls-files -z`.split("\x0").reject { |f| f.match(%r{^(test|spec|features)/}) }
   end
-  spec.bindir        = "exe"
+  spec.bindir        = 'exe'
   spec.executables   = spec.files.grep(%r{^exe/}) { |f| File.basename(f) }
-  spec.require_paths = ["lib"]
+  spec.require_paths = ['lib']
 
   # Runtime dependencies
-  spec.add_dependency "faraday", ">= 2.0"
-  spec.add_dependency "faraday-multipart", ">= 1.0"
-  spec.add_dependency "zeitwerk", ">= 2.6"
+  spec.add_dependency 'faraday', '>= 2.0'
+  spec.add_dependency 'faraday-multipart', '>= 1.0'
+  spec.add_dependency 'zeitwerk', '>= 2.6'
 
   # Rails integration dependencies
-  spec.add_development_dependency "activesupport", ">= 6.0", "< 9.0"
-  spec.add_development_dependency "activerecord", ">= 6.0", "< 9.0"
+  spec.add_development_dependency 'activerecord', '>= 6.0', '< 9.0'
+  spec.add_development_dependency 'activesupport', '>= 6.0', '< 9.0'
 
   # Development dependencies
-  spec.add_development_dependency "bundler", ">= 2.0"
-  spec.add_development_dependency "rake", ">= 13.0"
-  spec.add_development_dependency "rubocop", ">= 1.0"
-  spec.add_development_dependency "yard", ">= 0.9"
-  spec.add_development_dependency "simplecov", ">= 0.21"
-  spec.add_development_dependency "pry", ">= 0.14"
-  spec.add_development_dependency "overcommit", ">= 0.66"
+  spec.add_development_dependency 'bundler', '>= 2.0'
+  spec.add_development_dependency 'dotenv'
+  spec.add_development_dependency 'irb'
+  spec.add_development_dependency 'overcommit', '>= 0.66'
+  spec.add_development_dependency 'pry', '>= 0.14'
+  spec.add_development_dependency 'rake', '>= 13.0'
+  spec.add_development_dependency 'rdoc'
+  spec.add_development_dependency 'reline'
+  spec.add_development_dependency 'rspec', '~> 3.12'
+  spec.add_development_dependency 'rubocop', '>= 1.0'
+  spec.add_development_dependency 'rubocop-rake', '>= 0.6'
+  spec.add_development_dependency 'rubocop-rspec', '>= 3.4'
+  spec.add_development_dependency 'simplecov', '>= 0.21'
+  spec.add_development_dependency 'webmock', '~> 3.18'
+  spec.add_development_dependency 'yard', '>= 0.9'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ruby_llm
 version: !ruby/object:Gem::Version
-  version: 0.1.0.pre
+  version: 0.1.0.pre3
 platform: ruby
 authors:
 - Carmine Paolino
@@ -52,7 +52,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '2.6'
 - !ruby/object:Gem::Dependency
-  name: activesupport
+  name: activerecord
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -72,7 +72,7 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '9.0'
 - !ruby/object:Gem::Dependency
-  name: activerecord
+  name: activesupport
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
@@ -105,6 +105,62 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '2.0'
+- !ruby/object:Gem::Dependency
+  name: dotenv
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: irb
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: overcommit
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0.66'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0.66'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0.14'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0.14'
 - !ruby/object:Gem::Dependency
   name: rake
   requirement: !ruby/object:Gem::Requirement
@@ -119,6 +175,48 @@ dependencies:
     - - ">="
       - !ruby/object:Gem::Version
         version: '13.0'
+- !ruby/object:Gem::Dependency
+  name: rdoc
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: reline
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.12'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.12'
 - !ruby/object:Gem::Dependency
   name: rubocop
   requirement: !ruby/object:Gem::Requirement
@@ -134,19 +232,33 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '1.0'
 - !ruby/object:Gem::Dependency
-  name: yard
+  name: rubocop-rake
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0.9'
+        version: '0.6'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0.9'
+        version: '0.6'
+- !ruby/object:Gem::Dependency
+  name: rubocop-rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '3.4'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '3.4'
 - !ruby/object:Gem::Dependency
   name: simplecov
   requirement: !ruby/object:Gem::Requirement
@@ -162,36 +274,36 @@ dependencies:
       - !ruby/object:Gem::Version
         version: '0.21'
 - !ruby/object:Gem::Dependency
-  name: pry
+  name: webmock
   requirement: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.14'
+        version: '3.18'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
-    - - ">="
+    - - "~>"
       - !ruby/object:Gem::Version
-        version: '0.14'
+        version: '3.18'
 - !ruby/object:Gem::Dependency
-  name: overcommit
+  name: yard
   requirement: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0.66'
+        version: '0.9'
   type: :development
   prerelease: false
   version_requirements: !ruby/object:Gem::Requirement
     requirements:
     - - ">="
       - !ruby/object:Gem::Version
-        version: '0.66'
-description: |-
-  Complete Ruby library for working with Large Language Models (LLMs). Supports OpenAI, Anthropic,
-      and others with a consistent interface. Features include tool usage, token tracking, and seamless Rails integration.
+        version: '0.9'
+description: Complete Ruby library for working with Large Language Models (LLMs).
+  Supports OpenAI, Anthropic, and others with a consistent interface. Features include
+  tool usage, token tracking, and seamless Rails integration.
 email:
 - carmine@paolino.me
 executables: []
@@ -199,6 +311,12 @@ extensions: []
 extra_rdoc_files: []
 files:
 - ".github/workflows/gem-push.yml"
+- ".github/workflows/test.yml"
+- ".gitignore"
+- ".overcommit.yml"
+- ".rspec"
+- ".rubocop.yml"
+- Gemfile
 - README.md
 - Rakefile
 - bin/console
@@ -209,8 +327,15 @@ files:
 - lib/ruby_llm/configuration.rb
 - lib/ruby_llm/conversation.rb
 - lib/ruby_llm/message.rb
+- lib/ruby_llm/model_capabilities/anthropic.rb
+- lib/ruby_llm/model_capabilities/base.rb
+- lib/ruby_llm/model_capabilities/openai.rb
+- lib/ruby_llm/model_info.rb
+- lib/ruby_llm/providers/anthropic.rb
 - lib/ruby_llm/providers/base.rb
+- lib/ruby_llm/providers/openai.rb
 - lib/ruby_llm/railtie.rb
+- lib/ruby_llm/tool.rb
 - lib/ruby_llm/version.rb
 - ruby_llm.gemspec
 homepage: https://github.com/crmne/ruby_llm