gemini_cache 0.0.3 → 0.0.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '04685d49a72d97ba748888a2c1a90ffae49dd52d5bdea582e5cc93c5278ee05d'
-  data.tar.gz: 1fbf73ef5d37c9eeffee36e2667faf3846a326a217c4ae3dcc9bd6f781bec0ed
+  metadata.gz: bd9c1ef68b516dfab654ac65c1a211494e21573c772cc244dd13071cd951e097
+  data.tar.gz: caab21e5e9c2dccadc0dc332aeef23fd6a79978dcfdbbd96b3af341eb5809458
 SHA512:
-  metadata.gz: 9397ae965759c13a8b4f23cd26e17fbd66807cf6ef09ec299b5ff6695fdbc32f74c9858574f727da489d7039d761be2d900220c39d8d3d9a6cfb22360eb6bcdb
-  data.tar.gz: efa123d59d17f71e2193b82de461b0648d5a887377175331993b9bc48ef17872b0987f9a9622311c40062a6639f3c72e792d69b5a2345568f4985601eb0de624
+  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

metadata

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