Class: Legion::Extensions::Llm::Model::Modalities

Inherits:
Object
  • Object
show all
Defined in:
lib/legion/extensions/llm/model/modalities.rb

Overview

Holds and manages input and output modalities for a language model

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(data) ⇒ Modalities

Returns a new instance of Modalities.



11
12
13
14
 
# File 'lib/legion/extensions/llm/model/modalities.rb', line 11

def initialize(data)
  @input = Array(data[:input]).map(&:to_s)
  @output = Array(data[:output]).map(&:to_s)
end

Instance Attribute Details

#input ⇒ Object (readonly)

Returns the value of attribute input.



9
10
11
 
# File 'lib/legion/extensions/llm/model/modalities.rb', line 9

def input
  @input
end

#output ⇒ Object (readonly)

Returns the value of attribute output.



9
10
11
 
# File 'lib/legion/extensions/llm/model/modalities.rb', line 9

def output
  @output
end

Instance Method Details

#to_h ⇒ Object 



16
17
18
19
20
21
 
# File 'lib/legion/extensions/llm/model/modalities.rb', line 16

def to_h
  {
    input: input,
    output: output
  }
end