geminiext 0.0.1 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0217cac9967438f0bbd4fe0dc38b6c0aba1d21475b937c6b4d1b4d41d51a12b6
-  data.tar.gz: df412fa82b43d42dcaca12b993cfa76f6c9e49dea76e55dfa9cbcddc79a9bee6
+  metadata.gz: 226e4ae21c091df3b4f0a02141097e2f55d475e0b34336fe7415e3517c8d7c73
+  data.tar.gz: 99d3eba31272bd7a04095e9fa1b9b56a897ecf058fd5c3ed67bc1e3530e489b1
 SHA512:
-  metadata.gz: 884cdc51610f1d2beefd9884a4b779560adfff38c14deafc3bad84b3dc2ce84079cabf29eae88d96e6688d73dc25824d23eaea7fc2b379fac327de87974e29ea
-  data.tar.gz: e9f69ffe68c4434738612884e54f89d5e471016ee1425846d374f9efa80b13e3fb4210f3f21a3b088e292861717fe0fa3f45473414d28307ec5b63781fe84b3b
+  metadata.gz: cdd1c32369e78fd98d74a432b85035f55bc2e2104bfd5d98ecfc5e3dd8c5066313767e7ad6d307cda80a41a37aa57723ff78214dae319f3e97cc4fb8a3d9a86c
+  data.tar.gz: 344c36bd43bddd92d7867e7d14789a0bd2919de308f86c2ed7dc1f8b7be36c8b62a4707a71d832e5375580802eb7c5fd776838c40f92ffa6cc5736c529433074

data/lib/geminiext/cache.rb

@@ -0,0 +1,116 @@
+require 'faraday'
+require 'json'
+
+module GeminiExt
+  module Cache
+    def self.create(contents:, display_name:, model: 'gemini-1.5-flash-001', 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
+      
+      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' }
+          )
+
+          response = conn.post("/v1beta/models/#{self['model'].split('/').last}:generateContent") do |req|
+            req.params['key'] = ENV.fetch('GEMINI_API_KEY')
+            req.body = {
+              contents:,
+              cached_content: self['name']
+            }.to_json
+          end
+
+          return JSON.parse(response.body) if response.status == 200
+
+          raise "Erro ao gerar conteúdo: #{response.body}"
+        rescue Faraday::Error => e
+          raise "Erro na requisição: #{e.message}"
+        end
+
+        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.rb

@@ -1,15 +1,17 @@
 require 'gemini-ai'
+#require 'faraday'
 
-# require 'geminiext/model'
+require 'geminiext/cache'
 # require 'geminiext/messages'
 # require 'geminiext/response_extender'
 
 module GeminiExt
   MAX_TOKENS = ENV.fetch('GEMINI_MAX_TOKENS', 8_192).to_i
-  
+
   def self.new(model: 'gemini-1.5-flash-001')
     Gemini.new(load_config(model: model))
   end
+ 
 
   def self.load_config(model: 'gemini-1.5-flash-001')
     {

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: geminiext
 version: !ruby/object:Gem::Version
-  version: 0.0.1
+  version: 0.0.3
 platform: ruby
 authors:
 - Gedean Dias
@@ -46,9 +46,10 @@ extra_rdoc_files: []
 files:
 - README.md
 - lib/geminiext.rb
-- lib/openaiext/messages.rb.disabled
-- lib/openaiext/model.rb.disabled
-- lib/openaiext/response_extender.rb.disabled
+- 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/geminiext
 licenses:
 - MIT