gemini_cache 0.0.4 → 0.0.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bd9c1ef68b516dfab654ac65c1a211494e21573c772cc244dd13071cd951e097
-  data.tar.gz: caab21e5e9c2dccadc0dc332aeef23fd6a79978dcfdbbd96b3af341eb5809458
+  metadata.gz: 276c3ca26a6767aad15542bae697d58b6f6c546827c668bc922632d391ea0598
+  data.tar.gz: 672b16527768229e7d84819b7d2b2844c41fe57c57e1bf1171414c182e8cc101
 SHA512:
-  metadata.gz: da9d5c70a85fb76adf656a8b53d53123ab480451e2042e6edc91d4a40c935bfd33cfc7430270f68cc56d8edc507b19ed54eaf743c1403c6e3735d8236dde29f8
-  data.tar.gz: 6644a2f79ff8013674136cb533fcf5df679d54987f806f7709d0e0f5de2dd20bcfb1bbb965bccfd3fee83753ad66635bfb0700874bdb5483a422c8553f7fe160
+  metadata.gz: 69700db12b8f6c8e78e3a3577843bc01f6fda112aec173653419c0674dd91baa59aee0a03be37eda62d9f1cf07e8a3892c47f28ec29d7a539d29b0d984988e3e
+  data.tar.gz: 6992ec6e1537b00dc69de4afe0f13457a75758d6327f8c2bafecd9fb868e833d26a499363649e78e08ba9ac680641453fd572aaa52e4075ebac16885897e57a8

data/lib/gemini_cache.rb

@@ -1,12 +1,16 @@
 require 'faraday'
+require 'open-uri'
+require 'nokogiri'
 require 'json'
 
 module GeminiCache
-  def self.create(contents:, display_name:, model: 'gemini-1.5-flash-8b', ttl: 600)
+  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:)
+
     content = {
       model: "models/#{model}",
       display_name:,
-      contents:,
+      contents: [parts:, role: 'user'],
       ttl: "#{ttl}s"
     }.to_json
   
@@ -83,7 +87,10 @@ module GeminiCache
         raise "Erro na requisição: #{e.message}"
       end
       
-      def item.single_prompt(prompt:, generation_config: nil)
+      def item.single_prompt(prompt:, generation_config: :accurate_mode)
+        # accurate_mode: less creative, more accurate
+        generation_config = { temperature: 0, topP: 0, topK: 1 } if generation_config.eql?(:accurate_mode)
+
         generate_content(contents: [{ parts: [{ text: prompt }], role: 'user' }], generation_config:).content
       end
 
@@ -136,4 +143,13 @@ 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,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: gemini_cache
 version: !ruby/object:Gem::Version
-  version: 0.0.4
+  version: 0.0.6
 platform: ruby
 authors:
 - Gedean Dias
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2024-11-10 00:00:00.000000000 Z
+date: 2024-11-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday