raix 0.4.3 → 0.4.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 65adb56d006baa2b25193eaa5f87f52b24366876139d1e023c9031e7707f62ea
-  data.tar.gz: 545da8c28e699f87e53547b53d07fb934727948945f45998f3af5d34752a2854
+  metadata.gz: 50d0de4d7ec7fdd83776e539dbed2cc73ba3097c96752050eb768163dd5f510a
+  data.tar.gz: c82955632f789a0683e30553ecc8aef8b2f96aa4bc003e0b2e12953e6698adf9
 SHA512:
-  metadata.gz: 71a14254199a7e0eeb8195b842fe212516091743ab4799f349caadfeffa654eede58dc066813e4d1360c24b2384406f934086d9b167b45db70a618470c94df84
-  data.tar.gz: 3d5a8033107bc8037862d11cd88fb3cd7a67a6884b4f7d2d8756244ac676e938094a6580bd55275b353c0b12389f292cbe958138d8d13f2d1001c5befb81662d
+  metadata.gz: 48977e0c2265105f22da2c1508a1aced37337ab1a914c7297bca4995cc94d6839708a51edb9f63222e4531bc0dccce305448ad70f78a858f30d7f581ae289114
+  data.tar.gz: c52f4078787b2570a6a11eb92e0a0f6172a09e083219d0b8562349fee4efc14012674e25591ef2839e41fd2675bfc834361bf1a64d3b74614854ce944b6cf8ab

data/lib/raix/chat_completion.rb

@@ -12,10 +12,22 @@ module Raix
   # with the OpenRouter Chat Completion API via its client. The module includes a few
   # methods that allow you to build a transcript of messages and then send them to
   # the API for completion. The API will return a response that you can use however
-  # you see fit. If the response includes a function call, the module will dispatch
-  # the function call and return the result. Which implies that function calls need
-  # to be defined on the class that includes this module. (Note: You should probably
-  # use the `FunctionDispatch` module to define functions instead of doing it manually.)
+  # you see fit.
+  #
+  # If the response includes a function call, the module will dispatch the function
+  # call and return the result. Which implies that function calls need to be defined
+  # on the class that includes this module. The `FunctionDispatch` module provides a
+  # Rails-like DSL for declaring and implementing tool functions at the top of your
+  # class instead of having to manually implement them as instance methods. The
+  # primary benefit of using the `FunctionDispatch` module is that it handles
+  # adding the function call results to the ongoing conversation transcript for you.
+  # It also triggers a new chat completion automatically if you've set the `loop`
+  # option to `true`, which is useful for implementing conversational chatbots that
+  # include tool calls.
+  #
+  # Note that some AI models can make more than a single tool function call in a
+  # single response. When that happens, the module will dispatch all of the function
+  # calls sequentially and return an array of results.
   module ChatCompletion
     extend ActiveSupport::Concern
 
@@ -92,13 +104,13 @@ module Raix
         # TODO: add a standardized callback hook for usage events
         # broadcast(:usage_event, usage_subject, self.class.name.to_s, response, premium?)
 
-        # TODO: handle parallel tool calls
-        if (function = response.dig("choices", 0, "message", "tool_calls", 0, "function"))
-          @current_function = function["name"]
-          # dispatch the called function
-          arguments = JSON.parse(function["arguments"].presence || "{}")
-          arguments[:bot_message] = bot_message if respond_to?(:bot_message)
-          return send(function["name"], arguments.with_indifferent_access)
+        tool_calls = response.dig("choices", 0, "message", "tool_calls") || []
+        if tool_calls.any?
+          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)
+          end
         end
 
         response.tap do |res|

data/lib/raix/version.rb

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

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: raix
 version: !ruby/object:Gem::Version
-  version: 0.4.3
+  version: 0.4.4
 platform: ruby
 authors:
 - Obie Fernandez