ox-ai-workers 0.9.7 → 0.9.8

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d07816453560dd2bdb34d054ae67f111291b09f587e743aeab80bd7143d5681b
-  data.tar.gz: 43b0ca3a4866147b2f412e02c22e626964ff96d0968a5957b9ba132765370c3f
+  metadata.gz: cea51225b7b09bfd512c62960fc5e01f3bb6ed848e8efc6d72115ac6765ac254
+  data.tar.gz: 3f355bf99d4a5d3101c3d8b2f8e7f4859509489bb857c50b49d3efba15dc086f
 SHA512:
-  metadata.gz: c24890557276fa3a358ff0f633a6dbbf01ce8aaf769abef008d8614755f41fae3b4c55d766b32be17c11e6976c93864ef1c95bea9dafd8b0fe456cfab0256d0e
-  data.tar.gz: e6b08fa61b8b3c4a05c2a6d2b9c3f742d9c86a2865923b6448bb5fc659f561f8d499e0953a392567967fb1d3b28471c81245fd68e8cd63c920d86a407248121b
+  metadata.gz: b665b7c498647dc8c53fe9bd538f3db34c424fc2277b4226e08e66faa0fbc3f3df5cc8d9d8a65301cca94c0b30b0241140565675486a0f2e70be55cbaf43ceef
+  data.tar.gz: dcf0ef347b4e2017f69e09bc85bc99ffa785a4bf0d6ff565c292a0d8ee50fde675d49851766a25d9c473f67126d41185955365da303e4fe0272ad70e0b9a4b3c

data/CHANGELOG.md

@@ -1,4 +1,8 @@
 
+## [0.9.8] - 2025-05-12
+
+- Added `strict` parameter for `define_function`
+
 ## [0.9.6] - 2025-05-10
 
 - Added `add_file` for `Iterator` (only pdf for now)

data/README.md

@@ -397,6 +397,8 @@ class MyTool
 end
 ```
 
+The `define_function` method accepts an optional `strict` parameter (defaults to `true`) that controls whether additional properties are allowed in the input. When `strict: true` (default), the schema will include `additionalProperties: false`, enforcing that only defined properties can be used.
+
 ### Working with Files and Images
 
 You can easily add files and images to your assistants:

data/lib/oxaiworkers/tool/pipeline.rb

@@ -33,8 +33,10 @@ module OxAiWorkers
         context = context_for(to_id)
         @assistants[to_id].replace_context(context)
         @assistants[to_id].add_task message
-        @assistants[to_id].add_task "#{I18n.t('oxaiworkers.tool.pipeline.send_message.result')}: #{result}"
-        @assistants[to_id].add_task "#{I18n.t('oxaiworkers.tool.pipeline.send_message.example')}: #{example}"
+        with_locale do
+          @assistants[to_id].add_task "#{I18n.t('oxaiworkers.tool.pipeline.send_message.result')}: #{result}"
+          @assistants[to_id].add_task "#{I18n.t('oxaiworkers.tool.pipeline.send_message.example')}: #{example}"
+        end
         @assistants[to_id].execute
         nil
       end

data/lib/oxaiworkers/tool_definition.rb

@@ -53,10 +53,10 @@ module OxAiWorkers
     # @param method_name [Symbol] Name of the method to define
     # @param description [String] Description of the function
     # @yield Block that defines the parameters for the function
-    def define_function(method_name, description:, &)
+    def define_function(method_name, description:, strict: true, &)
       return unless @white_list.nil? || @white_list == method_name || @white_list.include?(method_name)
 
-      function_schemas.add_function(method_name:, description:, &)
+      function_schemas.add_function(method_name:, description:, strict:, &)
     end
 
     # Returns the FunctionSchemas instance for this tool
@@ -105,11 +105,11 @@ module OxAiWorkers
       # @param description [String] Description of the function
       # @yield Block that defines the parameters for the function
       # @raise [ArgumentError] If a block is defined and no parameters are specified for the function
-      def add_function(method_name:, description:, &)
+      def add_function(method_name:, description:, strict:, &)
         name = function_name(method_name)
 
         if block_given?
-          parameters = ParameterBuilder.new(parent_type: 'object').build(&)
+          parameters = ParameterBuilder.new(parent_type: 'object', strict:).build(&)
 
           if parameters[:properties].empty?
             raise ArgumentError,
@@ -117,9 +117,12 @@ module OxAiWorkers
           end
         end
 
+        function_params = { name:, description:, parameters: }
+        function_params[:strict] = true if strict
+
         @schemas[method_name] = {
           type: 'function',
-          function: { name:, description:, parameters:, strict: !parameters.nil? }.compact
+          function: function_params.compact
         }
       end
 
@@ -163,9 +166,10 @@ module OxAiWorkers
     class ParameterBuilder
       VALID_TYPES = %w[object array string number integer boolean].freeze
 
-      def initialize(parent_type:)
+      def initialize(parent_type:, strict: true)
         @schema = parent_type == 'object' ? { type: 'object', properties: {}, required: [] } : {}
         @parent_type = parent_type
+        @strict = strict
       end
 
       # Builds the parameter schema
@@ -192,7 +196,7 @@ module OxAiWorkers
         prop = { type:, description:, enum: }.compact
 
         if block_given?
-          nested_schema = ParameterBuilder.new(parent_type: type).build(&)
+          nested_schema = ParameterBuilder.new(parent_type: type, strict: @strict).build(&)
 
           case type
           when 'object'
@@ -214,7 +218,7 @@ module OxAiWorkers
         if @parent_type == 'object'
           @schema[:properties][name] = prop
           @schema[:required] << name.to_s if required
-          @schema[:additionalProperties] = false
+          @schema[:additionalProperties] = false if @strict
         else
           @schema = prop
         end

data/lib/oxaiworkers/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module OxAiWorkers
-  VERSION = '0.9.7'
+  VERSION = '0.9.8'
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ox-ai-workers
 version: !ruby/object:Gem::Version
-  version: 0.9.7
+  version: 0.9.8
 platform: ruby
 authors:
 - Denis Smolev