Class: OmniAI::Chat::Choice

Inherits:
Object
  • Object
show all
Defined in:
lib/omniai/chat/choice.rb

Overview

A choice wraps a message and index returned by an LLM. The default is to generate a single choice. Some LLMs support generating multiple choices at once (e.g. giving you multiple options to choose from).

Instance Attribute Summary collapse

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(message:, index: DEFAULT_INDEX) ⇒ Choice

Returns a new instance of Choice.

Parameters:

  • message (Message)
  • index (Integer) (defaults to: DEFAULT_INDEX) 


21
22
23
24
 
# File 'lib/omniai/chat/choice.rb', line 21

def initialize(message:, index: DEFAULT_INDEX)
  @message = message
  @index = index
end

Instance Attribute Details

#index ⇒ Integer

Returns:

  • (Integer) 


13
14
15
 
# File 'lib/omniai/chat/choice.rb', line 13

def index
  @index
end

#message ⇒ Message

Returns:



17
18
19
 
# File 'lib/omniai/chat/choice.rb', line 17

def message
  @message
end

Class Method Details

.deserialize(data, context: nil) ⇒ Choice

Parameters:

Returns:



35
36
37
38
39
40
41
42
43
 
# File 'lib/omniai/chat/choice.rb', line 35

def self.deserialize(data, context: nil)
  deserialize = context&.deserializer(:choice)
  return deserialize.call(data, context:) if deserialize

  index = data["index"] || DEFAULT_INDEX
  message = Message.deserialize(data["message"] || data["delta"], context:)

  new(message:, index:)
end

Instance Method Details

#content ⇒ Array<Content>, String

Returns:



63
64
65
 
# File 'lib/omniai/chat/choice.rb', line 63

def content
  message.content
end

#delta ⇒ Message

Returns:



58
59
60
 
# File 'lib/omniai/chat/choice.rb', line 58

def delta
  message
end

#inspect ⇒ String

Returns:

  • (String) 


27
28
29
 
# File 'lib/omniai/chat/choice.rb', line 27

def inspect
  "#<#{self.class.name} index=#{@index} message=#{@message.inspect}>"
end

#serialize(context: nil) ⇒ Hash

Parameters:

Returns:

  • (Hash) 


47
48
49
50
51
52
53
54
55
 
# File 'lib/omniai/chat/choice.rb', line 47

def serialize(context: nil)
  serialize = context&.serializer(:choice)
  return serialize.call(self, context:) if serialize

  {
    index:,
    message: message.serialize(context:),
  }
end

#tool_call_list ⇒ Array<ToolCall>?

Returns:



68
69
70
 
# File 'lib/omniai/chat/choice.rb', line 68

def tool_call_list
  message.tool_call_list
end