raix 0.8.2 → 0.8.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 04e086a8f8f010507fef64ba8f2a1fa7c4508b34d85b39d8d045088410ba041c
-  data.tar.gz: b6ddc5948d4d71a967321cfe664a6fb2ad6f38833c7067e0915216ce7fc64646
+  metadata.gz: 77197d6de1d77373260029d955fd0110990b7c562a279d887130ee4f95909377
+  data.tar.gz: d811d21316bba7c92797dff98c044871ec4382998b77090f29348217156698df
 SHA512:
-  metadata.gz: 1558de827080ea2fb1c0ab0a953ae4d4cb1a9f8ad1aaae0690942f78bfde31275739857948b064721aa83b0f9b963c91c4bcb8dc5be460283e88ee2294e10e64
-  data.tar.gz: 001c63730739473375f0bff5c7e4edb78e4247000825799cff5cd98f9ae56bf6f2a5c436f1334b03e05de3257ef081498e2c456567a70e65417c19b70ab8918b
+  metadata.gz: 9a513c9c82ef57ba4beb9681a02111a5207c9418b6f0b26015e9ad299d889b8a65644d594c11246ff7c70a5354419fc3932a8a776072757e615b8afbfc48ca98
+  data.tar.gz: ab0743ed3d99efd20886053b2d7648fc6dec7b551d48c5f86fee9689441dfbda0280e0c693766fa8f03c944f7dbfc459c8301f2b3bb969de47e3f4bbfbf23d0b

data/CHANGELOG.md

@@ -1,3 +1,6 @@
+## [0.8.3] - 2025-04-30
+- Adds optional ActiveSupport Cache parameter to `dispatch_tool_function` for caching tool calls
+
 ## [0.8.2] - 2025-04-29
 - Extracts function call dispatch into a public `dispatch_tool_function` that can be overridden in subclasses
 - Uses `public_send` instead of `send` for better security and explicitness

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    raix (0.8.2)
+    raix (0.8.4)
       activesupport (>= 6.0)
       faraday-retry (~> 2.0)
       open_router (~> 0.2)

data/README.md

@@ -228,6 +228,38 @@ class CustomDispatchExample
 end
 ```
 
+#### Function Call Caching
+
+You can use ActiveSupport's Cache to cache function call results, which can be particularly useful for expensive operations or external API calls that don't need to be repeated frequently.
+
+```ruby
+class CachedFunctionExample
+  include Raix::ChatCompletion
+  include Raix::FunctionDispatch
+
+  function :expensive_operation do |arguments|
+    "Result of expensive operation with #{arguments}"
+  end
+
+  # Override dispatch_tool_function to enable caching for all functions
+  def dispatch_tool_function(function_name, arguments)
+    # Pass the cache to the superclass implementation
+    super(function_name, arguments, cache: Rails.cache)
+  end
+end
+```
+
+The caching mechanism works by:
+1. Passing the cache object through `dispatch_tool_function` to the function implementation
+2. Using the function name and arguments as cache keys
+3. Automatically fetching from cache when available or executing the function when not cached
+
+This is particularly useful for:
+- Expensive database operations
+- External API calls
+- Resource-intensive computations
+- Functions with deterministic outputs for the same inputs
+
 #### Manually Stopping a Loop
 
 To loop AI components that don't interact with end users, at least one function block should invoke `stop_looping!` whenever you're ready to stop processing.

data/lib/raix/chat_completion.rb

@@ -140,7 +140,7 @@ module Raix
           content = res.dig("choices", 0, "message", "content")
 
           transcript << { assistant: content } if save_response
-          content = content.squish
+          content = content.strip
 
           if json
             # Make automatic JSON parsing available to non-OpenAI providers that don't support the response_format parameter
@@ -194,9 +194,10 @@ module Raix
     #
     # @param function_name [String] The name of the function to call
     # @param arguments [Hash] The arguments to pass to the function
+    # @param cache [ActiveSupport::Cache] Optional cache object
     # @return [Object] The result of the function call
-    def dispatch_tool_function(function_name, arguments)
-      public_send(function_name, arguments)
+    def dispatch_tool_function(function_name, arguments, cache: nil)
+      public_send(function_name, arguments, cache)
     end
 
     private

data/lib/raix/function_dispatch.rb

@@ -54,35 +54,41 @@ module Raix
           end
         end
 
-        define_method(name) do |arguments|
+        define_method(name) do |arguments, cache|
           id = SecureRandom.uuid[0, 23]
-          instance_exec(arguments, &block).tap do |content|
-            # add in one operation to prevent race condition and potential wrong
-            # interleaving of tool calls in multi-threaded environments
-            transcript << [
-              {
-                role: "assistant",
-                content: nil,
-                tool_calls: [
-                  {
-                    id:,
-                    type: "function",
-                    function: {
-                      name:,
-                      arguments: arguments.to_json
-                    }
+
+          content = if cache.present?
+                      cache.fetch([name, arguments]) do
+                        instance_exec(arguments, &block)
+                      end
+                    else
+                      instance_exec(arguments, &block)
+                    end
+
+          # add in one operation to prevent race condition and potential wrong
+          # interleaving of tool calls in multi-threaded environments
+          transcript << [
+            {
+              role: "assistant",
+              content: nil,
+              tool_calls: [
+                {
+                  id:,
+                  type: "function",
+                  function: {
+                    name:,
+                    arguments: arguments.to_json
                   }
-                ]
-              },
-              {
-                role: "tool",
-                tool_call_id: id,
-                name:,
-                content: content.to_s
-              }
-            ]
-            # TODO: add on_error handler as optional parameter to function
-          end
+                }
+              ]
+            },
+            {
+              role: "tool",
+              tool_call_id: id,
+              name:,
+              content: content.to_s
+            }
+          ]
 
           chat_completion(**chat_completion_args) if loop
         end

data/lib/raix/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Raix
-  VERSION = "0.8.2"
+  VERSION = "0.8.4"
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: raix
 version: !ruby/object:Gem::Version
-  version: 0.8.2
+  version: 0.8.4
 platform: ruby
 authors:
 - Obie Fernandez
 bindir: exe
 cert_chain: []
-date: 2025-04-30 00:00:00.000000000 Z
+date: 2025-05-07 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport