llm.rb 4.5.0 → 4.6.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c8252114d7ab58f00fd2d14389e932cfa9b2f0a71d96d9ea2c261f1f8b67d721
-  data.tar.gz: 8c00745ba750d0e271d8a4e5d5d9418a13556847c181e912aa8a8a1e7a9344b5
+  metadata.gz: c26db042940fc118505804bbec91d43dd39313b6625b4e3d7df51f43f01d58ae
+  data.tar.gz: 5780f3e3d0db3579772d4f7ba141052314d34b7fd0db06ba029e4b4fd6fd84dc
 SHA512:
-  metadata.gz: 9b45173644a3803db844c1cad679a14ec647e058bd4517ce0160b7f2f352480877375748a4abb727007d3c76e7b977a6c106a5bfa438575f69ef8cd60a5612d2
-  data.tar.gz: bebdb107819b1410bc6644529ad0f56544deadbae9661ab8ed2c84a420c74097b357113650de7ed7da1d0b388c601a9b7e60e30d887b62d72e7cfb84f3f0dd1a
+  metadata.gz: 864ba517e81d8fb4d248d754c5a9fc4bdcbf162f9ac4fb7691a54c015006ed40035300fd19c2057f755678676534a568204b0e0bc29b8e058cf2334b8625465e
+  data.tar.gz: cd8b18c4e780a26c83201ea39f81077ad2535db966efcb5fe860f4af55808f96a8bd8a6bdd4c25f38a9012b3092eb2611bbfd31552969b00b15579af5e95354a

data/README.md

@@ -4,7 +4,7 @@
 <p align="center">
   <a href="https://0x1eef.github.io/x/llm.rb?rebuild=1"><img src="https://img.shields.io/badge/docs-0x1eef.github.io-blue.svg" alt="RubyDoc"></a>
   <a href="https://opensource.org/license/0bsd"><img src="https://img.shields.io/badge/License-0BSD-orange.svg?" alt="License"></a>
-  <a href="https://github.com/llmrb/llm.rb/tags"><img src="https://img.shields.io/badge/version-4.5.0-green.svg?" alt="Version"></a>
+  <a href="https://github.com/llmrb/llm.rb/tags"><img src="https://img.shields.io/badge/version-4.6.0-green.svg?" alt="Version"></a>
 </p>
 
 ## About

data/lib/llm/tool/param.rb

@@ -21,7 +21,7 @@ class LLM::Tool
     ##
     # @param name [Symbol]
     #   The name of a parameter
-    # @param type [Class, Symbol]
+    # @param type [LLM::Schema::Leaf, Class]
     #   The parameter type (eg String)
     # @param description [String]
     #   The description of a property
@@ -36,9 +36,8 @@ class LLM::Tool
     def param(name, type, description, options = {})
       lock do
         function.params do |schema|
-          leaf = schema.public_send(Utils.resolve(type))
-          leaf = Utils.setup(leaf, description, options)
-          schema.object(name => leaf)
+          resolved = Utils.resolve(schema, type)
+          schema.object(name => Utils.setup(resolved, description, options))
         end
       end
     end
@@ -48,15 +47,14 @@ class LLM::Tool
     module Utils
       extend self
 
-      def resolve(type)
-        if type == String
-          :string
-        elsif type == Integer
-          :integer
-        elsif type == Float
-          :number
-        else
+      def resolve(schema, type)
+        if LLM::Schema::Leaf === type
           type
+        elsif Class === type && type.respond_to?(:object)
+          type.object
+        else
+          target = type.name.split("::").last.downcase
+          schema.public_send(target)
         end
       end
 

data/lib/llm/tool.rb

@@ -21,6 +21,15 @@ class LLM::Tool
   require_relative "tool/param"
   extend LLM::Tool::Param
 
+  types = [
+    :Leaf, :String, :Array,
+    :Object, :Integer, :Number,
+    :Boolean, :Null
+  ]
+  types.each do |constant|
+    const_set constant, LLM::Schema.const_get(constant)
+  end
+
   ##
   # Registers the tool as a function when inherited
   # @param [Class] klass The subclass

data/lib/llm/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module LLM
-  VERSION = "4.5.0"
+  VERSION = "4.6.0"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: llm.rb
 version: !ruby/object:Gem::Version
-  version: 4.5.0
+  version: 4.6.0
 platform: ruby
 authors:
 - Antar Azri