Class: LLM::Gemini::Files

Inherits:
Object
  • Object
show all
Defined in:
lib/llm/providers/gemini/files.rb

Overview

The LLM::Gemini::Files class provides a files object for interacting with Gemini’s Files API. The files API allows a client to reference media files in prompts where they can be referenced by their URL.

The files API is intended to preserve bandwidth and latency, especially for large files but it can be helpful for smaller files as well because it does not require the client to include a file in the prompt over and over again (which could be the case in a multi-turn conversation).

Examples:

#!/usr/bin/env ruby
require "llm"

llm = LLM.gemini(ENV["KEY"])
bot = LLM::Chat.new(llm).lazy
file = llm.files.create file: LLM::File("/audio/haiku.mp3")
bot.chat(file)
bot.chat("Describe the audio file I sent to you")
bot.chat("The audio file is the first message I sent to you.")
bot.messages.select(&:assistant?).each { print "[#{_1.role}]", _1.content, "\n" }
#!/usr/bin/env ruby
require "llm"

llm = LLM.gemini(ENV["KEY"])
bot = LLM::Chat.new(llm).lazy
file = llm.files.create file: LLM::File("/audio/haiku.mp3")
bot.chat(["Describe the audio file I sent to you", file])
bot.messages.select(&:assistant?).each { print "[#{_1.role}]", _1.content, "\n" }

Instance Method Summary collapse

Constructor Details

#initialize(provider) ⇒ LLM::Gemini::Files

Returns a new Files object

Parameters:



41
42
43
 
# File 'lib/llm/providers/gemini/files.rb', line 41

def initialize(provider)
  @provider = provider
end

Instance Method Details

#all(**params) ⇒ LLM::Response::FileList

List all files

Examples:

llm = LLM.gemini(ENV["KEY"])
res = llm.files.all
res.each do |file|
  print "name: ", file.name, "\n"
end

Parameters:

  • params (Hash) —

    Other parameters (see Gemini docs)

Returns:

Raises:

See Also:



57
58
59
60
61
62
63
64
65
66
67
68
 
# File 'lib/llm/providers/gemini/files.rb', line 57

def all(**params)
  query = URI.encode_www_form(params.merge!(key: secret))
  req = Net::HTTP::Get.new("/v1beta/files?#{query}", headers)
  res = request(http, req)
  LLM::Response::FileList.new(res).tap { |filelist|
    files = filelist.body["files"]&.map do |file|
      file = file.transform_keys { snakecase(_1) }
      OpenStruct.from_hash(file)
    end || []
    filelist.files = files
  }
end

#create(file:, **params) ⇒ LLM::Response::File

Create a file

Examples:

llm = LLM.gemini(ENV["KEY"])
res = llm.files.create file: LLM::File("/audio/haiku.mp3"),

Parameters:

  • file (File) —

    The file

  • params (Hash) —

    Other parameters (see Gemini docs)

Returns:

Raises:

See Also:



80
81
82
83
84
85
86
87
88
89
90
 
# File 'lib/llm/providers/gemini/files.rb', line 80

def create(file:, **params)
  req = Net::HTTP::Post.new(request_upload_url(file:), {})
  req["content-length"] = file.bytesize
  req["X-Goog-Upload-Offset"] = 0
  req["X-Goog-Upload-Command"] = "upload, finalize"
  file.with_io do |io|
    req.body_stream = io
    res = request(http, req)
    LLM::Response::File.new(res)
  end
end

#get(file:, **params) ⇒ LLM::Response::File

Get a file

Examples:

llm = LLM.gemini(ENV["KEY"])
res = llm.files.get(file: "files/1234567890")
print "name: ", res.name, "\n"

Parameters:

  • file (#name, String) —

    The file to get

  • params (Hash) —

    Other parameters (see Gemini docs)

Returns:

Raises:

See Also:



103
104
105
106
107
108
109
 
# File 'lib/llm/providers/gemini/files.rb', line 103

def get(file:, **params)
  file_id = file.respond_to?(:name) ? file.name : file.to_s
  query = URI.encode_www_form(params.merge!(key: secret))
  req = Net::HTTP::Get.new("/v1beta/#{file_id}?#{query}", headers)
  res = request(http, req)
  LLM::Response::File.new(res)
end

#delete(file:, **params) ⇒ LLM::Response::File

Delete a file

Examples:

llm = LLM.gemini(ENV["KEY"])
res = llm.files.delete(file: "files/1234567890")

Parameters:

  • file (#name, String) —

    The file to delete

  • params (Hash) —

    Other parameters (see Gemini docs)

Returns:

Raises:

See Also:



121
122
123
124
125
126
 
# File 'lib/llm/providers/gemini/files.rb', line 121

def delete(file:, **params)
  file_id = file.respond_to?(:name) ? file.name : file.to_s
  query = URI.encode_www_form(params.merge!(key: secret))
  req = Net::HTTP::Delete.new("/v1beta/#{file_id}?#{query}", headers)
  request(http, req)
end

#download ⇒ Object

Raises:

  • (NotImplementedError) —

    This method is not implemented by Gemini



131
132
133
 
# File 'lib/llm/providers/gemini/files.rb', line 131

def download
  raise NotImplementedError
end