Class: ActiveAgent::Providers::Mock::Messages::MessagesType

Inherits:
ActiveModel::Type::Value
  • Object
show all
Defined in:
lib/active_agent/providers/mock/messages/_types.rb

Overview

Type for Messages array

Instance Method Summary collapse

Constructor Details

#initialize ⇒ MessagesType

Returns a new instance of MessagesType.



14
15
16
17
 
# File 'lib/active_agent/providers/mock/messages/_types.rb', line 14

def initialize
  super
  @message_type = MessageType.new
end

Instance Method Details

#cast(value) ⇒ Object 



19
20
21
22
23
24
25
26
27
28
 
# File 'lib/active_agent/providers/mock/messages/_types.rb', line 19

def cast(value)
  case value
  when Array
    value.map { |v| @message_type.cast(v) }
  when nil
    nil
  else
    raise ArgumentError, "Cannot cast #{value.class} to Messages array"
  end
end

#deserialize(value) ⇒ Object 



51
52
53
 
# File 'lib/active_agent/providers/mock/messages/_types.rb', line 51

def deserialize(value)
  cast(value)
end

#serialize(value) ⇒ Object 



30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
 
# File 'lib/active_agent/providers/mock/messages/_types.rb', line 30

def serialize(value)
  case value
  when Array
    grouped = []

    value.each do |message|
      if grouped.empty? || grouped.last.role != message.role
        grouped << message.deep_dup
      else
        grouped.last.content += message.content.deep_dup
      end
    end

    grouped.map { |v| @message_type.serialize(v) }
  when nil
    nil
  else
    raise ArgumentError, "Cannot serialize #{value.class}"
  end
end