Module: LLM::Tool::Param

Included in:
LLM::Tool
Defined in:
lib/llm/tool/param.rb

Overview

The LLM::Tool::Param module extends the LLM::Tool class with a “param” method that can define a parameter for simple types. For complex types, use LLM::Tool.params instead.

Examples:

class Greeter < LLM::Tool
  name "greeter"
  description "Greets the user"
  parameter :name, String, "The user's name"
  required %i[name]

  def call(name:)
    puts "Hello, #{name}!"
  end
end

Defined Under Namespace

Modules: Utils

Instance Method Summary collapse

Instance Method Details

#param(name, type, description, options = {}) ⇒ Object Also known as: parameter

Parameters:

  • name (Symbol) —

    The name of a parameter

  • type (LLM::Schema::Leaf, Class) —

    The parameter type (eg String)

  • description (String) —

    The description of a property

  • options (Hash) (defaults to: {}) —

    A hash of options for the parameter

Options Hash (options):

  • :required (Boolean) —

    Whether or not the parameter is required

  • :default (Object) —

    The default value for a given property

  • :enum (Array<String>) —

    One or more possible values for a param



37
38
39
40
41
42
43
44
 
# File 'lib/llm/tool/param.rb', line 37

def param(name, type, description, options = {})
  lock do
    function.params do |schema|
      resolved = Utils.resolve(schema, type)
      schema.object(name => Utils.setup(resolved, description, options))
    end
  end
end

#required(names) ⇒ LLM::Schema::Object

Mark existing parameters as required.

Parameters:

  • names (Array<Symbol,String>)

Returns:



51
52
53
54
55
56
57
 
# File 'lib/llm/tool/param.rb', line 51

def required(names)
  lock do
    function.params.tap do |schema|
      [*names].each { Utils.fetch(schema.properties, _1).required }
    end
  end
end