gemini_cache 0.0.2 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 16f6934eba6497750a226af8f1ec7e769ccb40da707fb066d459472461c4f8d3
-  data.tar.gz: 13e80103252e7f59af3c313af528a2f4d300172d09e93d856beaa4b755069495
+  metadata.gz: bd9c1ef68b516dfab654ac65c1a211494e21573c772cc244dd13071cd951e097
+  data.tar.gz: caab21e5e9c2dccadc0dc332aeef23fd6a79978dcfdbbd96b3af341eb5809458
 SHA512:
-  metadata.gz: f5719b1e1bd3ffa26d09bfd013a2bd1d1a1b80481c20aef57c93329e949c0ed113389c2fd926f344f62bc71a59bcf155742ddc7e2f0358c543c216fdc7117d95
-  data.tar.gz: 6aaad0d3a575fb0219ab3568c8e346a0d345eab9ab8a83a1685b93548274db1c502ccb176200c860169a8aa8439955748e46679483eb934407117dde74cefa5a
+  metadata.gz: da9d5c70a85fb76adf656a8b53d53123ab480451e2042e6edc91d4a40c935bfd33cfc7430270f68cc56d8edc507b19ed54eaf743c1403c6e3735d8236dde29f8
+  data.tar.gz: 6644a2f79ff8013674136cb533fcf5df679d54987f806f7709d0e0f5de2dd20bcfb1bbb965bccfd3fee83753ad66635bfb0700874bdb5483a422c8553f7fe160

data/lib/gemini_cache.rb

@@ -51,7 +51,7 @@ module GeminiCache
       def item.delete = GeminiCache.delete(name: self['name'])
       def item.set_ttl(ttl = 120) = GeminiCache.update(name: self['name'], content: { ttl: "#{ttl}s" })
 
-      def item.generate_content(contents:)
+      def item.generate_content(contents:, generation_config: nil)
         conn = Faraday.new(
           url: 'https://generativelanguage.googleapis.com',
           headers: { 'Content-Type' => 'application/json' }
@@ -60,12 +60,16 @@ module GeminiCache
           f.options.open_timeout = 300   # timeout em segundos para abrir a conexão
         end
 
+        body = {
+          cached_content: self['name'],
+          contents:
+        }
+
+        body[:generation_config] = generation_config if !generation_config.nil?
+
         response = conn.post("/v1beta/models/#{self['model'].split('/').last}:generateContent") do |req|
           req.params['key'] = ENV.fetch('GEMINI_API_KEY')
-          req.body = {
-            cached_content: self['name'],
-            contents:
-          }.to_json
+          req.body = body.to_json
         end
         
         if response.status == 200
@@ -79,7 +83,9 @@ module GeminiCache
         raise "Erro na requisição: #{e.message}"
       end
       
-      def item.single_prompt(prompt:) = generate_content(contents: [{ parts: [{ text: prompt }], role: 'user' }]).content
+      def item.single_prompt(prompt:, generation_config: nil)
+        generate_content(contents: [{ parts: [{ text: prompt }], role: 'user' }], generation_config:).content
+      end
 
       item
     end
@@ -122,4 +128,12 @@ module GeminiCache
   rescue Faraday::Error => e
     raise "Erro na requisição: #{e.message}"
   end
+
+  def self.delete_all
+    GeminiCache.list.each { |item| item.delete }
+  end
+
+  class << self
+    alias clear delete_all
+  end
 end

metadata

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