omniai 1.8.0 → 1.8.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7325d58430db1dd52ecbbf7964ad9a02a3fe5ae89be51779582a672bbc8c8222
-  data.tar.gz: 2464d2ae08fc94fad891acae6eb1ec7fbe163414a3dfb6583fd89bca52dc92d1
+  metadata.gz: cca8dd5cc19989ca0a3fc89d1e8f2c5a56611378a202ea8da2fd49528454ea6c
+  data.tar.gz: ee0abcf49b13df01ed5b63ed557de44d42d640fb966bcfb5c7592e19aff06f93
 SHA512:
-  metadata.gz: 2d966778690e811bab403217378a748c778e043c48cdb946d9f2b3503ae3b48a165c63ab57bbba65ffa864cd9395388181b57b3c689efb548b864d58f931a757
-  data.tar.gz: a82e9b1b092c4df179a4d2b931f6d2c217ba1976504fd40179bc9c0549ab9c9fd349e3eedb79df2c3e30cde1a1ea3b236b99449074c10da381171e1d560314e8
+  metadata.gz: eb48e8f76d7cd1836ad8dc0ceb577362a3f289026749e2cc49044e31d25758a625ff66fe8003be12dc09585d9ee571671dd4f6d0b8c355cc7962eb4c136ec9f0
+  data.tar.gz: 41cf6419a3cf3c1662e2f6c5925ae5a05e991f6f2e8f81bcf3d5dfb62ad029bf3d6c32565892ef81acc651a43533dc3a7e9efb2c75922bbdddf28bb29e18b3c2

data/README.md

@@ -179,12 +179,12 @@ tool = OmniAI::Tool.new(
   parameters: OmniAI::Tool::Parameters.new(
     properties: {
       location: OmniAI::Tool::Property.string(description: 'e.g. Toronto'),
-      unit: OmniAI::Tool::Property.string(enum: %w[celcius farenheit]),
+      unit: OmniAI::Tool::Property.string(enum: %w[celcius fahrenheit]),
     },
     required: %i[location]
   )
 )
-client.chat('What is the weather in "London" and "Madrid"?', tools: [tool])
+client.chat('What is the weather in "London" in celcius and "Paris" in fahrenheit?', tools: [tool])
 ```
 
 ### Transcribe

data/lib/omniai/tool/parameters.rb

@@ -24,10 +24,10 @@ module OmniAI
       end
 
       # @return [Hash]
-      def prepare
+      def serialize
         {
           type: @type,
-          properties: @properties.transform_values(&:prepare),
+          properties: @properties.transform_values(&:serialize),
           required: @required,
         }.compact
       end

data/lib/omniai/tool/property.rb

@@ -60,7 +60,7 @@ module OmniAI
       end
 
       # @example
-      #   property.prepare
+      #   property.serialize
       #   # {
       #   #   type: 'string',
       #   #   description: 'The unit (e.g. "fahrenheit" or "celsius").'
@@ -68,7 +68,7 @@ module OmniAI
       #   # }
       #
       # @return [Hash]
-      def prepare
+      def serialize
         {
           type: @type,
           description: @description,

data/lib/omniai/tool.rb

@@ -53,7 +53,10 @@ module OmniAI
     #   #     parameters: {
     #   #       type: 'object',
     #   #       properties: {
-    #   #         n: OmniAI::Tool::Property.integer(description: 'The nth Fibonacci number to calculate')
+    #   #         n: {
+    #   #           description: 'The nth Fibonacci number to calculate')
+    #   #           type: 'integer'
+    #   #         }
     #   #       },
     #   #       required: ['n']
     #   #     }
@@ -71,7 +74,7 @@ module OmniAI
         function: {
           name: @name,
           description: @description,
-          parameters: @parameters&.prepare,
+          parameters: @parameters.is_a?(Tool::Parameters) ? @parameters.serialize : @parameters,
         }.compact,
       }
     end
@@ -83,7 +86,7 @@ module OmniAI
     # @param args [Hash]
     # @return [String]
     def call(args = {})
-      @function.call(**(@parameters ? @parameters.parse(args) : args))
+      @function.call(**(@parameters.is_a?(Tool::Parameters) ? @parameters.parse(args) : args))
     end
   end
 end

data/lib/omniai/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module OmniAI
-  VERSION = '1.8.0'
+  VERSION = '1.8.1'
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: omniai
 version: !ruby/object:Gem::Version
-  version: 1.8.0
+  version: 1.8.1
 platform: ruby
 authors:
 - Kevin Sylvestre
 autorequire: 
 bindir: exe
 cert_chain: []
-date: 2024-08-16 00:00:00.000000000 Z
+date: 2024-08-17 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: event_stream_parser