gemini_cache 0.0.6 → 0.0.7

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 276c3ca26a6767aad15542bae697d58b6f6c546827c668bc922632d391ea0598
-  data.tar.gz: 672b16527768229e7d84819b7d2b2844c41fe57c57e1bf1171414c182e8cc101
+  metadata.gz: ba751abef4a7f8f6f6c856d782c9e43dfcc27cfb64e6d69a6b6b095522fba338
+  data.tar.gz: 5dbef856507a550b7b739a5d972991ab822f36ca0aafef4bd5576df3b920c9c6
 SHA512:
-  metadata.gz: 69700db12b8f6c8e78e3a3577843bc01f6fda112aec173653419c0674dd91baa59aee0a03be37eda62d9f1cf07e8a3892c47f28ec29d7a539d29b0d984988e3e
-  data.tar.gz: 6992ec6e1537b00dc69de4afe0f13457a75758d6327f8c2bafecd9fb868e833d26a499363649e78e08ba9ac680641453fd572aaa52e4075ebac16885897e57a8
+  metadata.gz: 032fc3e2f78e791b55d82a63c15094eb325f5778ac4232dfb088e38b99bbd2402f9381ed79c12e03051768a2d488719f6be786cc815dd927927320942f2e8800
+  data.tar.gz: 31944aa6d6cbe592fd98909b465a1521b2049772d9436d09ad1663c9088473e7aae954ad7e224e4393670ca7d4cb7791295adff9be40a0b1536ecb5ab3b3f3cb

data/lib/gemini_cache.rb

@@ -4,6 +4,15 @@ require 'nokogiri'
 require 'json'
 
 module GeminiCache
+  def self.read_local_file(file_path) = Base64.strict_encode64(File.read(file_path))
+  def self.read_remote_file(file_url) = Base64.strict_encode64(URI.open(file_url).read)
+  def self.read_html(url, default_remover: true)
+    doc = Nokogiri::HTML(URI.open(url))
+    %w[script style].each { |element| doc.css(element).each(&:remove) } if default_remover
+
+    doc
+  end
+
   def self.create(parts:, display_name:, model: 'gemini-1.5-flash-8b', ttl: 600)
     raise "Cache name already exist: '#{display_name}'" if GeminiCache.get(display_name:)
 
@@ -24,7 +33,7 @@ module GeminiCache
       req.body = content
     end
   
-    return JSON.parse(response.body) if response.status == 200
+    return get(name: JSON.parse(response.body)['name']) if response.status == 200
   
     raise "Erro ao criar cache: #{response.status} - #{response.body}"
   rescue Faraday::Error => e
@@ -39,6 +48,26 @@ module GeminiCache
     return GeminiCache.list.find { |item| item['displayName'].eql? display_name } if !display_name.nil?
   end
 
+  def self.create_from_text(text:, display_name:, model: 'gemini-1.5-flash-8b', ttl: 600)
+    GeminiCache.create(parts: [{ text: }], display_name:, model:, ttl:)
+  end
+
+  def self.create_from_webpage(url:, display_name:, model: 'gemini-1.5-flash-8b', ttl: 600)
+    GeminiCache.create(parts: [{ text: GeminiCache.read_html(url).inner_text }], display_name:, model:, ttl:)
+  end
+
+  def self.create_from_local_file(file_path:, mime_type:, display_name:, model: 'gemini-1.5-flash-8b', ttl: 600)
+    parts = [{ inline_data: { mime_type:, data: GeminiCache.read_local_file(file_path) } }]
+
+    GeminiCache.create(parts:, display_name:, model:, ttl:)
+  end
+
+  def self.create_from_remote_file(file_url:, mime_type:, display_name:, model: 'gemini-1.5-flash-8b', ttl: 600)
+    parts = [{ inline_data: { mime_type:, data: GeminiCache.read_remote_file(file_url) } }]
+
+    GeminiCache.create(parts:, display_name:, model:, ttl:)
+  end
+  
   def self.list
     conn = Faraday.new(
       url: 'https://generativelanguage.googleapis.com',
@@ -143,13 +172,4 @@ module GeminiCache
   class << self
     alias clear delete_all
   end
-
-  def self.read_local_file(file_path) = Base64.strict_encode64(File.read(file_path))
-  def self.read_remote_file(file_url) = Base64.strict_encode64(URI.open(file_url).read)
-
-  def self.read_nokogiri_html(url, default_remover: true)
-    doc = Nokogiri::HTML(URI.open(url))
-    %w[script style].each { |element| doc.css(element).each(&:remove) } if default_remover
-    doc
-  end
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: gemini_cache
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.0.7
 platform: ruby
 authors:
 - Gedean Dias