gemini_cache 0.0.6 → 0.0.9

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 276c3ca26a6767aad15542bae697d58b6f6c546827c668bc922632d391ea0598
-  data.tar.gz: 672b16527768229e7d84819b7d2b2844c41fe57c57e1bf1171414c182e8cc101
+  metadata.gz: ae4847ce3abf808c2779601462af28a47989301593355a8a39977a5df495f322
+  data.tar.gz: 71e72d0e2750d6b530d407f97a7490c65b6acbf342ad10803916b4b24d418b71
 SHA512:
-  metadata.gz: 69700db12b8f6c8e78e3a3577843bc01f6fda112aec173653419c0674dd91baa59aee0a03be37eda62d9f1cf07e8a3892c47f28ec29d7a539d29b0d984988e3e
-  data.tar.gz: 6992ec6e1537b00dc69de4afe0f13457a75758d6327f8c2bafecd9fb868e833d26a499363649e78e08ba9ac680641453fd572aaa52e4075ebac16885897e57a8
+  metadata.gz: 8ab29a93e9555626f6d1eecb189351b5d62b28805e0e90e0ba9c29cb9e5bcf8698f72dadb0621ffbf196503d186a2c341b45c387bc25f940d055bb110455ff78
+  data.tar.gz: 50a917b60662ecd5bc1cef90c0a633febfcfbe294c297525d397f60863dfaa29977e43c0743c9f717452fa7d45ce5fe2da8f11f3816d0f65fa5e06f6632675ff

data/lib/gemini_cache/item_extender.rb

@@ -0,0 +1,46 @@
+module ItemExtender
+  def delete = GeminiCache.delete(name: self['name'])
+
+  def ttl=(new_ttl)
+    GeminiCache.update(name: self['name'], content: { ttl: "#{new_ttl}s" })
+    GeminiCache.get_by_name(name: self['name'])
+  end
+
+  def generate_content(contents:, generation_config: nil)
+    conn = Faraday.new(
+      url: 'https://generativelanguage.googleapis.com',
+      headers: { 'Content-Type' => 'application/json' }
+    ) do |f|
+      f.options.timeout = 300        # timeout em segundos para a requisição completa
+      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 = body.to_json
+    end
+    
+    if response.status == 200
+      resp = JSON.parse(response.body)
+      def resp.content = dig('candidates', 0, 'content', 'parts', 0, 'text')
+      return resp
+    end
+
+    raise "Erro ao gerar conteúdo: #{response.body}"
+  rescue Faraday::Error => e
+    raise "Erro na requisição: #{e.message}"
+  end
+    
+  def 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      
+end

data/lib/gemini_cache.rb

@@ -2,10 +2,22 @@ require 'faraday'
 require 'open-uri'
 require 'nokogiri'
 require 'json'
+require 'base64'
+
+require 'gemini_cache/item_extender'
 
 module GeminiCache
-  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:)
+  def self.read_local_file(path:, mime_type:) = { inline_data: { mime_type:, data: Base64.strict_encode64(File.read(path)) } } 
+  def self.read_remote_file(url:, mime_type:) = { inline_data: { mime_type:, data: Base64.strict_encode64(URI.open(url).read) } } 
+  def self.read_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
+
+  def self.create(parts:, display_name:, model: 'gemini-1.5-flash-8b', ttl: 300)
+    raise "Cache name already exist: '#{display_name}'" if GeminiCache.get_by_display_name(display_name:)
 
     content = {
       model: "models/#{model}",
@@ -24,21 +36,32 @@ module GeminiCache
       req.body = content
     end
   
-    return JSON.parse(response.body) if response.status == 200
+    return get_by_name(name: JSON.parse(response.body)['name']) if response.status == 200
   
     raise "Erro ao criar cache: #{response.status} - #{response.body}"
   rescue Faraday::Error => e
     raise "Erro na requisição: #{e.message}"
   end
 
-  def self.get(name: nil, display_name: nil)
-    raise 'Nome do cache ou display name é obrigatório' if name.nil? && display_name.nil?
-    raise 'Nome do cache e display name não podem ser informados juntos' if !name.nil? && !display_name.nil?
-    
-    return GeminiCache.list.find { |item| item['name'].eql? name } if !name.nil?
-    return GeminiCache.list.find { |item| item['displayName'].eql? display_name } if !display_name.nil?
+  def self.create_from_text(text:, display_name:, model: 'gemini-1.5-flash-8b', ttl: 300)
+    GeminiCache.create(parts: [{ text: }], display_name:, model:, ttl:)
   end
 
+  def self.create_from_webpage(url:, display_name:, model: 'gemini-1.5-flash-8b', ttl: 300)
+    create_from_text(text: GeminiCache.read_html(url:).inner_text, display_name:, model:, ttl:)
+  end
+
+  def self.create_from_local_file(path:, mime_type:, display_name:, model: 'gemini-1.5-flash-8b', ttl: 300)
+    GeminiCache.create(parts: GeminiCache.read_local_file(path:, mime_type:), display_name:, model:, ttl:)
+  end
+
+  def self.create_from_remote_file(url:, mime_type:, display_name:, model: 'gemini-1.5-flash-8b', ttl: 300)
+    GeminiCache.create(parts: GeminiCache.read_remote_file(url:, mime_type:), display_name:, model:, ttl:)
+  end
+
+  def self.get_by_name(name: nil) = GeminiCache.list.find { |item| item['name'].eql? name }
+  def self.get_by_display_name(display_name: nil) = GeminiCache.list.find { |item| item['displayName'].eql? display_name }
+
   def self.list
     conn = Faraday.new(
       url: 'https://generativelanguage.googleapis.com',
@@ -51,51 +74,7 @@ module GeminiCache
     
     return [] if JSON.parse(response.body).empty?
 
-    JSON.parse(response.body)['cachedContents'].map do |item|
-      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:, generation_config: nil)
-        conn = Faraday.new(
-          url: 'https://generativelanguage.googleapis.com',
-          headers: { 'Content-Type' => 'application/json' }
-        ) do |f|
-          f.options.timeout = 300        # timeout em segundos para a requisição completa
-          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 = body.to_json
-        end
-        
-        if response.status == 200
-          resp = JSON.parse(response.body)
-          def resp.content = dig('candidates', 0, 'content', 'parts', 0, 'text')
-          return resp
-        end
-
-        raise "Erro ao gerar conteúdo: #{response.body}"
-      rescue Faraday::Error => e
-        raise "Erro na requisição: #{e.message}"
-      end
-      
-      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
+    JSON.parse(response.body)['cachedContents'].map { |item| item.extend(ItemExtender) }
 
   rescue Faraday::Error => e
     raise "Erro na requisição: #{e.message}"
@@ -143,13 +122,4 @@ 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,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: gemini_cache
 version: !ruby/object:Gem::Version
-  version: 0.0.6
+  version: 0.0.9
 platform: ruby
 authors:
 - Gedean Dias
@@ -24,6 +24,34 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '2'
+- !ruby/object:Gem::Dependency
+  name: base64
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.2.0
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: 0.2.0
+- !ruby/object:Gem::Dependency
+  name: nokogiri
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '1'
 description: Ruby's Gemini Context Caching wrapper
 email: gedean.dias@gmail.com
 executables: []
@@ -31,6 +59,7 @@ extensions: []
 extra_rdoc_files: []
 files:
 - lib/gemini_cache.rb
+- lib/gemini_cache/item_extender.rb
 homepage: https://github.com/gedean/gemini_cache
 licenses:
 - MIT