tool_tailor 0.1.1 → 0.1.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 5b242798d32a1e165bcfee93e5562b97015ad554acdd6837e3ee729b3b27e6ed
-  data.tar.gz: 3a9a097551149899b80e71b00dc936c6942dd213d66d97fb040485aeadd1408d
+  metadata.gz: abc31d6d69830be0126f8b6f61aee408be9e17bdb91423b68e9166f30efaf21f
+  data.tar.gz: 7eb42498a63836eeac46c6996d2e32f34303430b58dcdd36cb89d833cbadeaef
 SHA512:
-  metadata.gz: 342c3b07cb9ee1b01ead72205ce15b7b1cde92f10b3ba3d16cdfc18114d76c04b6c4bdd64a3a8c10fa0c1c3927248d4b3d7e8586edaca1c0d7e13ac1f49b018e
-  data.tar.gz: 347ff4fe7ab0d29164d48e97a38400e4d8bb7816c81b791baab903ebdc764f8b4fcf97156dc256e56d66d6eddca92185c80740c488889342ffb9efaeabbd8aec
+  metadata.gz: 04d10928c7453fbd20980d3c23a6bbbe8cf000baa25f2ffce903910b21607088762f300fde528ee4c7f0b8673ba58a65224926c261486ff033d87c5d25601468
+  data.tar.gz: 779238cd2409d9d6c669cab52f0d286fc79cf3c9335082d88bf62d871f23ec6439fe479f465d25fab0bd13d2285e6522ee026ea9e5257b019cda483244c28664

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    tool_tailor (0.1.1)
+    tool_tailor (0.1.2)
       yard (~> 0.9.36)
 
 GEM

data/README.md

@@ -26,12 +26,12 @@ class TestClass
   #
   # @param location [String] The city and state, e.g., San Francisco, CA.
   # @param unit [String] The unit of temperature, either 'celsius' or 'fahrenheit'.
-  # @param api_key [Float] The API key for the weather service.
-  def get_current_weather(location, unit = 'celsius', api_key: nil)
+  def get_current_weather(location, unit = 'celsius')
     # Function implementation goes here
   end
 end
 
+# Unbound method
 TestClass.instance_method(:get_current_weather).to_json_schema # => {
 #   "type" => "function",
 #   "function" => {
@@ -56,6 +56,77 @@ TestClass.instance_method(:get_current_weather).to_json_schema # => {
 #       "required" => ["location", "unit", "api_key"]
 #     }
 #   }
+
+# Bound method
+example_instance = TestClass.new
+example_instance.method(:get_current_weather).to_json_schema # => {
+#   "type" => "function",
+#   "function" => {
+#     "name" => "get_current_weather",
+#     "description" => "Get the current weather in a given location.",
+#     "parameters" => {
+#       "type" => "object",
+#       "properties" => {
+#         "location" => {
+#           "type" => "string",
+#           "description" => "The city and state, e.g., San Francisco, CA."
+#         },
+#         "unit" => {
+#           "type" => "string",
+#           "description" => "The unit of temperature, either 'celsius' or 'fahrenheit'."
+#         },
+#         "api_key" => {
+#           "type" => "number",
+#           "description" => "The API key for the weather service."
+#         }
+#       },
+#       "required" => ["location", "unit", "api_key"]
+#     }
+#   }
+# }
+```
+
+And with [ruby-openai](https://github.com/alexrudall/ruby-openai):
+
+```rb
+response =
+  client.chat(
+    parameters: {
+      model: "gpt-4o",
+      messages: [
+        {
+          "role": "user",
+          "content": "What is the weather like in San Francisco?",
+        },
+      ],
+      tools: [
+        TestClass.instance_method(:get_current_weather).to_json_schema
+      ],
+      tool_choice: {
+        type: "function",
+        function: {
+          name: "get_current_weather"
+        }
+      }
+    },
+  )
+
+message = response.dig("choices", 0, "message")
+
+if message["role"] == "assistant" && message["tool_calls"]
+  function_name = message.dig("tool_calls", 0, "function", "name")
+  args =
+    JSON.parse(
+      message.dig("tool_calls", 0, "function", "arguments"),
+      { symbolize_names: true },
+    )
+
+  case function_name
+  when "get_current_weather"
+    TestClass.get_current_weather(**args)
+  end
+end
+# => "The weather is nice 🌞"
 ```
 
 ## Development

data/lib/tool_tailor/version.rb

@@ -1,3 +1,3 @@
 module ToolTailor
-  VERSION = "0.1.1"
+  VERSION = "0.1.2"
 end

data/lib/tool_tailor.rb

@@ -1,13 +1,13 @@
 require "tool_tailor/version"
-require 'json'
-require 'yard'
+require "json"
+require "yard"
 
 module ToolTailor
   class Error < StandardError; end
 
   # Converts a function to a JSON schema representation.
   #
-  # @param function [Method] The function to convert.
+  # @param function [Method, UnboundMethod] The function to convert.
   # @return [String] The JSON schema representation of the function.
   # @raise [ArgumentError] If the provided object is not a Method or UnboundMethod.
   def self.convert(function)
@@ -18,14 +18,13 @@ module ToolTailor
     file_path, line_number = function.source_location
     YARD.parse(file_path)
 
-    # Construct the correct identifier for the YARD object
     method_path = "#{function.owner}##{function.name}"
     yard_object = YARD::Registry.at(method_path)
     raise "Documentation for #{method_path} not found." if yard_object.nil?
 
     function_description = yard_object.docstring
 
-    parameters = yard_object.tags('param').map do |tag|
+    parameters = yard_object.tags("param").map do |tag|
       {
         name: tag.name,
         type: type_mapping(tag.types.first),
@@ -57,33 +56,101 @@ module ToolTailor
 
   # Maps Ruby types to JSON schema types.
   #
-  # @param type [Class] The Ruby type to map.
+  # @param type [String] The Ruby type to map.
   # @return [String] The corresponding JSON schema type.
   # @raise [ArgumentError] If the provided type is not supported.
   def self.type_mapping(type)
     case type
-    when "String"
-      'string'
-    when "Integer"
-      'integer'
-    when "Float"
-      'number'
-    when "TrueClass", "FalseClass"
-      'boolean'
-    when "Array"
-      'array'
-    when "Hash"
-      'object'
-    when "NilClass"
-      'null'
+    when "String"    then "string"
+    when "Integer"   then "integer"
+    when "Float"     then "number"
+    when "TrueClass", "FalseClass" then "boolean"
+    when "Array"     then "array"
+    when "Hash"      then "object"
+    when "NilClass"  then "null"
     else
-      # raise ArgumentError, "Unsupported type: #{type} #{type.class}"
-      'string'
+      raise ArgumentError, "Unsupported type: #{type} #{type.class}"
     end
   end
 end
 
 class UnboundMethod
+  # Converts an UnboundMethod to a JSON schema.
+  #
+  # @example
+  #   class ExampleClass
+  #     # @param name [String] The name of the person.
+  #     # @param age [Integer] The age of the person.
+  #     def greet(name, age)
+  #       puts "Hello, #{name}! You are #{age} years old."
+  #     end
+  #   end
+  #
+  #   ExampleClass.instance_method(:greet).to_json_schema
+  #   # => {
+  #   #   "type" => "function",
+  #   #   "function" => {
+  #   #     "name" => "greet",
+  #   #     "description" => "",
+  #   #     "parameters" => {
+  #   #       "type" => "object",
+  #   #       "properties" => {
+  #   #         "name" => {
+  #   #           "type" => "string",
+  #   #           "description" => "The name of the person."
+  #   #         },
+  #   #         "age" => {
+  #   #           "type" => "integer",
+  #   #           "description" => "The age of the person."
+  #   #         }
+  #   #       },
+  #   #       "required" => ["name", "age"]
+  #   #     }
+  #   #   }
+  #   # }
+  #
+  # @return [String] The JSON schema representation of the method.
+  def to_json_schema
+    ToolTailor.convert(self)
+  end
+end
+
+class Method
+  # Converts a Method to a JSON schema.
+  #
+  # @example
+  #   class ExampleClass
+  #     # @param name [String] The name of the person.
+  #     # @param age [Integer] The age of the person.
+  #     def greet(name, age)
+  #       puts "Hello, #{name}! You are #{age} years old."
+  #     end
+  #   end
+  #
+  #   ExampleClass.new.method(:greet).to_json_schema
+  #   # => {
+  #   #   "type" => "function",
+  #   #   "function" => {
+  #   #     "name" => "greet",
+  #   #     "description" => "",
+  #   #     "parameters" => {
+  #   #       "type" => "object",
+  #   #       "properties" => {
+  #   #         "name" => {
+  #   #           "type" => "string",
+  #   #           "description" => "The name of the person."
+  #   #         },
+  #   #         "age" => {
+  #   #           "type" => "integer",
+  #   #           "description" => "The age of the person."
+  #   #         }
+  #   #       },
+  #   #       "required" => ["name", "age"]
+  #   #     }
+  #   #   }
+  #   # }
+  #
+  # @return [String] The JSON schema representation of the method.
   def to_json_schema
     ToolTailor.convert(self)
   end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: tool_tailor
 version: !ruby/object:Gem::Version
-  version: 0.1.1
+  version: 0.1.2
 platform: ruby
 authors:
 - Kieran Klaassen