Class: OmniAI::Tool::Property

Inherits:
Object
  • Object
show all
Defined in:
lib/omniai/tool/property.rb

Overview

A property used for a tool parameter.

Examples:

OmniAI::Tool::Property.array(description: '...', items: ...)
OmniAI::Tool::Property.object(description: '...', properties: { ... }, required: %i[...])
OmniAI::Tool::Property.string(description: '...')
OmniAI::Tool::Property.integer(description: '...')
OmniAI::Tool::Property.number(description: '...')
OmniAI::Tool::Property.boolean(description: '...')

Defined Under Namespace

Modules: Type

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(type:, description: nil, enum: nil) ⇒ OmniAI::Tool::Property

Parameters:

  • type (String) —

    required - the type of the property

  • description (String) (defaults to: nil) —

    optional - a description of the property

  • enum (Array) (defaults to: nil) —

    optional - the possible values of the property



119
120
121
122
123
 
# File 'lib/omniai/tool/property.rb', line 119

def initialize(type:, description: nil, enum: nil)
  @type = type
  @description = description
  @enum = enum
end

Instance Attribute Details

#description ⇒ String? (readonly)

Returns:

  • (String, nil) 


26
27
28
 
# File 'lib/omniai/tool/property.rb', line 26

def description
  @description
end

#enum ⇒ Array<String>? (readonly)

Returns:



29
30
31
 
# File 'lib/omniai/tool/property.rb', line 29

def enum
  @enum
end

#type ⇒ String (readonly)

Returns:

  • (String) 


23
24
25
 
# File 'lib/omniai/tool/property.rb', line 23

def type
  @type
end

Class Method Details

.array(items:, min_items: nil, max_items: nil, description: nil) ⇒ OmniAI::Tool::Array

Examples:

property = OmniAI::Tool::Property.array(
  items: OmniAI::Tool::Property.string(description: 'The name of the person.'),
  description: 'A list of names.'
  min_items: 1,
  max_items: 5,
)

Parameters:

  • items (OmniAI::Tool::Property) —

    required - the items in the array

  • min_items (Integer) (defaults to: nil) —

    optional - the minimum number of items

  • max_items (Integer) (defaults to: nil) —

    optional - the maximum number of items

  • description (String) (defaults to: nil) —

    optional - a description of the array

Returns:



58
59
60
 
# File 'lib/omniai/tool/property.rb', line 58

def self.array(items:, min_items: nil, max_items: nil, description: nil)
  OmniAI::Tool::Array.new(items:, description:, min_items:, max_items:)
end

.boolean(description: nil, enum: nil) ⇒ OmniAI::Tool::Property

Parameters:

  • description (String) (defaults to: nil) —

    optional - a description of the property

  • enum (Array<Boolean>) (defaults to: nil) —

    optional - the possible values of the property

Returns:



86
87
88
 
# File 'lib/omniai/tool/property.rb', line 86

def self.boolean(description: nil, enum: nil)
  new(type: Type::BOOLEAN, description:, enum:)
end

.build(kind, **args) ⇒ OmniAI::Tool::Property

Parameters:

  • kind (Symbol)

Returns:



33
34
35
36
37
38
39
40
41
42
 
# File 'lib/omniai/tool/property.rb', line 33

def self.build(kind, **args)
  case kind
  when :array then array(**args)
  when :object then object(**args)
  when :boolean then boolean(**args)
  when :integer then integer(**args)
  when :string then string(**args)
  when :number then number(**args)
  end
end

.integer(description: nil, enum: nil) ⇒ OmniAI::Tool::Property

Parameters:

  • description (String) (defaults to: nil) —

    optional - a description of the property

  • enum (Array<Integer>) (defaults to: nil) —

    optinoal - the possible values of the property

Returns:



94
95
96
 
# File 'lib/omniai/tool/property.rb', line 94

def self.integer(description: nil, enum: nil)
  new(type: Type::INTEGER, description:, enum:)
end

.number(description: nil, enum: nil) ⇒ OmniAI::Tool::Property

Parameters:

  • description (String) (defaults to: nil) —

    optional - a description of the property

  • enum (Array<Number>) (defaults to: nil) —

    optional - the possible values of the property

Returns:



110
111
112
 
# File 'lib/omniai/tool/property.rb', line 110

def self.number(description: nil, enum: nil)
  new(type: Type::NUMBER, description:, enum:)
end

.object(properties: {}, required: [], description: nil) ⇒ OmniAI::Tool::Array

Examples:

property = OmniAI::Tool::Property.object(
  properties: {
    name: OmniAI::Tool::Property.string(description: 'The name of the person.'),
    age: OmniAI::Tool::Property.integer(description: 'The age of the person.'),
    employed: OmniAI::Tool::Property.boolean(description: 'Is the person employed?'),
  },
  description: 'A person.'
  required: %i[name]
)

Parameters:

  • properties (Hash<String, OmniAI::Tool::Property>) (defaults to: {}) —

    required - the properties of the object

  • required (Array<Symbol>) (defaults to: []) —

    optional - the required properties

  • description (String) (defaults to: nil) —

    optional - a description of the object

Returns:



78
79
80
 
# File 'lib/omniai/tool/property.rb', line 78

def self.object(properties: {}, required: [], description: nil)
  OmniAI::Tool::Object.new(properties:, required:, description:)
end

.string(description: nil, enum: nil) ⇒ OmniAI::Tool::Property

Parameters:

  • description (String) (defaults to: nil) —

    optional - a description of the property

  • enum (Array<String>) (defaults to: nil) —

    optional - the possible values of the property

Returns:



102
103
104
 
# File 'lib/omniai/tool/property.rb', line 102

def self.string(description: nil, enum: nil)
  new(type: Type::STRING, description:, enum:)
end

Instance Method Details

#parse(value) ⇒ String, ...

Examples:

property.parse('123') #=> 123

Returns:

  • (String, Integer, Float, Boolean, Object) 


141
142
143
144
145
146
147
148
 
# File 'lib/omniai/tool/property.rb', line 141

def parse(value)
  case @type
  when Type::INTEGER then Integer(value)
  when Type::STRING then String(value)
  when Type::NUMBER then Float(value)
  else value
  end
end

#serialize ⇒ Hash

Examples:

property.serialize #=> { type: 'string' }

Returns:

  • (Hash) 


129
130
131
132
133
134
135
 
# File 'lib/omniai/tool/property.rb', line 129

def serialize
  {
    type: @type,
    description: @description,
    enum: @enum,
  }.compact
end