gemini_cache 0.0.3 → 0.0.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: '04685d49a72d97ba748888a2c1a90ffae49dd52d5bdea582e5cc93c5278ee05d'
-  data.tar.gz: 1fbf73ef5d37c9eeffee36e2667faf3846a326a217c4ae3dcc9bd6f781bec0ed
+  metadata.gz: db225c3f8c93563f2bd4f1c8bd4fec3df644609ca9268bb7eeec996f42eaf48b
+  data.tar.gz: bb4f2198799740b227c16621396475f171b89806aae3f9644631651c22dfc049
 SHA512:
-  metadata.gz: 9397ae965759c13a8b4f23cd26e17fbd66807cf6ef09ec299b5ff6695fdbc32f74c9858574f727da489d7039d761be2d900220c39d8d3d9a6cfb22360eb6bcdb
-  data.tar.gz: efa123d59d17f71e2193b82de461b0648d5a887377175331993b9bc48ef17872b0987f9a9622311c40062a6639f3c72e792d69b5a2345568f4985601eb0de624
+  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
   
@@ -51,7 +55,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 +64,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 +87,12 @@ 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: :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
 
       item
     end
@@ -130,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.3
+  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