Class: RubyLLM::Attachment

Inherits:
Object
  • Object
show all
Defined in:
lib/ruby_llm/attachment.rb

Overview

A class representing a file attachment.

Instance Attribute Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(source, filename: nil) ⇒ Attachment

Returns a new instance of Attachment.



11
12
13
14
15
16
17
 
# File 'lib/ruby_llm/attachment.rb', line 11

def initialize(source, filename: nil)
  @source = source
  @source = source_type_cast
  @filename = filename || source_filename

  determine_mime_type
end

Instance Attribute Details

#filename ⇒ Object (readonly)

Returns the value of attribute filename.



9
10
11
 
# File 'lib/ruby_llm/attachment.rb', line 9

def filename
  @filename
end

#mime_type ⇒ Object (readonly)

Returns the value of attribute mime_type.



9
10
11
 
# File 'lib/ruby_llm/attachment.rb', line 9

def mime_type
  @mime_type
end

#source ⇒ Object (readonly)

Returns the value of attribute source.



9
10
11
 
# File 'lib/ruby_llm/attachment.rb', line 9

def source
  @source
end

Instance Method Details

#active_storage? ⇒ Boolean

Returns:

  • (Boolean) 


31
32
33
34
35
36
37
 
# File 'lib/ruby_llm/attachment.rb', line 31

def active_storage?
  return false unless defined?(ActiveStorage)

  @source.is_a?(ActiveStorage::Blob) ||
    @source.is_a?(ActiveStorage::Attached::One) ||
    @source.is_a?(ActiveStorage::Attached::Many)
end

#audio? ⇒ Boolean

Returns:

  • (Boolean) 


97
98
99
 
# File 'lib/ruby_llm/attachment.rb', line 97

def audio?
  RubyLLM::MimeType.audio? mime_type
end

#content ⇒ Object 



39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
 
# File 'lib/ruby_llm/attachment.rb', line 39

def content
  return @content if defined?(@content) && !@content.nil?

  if url?
    fetch_content
  elsif path?
    load_content_from_path
  elsif active_storage?
    load_content_from_active_storage
  elsif io_like?
    load_content_from_io
  else
    RubyLLM.logger.warn "Source is neither a URL, path, ActiveStorage, nor IO-like: #{@source.class}"
    nil
  end

  @content
end

#encoded ⇒ Object 



58
59
60
 
# File 'lib/ruby_llm/attachment.rb', line 58

def encoded
  Base64.strict_encode64(content)
end

#for_llm ⇒ Object 



70
71
72
73
74
75
76
77
 
# File 'lib/ruby_llm/attachment.rb', line 70

def for_llm
  case type
  when :text
    "<file name='#{filename}' mime_type='#{mime_type}'>#{content}</file>"
  else
    "data:#{mime_type};base64,#{encoded}"
  end
end

#format ⇒ Object 



101
102
103
104
105
106
107
108
109
110
 
# File 'lib/ruby_llm/attachment.rb', line 101

def format
  case mime_type
  when 'audio/mpeg'
    'mp3'
  when 'audio/wav', 'audio/wave', 'audio/x-wav'
    'wav'
  else
    mime_type.split('/').last
  end
end

#image? ⇒ Boolean

Returns:

  • (Boolean) 


89
90
91
 
# File 'lib/ruby_llm/attachment.rb', line 89

def image?
  RubyLLM::MimeType.image? mime_type
end

#io_like? ⇒ Boolean

Returns:

  • (Boolean) 


27
28
29
 
# File 'lib/ruby_llm/attachment.rb', line 27

def io_like?
  @source.respond_to?(:read) && !path? && !active_storage?
end

#path? ⇒ Boolean

Returns:

  • (Boolean) 


23
24
25
 
# File 'lib/ruby_llm/attachment.rb', line 23

def path?
  @source.is_a?(Pathname) || (@source.is_a?(String) && !url?)
end

#pdf? ⇒ Boolean

Returns:

  • (Boolean) 


112
113
114
 
# File 'lib/ruby_llm/attachment.rb', line 112

def pdf?
  RubyLLM::MimeType.pdf? mime_type
end

#save(path) ⇒ Object 



62
63
64
65
66
67
68
 
# File 'lib/ruby_llm/attachment.rb', line 62

def save(path)
  return unless io_like?

  File.open(path, 'w') do |f|
    f.puts(@source.read)
  end
end

#text? ⇒ Boolean

Returns:

  • (Boolean) 


116
117
118
 
# File 'lib/ruby_llm/attachment.rb', line 116

def text?
  RubyLLM::MimeType.text? mime_type
end

#to_h ⇒ Object 



120
121
122
 
# File 'lib/ruby_llm/attachment.rb', line 120

def to_h
  { type: type, source: @source }
end

#type ⇒ Object 



79
80
81
82
83
84
85
86
87
 
# File 'lib/ruby_llm/attachment.rb', line 79

def type
  return :image if image?
  return :video if video?
  return :audio if audio?
  return :pdf if pdf?
  return :text if text?

  :unknown
end

#url? ⇒ Boolean

Returns:

  • (Boolean) 


19
20
21
 
# File 'lib/ruby_llm/attachment.rb', line 19

def url?
  @source.is_a?(URI) || (@source.is_a?(String) && @source.match?(%r{^https?://}))
end

#video? ⇒ Boolean

Returns:

  • (Boolean) 


93
94
95
 
# File 'lib/ruby_llm/attachment.rb', line 93

def video?
  RubyLLM::MimeType.video? mime_type
end