gemini_cache 0.0.1 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 7bc6c689b1bad80c2f7865816137cd8cbadd3a3c112938d7a0f8982820f54065
-  data.tar.gz: c18e33fe4733104d6945cd1e72255066e2e49e339cc18873cfad3276721e7c9a
+  metadata.gz: '04685d49a72d97ba748888a2c1a90ffae49dd52d5bdea582e5cc93c5278ee05d'
+  data.tar.gz: 1fbf73ef5d37c9eeffee36e2667faf3846a326a217c4ae3dcc9bd6f781bec0ed
 SHA512:
-  metadata.gz: 6f8659b09c15ab666ca8e858389eb39a4e2efabb9368a98685bce34bd45049359cf5a1c0c8e9080ea5bcf5d05f113df17d784f768154d720c9a87f135f0b4de5
-  data.tar.gz: eb06c97493baf8a5cf61d6020fbb47094e9b831ec257afae63148ba5127867a9f8c5ba8b3241fe548e6727766afe3e3a8f9137d0def687c1cfa13f9638837b43
+  metadata.gz: 9397ae965759c13a8b4f23cd26e17fbd66807cf6ef09ec299b5ff6695fdbc32f74c9858574f727da489d7039d761be2d900220c39d8d3d9a6cfb22360eb6bcdb
+  data.tar.gz: efa123d59d17f71e2193b82de461b0648d5a887377175331993b9bc48ef17872b0987f9a9622311c40062a6639f3c72e792d69b5a2345568f4985601eb0de624

data/lib/gemini_cache.rb

@@ -31,8 +31,8 @@ module GeminiCache
     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::Cache.list.find { |item| item['name'].eql? name } if !name.nil?
-    return GeminiCache::Cache.list.find { |item| item['displayName'].eql? display_name } if !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?
   end
 
   def self.list
@@ -48,8 +48,8 @@ module GeminiCache
     return [] if JSON.parse(response.body).empty?
 
     JSON.parse(response.body)['cachedContents'].map do |item|
-      def item.delete = GeminiCache::Cache.delete(name: self['name'])
-      def item.set_ttl(ttl = 120) = GeminiCache::Cache.update(name: self['name'], content: { ttl: "#{ttl}s" })
+      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:)
         conn = Faraday.new(
@@ -122,4 +122,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.1
+  version: 0.0.3
 platform: ruby
 authors:
 - Gedean Dias
@@ -31,10 +31,6 @@ extensions: []
 extra_rdoc_files: []
 files:
 - lib/gemini_cache.rb
-- lib/geminiext/cache.rb
-- lib/geminiext/messages.rb.disabled
-- lib/geminiext/model.rb.disabled
-- lib/geminiext/response_extender.rb.disabled
 homepage: https://github.com/gedean/gemini_cache
 licenses:
 - MIT

data/lib/geminiext/cache.rb

@@ -1,127 +0,0 @@
-require 'faraday'
-require 'json'
-
-module GeminiExt
-  module Cache
-    def self.create(contents:, display_name:, model: 'gemini-1.5-flash-8b', ttl: 600)
-      content = {
-        model: "models/#{model}",
-        display_name:,
-        contents:,
-        ttl: "#{ttl}s"
-      }.to_json
-    
-      conn = Faraday.new(
-        url: 'https://generativelanguage.googleapis.com',
-        headers: { 'Content-Type' => 'application/json' }
-      )
-    
-      response = conn.post('/v1beta/cachedContents') do |req|
-        req.params['key'] = ENV.fetch('GEMINI_API_KEY')
-        req.body = content
-      end
-    
-      return JSON.parse(response.body) 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 GeminiExt::Cache.list.find { |item| item['name'].eql? name } if !name.nil?
-      return GeminiExt::Cache.list.find { |item| item['displayName'].eql? display_name } if !display_name.nil?
-    end
-
-    def self.list
-      conn = Faraday.new(
-        url: 'https://generativelanguage.googleapis.com',
-        headers: { 'Content-Type' => 'application/json' }
-      )
-
-      response = conn.get("/v1beta/cachedContents") do |req|
-        req.params['key'] = ENV.fetch('GEMINI_API_KEY')
-      end
-      
-      return [] if JSON.parse(response.body).empty?
-
-      JSON.parse(response.body)['cachedContents'].map do |item|
-        def item.delete = GeminiExt::Cache.delete(name: self['name'])
-        def item.set_ttl(ttl = 120) = GeminiExt::Cache.update(name: self['name'], content: { ttl: "#{ttl}s" })
-
-        def item.generate_content(contents:)
-          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
-
-          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
-          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: ) = generate_content(contents: [{ parts: [{ text: prompt }], role: 'user' }])
-
-        item
-      end
-
-    rescue Faraday::Error => e
-      raise "Erro na requisição: #{e.message}"
-    end
-    
-    def self.update(name:, content:)
-      conn = Faraday.new(
-        url: 'https://generativelanguage.googleapis.com',
-        headers: { 'Content-Type' => 'application/json' }
-      )
-
-      response = conn.patch("/v1beta/#{name}") do |req|
-        req.params['key'] = ENV.fetch('GEMINI_API_KEY')
-        req.body = content.to_json
-      end
-
-      return JSON.parse(response.body) if response.status == 200
-      
-      raise "Erro ao atualizar cache: #{response.body}"
-    rescue Faraday::Error => e
-      raise "Erro na requisição: #{e.message}"
-    end
-
-    def self.delete(name:)
-      conn = Faraday.new(
-        url: 'https://generativelanguage.googleapis.com',
-        headers: { 'Content-Type' => 'application/json' }
-      )
-
-      response = conn.delete("/v1beta/#{name}") do |req|
-        req.params['key'] = ENV.fetch('GEMINI_API_KEY')
-      end
-
-      return true if response.status == 200
-      
-      raise "Erro ao deletar cache: #{response.body}"
-    rescue Faraday::Error => e
-      raise "Erro na requisição: #{e.message}"
-    end
-  end
-end

data/lib/geminiext/messages.rb.disabled

@@ -1,34 +0,0 @@
-module geminiext
-  class Messages < Array
-    def initialize messages = nil
-      super parse_messages(messages)
-    end
-
-    def add(message) = concat(parse_messages(message))
-
-    private
-    def parse_messages(messages)
-      return [] if messages.nil?
-
-      messages = [messages] unless messages.is_a?(Array)
-      
-      # if first element is ok, then do not parse the rest
-      return messages if messages.first in { role: String | Symbol, content: String | Array | Hash}
-
-      messages.flat_map do |msg|
-        if msg.is_a?(Hash)
-          if msg.keys.size == 1
-            role, content = msg.first
-            { role: role.to_s, content: content }
-          elsif msg.key?(:role) && msg.key?(:content)
-            { role: msg[:role].to_s, content: msg[:content] }
-          else
-            msg.map { |role, content| { role: role.to_s, content: content } }
-          end
-        else
-          raise ArgumentError, "Invalid message format: #{msg}"
-        end
-      end
-    end
-  end
-end

data/lib/geminiext/model.rb.disabled

@@ -1,27 +0,0 @@
-module geminiext
-  module Model
-    GPT_BASIC_MODEL = ENV.fetch('OPENAI_GPT_BASIC_MODEL', 'gpt-4o-mini')
-    GPT_ADVANCED_MODEL = ENV.fetch('OPENAI_GPT_ADVANCED_MODEL', 'gpt-4o')
-    GPT_ADVANCED_MODEL_LATEST = ENV.fetch('OPENAI_GPT_ADVANCED_MODEL_LATEST', 'chatgpt-4o-latest')
-
-    O1_BASIC_MODEL = ENV.fetch('OPENAI_O1_BASIC_MODEL', 'o1-mini')
-    O1_ADVANCED_MODEL = ENV.fetch('OPENAI_O1_ADVANCED_MODEL', 'o1-preview')
-
-    def self.select(model)
-      case model
-      when :gpt_basic
-        GPT_BASIC_MODEL
-      when :gpt_advanced
-        GPT_ADVANCED_MODEL
-      when :gpt_advanced_latest
-        GPT_ADVANCED_MODEL_LATEST
-      when :o1_basic
-        O1_BASIC_MODEL
-      when :o1_advanced
-        O1_ADVANCED_MODEL
-      else
-        model
-      end
-    end
-  end
-end

data/lib/geminiext/response_extender.rb.disabled

@@ -1,44 +0,0 @@
-module ResponseExtender
-  def chat_params = self[:chat_params]
-
-  def message = dig('choices', 0, 'message')
-
-  def content = dig('choices', 0, 'message', 'content')
-  def content? = !content.nil?
-
-  def tool_calls = dig('choices', 0, 'message', 'tool_calls')
-  def tool_calls? = !tool_calls.nil?
-
-  def functions
-    return if tool_calls.nil?
-    
-    functions = tool_calls.filter { |tool| tool['type'].eql? 'function' }
-    return if functions.empty?
-    
-    functions_list = []
-    functions.map.with_index do |function, function_index|
-      function_info = tool_calls.dig(function_index, 'function')
-      function_def = { id: function['id'], name: function_info['name'], arguments: Oj.load(function_info['arguments'], symbol_keys: true) }
-      
-      def function_def.run(context:)
-        {
-          tool_call_id: self[:id],
-          role: :tool,
-          name: self[:name],
-          content: context.send(self[:name], **self[:arguments])
-        }
-      end
-
-      functions_list << function_def
-    end
-
-    functions_list
-  end
-
-  def functions_run_all(context:)
-    raise 'No functions to run' if functions.nil?
-    functions.map { |function| function.run(context:) }
-  end
-
-  def functions? = !functions.nil?
-end