geminiext 0.0.1 → 0.0.2

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 0217cac9967438f0bbd4fe0dc38b6c0aba1d21475b937c6b4d1b4d41d51a12b6
-  data.tar.gz: df412fa82b43d42dcaca12b993cfa76f6c9e49dea76e55dfa9cbcddc79a9bee6
+  metadata.gz: 8584393659f6094698f8efc8f19d5bb2c1a1608641614cc2bbbf69734853328a
+  data.tar.gz: 586b5fb85e7b7074ead525b376379d79e01a471facf54c009b7575eb2dbabe10
 SHA512:
-  metadata.gz: 884cdc51610f1d2beefd9884a4b779560adfff38c14deafc3bad84b3dc2ce84079cabf29eae88d96e6688d73dc25824d23eaea7fc2b379fac327de87974e29ea
-  data.tar.gz: e9f69ffe68c4434738612884e54f89d5e471016ee1425846d374f9efa80b13e3fb4210f3f21a3b088e292861717fe0fa3f45473414d28307ec5b63781fe84b3b
+  metadata.gz: 20fc9f1d721df29f89f419e6ac9e9d76af4220b1e901e548562681f6d612f557f59e1a8415bb1a013575da14f055026a838b995096e2fce2bf566d4831b53731
+  data.tar.gz: 4d7b0de55589e664125455afc6c5a6a4fac115e610f4faa271a770e692b4ebbdd96e3a100c993fc2aeeadc9b93ce2dcd59152373ce0e6e8ee5eab425611091c3

data/lib/geminiext/cache.rb

@@ -0,0 +1,81 @@
+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.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)
+    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
+      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.2
 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