raix 0.4.8 → 0.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 205dfe2efa97dd596884ca2063e33e0c76da34ae052f98e18e8c75c423752b3a
-  data.tar.gz: 24bfe7d9b6d20a701d70a96366937bf96c4999ea26469c8cc645e4ae874ce5c8
+  metadata.gz: 4bb905086f308c10375b54337b9cb4effe9387ed7e7a6a0dd84875176984e29c
+  data.tar.gz: dbc3b0d67aa91d8dc9b8ba9504b84d52617529c2a8e69fd0d21f3ad4e59df31d
 SHA512:
-  metadata.gz: ea57058bb1e8614decd24d295a641b4e2e6d90ac2d5b2d744b3531561f13304e4e8e41a679c9e0f80b16fdcf0477c8c61262ed9b4fbb451d507db24469540037
-  data.tar.gz: f6c255abef3c7d2e7440075d4114f564050ffcf95e431dc5301ea6a516cd89f1baf3d96346e86614192d85330e4f234673b89eb9d19930043f0882c8ee2451eb
+  metadata.gz: cbbaebe7bd8d18d2cd82b41b7a4bcf468eda1483485b2a707ff980d51068dea1e9ef5e20510dc7dd46bf1982fc7db14054399919842a28930cb12e37fa016022
+  data.tar.gz: 61037ede25456a15053e96dc1bae11125a9d3e0d5dc08e255087c9b69a2c4d096fc2e95d27f600a955bd93b421d8ed703b7c7b0df8de35812b82dbf9f0818858

data/.rubocop.yml

@@ -30,3 +30,6 @@ Metrics/CyclomaticComplexity:
 
 Metrics/PerceivedComplexity:
   Enabled: false
+
+Metrics/ParameterLists:
+  Enabled: false

data/CHANGELOG.md

@@ -1,5 +1,9 @@
 ## [Unreleased]
 
+## [0.6.0] - 2024-11-12
+- adds `save_response` option to `chat_completion` to control transcript updates
+- fixes potential race conditions in transcript handling
+
 ## [0.1.0] - 2024-04-03
 
 - Initial release, placeholder gem

data/Gemfile.lock

@@ -1,7 +1,7 @@
 PATH
   remote: .
   specs:
-    raix (0.4.8)
+    raix (0.6)
       activesupport (>= 6.0)
       open_router (~> 0.2)
       ruby-openai (~> 7.0)

data/README.md

@@ -101,6 +101,8 @@ RSpec.describe WhatIsTheWeather do
 end
 ```
 
+Note that for security reasons, dispatching functions only works with functions implemented using `Raix::FunctionDispatch#function` or directly on the class.
+
 #### Multiple Tool Calls
 
 Some AI models (like GPT-4) can make multiple tool calls in a single response. When this happens, Raix will automatically handle all the function calls sequentially and return an array of their results. Here's an example:

data/lib/raix/chat_completion.rb

@@ -44,7 +44,7 @@ module Raix
     # @option params [Boolean] :openai (false) Whether to use OpenAI's API instead of OpenRouter's.
     # @option params [Boolean] :raw (false) Whether to return the raw response or dig the text content.
     # @return [String|Hash] The completed chat response.
-    def chat_completion(params: {}, loop: false, json: false, raw: false, openai: false)
+    def chat_completion(params: {}, loop: false, json: false, raw: false, openai: false, save_response: true)
       # set params to default values if not provided
       params[:cache_at] ||= cache_at.presence
       params[:frequency_penalty] ||= frequency_penalty.presence
@@ -84,7 +84,10 @@ module Raix
       self.model ||= Raix.configuration.model
 
       adapter = MessageAdapters::Base.new(self)
-      messages = transcript.flatten.compact.map { |msg| adapter.transform(msg) }
+
+      # duplicate the transcript to avoid race conditions in situations where
+      # chat_completion is called multiple times in parallel
+      messages = transcript.flatten.compact.map { |msg| adapter.transform(msg) }.dup
       raise "Can't complete an empty transcript" if messages.blank?
 
       begin
@@ -110,12 +113,18 @@ module Raix
           return tool_calls.map do |tool_call|
             # dispatch the called function
             arguments = JSON.parse(tool_call["function"]["arguments"].presence || "{}")
-            send(tool_call["function"]["name"], arguments.with_indifferent_access)
+            function_name = tool_call["function"]["name"]
+            raise "Unauthorized function call: #{function_name}" unless self.class.functions.map { |f| f[:name].to_sym }.include?(function_name.to_sym)
+
+            send(function_name, arguments.with_indifferent_access)
           end
         end
 
         response.tap do |res|
           content = res.dig("choices", 0, "message", "content")
+
+          transcript << { assistant: content } if save_response
+
           if json
             content = content.squish
             return JSON.parse(content)
@@ -174,7 +183,7 @@ module Raix
       params[:stream] ||= stream.presence
       params[:stream_options] = { include_usage: true } if params[:stream]
 
-      params.delete(:temperature) if model == "o1-preview"
+      params.delete(:temperature) if model.start_with?("o")
 
       Raix.configuration.openai_client.chat(parameters: params.compact.merge(model:, messages:))
     end

data/lib/raix/version.rb

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

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: raix
 version: !ruby/object:Gem::Version
-  version: 0.4.8
+  version: '0.6'
 platform: ruby
 authors:
 - Obie Fernandez
 autorequire:
 bindir: exe
 cert_chain: []
-date: 2024-12-06 00:00:00.000000000 Z
+date: 2025-04-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport
@@ -102,7 +102,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.4.10
+rubygems_version: 3.5.21
 signing_key:
 specification_version: 4
 summary: Ruby AI eXtensions