Class: LLM::Buffer

Inherits:
Object
  • Object
show all
Includes:
Enumerable
Defined in:
lib/llm/buffer.rb

Overview

LLM::Buffer provides an Enumerable object that tracks messages in a conversation thread.

Instance Method Summary collapse

Constructor Details

#initialize(provider) ⇒ LLM::Buffer

Parameters:



13
14
15
16
 
# File 'lib/llm/buffer.rb', line 13

def initialize(provider)
  @provider = provider
  @messages = []
end

Instance Method Details

#<<(item) ⇒ void Also known as: push

This method returns an undefined value.

Parameters:



76
77
78
79
 
# File 'lib/llm/buffer.rb', line 76

def <<(item)
  @messages << item
  self
end

#[](index) ⇒ LLM::Message?

Returns a message, or nil

Parameters:

  • index (Integer, Range) —

    The message index

Returns:



87
88
89
 
# File 'lib/llm/buffer.rb', line 87

def [](index)
  @messages[index]
end

#concat(ary) ⇒ Object

Append an array

Parameters:



22
23
24
 
# File 'lib/llm/buffer.rb', line 22

def concat(ary)
  @messages.concat(ary)
end

#each {|LLM::Message| ... } ⇒ void

This method returns an undefined value.

Yields:

  • (LLM::Message) —

    Yields each message in the conversation thread



40
41
42
43
44
45
46
 
# File 'lib/llm/buffer.rb', line 40

def each(...)
  if block_given?
    @messages.each { yield(_1) }
  else
    enum_for(:each, ...)
  end
end

#empty? ⇒ Boolean

Returns true when the buffer is empty

Returns:

  • (Boolean) 


114
115
116
 
# File 'lib/llm/buffer.rb', line 114

def empty?
  @messages.empty?
end

#find ⇒ LLM::Message?

Find a message (in descending order)

Returns:



51
52
53
 
# File 'lib/llm/buffer.rb', line 51

def find(...)
  reverse_each.find(...)
end

#inspect ⇒ String

Returns:

  • (String) 


99
100
101
102
 
# File 'lib/llm/buffer.rb', line 99

def inspect
  "#<#{self.class.name}:0x#{object_id.to_s(16)} " \
  "message_count=#{@messages.size}>"
end

#last(n = nil) ⇒ LLM::Message, ...

Returns the last message(s) in the buffer

Parameters:

  • n (Integer, nil) (defaults to: nil) —

    The number of messages to return

Returns:



68
69
70
 
# File 'lib/llm/buffer.rb', line 68

def last(n = nil)
  n.nil? ? @messages.last : @messages.last(n)
end

#replace(messages) ⇒ LLM::Buffer

Replace the tracked messages

Parameters:

  • messages (Array<LLM::Message>) —

    The replacement messages

Returns:



31
32
33
34
 
# File 'lib/llm/buffer.rb', line 31

def replace(messages)
  @messages.replace(messages)
  self
end

#rindex {|LLM::Message| ... } ⇒ Integer?

Returns the index of the last message matching the given block.

Yields:

Returns:

  • (Integer, nil) 


59
60
61
 
# File 'lib/llm/buffer.rb', line 59

def rindex(...)
  @messages.rindex(...)
end

#size ⇒ Integer

Returns the number of messages in the buffer

Returns:

  • (Integer) —

    Returns the number of messages in the buffer



107
108
109
 
# File 'lib/llm/buffer.rb', line 107

def size
  @messages.size
end

#to_json ⇒ String

Returns:

  • (String) 


93
94
95
 
# File 'lib/llm/buffer.rb', line 93

def to_json(...)
  LLM.json.dump(@messages, ...)
end