gemini_cache 0.0.4 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bd9c1ef68b516dfab654ac65c1a211494e21573c772cc244dd13071cd951e097
-  data.tar.gz: caab21e5e9c2dccadc0dc332aeef23fd6a79978dcfdbbd96b3af341eb5809458
+  metadata.gz: db225c3f8c93563f2bd4f1c8bd4fec3df644609ca9268bb7eeec996f42eaf48b
+  data.tar.gz: bb4f2198799740b227c16621396475f171b89806aae3f9644631651c22dfc049
 SHA512:
-  metadata.gz: da9d5c70a85fb76adf656a8b53d53123ab480451e2042e6edc91d4a40c935bfd33cfc7430270f68cc56d8edc507b19ed54eaf743c1403c6e3735d8236dde29f8
-  data.tar.gz: 6644a2f79ff8013674136cb533fcf5df679d54987f806f7709d0e0f5de2dd20bcfb1bbb965bccfd3fee83753ad66635bfb0700874bdb5483a422c8553f7fe160
+  metadata.gz: d5f925f85d3c1bbe71b1b2c44febdce091ca508db753c26bc543613c81fffdefc2a4636f7bdb7ac06152f25d759bb805d6279e4a74d1bcf9452d18e7d0889f5c
+  data.tar.gz: 228a2728324e5d10210ec3c9691c1ce505cd3b13fa4914ba889bb6348dc24c43ddcf35c6c8821af6c583a3f0893a7d1467ce521bbf8bbec33454cd631e6fc420

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,8 @@ 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) = Nokogiri::HTML(URI.open(url))
 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.5
 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