gemini_cache 0.0.12 → 0.0.13

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c7e9a54d31ede523084b82ed3d94912bcbbd33ac2f0452bbc0a87d22ddc3d749
-  data.tar.gz: 7ca621fad09fc18f1a87938d2a42b289841ed7c9a6eaadd290869af4cd132613
+  metadata.gz: ebf54e6bff38c64454e4914eb2523b8e7f31afdc8679047a7f6d98cde885fea1
+  data.tar.gz: 2018cfa7beac9143381785b9bbabec1eb4df0f4d9faafd0dfc4b7c7900376511
 SHA512:
-  metadata.gz: 7281d91213fa3b6243e6013e352653b89e15c0be1545ad62c11c97f3259f7afcdb673ffd82d9130c3fd61087ef555844ebe35f684c5ebb0ccd1e9fa0a9a65f7f
-  data.tar.gz: 9648a180a9c48e3ddddc52d3006603b9a8b8f418712474311c578660d8c329ceb4ca5ee23f89ab6262eb4058818e2289910dc58979702662852d50aa3ed2a85f
+  metadata.gz: 79d93c0baad884c33cfef53b53e6e486c0942ce8e68cddd045dfdfcbbfe6ff9e7bec404e37a56778d756af0a86a1c4cb450eaffd04d9892bcdf449a4f0b9c7b2
+  data.tar.gz: b330e9737286d3997eda16126b6bc2c20e906b4825686e1e7f3d1e4530c621db534bc52d25b3899e6ac8ba96c341d6db2a7f7f964bb9b1060e118218a5f08967

data/README.md

@@ -148,6 +148,20 @@ The methods use configurations defined in the `gemini_cache/configuration` modul
 ## Errors
 In case of conflicts or API errors, the module raises a custom exception `GeminiCache::Error`.
 
+## Examples
+
+```ruby
+require 'gemini_cache'
+
+parts = [
+  { text: 'Some text' },
+  GeminiCache.read_remote_file(url: '<remote_file_url>', mime_type: 'application/pdf'),
+  GeminiCache.read_webpage_text(url: '<webpage url>')
+]
+
+cache = GeminiCache.create parts: parts, display_name: 'mycache'
+```	
+
 ## Conclusion
 This documentation covers the main functionalities of the `GeminiCache` module. For more details on specific configurations, refer to the source code or official documentation.
 

data/lib/gemini_cache/api_client.rb

@@ -52,7 +52,12 @@ module GeminiCache
     def handle_response(response)
       return JSON.parse(response.body) if response.status == 200
       
-      error_message = JSON.parse(response.body)['error'] rescue response.body
+      error_message = begin
+        JSON.parse(response.body)['error']
+      rescue
+        response.body
+      end
+
       raise ApiError, "API request failed (#{response.status}): #{error_message}"
     rescue Faraday::Error => e
       raise ApiError, "Network error: #{e.message}"

data/lib/gemini_cache/configuration.rb

@@ -1,21 +1,20 @@
 module GeminiCache
   class Configuration
-    attr_accessor :api_key, :api_base_url, :default_model, :default_ttl
+    attr_accessor :api_key, :api_base_url, :default_model, :default_ttl, :default_timeout
 
     def initialize
       @api_base_url = 'https://generativelanguage.googleapis.com'
-      @default_model = 'gemini-1.5-flash-8b'
+      @default_model = 'gemini-flash-lite-latest'
       @default_ttl = 300
+      @default_timeout = 300
     end
   end
 
-  class << self
-    def configuration
-      @configuration ||= Configuration.new
-    end
+  def self.configuration
+    @configuration ||= Configuration.new
+  end
 
-    def configure
-      yield(configuration)
-    end
+  def self.configure
+    yield configuration
   end
 end

data/lib/gemini_cache/item_extender.rb

@@ -1,11 +1,9 @@
-# frozen_string_literal: true
-
 module ItemExtender
-  GEMINI_API_BASE_URL = 'https://generativelanguage.googleapis.com'
-  DEFAULT_TIMEOUT = 300 # seconds
   ACCURATE_MODE_CONFIG = { temperature: 0, topP: 0, topK: 1 }.freeze
 
-  def delete = GeminiCache.delete(name: self['name'])
+  def delete
+    GeminiCache.delete(name: self['name'])
+  end
 
   def ttl=(new_ttl)
     GeminiCache.update(name: self['name'], content: { ttl: "#{new_ttl}s" }.to_json)
@@ -23,7 +21,7 @@ module ItemExtender
   end
 
   def single_prompt(prompt:, generation_config: :accurate_mode)
-    config = generation_config.eql?(:accurate_mode) ? ACCURATE_MODE_CONFIG : generation_config
+    config = generation_config == :accurate_mode ? ACCURATE_MODE_CONFIG : generation_config
     
     generate_content(
       contents: [{ parts: [{ text: prompt }], role: 'user' }],
@@ -34,12 +32,15 @@ module ItemExtender
   private
 
   def api_client
-    @api_client ||= Faraday.new(
-      url: GEMINI_API_BASE_URL,
-      headers: { 'Content-Type' => 'application/json' }
-    ) do |f|
-      f.options.timeout = DEFAULT_TIMEOUT
-      f.options.open_timeout = DEFAULT_TIMEOUT
+    @api_client ||= begin
+      timeout = GeminiCache.configuration.default_timeout
+      Faraday.new(
+        url: GeminiCache.configuration.api_base_url,
+        headers: { 'Content-Type' => 'application/json' }
+      ) do |f|
+        f.options.timeout = timeout
+        f.options.open_timeout = timeout
+      end
     end
   end
 
@@ -50,8 +51,8 @@ module ItemExtender
   def build_request_body(contents, generation_config)
     {
       cached_content: self['name'],
-      contents: contents,
-      generation_config: generation_config
+      contents:,
+      generation_config:
     }.compact.to_json
   end
 

data/lib/gemini_cache.rb

@@ -11,76 +11,88 @@ require 'gemini_cache/item_extender'
 module GeminiCache
   class Error < StandardError; end
 
-  class << self
-    def parse_html(url:, default_remover: true)
-      doc = Nokogiri::HTML(URI.open(url, "User-Agent" => "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"))
-      %w[script style].each { |element| doc.css(element).each(&:remove) } if default_remover
-      doc
-    end
+  def self.parse_html(url:, default_remover: true)
+    doc = Nokogiri::HTML(URI.open(url, 'User-Agent' => 'Mozilla/5.0'))
+    %w[script style].each { doc.css(it).remove } if default_remover
+    doc
+  end
 
-    def read_local_file(path:, mime_type:)
-      { inline_data: { mime_type:, data: Base64.strict_encode64(File.read(path)) } }
-    end
+  def self.read_local_file(path:, mime_type:)
+    { inline_data: { mime_type:, data: Base64.strict_encode64(File.read(path)) } }
+  end
 
-    def read_remote_file(url:, mime_type:)
-      { inline_data: { mime_type:, data: Base64.strict_encode64(URI.open(url).read) } }
-    end
+  def self.read_remote_file(url:, mime_type:)
+    { inline_data: { mime_type:, data: Base64.strict_encode64(URI.open(url).read) } }
+  end
 
-    def read_webpage_text(url:, default_remover: true) = { text: parse_html(url:, default_remover:).inner_text }
-
-    def create(parts:, display_name:, on_conflict: :raise_error, model: nil, ttl: nil)
-      existing_cache = find_by_display_name(display_name:)
-      
-      if existing_cache
-        case on_conflict
-        when :raise_error
-          raise Error, "Cache with display name '#{display_name}' already exists"
-        when :get_existing
-          return existing_cache
-        end
-      end
-
-      content = {
-        model: "models/#{model || configuration.default_model}",
-        display_name: display_name,
-        contents: [{ parts:, role: 'user' }],
-        ttl: "#{ttl || configuration.default_ttl}s"
-      }
-
-      response = api_client.create_cache(content.to_json)
-      find_by_name(name: response['name'])
-    end
+  def self.read_webpage_text(url:, default_remover: true)
+    { text: parse_html(url:, default_remover:).inner_text }
+  end
 
-    def create_from_text(text:, **options) = create(parts: [{ text: }], **options)
-    def create_from_webpage(url:, **options) = create_from_text(text: read_webpage_text(url:)[:text], **options)
-    def create_from_local_file(path:, mime_type:, **options)
-      file_data = read_local_file(path: path, mime_type: mime_type)
-      create(parts: [file_data], **options)
+  def self.create(parts:, display_name:, on_conflict: :raise_error, model: nil, ttl: nil)
+    existing_cache = find_by_display_name(display_name:)
+    
+    if existing_cache
+      return existing_cache if on_conflict == :get_existing
+      raise Error, "Cache with display name '#{display_name}' already exists" if on_conflict == :raise_error
     end
 
-    def create_from_remote_file(url:, mime_type:, **options)
-      file_data = read_remote_file(url: url, mime_type: mime_type)
-      create(parts: [file_data], **options)
-    end
+    content = {
+      model: "models/#{model || configuration.default_model}",
+      display_name:,
+      contents: [{ parts:, role: 'user' }],
+      ttl: "#{ttl || configuration.default_ttl}s"
+    }
 
-    def list
-      response = api_client.list_caches
-      return [] if response.empty?
+    response = api_client.create_cache(content.to_json)
+    find_by_name(name: response['name'])
+  end
 
-      response['cachedContents'].map { |item| item.extend(ItemExtender) }
-    end
+  def self.create_from_text(text:, **options)
+    create(parts: [{ text: }], **options)
+  end
 
-    def find_by_name(name:) = list.find { |item| item['name'].eql?(name) }
-    def find_by_display_name(display_name:) = list.find { |item| item['displayName'].eql?(display_name) }
+  def self.create_from_webpage(url:, **options)
+    create_from_text(text: read_webpage_text(url:)[:text], **options)
+  end
 
-    def update(name:, content:) = api_client.update_cache(name, content)
+  def self.create_from_local_file(path:, mime_type:, **options)
+    create(parts: [read_local_file(path:, mime_type:)], **options)
+  end
 
-    def delete(name:)
-      api_client.delete_cache(name)
-      true
-    end
+  def self.create_from_remote_file(url:, mime_type:, **options)
+    create(parts: [read_remote_file(url:, mime_type:)], **options)
+  end
+
+  def self.list
+    response = api_client.list_caches
+    return [] if response.empty?
+
+    response['cachedContents'].map { it.extend(ItemExtender) }
+  end
+
+  def self.find_by_name(name:)
+    list.find { it['name'] == name }
+  end
+
+  def self.find_by_display_name(display_name:)
+    list.find { it['displayName'] == display_name }
+  end
 
-    def delete_all() = list.each { |item| item.delete }
+  def self.update(name:, content:)
+    api_client.update_cache(name, content)
+  end
+
+  def self.delete(name:)
+    api_client.delete_cache(name)
+    true
+  end
+
+  def self.delete_all
+    list.each(&:delete)
+  end
+
+  class << self
     alias clear delete_all
 
     private

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: gemini_cache
 version: !ruby/object:Gem::Version
-  version: 0.0.12
+  version: 0.0.13
 platform: ruby
 authors:
 - Gedean Dias
 bindir: bin
 cert_chain: []
-date: 2025-01-06 00:00:00.000000000 Z
+date: 2025-09-29 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: faraday
@@ -80,7 +80,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.6.2
+rubygems_version: 3.7.2
 specification_version: 4
 summary: Ruby Gemini Context Caching
 test_files: []