geminiext 0.0.2 → 0.0.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 8584393659f6094698f8efc8f19d5bb2c1a1608641614cc2bbbf69734853328a
-  data.tar.gz: 586b5fb85e7b7074ead525b376379d79e01a471facf54c009b7575eb2dbabe10
+  metadata.gz: 226e4ae21c091df3b4f0a02141097e2f55d475e0b34336fe7415e3517c8d7c73
+  data.tar.gz: 99d3eba31272bd7a04095e9fa1b9b56a897ecf058fd5c3ed67bc1e3530e489b1
 SHA512:
-  metadata.gz: 20fc9f1d721df29f89f419e6ac9e9d76af4220b1e901e548562681f6d612f557f59e1a8415bb1a013575da14f055026a838b995096e2fce2bf566d4831b53731
-  data.tar.gz: 4d7b0de55589e664125455afc6c5a6a4fac115e610f4faa271a770e692b4ebbdd96e3a100c993fc2aeeadc9b93ce2dcd59152373ce0e6e8ee5eab425611091c3
+  metadata.gz: cdd1c32369e78fd98d74a432b85035f55bc2e2104bfd5d98ecfc5e3dd8c5066313767e7ad6d307cda80a41a37aa57723ff78214dae319f3e97cc4fb8a3d9a86c
+  data.tar.gz: 344c36bd43bddd92d7867e7d14789a0bd2919de308f86c2ed7dc1f8b7be36c8b62a4707a71d832e5375580802eb7c5fd776838c40f92ffa6cc5736c529433074

data/lib/geminiext/cache.rb

@@ -28,6 +28,14 @@ module GeminiExt
       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',
@@ -37,8 +45,35 @@ module GeminiExt
       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
 
-      JSON.parse(response.body)
     rescue Faraday::Error => e
       raise "Erro na requisição: #{e.message}"
     end
@@ -51,7 +86,7 @@ module GeminiExt
 
       response = conn.patch("/v1beta/#{name}") do |req|
         req.params['key'] = ENV.fetch('GEMINI_API_KEY')
-        req.body = content
+        req.body = content.to_json
       end
 
       return JSON.parse(response.body) if response.status == 200

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: geminiext
 version: !ruby/object:Gem::Version
-  version: 0.0.2
+  version: 0.0.3
 platform: ruby
 authors:
 - Gedean Dias