raix 0.8.0 → 0.8.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '08737974243ed41d4db2406405a4d081e31972cf173cfc3b2859cb97eabf0848'
-  data.tar.gz: 0d78927f5a35fad586f300bff4d86ce773d7bb7cd9f6787320e1cff6297f1aed
+  metadata.gz: 2de11b289a7c245c49e865735e1abc67ff671f02a9f23fb89d262d3ff83e6157
+  data.tar.gz: f8cd493f6f018ddd6b981f944786635b6800ca9beac4480e0a4c3c08bb7b8392
 SHA512:
-  metadata.gz: c0eec6ef99a37cd8ba69dfec62c8be70c9a8f6ed0be78b24ed184bde364c78f5189ccac71e3196f56af4c85390844b044c30be1b69172fe4e0f3c584903ea7c0
-  data.tar.gz: 90f084213b11dd737b55ed89fd21e75cf23b4c0ef5ab0daca8652f974b9fdb0f40a68ee07ea2abcd0b6af715ae3edc684643034b7e7e3455f7d05fae9ee1e473
+  metadata.gz: 4d26f6ffad99abb7eaf6c8ef072393d75625e4762206d0d1f94c811e9d44747fa2e9048d5a8951484ded2861ecd5ebac69ea39bf6163432a411e2f9e427b0533
+  data.tar.gz: 893d062ab1f23b6c6a47dc715a7c98a3886dbe092645642e74bdb97ad80b2b4f637305b95acc6e6a8984cfe6defcdf020410233f5f818cba16bbbd0c5f49b5ed

data/CHANGELOG.md

@@ -1,4 +1,7 @@
-## [0.8.0] - 2024-05-08
+## [0.8.1] - 2025-04-24
+Added ability to filter tool functions (or disable completely) when calling `chat_completion`. Thanks to @parruda for the contribution.
+
+## [0.8.0] - 2025-04-23
 ### Added
 * **MCP integration (Experimental)** — new `Raix::MCP` concern and `mcp` DSL for declaring remote MCP servers.
   * Automatically fetches `tools/list`, registers remote tools as OpenAI‑compatible function schemas, and defines proxy methods that forward `tools/call`.

data/Gemfile.lock

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

data/README.md

@@ -134,6 +134,41 @@ end
 
 Note that for security reasons, dispatching functions only works with functions implemented using `Raix::FunctionDispatch#function` or directly on the class.
 
+#### Tool Filtering
+
+You can control which tools are available to the AI on a given chat completion request using the `tools` parameter in the `chat_completion` method:
+
+```ruby
+class WeatherAndTime
+  include Raix::ChatCompletion
+  include Raix::FunctionDispatch
+
+  function :check_weather, "Check the weather for a location", location: { type: "string" } do |arguments|
+    "The weather in #{arguments[:location]} is sunny"
+  end
+
+  function :get_time, "Get the current time" do |_arguments|
+    "The time is 12:00 PM"
+  end
+end
+
+weather = WeatherAndTime.new
+
+# Don't pass any tools to the LLM
+weather.chat_completion(tools: false)
+
+# Only pass specific tools to the LLM
+weather.chat_completion(tools: [:check_weather])
+
+# Pass all declared tools (default behavior)
+weather.chat_completion
+```
+
+The `tools` parameter accepts three types of values:
+- `false`: No tools are passed to the LLM
+- An array of symbols: Only the specified tools are passed (raises `Raix::UndeclaredToolError` if any tool is not declared)
+- Not provided: All declared tools are passed (default behavior)
+
 #### 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.

data/lib/raix/chat_completion.rb

@@ -9,6 +9,8 @@ require "openai"
 require_relative "message_adapters/base"
 
 module Raix
+  class UndeclaredToolError < StandardError; end
+
   # The `ChatCompletion`` module is a Rails concern that provides a way to interact
   # 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
@@ -44,8 +46,9 @@ 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.
     # @option params [Array] :messages (nil) An array of messages to use instead of the transcript.
+    # @option tools [Array|false] :tools (nil) Tools to pass to the LLM. If false, no tools are passed. If an array, only declared tools in the array are passed.
     # @return [String|Hash] The completed chat response.
-    def chat_completion(params: {}, loop: false, json: false, raw: false, openai: false, save_response: true, messages: nil)
+    def chat_completion(params: {}, loop: false, json: false, raw: false, openai: false, save_response: true, messages: nil, tools: nil)
       # set params to default values if not provided
       params[:cache_at] ||= cache_at.presence
       params[:frequency_penalty] ||= frequency_penalty.presence
@@ -63,7 +66,13 @@ module Raix
       params[:stop] ||= stop.presence
       params[:temperature] ||= temperature.presence || Raix.configuration.temperature
       params[:tool_choice] ||= tool_choice.presence
-      params[:tools] ||= tools.presence
+      params[:tools] = if tools == false
+                         nil
+                       elsif tools.is_a?(Array)
+                         filtered_tools(tools)
+                       else
+                         self.tools.presence
+                       end
       params[:top_a] ||= top_a.presence
       params[:top_k] ||= top_k.presence
       params[:top_logprobs] ||= top_logprobs.presence
@@ -182,6 +191,18 @@ module Raix
 
     private
 
+    def filtered_tools(tool_names)
+      return nil if tool_names.blank?
+
+      requested_tools = tool_names.map(&:to_sym)
+      available_tool_names = tools.map { |tool| tool.dig(:function, :name).to_sym }
+
+      undeclared_tools = requested_tools - available_tool_names
+      raise UndeclaredToolError, "Undeclared tools: #{undeclared_tools.join(", ")}" if undeclared_tools.any?
+
+      tools.select { |tool| requested_tools.include?(tool.dig(:function, :name).to_sym) }
+    end
+
     def openai_request(params:, model:, messages:)
       if params[:prediction]
         params.delete(:max_completion_tokens)

data/lib/raix/function_dispatch.rb

@@ -94,10 +94,7 @@ module Raix
     end
 
     def chat_completion(**chat_completion_args)
-      raise "No functions defined" if self.class.functions.blank?
-
       self.chat_completion_args = chat_completion_args
-
       super
     end
 
@@ -109,6 +106,8 @@ module Raix
     end
 
     def tools
+      return [] unless self.class.functions
+
       self.class.functions.map { |function| { type: "function", function: } }
     end
   end

data/lib/raix/version.rb

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

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: raix
 version: !ruby/object:Gem::Version
-  version: 0.8.0
+  version: 0.8.1
 platform: ruby
 authors:
 - Obie Fernandez
 bindir: exe
 cert_chain: []
-date: 2025-04-23 00:00:00.000000000 Z
+date: 2025-04-25 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: activesupport