Module: Raix::FunctionDispatch

Extended by:
ActiveSupport::Concern
Defined in:
lib/raix/function_dispatch.rb

Overview

Provides declarative function definition for ChatCompletion classes.

Example:

class MeaningOfLife
include Raix::ChatCompletion
include Raix::FunctionDispatch

function :ask_deep_thought do
  wait 236_682_000_000_000
  "The meaning of life is 42"
end

def initialize
  transcript << { user: "What is the meaning of life?" }
  chat_completion
end
end

Instance Method Summary collapse

Instance Method Details

#chat_completion(**chat_completion_args) ⇒ Object 



114
115
116
117
 
# File 'lib/raix/function_dispatch.rb', line 114

def chat_completion(**chat_completion_args)
  self.chat_completion_args = chat_completion_args
  super
end

#stop_tool_calls_and_respond! ⇒ Object

Stops the automatic continuation of chat completions after this function call. Useful when you want to halt processing within a function and force the AI to provide a text response without making additional tool calls.



122
123
124
 
# File 'lib/raix/function_dispatch.rb', line 122

def stop_tool_calls_and_respond!
  @stop_tool_calls_and_respond = true
end

#tools ⇒ Object 



126
127
128
129
130
 
# File 'lib/raix/function_dispatch.rb', line 126

def tools
  return [] unless self.class.functions

  self.class.functions.map { |function| { type: "function", function: } }
end