ox-ai-workers 0.9.7 → 0.9.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: d07816453560dd2bdb34d054ae67f111291b09f587e743aeab80bd7143d5681b
-  data.tar.gz: 43b0ca3a4866147b2f412e02c22e626964ff96d0968a5957b9ba132765370c3f
+  metadata.gz: 9fde2ab33d3d1b89a4b351ff58f6c9f4d85c9d667770d6d258e5ee8215655666
+  data.tar.gz: dbbd0c85cf70948b8c92342e3b57804bf0b62828a5276b29650e66a79ac34742
 SHA512:
-  metadata.gz: c24890557276fa3a358ff0f633a6dbbf01ce8aaf769abef008d8614755f41fae3b4c55d766b32be17c11e6976c93864ef1c95bea9dafd8b0fe456cfab0256d0e
-  data.tar.gz: e6b08fa61b8b3c4a05c2a6d2b9c3f742d9c86a2865923b6448bb5fc659f561f8d499e0953a392567967fb1d3b28471c81245fd68e8cd63c920d86a407248121b
+  metadata.gz: a6b1b2e7494c478266ba2044db4545006a28d2d7be61eeb62309b40eb360f8b5eed41672657c0e5b87c5efb0d2424572b5618df128b093f1e58b4eee9e49ea29
+  data.tar.gz: '00697510a6c17ce5a98b9460b4446fece185a5f618286e18c7ae28907d3c7ddaa17a480648342dd947a73373bb61a7fa34353b967e22449f8acd7812e4ead33e'

data/CHANGELOG.md

@@ -1,4 +1,9 @@
 
+## [0.9.9] - 2025-05-12
+
+- Added `strict` parameter for `define_function`
+- Added `Wolfram` tool
+
 ## [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:
@@ -567,6 +569,26 @@ OxAiWorkers provides several specialized tools to extend functionality:
 
   Provides functions for generating images with customizable parameters like size and quality, with ability to save generated images to disk.
 
+- **Wolfram**: Query the Wolfram Alpha computational knowledge engine
+
+  ```ruby
+  `gem "wolfram-alpha", github: "neonix20b/wolfram-alpha"`
+  ```
+
+  ```ruby
+  OxAiWorkers.configuration.access_token_wolfram = 'YOUR_WOLFRAM_API_KEY'
+  ```
+
+  ```ruby
+  # Initialize with optional parameters
+  wolfram = OxAiWorkers::Tool::Wolfram.new(
+    access_token: 'YOUR_WOLFRAM_API_KEY', # Optional: API key
+    location: 'Berlin' # Optional: Location to use for the query
+  )
+  ```
+
+  This tool enables access to Wolfram Alpha's vast computational intelligence for performing complex mathematical calculations, solving equations, accessing scientific data, and answering knowledge-based queries with precise, authoritative results.
+
 - **Pipeline**: Assistant coordination and communication tool
 
   ```ruby

data/config/locales/en.oxaiworkers.tool.yml

@@ -44,4 +44,9 @@ en:
           prompt: 'Text description of the image to generate on **English language**'
           size: 'Size of the generated image'
           quality: 'Quality of the generated image'
-          file_name: 'File name for saving the generated image in png format. For example: image.png'
+          file_name: 'File name for saving the generated image in png format. For example: image.png'
+      wolfram_alpha:
+        ask:
+          description: 'Wolfram Alpha Tool: Query the Wolfram Alpha computational knowledge engine'
+          prompt: 'The query to send to Wolfram Alpha'
+          location: 'Location information to provide context for the query'

data/config/locales/ru.oxaiworkers.tool.yml

@@ -49,3 +49,8 @@ ru:
           description: 'Инструмент редактирования изображений: Редактирует изображение на основе предоставленного текстового запроса.'
           input_image: 'URL или путь к изображению для редактирования'
           prompt: 'Текстовое описание действий для редактирования изображения'
+      wolfram_alpha:
+        ask:
+          description: 'Инструмент Wolfram Alpha: Запрос к вычислительной базе знаний Wolfram Alpha'
+          prompt: 'Запрос для отправки в Wolfram Alpha'
+          location: 'Информация о местоположении для контекста запроса'

data/lib/ox-ai-workers.rb

@@ -28,6 +28,7 @@ require_relative 'oxaiworkers/tool/database'
 require_relative 'oxaiworkers/tool/file_system'
 require_relative 'oxaiworkers/tool/pipeline'
 require_relative 'oxaiworkers/tool/pixels'
+require_relative 'oxaiworkers/tool/wolfram'
 
 require_relative 'oxaiworkers/assistant/module_base'
 require_relative 'oxaiworkers/assistant/sysop'
@@ -58,7 +59,7 @@ module OxAiWorkers
 
   class Configuration
     attr_accessor :max_tokens, :temperature, :wait_for_complete, :access_token_deepseek, :access_token_openai,
-                  :access_token_stability
+                  :access_token_stability, :access_token_wolfram
 
     def initialize
       @max_tokens = DEFAULT_MAX_TOKEN
@@ -68,6 +69,7 @@ module OxAiWorkers
       @access_token_deepseek = nil
       @access_token_openai = nil
       @access_token_stability = nil
+      @access_token_wolfram = nil
 
       [Array, NilClass, String, Symbol, Hash].each do |c|
         c.send(:include, OxAiWorkers::PresentCompat) unless c.method_defined?(:present?)

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/wolfram.rb

@@ -0,0 +1,52 @@
+require 'fileutils'
+require 'open-uri'
+
+module OxAiWorkers
+  module Tool
+    class Wolfram
+      include OxAiWorkers::ToolDefinition
+      include OxAiWorkers::DependencyHelper
+      include OxAiWorkers::LoadI18n
+
+      attr_accessor :access_token, :location
+
+      def initialize(only: nil, access_token: nil, location: nil)
+        store_locale
+
+        init_white_list_with only
+
+        define_function :ask, description: I18n.t('oxaiworkers.tool.wolfram_alpha.ask.description') do
+          property :prompt, type: 'string', description: I18n.t('oxaiworkers.tool.wolfram_alpha.ask.prompt'),
+                            required: true
+          if location.nil?
+            property :location, type: 'string', description: I18n.t('oxaiworkers.tool.wolfram_alpha.ask.location'),
+                                required: true
+          end
+        end
+
+        @access_token = access_token
+        @location = location
+      end
+
+      def ask(prompt:, location: nil)
+        location ||= @location
+        options = { location: }
+        api_key = @access_token || OxAiWorkers.configuration.access_token_wolfram
+        @wolfram ||= WolframAlpha::Client.new api_key, options.compact
+        response = @wolfram.query prompt
+        input = response['Input'] # Get the input interpretation pod.
+        result = ''
+        result = "Input interpretation:\n#{input.subpods.map(&:plaintext).join("\n")}\n\n" unless input.nil?
+        result += response.pods.map { |p| [p.title, p.subpods.map(&:plaintext).join("\n")].join(":\n") }.join("\n\n")
+        images = []
+        %w[Plots Result Image].each do |title|
+          plot = response.find { |pod| pod.title == title }
+          images += plot.subpods.map { |img| img&.image } unless plot.nil?
+        end
+        images = images.compact.filter { |i| i[:type] != 'Default' }.uniq
+        result += "\n\n" + images.map { |img| "![#{img[:alt]}](\"#{img[:src]}\")" }.join("\n") unless images.empty?
+        result
+      end
+    end
+  end
+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.9'
 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.9
 platform: ruby
 authors:
 - Denis Smolev
@@ -177,6 +177,7 @@ files:
 - lib/oxaiworkers/tool/file_system.rb
 - lib/oxaiworkers/tool/pipeline.rb
 - lib/oxaiworkers/tool/pixels.rb
+- lib/oxaiworkers/tool/wolfram.rb
 - lib/oxaiworkers/tool_definition.rb
 - lib/oxaiworkers/version.rb
 - lib/ruby/ox-ai-workers.rb